@andypai/agent-kanban 0.2.0 → 0.3.1

package/src/providers/linear-client.ts

@@ -1,5 +1,5 @@
-import { ErrorCode } from '../errors.ts'
-import { providerUpstreamError } from './errors.ts'
+import { ErrorCode } from '../errors'
+import { providerUpstreamError } from './errors'
 
 interface GraphQLResponse<T> {
   data?: T
@@ -31,6 +31,16 @@ export interface LinearIssue {
   assignee?: { id: string; name?: string | null; displayName?: string | null } | null
   project?: { id: string; name: string; url?: string | null; state?: string | null } | null
   state: { id: string; name: string; position: number }
+  labels?: string[]
+  commentCount?: number
+}
+
+export interface LinearComment {
+  id: string
+  body: string
+  createdAt: string
+  updatedAt: string
+  user?: { id: string; name?: string | null; displayName?: string | null } | null
 }
 
 interface LinearIssueNode {
@@ -45,8 +55,43 @@ interface LinearIssueNode {
   assignee?: { id: string; name?: string | null; displayName?: string | null } | null
   project?: { id: string; name: string; url?: string | null; state?: string | null } | null
   state: { id: string; name: string; position: number }
+  labels?: { nodes: Array<{ id: string; name: string }> }
+  comments?: {
+    nodes: Array<{ id: string }>
+    pageInfo?: { hasNextPage: boolean; endCursor: string | null }
+  } | null
 }
 
+interface LinearCommentNode {
+  id: string
+  body: string
+  createdAt: string
+  updatedAt: string
+  user?: { id: string; name?: string | null; displayName?: string | null } | null
+}
+
+function toLinearIssue(node: LinearIssueNode): LinearIssue {
+  return {
+    ...node,
+    assignee: node.assignee
+      ? {
+          id: node.assignee.id,
+          name: node.assignee.displayName || node.assignee.name,
+        }
+      : null,
+    labels: node.labels?.nodes.map((label) => label.name) ?? [],
+    commentCount: node.comments?.nodes?.length ?? undefined,
+  }
+}
+
+const COMMENT_FIELDS = `
+  id
+  body
+  createdAt
+  updatedAt
+  user { id name displayName }
+`
+
 export class LinearClient {
   private readonly endpoint = 'https://api.linear.app/graphql'
 
@@ -234,6 +279,13 @@ export class LinearClient {
                   name
                   position
                 }
+                labels {
+                  nodes { id name }
+                }
+                comments(first: 250) {
+                  nodes { id }
+                  pageInfo { hasNextPage endCursor }
+                }
               }
               pageInfo {
                 hasNextPage
@@ -244,17 +296,7 @@ export class LinearClient {
         `,
         { teamId, after, updatedAfter: updatedAfter ?? '1970-01-01T00:00:00.000Z' },
       )
-      issues.push(
-        ...data.issues.nodes.map((issue: LinearIssueNode) => ({
-          ...issue,
-          assignee: issue.assignee
-            ? {
-                id: issue.assignee.id,
-                name: issue.assignee.displayName || issue.assignee.name,
-              }
-            : null,
-        })),
-      )
+      issues.push(...data.issues.nodes.map(toLinearIssue))
       after = data.issues.pageInfo.hasNextPage ? data.issues.pageInfo.endCursor : null
     } while (after)
 
@@ -271,7 +313,7 @@ export class LinearClient {
     projectId?: string
   }): Promise<{ success: boolean; issue: LinearIssue | null }> {
     const data = await this.query<{
-      issueCreate: { success: boolean; issue: LinearIssue | null }
+      issueCreate: { success: boolean; issue: LinearIssueNode | null }
     }>(
       `
         mutation CreateIssue($input: IssueCreateInput!) {
@@ -289,6 +331,11 @@ export class LinearClient {
               assignee { id name displayName }
               project { id name url state }
               state { id name position }
+              labels { nodes { id name } }
+              comments(first: 250) {
+                nodes { id }
+                pageInfo { hasNextPage endCursor }
+              }
             }
           }
         }
@@ -305,7 +352,11 @@ export class LinearClient {
         },
       },
     )
-    return data.issueCreate
+    const node = data.issueCreate.issue
+    return {
+      success: data.issueCreate.success,
+      issue: node ? toLinearIssue(node) : null,
+    }
   }
 
   async updateIssue(
@@ -326,4 +377,193 @@ export class LinearClient {
     )
     return data.issueUpdate
   }
