From d4dbadee1059106dadcdfad5ff2658499cef9624 Mon Sep 17 00:00:00 2001
From: Bryan Frimin <bryan@getprobo.com>
Date: Thu, 3 Jul 2025 08:26:25 +0200
Subject: [PATCH] Fix 5xx on document type order

Signed-off-by: Bryan Frimin <bryan@getprobo.com>
---
 pkg/coredata/document.go                   |  2 ++
 pkg/coredata/document_order_field.go       |  5 +++--
 pkg/server/api/console/v1/schema.graphql   |  4 ++++
 pkg/server/api/console/v1/schema/schema.go | 14 ++++++++++----
 4 files changed, 19 insertions(+), 6 deletions(-)

diff --git a/pkg/coredata/document.go b/pkg/coredata/document.go
index 13b2f66aa..5c49d5c47 100644
--- a/pkg/coredata/document.go
+++ b/pkg/coredata/document.go
@@ -47,6 +47,8 @@ func (p Document) CursorKey(orderBy DocumentOrderField) page.CursorKey {
 		return page.NewCursorKey(p.ID, p.CreatedAt)
 	case DocumentOrderFieldTitle:
 		return page.NewCursorKey(p.ID, p.Title)
+	case DocumentOrderFieldDocumentType:
+		return page.NewCursorKey(p.ID, p.DocumentType)
 	}
 
 	panic(fmt.Sprintf("unsupported order by: %s", orderBy))
diff --git a/pkg/coredata/document_order_field.go b/pkg/coredata/document_order_field.go
index e71c717da..06112a188 100644
--- a/pkg/coredata/document_order_field.go
+++ b/pkg/coredata/document_order_field.go
@@ -19,8 +19,9 @@ type (
 )
 
 const (
-	DocumentOrderFieldCreatedAt DocumentOrderField = "CREATED_AT"
-	DocumentOrderFieldTitle     DocumentOrderField = "TITLE"
+	DocumentOrderFieldCreatedAt    DocumentOrderField = "CREATED_AT"
+	DocumentOrderFieldTitle        DocumentOrderField = "TITLE"
+	DocumentOrderFieldDocumentType DocumentOrderField = "DOCUMENT_TYPE"
 )
 
 func (p DocumentOrderField) Column() string {
diff --git a/pkg/server/api/console/v1/schema.graphql b/pkg/server/api/console/v1/schema.graphql
index ed7df01f5..9c094adc3 100644
--- a/pkg/server/api/console/v1/schema.graphql
+++ b/pkg/server/api/console/v1/schema.graphql
@@ -213,6 +213,10 @@ enum DocumentOrderField
     @goEnum(
       value: "github.com/getprobo/probo/pkg/coredata.DocumentOrderFieldCreatedAt"
     )
+  DOCUMENT_TYPE
+    @goEnum(
+      value: "github.com/getprobo/probo/pkg/coredata.DocumentOrderFieldDocumentType"
+    )
 }
 
 enum RiskOrderField
diff --git a/pkg/server/api/console/v1/schema/schema.go b/pkg/server/api/console/v1/schema/schema.go
index 0e2db365a..7b5a0ccc7 100644
--- a/pkg/server/api/console/v1/schema/schema.go
+++ b/pkg/server/api/console/v1/schema/schema.go
@@ -5068,6 +5068,10 @@ enum DocumentOrderField
     @goEnum(
       value: "github.com/getprobo/probo/pkg/coredata.DocumentOrderFieldCreatedAt"
     )
+  DOCUMENT_TYPE
+    @goEnum(
+      value: "github.com/getprobo/probo/pkg/coredata.DocumentOrderFieldDocumentType"
+    )
 }
 
 enum RiskOrderField
@@ -51348,12 +51352,14 @@ func (ec *executionContext) marshalNDocumentOrderField2githubᚗcomᚋgetprobo
 
 var (
 	unmarshalNDocumentOrderField2githubᚗcomᚋgetproboᚋproboᚋpkgᚋcoredataᚐDocumentOrderField = map[string]coredata.DocumentOrderField{
-		"TITLE":      coredata.DocumentOrderFieldTitle,
-		"CREATED_AT": coredata.DocumentOrderFieldCreatedAt,
+		"TITLE":         coredata.DocumentOrderFieldTitle,
+		"CREATED_AT":    coredata.DocumentOrderFieldCreatedAt,
+		"DOCUMENT_TYPE": coredata.DocumentOrderFieldDocumentType,
 	}
 	marshalNDocumentOrderField2githubᚗcomᚋgetproboᚋproboᚋpkgᚋcoredataᚐDocumentOrderField = map[coredata.DocumentOrderField]string{
-		coredata.DocumentOrderFieldTitle:     "TITLE",
-		coredata.DocumentOrderFieldCreatedAt: "CREATED_AT",
+		coredata.DocumentOrderFieldTitle:        "TITLE",
+		coredata.DocumentOrderFieldCreatedAt:    "CREATED_AT",
+		coredata.DocumentOrderFieldDocumentType: "DOCUMENT_TYPE",
 	}
 )