glab-agent 0.2.5 → 0.2.7

package/src/local-agent/gitlab-glab-client.ts

@@ -17,6 +17,12 @@ export interface GitlabTodoItem {
   targetIid?: number;
   body?: string;
   sourceBranch?: string;
+  /** Note ID of the @mention comment that created this todo (for emoji reactions) */
+  noteId?: number;
+  /** GitLab user ID of the todo's author (the person who @mentioned the bot) */
+  authorId?: number;
+  /** GitLab username of the todo's author */
+  authorUsername?: string;
 }
 
 export interface GitlabIssue {
@@ -95,6 +101,15 @@ export interface GitlabClient {
   getWikiPage(projectId: number | string, slug: string): Promise<WikiPage>;
   createWikiPage(projectId: number | string, title: string, content: string): Promise<WikiPage>;
   updateWikiPage(projectId: number | string, slug: string, title: string, content: string): Promise<WikiPage>;
+  getProject(projectIdOrPath: number | string): Promise<{ id: number } | undefined>;
+  createProject(name: string, options?: { visibility?: string; initializeWithReadme?: boolean }): Promise<{ id: number }>;
+  getRepositoryFile(projectId: number | string, filePath: string, ref?: string): Promise<{ content: string } | undefined>;
+  createOrUpdateRepositoryFile(projectId: number | string, filePath: string, content: string, commitMessage: string, options?: { create?: boolean }): Promise<void>;
+  addAwardEmoji(projectId: number | string, issueIid: number, noteId: number, emojiName: string): Promise<{ id: number }>;
+  removeAwardEmoji(projectId: number | string, issueIid: number, noteId: number, awardId: number): Promise<void>;
+  listAwardEmoji(projectId: number | string, issueIid: number, noteId: number): Promise<Array<{ id: number; name: string; user: { id: number } }>>;
+  findMergeRequestByBranch(projectId: number | string, sourceBranch: string): Promise<{ iid: number; webUrl: string } | undefined>;
+  getUserById(userId: number): Promise<{ id: number; email?: string; username: string }>;
 }
 
 interface GitlabGlabClientOptions {
@@ -220,6 +235,13 @@ export function parseTodoUniversal(payload: Payload): GitlabTodoItem | undefined
     ? target.source_branch as string
     : undefined;
 
+  const note = payload.note as Payload | undefined;
+  const noteId = parseInteger(note?.id);
+
+  const author = payload.author as Payload | undefined;
+  const authorId = parseInteger(author?.id);
+  const authorUsername = typeof author?.username === "string" ? author.username as string : undefined;
+
   return {
     id,
     projectId,
@@ -233,7 +255,10 @@ export function parseTodoUniversal(payload: Payload): GitlabTodoItem | undefined
     createdAt: typeof payload.created_at === "string" ? payload.created_at : undefined,
     labels: parseLabels(target),
     body: typeof payload.body === "string" ? payload.body : undefined,
-    sourceBranch
+    sourceBranch,
+    noteId,
+    authorId,
+    authorUsername
   };
 }
 
@@ -693,6 +718,117 @@ export class GitlabGlabClient implements GitlabClient {
     };
   }
 
