feat: add support for evaluating all open MRs

Author: jippi

.gitlab-ci.yml

@@ -1,4 +1,4 @@
-run::cli:
+scm-engine::evaluate:
   image: ${CI_DEPENDENCY_PROXY_GROUP_IMAGE_PREFIX}/golang:1.22.3
   rules:
     - if: $CI_PIPELINE_SOURCE == 'merge_request_event'

.scm-engine.example.yml

@@ -2,20 +2,44 @@
 
 actions:
   - name: Warn that the ticket if older than 30 days
-    if: merge_request.state != "closed" && merge_request.time_since_last_commit > duration("30d") && not merge_request.has_label("do-not-close")
+    if: |
+      // ignore MRs already closed
+      merge_request.state != "closed"
+      // if last commit happened more than 30 days ago
+      && merge_request.time_since_last_commit > duration("30d")
+      // but still less than 45 days ago (where we close the MR)
+      && merge_request.time_since_last_commit < duration("45d")
+      // and the label to disable this feature is not on the MR
+      && not merge_request.has_label("do-not-close")
     then:
       - action: comment
-        message: "Hey, this MR is old, we will close it in 15 days if no activity has happened. If you want to disable this behavior, add the label 'do-not-close' on the MR."
+        message: |
+          Hello!
+
+          This Merge Request have not seen any commit activity for 30 days. In an effort to keep our project clean we will automatically close the Merge request after 45 days.
+
+          You can add the "do-not-close" label to the Merge Request to disable this behavior.
 
   - name: Close ticket if older than 45 days
-    if: merge_request.state != "closed" && merge_request.time_since_last_commit > duration("45d") && not merge_request.has_label("do-not-close")
+    if: |
+      merge_request.state != "closed"
+      && merge_request.time_since_last_commit > duration("45d")
+      && not merge_request.has_label("do-not-close")
     then:
       - action: close
       - action: comment
-        message: "As promised, we're closing the MR due to inactivity, bye bye"
+        message: |
+          Hello!
+
+          This Merge Request have not seen any commit activity for 45 days. In an effort to keep our project clean we will automatically close the Merge request.
+
+          You can add the "do-not-close" label to the Merge Request to disable this behavior.
 
   - name: Approve MR if the 'break-glass-approve' label is configured
-    if: merge_request.state != "closed" && not merge_request.approved && merge_request.has_label("break-glass-approve")
+    if: |
+      merge_request.state != "closed"
+      && not merge_request.approved
+      && merge_request.has_label("break-glass-approve")
     then:
       - action: approve
       - action: comment

cmd/cmd_evaluate.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 
 	"github.com/jippi/scm-engine/pkg/config"
+	"github.com/jippi/scm-engine/pkg/scm"
 	"github.com/jippi/scm-engine/pkg/scm/gitlab"
 	"github.com/jippi/scm-engine/pkg/state"
 	"github.com/urfave/cli/v2"
@@ -23,6 +24,19 @@ func Evaluate(cCtx *cli.Context) error {
 	}
 
 	switch {
+	// If first arg is 'all' we will find all opened MRs and apply the rules to them
+	case cCtx.Args().First() == "all":
+		res, err := client.MergeRequests().List(ctx, &scm.ListMergeRequestsOptions{State: "opened", First: 100})
+		if err != nil {
+			return err
+		}
+
+		for _, mr := range res {
+			if err := ProcessMR(ctx, client, cfg, mr.ID); err != nil {
+				return err
+			}
+		}
+
 	// If the flag is set, use that for evaluation
 	case cCtx.String(FlagMergeRequestID) != "":
 		return ProcessMR(ctx, client, cfg, cCtx.String(FlagMergeRequestID))
@@ -37,7 +51,7 @@ func Evaluate(cCtx *cli.Context) error {
 				return err
 			}
 		}
-
-		return nil
 	}
+
+	return nil
 }

pkg/scm/gitlab/client_merge_request.go

@@ -5,9 +5,11 @@ import (
 	"fmt"
 	"net/http"
 
+	"github.com/hasura/go-graphql-client"
 	"github.com/jippi/scm-engine/pkg/scm"
 	"github.com/jippi/scm-engine/pkg/state"
 	go_gitlab "github.com/xanzy/go-gitlab"
+	"golang.org/x/oauth2"
 )
 
 var _ scm.MergeRequestClient = (*MergeRequestClient)(nil)
