feat(ui): add structured_doc page graphql types (#4151)

* feat(ui): add structured_doc page graphql types

* [autofix.ci] apply automated fixes

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>

Author: zwpaper

ee/tabby-ui/app/pages/components/page-main.tsx

@@ -514,7 +514,7 @@ export function Page() {
           pageId,
           titlePrompt: title,
           docQuery: {
-            sourceIds: compact([page?.codeSourceId]),
+            sourceIds: compact([page?.codeSourceId, 'page']),
             content: title,
             searchPublic: true
           },
@@ -619,7 +619,7 @@ export function Page() {
               }
             : null,
           docQuery: {
-            sourceIds: compact([codeSourceId]),
+            sourceIds: compact([codeSourceId, 'page']),
             content: titlePrompt,
             searchPublic: true
           },

ee/tabby-ui/app/pages/lib/query.ts

@@ -121,6 +121,11 @@ export const createThreadToPageRunSubscription = graphql(/* GraphQL */ `
               }
               authorAt
             }
+            ... on AttachmentPageDoc {
+              link
+              title
+              content
+            }
           }
           score
         }
@@ -274,6 +279,11 @@ export const createPageRunSubscription = graphql(/* GraphQL */ `
               }
               authorAt
             }
+            ... on AttachmentPageDoc {
+              link
+              title
+              content
+            }
           }
           score
         }
@@ -383,6 +393,11 @@ export const createPageSectionRunSubscription = graphql(/* GraphQL */ `
               }
               authorAt
             }
+            ... on AttachmentPageDoc {
+              link
+              title
+              content
+            }
           }
           score
         }

ee/tabby-ui/app/search/components/search.tsx

@@ -1037,7 +1037,9 @@ function getSourceInputs(
   if (ctx) {
     sourceIdsForDocQuery = uniq(
       // Compatible with existing user messages
-      compact([repositorySourceId, ctx?.codeSourceId].concat(ctx.docSourceIds))
+      compact(
+        [repositorySourceId, ctx?.codeSourceId, 'page'].concat(ctx.docSourceIds)
+      )
     )
     searchPublic = ctx.searchPublic ?? false
     sourceIdForCodeQuery = repositorySourceId || ctx.codeSourceId || undefined

ee/tabby-ui/components/message-markdown/doc-detail-view.tsx

@@ -10,6 +10,7 @@ import {
   getAttachmentDocContent,
   isAttachmentCommitDoc,
   isAttachmentIssueDoc,
+  isAttachmentPageDoc,
   isAttachmentPullDoc,
   isAttachmentWebDoc
 } from '@/lib/utils'
@@ -51,9 +52,11 @@ export function DocDetailView({
     relevantDocument.title
   )
   const sourceUrl = link ? new URL(link) : null
-  const author = isAttachmentWebDoc(relevantDocument)
-    ? undefined
-    : relevantDocument.author
+  const author =
+    isAttachmentWebDoc(relevantDocument) ||
+    isAttachmentPageDoc(relevantDocument)
+      ? undefined
+      : relevantDocument.author
   const score = relevantDocument?.extra?.score
 
   return (

ee/tabby-ui/lib/hooks/use-thread-run.ts

@@ -104,6 +104,11 @@ const CreateThreadAndRunSubscription = graphql(/* GraphQL */ `
               }
               authorAt
             }
+            ... on MessageAttachmentPageDoc {
+              link
+              title
+              content
+            }
           }
           score
         }
@@ -205,6 +210,11 @@ const CreateThreadRunSubscription = graphql(/* GraphQL */ `
               }
               authorAt
             }
+            ... on MessageAttachmentPageDoc {
+              link
+              title
+              content
+            }
           }
           score
         }

ee/tabby-ui/lib/tabby/query.ts

@@ -465,6 +465,11 @@ export const listThreadMessages = graphql(/* GraphQL */ `
                 }
                 authorAt
               }
+              ... on MessageAttachmentPageDoc {
+                title
+                link
+                content
+              }
             }
             codeFileList {
               fileList
@@ -663,6 +668,11 @@ export const listPageSections = graphql(/* GraphQL */ `
                 }
                 authorAt
               }
+              ... on AttachmentPageDoc {
+                link
+                title
+                content
+              }
             }
           }
         }

ee/tabby-ui/lib/utils/attachment.ts

@@ -28,6 +28,13 @@ export function isAttachmentIssueDoc(attachment: AttachmentDocItem) {
   )
 }
 
+export function isAttachmentPageDoc(attachment: AttachmentDocItem) {
+  return (
+    attachment.__typename === 'AttachmentPageDoc' ||
+    attachment.__typename === 'MessageAttachmentPageDoc'
+  )
+}
+
 export function getAttachmentDocContent(attachment: AttachmentDocItem) {
   switch (attachment.__typename) {
     case 'MessageAttachmentWebDoc':