+  async getProject(projectIdOrPath: number | string): Promise<{ id: number } | undefined> {
+    try {
+      const encoded = typeof projectIdOrPath === "string" ? encodeURIComponent(projectIdOrPath) : projectIdOrPath;
+      const payload = await this.readJson(`projects/${encoded}`);
+      const p = (payload ?? {}) as Payload;
+      const id = Number(p.id);
+      return Number.isNaN(id) ? undefined : { id };
+    } catch {
+      return undefined;
+    }
+  }
+
+  async createProject(name: string, options?: { visibility?: string; initializeWithReadme?: boolean }): Promise<{ id: number }> {
+    const stdout = await this.request("projects", {
+      method: "POST",
+      fields: {
+        name,
+        path: name,
+        visibility: options?.visibility ?? "public",
+        initialize_with_readme: options?.initializeWithReadme ? "true" : "false"
+      }
+    });
+    const payload = JSON.parse(stdout) as Payload;
+    return { id: Number(payload.id) };
+  }
+
+  async getRepositoryFile(projectId: number | string, filePath: string, ref?: string): Promise<{ content: string } | undefined> {
+    try {
+      const encoded = typeof projectId === "string" ? encodeURIComponent(projectId) : projectId;
+      const endpoint = this.withQuery(`projects/${encoded}/repository/files/${encodeURIComponent(filePath)}`, { ref: ref ?? "main" });
+      const payload = await this.readJson(endpoint);
+      const p = (payload ?? {}) as Payload;
+      const content = typeof p.content === "string" ? Buffer.from(p.content, "base64").toString("utf8") : "";
+      return { content };
+    } catch {
+      return undefined;
+    }
+  }
+
+  async createOrUpdateRepositoryFile(projectId: number | string, filePath: string, content: string, commitMessage: string, options?: { create?: boolean }): Promise<void> {
+    const encoded = typeof projectId === "string" ? encodeURIComponent(projectId) : projectId;
+    const method = options?.create ? "POST" : "PUT";
+    await this.request(`projects/${encoded}/repository/files/${encodeURIComponent(filePath)}`, {
+      method,
+      fields: {
+        branch: "main",
+        content,
+        commit_message: commitMessage
+      }
+    });
+  }
+
+  async addAwardEmoji(projectId: number | string, issueIid: number, noteId: number, emojiName: string): Promise<{ id: number }> {
+    const stdout = await this.request(`projects/${projectId}/issues/${issueIid}/notes/${noteId}/award_emoji`, {
+      method: "POST",
+      fields: { name: emojiName }
+    });
+    const payload = JSON.parse(stdout) as Payload;
+    return { id: Number(payload.id) };
+  }
+
+  async removeAwardEmoji(projectId: number | string, issueIid: number, noteId: number, awardId: number): Promise<void> {
+    await this.request(`projects/${projectId}/issues/${issueIid}/notes/${noteId}/award_emoji/${awardId}`, {
+      method: "DELETE",
+      fields: {}
+    });
+  }
+
+  async listAwardEmoji(projectId: number | string, issueIid: number, noteId: number): Promise<Array<{ id: number; name: string; user: { id: number } }>> {
+    const payload = await this.readJson(`projects/${projectId}/issues/${issueIid}/notes/${noteId}/award_emoji`);
+    if (!Array.isArray(payload)) return [];
+    return payload.map((item: unknown) => {
+      const p = (item ?? {}) as Record<string, unknown>;
+      return {
+        id: Number(p.id),
+        name: String(p.name ?? ""),
+        user: { id: Number((p.user as Record<string, unknown> | undefined)?.id ?? 0) }
+      };
+    });
+  }
+
+  async findMergeRequestByBranch(projectId: number | string, sourceBranch: string): Promise<{ iid: number; webUrl: string } | undefined> {
+    try {
+      const endpoint = this.withQuery(`projects/${projectId}/merge_requests`, {
+        source_branch: sourceBranch,
+        state: "opened",
+        per_page: "1"
+      });
+      const payload = await this.readJson(endpoint);
+      if (!Array.isArray(payload) || payload.length === 0) return undefined;
+      const mr = (payload[0] ?? {}) as Record<string, unknown>;
+      const iid = Number(mr.iid);
+      const webUrl = String(mr.web_url ?? "");
+      return (iid && webUrl) ? { iid, webUrl } : undefined;
+    } catch {
+      return undefined;
+    }
+  }
+
+  async getUserById(userId: number): Promise<{ id: number; email?: string; username: string }> {
+    const payload = (await this.readJson(`users/${userId}`)) as Payload;
+    const p = payload ?? {};
+    return {
+      id: Number(p.id),
+      email: typeof p.email === "string" && p.email
+        ? p.email
+        : (typeof p.public_email === "string" && p.public_email ? p.public_email : undefined),
+      username: typeof p.username === "string" ? p.username : ""
+    };
+  }
+
   /**
    * Direct HTTP call using Node's built-in fetch.
    * Used for endpoints where glab CLI's response parsing has issues (e.g. PUT /user/status).
@@ -742,7 +878,7 @@ export class GitlabGlabClient implements GitlabClient {
   private async request(
     endpoint: string,
     options: {
-      method?: "GET" | "POST" | "PUT";
+      method?: "GET" | "POST" | "PUT" | "DELETE";
       fields?: Record<string, string>;
     } = {}
   ): Promise<string> {

package/src/local-agent/notifier.ts

@@ -6,6 +6,7 @@ export interface WebhookPayload {
   title: string;
   message: string;
   url?: string;
+  status?: "accepted" | "completed" | "failed" | "busy";
 }
 
 function isFeishuUrl(url: string): boolean {
@@ -16,18 +17,58 @@ function isSlackUrl(url: string): boolean {
   return url.includes("hooks.slack.com");
 }
 
+function buildFeishuCard(payload: WebhookPayload): unknown {
+  const statusConfig = {
+    accepted: { color: "blue", title: "已接单" },
+    completed: { color: "green", title: "已完成" },
+    failed: { color: "red", title: "执行失败" },
+    busy: { color: "orange", title: "排队中" },
+  };
+  const cfg = statusConfig[payload.status ?? "accepted"];
+
+  const elements: unknown[] = [
+    {
+      tag: "div",
+      text: { tag: "lark_md", content: payload.message }
+    }
+  ];
+
+  // Add URL button if available
+  if (payload.url) {
+    elements.push({
+      tag: "action",
+      actions: [
+        {
+          tag: "button",
+          text: { tag: "plain_text", content: payload.status === "completed" ? "查看 MR" : "查看 Issue" },
+          type: "primary",
+          url: payload.url
+        }
+      ]
+    });
+  }
+
+  return {
+    msg_type: "interactive",
+    card: {
+      header: {
+        title: { tag: "plain_text", content: payload.title },
+        template: cfg.color
+      },
+      elements
+    }
+  };
+}
+
 function buildWebhookBody(url: string, payload: WebhookPayload): unknown {
+  if (isFeishuUrl(url)) {
+    return buildFeishuCard(payload);
+  }
+
   const text = payload.url
     ? `${payload.title}\n${payload.message}\n${payload.url}`
     : `${payload.title}\n${payload.message}`;
 
-  if (isFeishuUrl(url)) {
-    return {
-      msg_type: "text",
-      content: { text }
-    };
-  }
-
   if (isSlackUrl(url)) {
     return { text };
   }
@@ -69,12 +110,15 @@ export async function notifyAccepted(
   agentName: string,
   issueIid: number,
   issueTitle: string,
+  issueUrl?: string,
   webhookUrl?: string
 ): Promise<void> {
   if (!webhookUrl) return;
   await sendWebhook(webhookUrl, {
     title: `\u{1F916} ${agentName} 已接单`,
     message: `#${issueIid} ${issueTitle}`,
+    url: issueUrl,
+    status: "accepted",
   });
 }
 
@@ -90,6 +134,7 @@ export async function notifyCompleted(
     title: `\u2705 ${agentName} 已完成`,
     message: `#${issueIid} ${issueTitle}`,
     url: mrUrl,
+    status: "completed",
   });
 }
 
@@ -103,6 +148,24 @@ export async function notifyFailed(
   if (!webhookUrl) return;
   await sendWebhook(webhookUrl, {
     title: `\u274C ${agentName} 执行失败`,
-    message: `#${issueIid} ${issueTitle}\n${summary.slice(0, 100)}`,
+    message: `#${issueIid} ${issueTitle}\n${summary.slice(0, 200)}`,
+    status: "failed",
+  });
+}
+
+export async function notifyQueued(
+  agentName: string,
+  issueIid: number,
+  issueTitle: string,
+  queuePosition: number,
+  issueUrl?: string,
+  webhookUrl?: string
+): Promise<void> {
+  if (!webhookUrl) return;
+  await sendWebhook(webhookUrl, {
+    title: `\u23F3 ${agentName} 排队中`,
+    message: `#${issueIid} ${issueTitle}\n排在第 ${queuePosition} 位`,
+    url: issueUrl,
+    status: "busy",
   });
 }