Add sort fields for audit engagement dates

Expose AUDIT_START_DATE and AUDIT_END_DATE on AuditOrderField
so list queries can paginate like valid_from and valid_until.

Signed-off-by: Cursor Agent <cursoragent@cursor.com>

Co-authored-by: Bryan FRIMIN <bryan@frimin.fr>

pkg/cmd/audit/list/list.go

@@ -122,7 +122,7 @@ func NewCmdList(f *cmdutil.Factory) *cobra.Command {
 			}
 
 			if flagOrderBy != "" {
-				if err := cmdutil.ValidateEnum("order-by", flagOrderBy, []string{"CREATED_AT", "VALID_FROM", "VALID_UNTIL", "STATE"}); err != nil {
+				if err := cmdutil.ValidateEnum("order-by", flagOrderBy, []string{"CREATED_AT", "VALID_FROM", "VALID_UNTIL", "AUDIT_START_DATE", "AUDIT_END_DATE", "STATE"}); err != nil {
 					return err
 				}
 
@@ -224,7 +224,7 @@ func NewCmdList(f *cmdutil.Factory) *cobra.Command {
 
 	cmd.Flags().StringVar(&flagOrg, "org", "", "Organization ID")
 	cmd.Flags().IntVarP(&flagLimit, "limit", "L", 30, "Maximum number of audits to list")
-	cmd.Flags().StringVar(&flagOrderBy, "order-by", "", "Order by field (CREATED_AT, VALID_FROM, VALID_UNTIL, STATE)")
+	cmd.Flags().StringVar(&flagOrderBy, "order-by", "", "Order by field (CREATED_AT, VALID_FROM, VALID_UNTIL, AUDIT_START_DATE, AUDIT_END_DATE, STATE)")
 	cmd.Flags().StringVar(&flagOrderDir, "order-direction", "DESC", "Sort direction (ASC, DESC)")
 	flagOutput = cmdutil.AddOutputFlag(cmd)
 

pkg/coredata/audit.go

@@ -62,6 +62,10 @@ func (a *Audit) CursorKey(field AuditOrderField) page.CursorKey {
 		return page.NewCursorKey(a.ID, a.ValidFrom)
 	case AuditOrderFieldValidUntil:
 		return page.NewCursorKey(a.ID, a.ValidUntil)
+	case AuditOrderFieldAuditStartDate:
+		return page.NewCursorKey(a.ID, a.AuditStartDate)
+	case AuditOrderFieldAuditEndDate:
+		return page.NewCursorKey(a.ID, a.AuditEndDate)
 	case AuditOrderFieldState:
 		return page.NewCursorKey(a.ID, a.State)
 	}

pkg/coredata/audit_order_field.go

@@ -30,10 +30,12 @@ import (
 type AuditOrderField string
 
 const (
-	AuditOrderFieldCreatedAt  AuditOrderField = "CREATED_AT"
-	AuditOrderFieldValidFrom  AuditOrderField = "VALID_FROM"
-	AuditOrderFieldValidUntil AuditOrderField = "VALID_UNTIL"
-	AuditOrderFieldState      AuditOrderField = "STATE"
+	AuditOrderFieldCreatedAt      AuditOrderField = "CREATED_AT"
+	AuditOrderFieldValidFrom      AuditOrderField = "VALID_FROM"
+	AuditOrderFieldValidUntil     AuditOrderField = "VALID_UNTIL"
+	AuditOrderFieldAuditStartDate AuditOrderField = "AUDIT_START_DATE"
+	AuditOrderFieldAuditEndDate   AuditOrderField = "AUDIT_END_DATE"
+	AuditOrderFieldState          AuditOrderField = "STATE"
 )
 
 var (
@@ -48,6 +50,8 @@ func AuditOrderFields() []AuditOrderField {
 		AuditOrderFieldCreatedAt,
 		AuditOrderFieldValidFrom,
 		AuditOrderFieldValidUntil,
+		AuditOrderFieldAuditStartDate,
+		AuditOrderFieldAuditEndDate,
 		AuditOrderFieldState,
 	}
 }
@@ -58,6 +62,8 @@ func (v AuditOrderField) IsValid() bool {
 		AuditOrderFieldCreatedAt,
 		AuditOrderFieldValidFrom,
 		AuditOrderFieldValidUntil,
+		AuditOrderFieldAuditStartDate,
+		AuditOrderFieldAuditEndDate,
 		AuditOrderFieldState:
 		return true
 	}

pkg/server/api/console/v1/graphql/audit.graphql

@@ -89,6 +89,14 @@ enum AuditOrderField
         @goEnum(
             value: "go.probo.inc/probo/pkg/coredata.AuditOrderFieldValidUntil"
         )
+    AUDIT_START_DATE
+        @goEnum(
+            value: "go.probo.inc/probo/pkg/coredata.AuditOrderFieldAuditStartDate"
+        )
+    AUDIT_END_DATE
+        @goEnum(
+            value: "go.probo.inc/probo/pkg/coredata.AuditOrderFieldAuditEndDate"
+        )
     STATE @goEnum(value: "go.probo.inc/probo/pkg/coredata.AuditOrderFieldState")
 }
 

pkg/server/api/mcp/v1/specification.yaml

@@ -4620,6 +4620,8 @@ components:
         - CREATED_AT
         - VALID_FROM
         - VALID_UNTIL
+        - AUDIT_START_DATE
+        - AUDIT_END_DATE
         - STATE
       go.probo.inc/mcpgen/type: go.probo.inc/probo/pkg/coredata.AuditOrderField