+
+  async listIssueHistory(params: {
+    issueId: string
+    first?: number
+    after?: string | null
+  }): Promise<{
+    nodes: Array<{
+      id: string
+      createdAt: string
+      fromState?: { id: string } | null
+      toState?: { id: string } | null
+    }>
+    pageInfo: { hasNextPage: boolean; endCursor: string | null }
+  }> {
+    const data = await this.query<{
+      issue: {
+        history: {
+          nodes: Array<{
+            id: string
+            createdAt: string
+            fromState?: { id: string } | null
+            toState?: { id: string } | null
+          }>
+          pageInfo: { hasNextPage: boolean; endCursor: string | null }
+        }
+      } | null
+    }>(
+      `
+        query IssueHistory($issueId: String!, $first: Int, $after: String) {
+          issue(id: $issueId) {
+            history(first: $first, after: $after) {
+              nodes {
+                id
+                createdAt
+                fromState { id }
+                toState { id }
+              }
+              pageInfo { hasNextPage endCursor }
+            }
+          }
+        }
+      `,
+      {
+        issueId: params.issueId,
+        first: params.first ?? 50,
+        after: params.after ?? null,
+      },
+    )
+    if (!data.issue) {
+      return { nodes: [], pageInfo: { hasNextPage: false, endCursor: null } }
+    }
+    return data.issue.history
+  }
+
+  async getIssueTeam(issueId: string): Promise<{ id: string; key: string } | null> {
+    const data = await this.query<{
+      issue: {
+        team: {
+          id: string
+          key: string
+        } | null
+      } | null
+    }>(
+      `
+        query IssueTeam($issueId: String!) {
+          issue(id: $issueId) {
+            team {
+              id
+              key
+            }
+          }
+        }
+      `,
+      { issueId },
+    )
+
+    return data.issue?.team ?? null
+  }
+
+  async listComments(issueId: string): Promise<LinearComment[]> {
+    let after: string | null = null
+    const comments: LinearComment[] = []
+
+    do {
+      const data: {
+        issue: {
+          comments: {
+            nodes: LinearCommentNode[]
+            pageInfo: PageInfo
+          }
+        } | null
+      } = await this.query(
+        `
+          query IssueComments($issueId: String!, $after: String) {
+            issue(id: $issueId) {
+              comments(first: 100, after: $after) {
+                nodes { ${COMMENT_FIELDS} }
+                pageInfo {
+                  hasNextPage
+                  endCursor
+                }
+              }
+            }
+          }
+        `,
+        { issueId, after },
+      )
+      if (!data.issue) {
+        providerUpstreamError(`Linear issue '${issueId}' was not found`)
+      }
+      comments.push(...data.issue.comments.nodes)
+      after = data.issue.comments.pageInfo.hasNextPage
+        ? data.issue.comments.pageInfo.endCursor
+        : null
+    } while (after)
+
+    return comments
+  }
+
+  async getComment(commentId: string): Promise<LinearComment> {
+    const data = await this.query<{ comment: LinearCommentNode | null }>(
+      `
+        query Comment($id: String!) {
+          comment(id: $id) { ${COMMENT_FIELDS} }
+        }
+      `,
+      { id: commentId },
+    )
+    if (!data.comment) {
+      providerUpstreamError(`Linear comment '${commentId}' was not found`)
+    }
+    return data.comment
+  }
+
+  async commentCreate(
+    issueId: string,
+    body: string,
+  ): Promise<{ success: boolean; comment: LinearComment | null }> {
+    const data = await this.query<{
+      commentCreate: { success: boolean; comment: LinearCommentNode | null }
+    }>(
+      `
+        mutation CommentCreate($input: CommentCreateInput!) {
+          commentCreate(input: $input) {
+            success
+            comment { ${COMMENT_FIELDS} }
+          }
+        }
+      `,
+      {
+        input: {
+          issueId,
+          body,
+        },
+      },
+    )
+    return {
+      success: data.commentCreate.success,
+      comment: data.commentCreate.comment,
+    }
+  }
+
+  async commentUpdate(
+    commentId: string,
+    body: string,
+  ): Promise<{ success: boolean; comment: LinearComment | null }> {
+    const data = await this.query<{
+      commentUpdate: { success: boolean; comment: LinearCommentNode | null }
+    }>(
+      `
+        mutation CommentUpdate($id: String!, $input: CommentUpdateInput!) {
+          commentUpdate(id: $id, input: $input) {
+            success
+            comment { ${COMMENT_FIELDS} }
+          }
+        }
+      `,
+      {
+        id: commentId,
+        input: {
+          body,
+        },
+      },
+    )
+    return {
+      success: data.commentUpdate.success,
+      comment: data.commentUpdate.comment,
+    }
+  }
 }