@@ -43,3 +45,36 @@ func (client *MergeRequestClient) Update(ctx context.Context, opt *scm.UpdateMer
 
 	return convertResponse(resp), err
 }
+
+func (client *MergeRequestClient) List(ctx context.Context, options *scm.ListMergeRequestsOptions) ([]scm.ListMergeRequest, error) {
+	httpClient := oauth2.NewClient(
+		ctx,
+		oauth2.StaticTokenSource(
+			&oauth2.Token{
+				AccessToken: client.client.token,
+			},
+		),
+	)
+
+	graphqlClient := graphql.NewClient(graphqlBaseURL(client.client.wrapped.BaseURL())+"/api/graphql", httpClient)
+
+	var (
+		result    *ListMergeRequestsQuery
+		variables = map[string]any{
+			"project_id": graphql.ID(state.ProjectIDFromContext(ctx)),
+			"state":      MergeRequestState(options.State),
+			"first":      options.First,
+		}
+	)
+
+	if err := graphqlClient.Query(ctx, &result, variables); err != nil {
+		return nil, err
+	}
+
+	hits := []scm.ListMergeRequest{}
+	for _, x := range result.Project.MergeRequests.Nodes {
+		hits = append(hits, scm.ListMergeRequest{ID: x.ID})
+	}
+
+	return hits, nil
+}

pkg/scm/interfaces.go

@@ -19,6 +19,7 @@ type LabelClient interface {
 
 type MergeRequestClient interface {
 	Update(ctx context.Context, opt *UpdateMergeRequestOptions) (*Response, error)
+	List(ctx context.Context, options *ListMergeRequestsOptions) ([]ListMergeRequest, error)
 }
 
 type EvalContext interface {

pkg/scm/types.go

@@ -94,6 +94,16 @@ type ListOptions struct {
 	Sort string `json:"sort,omitempty" url:"sort,omitempty"`
 }
 
+type ListMergeRequestsOptions struct {
+	ListOptions
+	State string
+	First int
+}
+
+type ListMergeRequest struct {
+	ID string `expr:"id" graphql:"id"`
+}
+
 // Response is a GitLab API response. This wraps the standard http.Response
 // returned from GitLab and provides convenient access to things like
 // pagination links.

schema/gitlab.go

@@ -67,7 +67,12 @@ func main() {
 func nest(props []*Property) {
 	for _, field := range props {
 		if field.IsCustomType {
-			for _, nested := range PropMap[field.Type].Attributes {
+			attr, ok := PropMap[field.Type]
+			if !ok {
+				continue
+			}
+
+			for _, nested := range attr.Attributes {
 				field.AddAttribute(&Property{
 					Name:         nested.Name,
 					Description:  nested.Description,
@@ -149,6 +154,8 @@ func mutateHook(b *modelgen.ModelBuild) *modelgen.ModelBuild {
 			Description: model.Description,
 		}
 
+		fmt.Println("model", modelProperty.Name)
+
 		for _, field := range model.Fields {
 			tags, err := structtag.Parse(field.Tag)
 			if err != nil {

schema/gitlab.schema.graphqls

@@ -21,6 +21,7 @@ scalar Time
 # Add time.Duration support
 scalar Duration
 
+
 type Context {
   "The project the Merge Request belongs to"
   Project: ContextProject @graphql(key: "project(fullPath: $project_id)")
@@ -32,6 +33,37 @@ type Context {
   MergeRequest: ContextMergeRequest @generated
 }
 
+enum MergeRequestState {
+  all
+  closed
+  locked
+  merged
+  opened
+}
+
+input ListMergeRequestsQueryInput {
+  project_id: ID!
+  state: MergeRequestState! = "opened"
+  first: Int! = 100
+}
+
+type ListMergeRequestsQuery {
+  "The project the Merge Request belongs to"
+  Project: ListMergeRequestsProject @graphql(key: "project(fullPath: $project_id)")
+}
+
+type ListMergeRequestsProject {
+  MergeRequests: ListMergeRequestsProjectMergeRequestNodes @graphql(key: "mergeRequests(state: $state, first: $first)") @internal
+}
+
+type ListMergeRequestsProjectMergeRequestNodes {
+  Nodes: [ListMergeRequestsProjectMergeRequest!]
+}
+
+type ListMergeRequestsProjectMergeRequest {
+  ID: String! @graphql(key: "iid") @internal
+}
+
 type ContextProject {
   #
   # Native GraphQL fields - https://docs.gitlab.com/ee/api/graphql/reference/#project