add team ids filter config option for ticket schemas

Author: laurenleach

cmd/baton-linear/main.go

@@ -36,7 +36,7 @@ func main() {
 
 func getConnector(ctx context.Context, lc *cfg.Linear) (types.ConnectorServer, error) {
 	l := ctxzap.Extract(ctx)
-	cb, err := connector.New(ctx, lc.ApiKey, lc.SkipProjects)
+	cb, err := connector.New(ctx, lc.ApiKey, lc.SkipProjects, lc.TicketSchemaTeamIdsFilter)
 	if err != nil {
 		l.Error("error creating connector", zap.Error(err))
 		return nil, err

pkg/config/conf.gen.go

@@ -17,13 +17,22 @@ var (
 		field.WithDisplayName("Skip projects"),
 		field.WithDescription("Skip syncing projects."),
 	)
+	teamIDsTicketSchemaFilterField = field.StringSliceField(
+		"ticket-schema-team-ids-filter",
+		field.WithDisplayName("Teams"),
+		field.WithDescription("Comma-separated list of team IDs to use for tickets schemas."),
+	)
 )
 
 var externalTicketField = field.TicketingField.ExportAs(field.ExportTargetGUI)
+var configRelations = []field.SchemaFieldRelationship{
+	field.FieldsDependentOn([]field.SchemaField{teamIDsTicketSchemaFilterField}, []field.SchemaField{field.TicketingField}),
+}
 
 //go:generate go run ./gen
 var Config = field.NewConfiguration(
-	[]field.SchemaField{apiKey, externalTicketField, skipProjects},
+	[]field.SchemaField{apiKey, externalTicketField, skipProjects, teamIDsTicketSchemaFilterField},
+	field.WithConstraints(configRelations...),
 	field.WithConnectorDisplayName("Linear"),
 	field.WithHelpUrl("/docs/baton/linear"),
 	field.WithIconUrl("/static/app-icons/linear.svg"),

pkg/config/config.go

@@ -41,8 +41,9 @@ var (
 )
 
 type Linear struct {
-	client       *linear.Client
-	skipProjects bool
+	client              *linear.Client
+	skipProjects        bool
+	ticketSchemaTeamIDs []string
 }
 
 func (ln *Linear) ResourceSyncers(ctx context.Context) []connectorbuilder.ResourceSyncer {
@@ -79,14 +80,15 @@ func (ln *Linear) Validate(ctx context.Context) (annotations.Annotations, error)
 }
 
 // New returns the Linear connector.
-func New(ctx context.Context, apiKey string, skipProjects bool) (*Linear, error) {
+func New(ctx context.Context, apiKey string, skipProjects bool, ticketSchemaTeamIDs []string) (*Linear, error) {
 	client, err := linear.NewClient(ctx, apiKey)
 	if err != nil {
 		return nil, err
 	}
 
 	return &Linear{
-		client:       client,
-		skipProjects: skipProjects,
+		client:              client,
+		skipProjects:        skipProjects,
+		ticketSchemaTeamIDs: ticketSchemaTeamIDs,
 	}, nil
 }

pkg/connector/connector.go

@@ -154,7 +154,7 @@ func (ln *Linear) ListTicketSchemas(ctx context.Context, p *pagination.Token) ([
 	}
 	l := ctxzap.Extract(ctx)
 
-	teams, nextToken, _, rlData, err := ln.client.ListTeamWorkflowStates(ctx, linear.GetTeamsVars{After: bag.PageToken(), First: resourcePageSize})
+	teams, nextToken, _, rlData, err := ln.client.ListTeamWorkflowStates(ctx, linear.GetTeamsVars{TeamIDs: ln.ticketSchemaTeamIDs, After: bag.PageToken(), First: resourcePageSize})
 	annotations.WithRateLimiting(rlData)
 	if err != nil {
 		l.Debug("teams failed", zap.Error(err), zap.Any("rlData", rlData))