@desplega.ai/agent-swarm 1.20.0 → 1.51.2

package/src/gitlab/handlers.ts

@@ -0,0 +1,368 @@
+/**
+ * GitLab Webhook Event Handlers
+ *
+ * Mirrors the pattern from src/github/handlers.ts:
+ * - Parses GitLab webhook payloads
+ * - Creates agent tasks via createTaskExtended()
+ * - Deduplicates events via in-memory TTL map
+ * - Detects bot mentions
+ */
+
+import { createTaskExtended, failTask, findTaskByVcs, getAllAgents } from "../be/db";
+import { resolveTemplate } from "../prompts/resolver";
+import { GITLAB_BOT_NAME } from "./auth";
+import { addGitLabNoteReaction, addGitLabReaction } from "./reactions";
+// Side-effect import: registers all GitLab event templates in the in-memory registry
+import "./templates";
+import type { IssueEvent, MergeRequestEvent, NoteEvent, PipelineEvent } from "./types";
+
+// ── Dedup cache (same pattern as GitHub) ──
+const processedEvents = new Map<string, number>();
+const DEDUP_TTL_MS = 60_000;
+
+function isDuplicate(key: string): boolean {
+  const now = Date.now();
+  // Cleanup expired
+  for (const [k, ts] of processedEvents) {
+    if (now - ts > DEDUP_TTL_MS) processedEvents.delete(k);
+  }
+  if (processedEvents.has(key)) return true;
+  processedEvents.set(key, now);
+  return false;
+}
+
+// ── Helpers ──
+
+function detectMention(text: string | null | undefined): boolean {
+  if (!text) return false;
+  return new RegExp(`@${GITLAB_BOT_NAME}\\b`, "i").test(text);
+}
+
+function extractMentionContext(text: string): string {
+  return text.replace(new RegExp(`@${GITLAB_BOT_NAME}\\b`, "gi"), "").trim();
+}
+
+function findLeadAgent() {
+  const agents = getAllAgents();
+  return (
+    agents.find((a) => a.role === "lead" && a.status === "idle") ??
+    agents.find((a) => a.role === "lead") ??
+    null
+  );
+}
+
+// ── Event Handlers ──
+
+export async function handleMergeRequest(
+  event: MergeRequestEvent,
+): Promise<{ created: boolean; taskId?: string }> {
+  const { user, project, object_attributes: mr } = event;
+  const action = mr.action;
+  const repo = project.path_with_namespace;
+
+  console.log(`[GitLab] MR #${mr.iid} ${action} by ${user.username} in ${repo}`);
+
+  const dedupKey = `gitlab-mr-${repo}-${mr.iid}-${action}-${user.username}`;
+  if (isDuplicate(dedupKey)) {
+    console.log(`[GitLab] Skipping duplicate MR event`);
+    return { created: false };
+  }
+
+  const lead = findLeadAgent();
+
+  switch (action) {
+    case "open": {
+      // Check for bot mention in description
+      const hasMention = detectMention(mr.description);
+      if (!hasMention) {
+        console.log(`[GitLab] MR opened without bot mention, skipping`);
+        return { created: false };
+      }
+
+      const context = mr.description ? extractMentionContext(mr.description) : "";
+      const contextSection = context ? `Context: ${context}\n\n` : "";
+      const result = resolveTemplate("gitlab.merge_request.opened", {
+        mr_iid: mr.iid,
+        mr_title: mr.title,
+        repo,
+        username: user.username,
+        source_branch: mr.source_branch,
+        target_branch: mr.target_branch,
+        mr_url: mr.url,
+        context_section: contextSection,
+      });
+
+      if (result.skipped) {
+        return { created: false };
+      }
+
+      const task = createTaskExtended(result.text, {
+        agentId: lead?.id ?? null,
+        source: "gitlab",
+        vcsProvider: "gitlab",
+        taskType: "gitlab-mr",
+        vcsRepo: repo,
+        vcsEventType: "merge_request",
+        vcsNumber: mr.iid,
+        vcsAuthor: user.username,
+        vcsUrl: mr.url,
+      });
+
+      try {
+        await addGitLabReaction(repo, "mr", mr.iid, "eyes");
+      } catch {}
+
+      return { created: true, taskId: task.id };
+    }
+
+    case "close":
+    case "merge": {
+      // Cancel existing tasks for this MR
+      const existingTask = findTaskByVcs(repo, mr.iid);
+      if (existingTask) {
+        const reason = action === "merge" ? "MR was merged" : "MR was closed";
+        console.log(`[GitLab] Cancelling task ${existingTask.id} — ${reason}`);
+        failTask(existingTask.id, reason);
+      }
+      return { created: false };
+    }
+
+    case "update": {
+      // Check if there's an active task for this MR — if so, notify about the update
+      const task = findTaskByVcs(repo, mr.iid);
+      if (task) {
+        console.log(`[GitLab] MR #${mr.iid} updated, active task exists: ${task.id}`);
+        // Don't create a new task — the worker will see the changes
+        return { created: false };
+      }
+      return { created: false };
+    }
+
+    default:
+      console.log(`[GitLab] Ignoring MR action: ${action}`);
+      return { created: false };
+  }
+}
+
+export async function handleIssue(
+  event: IssueEvent,
+): Promise<{ created: boolean; taskId?: string }> {
+  const { user, project, object_attributes: issue } = event;
+  const action = issue.action;
+  const repo = project.path_with_namespace;
+
+  console.log(`[GitLab] Issue #${issue.iid} ${action} by ${user.username} in ${repo}`);
+
+  const dedupKey = `gitlab-issue-${repo}-${issue.iid}-${action}-${user.username}`;
+  if (isDuplicate(dedupKey)) {
+    return { created: false };
+  }
+
+  const lead = findLeadAgent();
+
+  switch (action) {
+    case "open": {
+      const hasMention = detectMention(issue.description);
+      if (!hasMention) {
+        // Check if bot is in assignees
+        const isBotAssigned = event.assignees?.some(
+          (a) => a.username.toLowerCase() === GITLAB_BOT_NAME.toLowerCase(),
+        );
+        if (!isBotAssigned) {
+          console.log(`[GitLab] Issue opened without bot mention/assignment, skipping`);
+          return { created: false };
+        }
+      }
+
+      const context = issue.description ? extractMentionContext(issue.description) : "";
+      const contextSection = context ? `Context: ${context}\n\n` : "";
+      const result = resolveTemplate("gitlab.issue.assigned", {
+        issue_iid: issue.iid,
+        issue_title: issue.title,
+        repo,
+        username: user.username,
+        issue_url: issue.url,
+        context_section: contextSection,
+      });
+
+      if (result.skipped) {
+        return { created: false };
+      }
+
+      const task = createTaskExtended(result.text, {
+        agentId: lead?.id ?? null,
+        source: "gitlab",
+        vcsProvider: "gitlab",
+        taskType: "gitlab-issue",
+        vcsRepo: repo,
+        vcsEventType: "issue",
+        vcsNumber: issue.iid,
+        vcsAuthor: user.username,
+        vcsUrl: issue.url,
+      });
+
+      try {
+        await addGitLabReaction(repo, "issue", issue.iid, "eyes");
+      } catch {}
+
+      return { created: true, taskId: task.id };
+    }
+
+    case "close": {
+      const existingTask = findTaskByVcs(repo, issue.iid);
+      if (existingTask) {
+        console.log(`[GitLab] Cancelling task ${existingTask.id} — issue closed`);
+        failTask(existingTask.id, "Issue was closed");
+      }
+      return { created: false };
+    }
+
+    default:
+      return { created: false };
+  }
+}
+
+export async function handleNote(event: NoteEvent): Promise<{ created: boolean; taskId?: string }> {
+  const { user, project, object_attributes: note } = event;
+  const repo = project.path_with_namespace;
+
+  // Only handle comments with bot mentions
+  if (!detectMention(note.note)) {
+    return { created: false };
+  }
+
+  console.log(`[GitLab] Note by ${user.username} in ${repo} (${note.noteable_type})`);
+
+  const dedupKey = `gitlab-note-${note.id}`;
+  if (isDuplicate(dedupKey)) {
+    return { created: false };
+  }
+
+  const lead = findLeadAgent();
+  const context = extractMentionContext(note.note);
+
+  // Determine the target entity (MR or issue)
+  let targetNumber: number | undefined;
+  let targetUrl: string;
+  let eventType: string;
+  let entityLabel: string;
+
+  if (note.noteable_type === "MergeRequest" && event.merge_request) {
+    targetNumber = event.merge_request.iid;
+    targetUrl = event.merge_request.url ?? note.url;
+    eventType = "note_on_mr";
+    entityLabel = `MR #${targetNumber}`;
+  } else if (note.noteable_type === "Issue" && event.issue) {
+    targetNumber = event.issue.iid;
+    targetUrl = event.issue.url ?? note.url;
+    eventType = "note_on_issue";
+    entityLabel = `Issue #${targetNumber}`;
+  } else {
+    console.log(`[GitLab] Ignoring note on ${note.noteable_type}`);
+    return { created: false };
+  }
+
+  // Check if there's already an active task for this entity
+  const existingTask = targetNumber ? findTaskByVcs(repo, targetNumber) : null;
+
+  const existingTaskNote = existingTask
+    ? `\n\n_Note: There's an active task (${existingTask.id}) for this ${entityLabel}._`
+    : "";
+
+  const noteResult = resolveTemplate("gitlab.comment.mentioned", {
+    entity_label: entityLabel,
+    username: user.username,
+    repo,
+    target_url: targetUrl,
+    context,
+    existing_task_note: existingTaskNote,
+  });
+
+  if (noteResult.skipped) {
+    return { created: false };
+  }
+
+  const task = createTaskExtended(noteResult.text, {
+    agentId: lead?.id ?? null,
+    source: "gitlab",
+    vcsProvider: "gitlab",
+    taskType: "gitlab-comment",
+    vcsRepo: repo,
+    vcsEventType: eventType,
+    vcsNumber: targetNumber,
+    vcsCommentId: note.id,
+    vcsAuthor: user.username,
+    vcsUrl: targetUrl,
+    parentTaskId: existingTask?.id,
+  });
+
+  try {
+    await addGitLabNoteReaction(
+      repo,
+      note.noteable_type === "MergeRequest" ? "mr" : "issue",
+      targetNumber!,
+      note.id,
+      "eyes",
+    );
+  } catch {}
+
+  return { created: true, taskId: task.id };
+}
+
+export async function handlePipeline(
+  event: PipelineEvent,
+): Promise<{ created: boolean; taskId?: string }> {
+  const { project, object_attributes: pipeline } = event;
+  const repo = project.path_with_namespace;
+
+  // Only handle failed pipelines that are associated with a merge request
+  if (pipeline.status !== "failed") {
+    return { created: false };
+  }
+
+  if (!event.merge_request) {
+    console.log(`[GitLab] Pipeline failed but no associated MR, skipping`);
+    return { created: false };
+  }
+
+  const mrIid = event.merge_request.iid;
+  const dedupKey = `gitlab-pipeline-${pipeline.id}`;
+  if (isDuplicate(dedupKey)) {
+    return { created: false };
+  }
+
+  // Only create task if there's already an active task for this MR
+  const existingTask = findTaskByVcs(repo, mrIid);
+  if (!existingTask) {
+    return { created: false };
+  }
+
+  const lead = findLeadAgent();
+  const pipelineResult = resolveTemplate("gitlab.pipeline.failed", {
+    pipeline_id: pipeline.id,
+    mr_iid: mrIid,
+    repo,
+    mr_title: event.merge_request.title,
+    mr_url: event.merge_request.url,
+    source_branch: event.merge_request.source_branch,
+  });
+
+  if (pipelineResult.skipped) {
+    return { created: false };
+  }
+
+  const task = createTaskExtended(pipelineResult.text, {
+    agentId: lead?.id ?? null,
+    source: "gitlab",
+    vcsProvider: "gitlab",
+    taskType: "gitlab-ci",
+    vcsRepo: repo,
+    vcsEventType: "pipeline",
+    vcsNumber: mrIid,
+    vcsAuthor: "",
+    vcsUrl: event.merge_request.url,
+    parentTaskId: existingTask.id,
+  });
+
+  return { created: true, taskId: task.id };
+}

package/src/gitlab/index.ts

@@ -0,0 +1,19 @@
+// GitLab Integration
+export {
+  GITLAB_BOT_NAME,
+  getGitLabToken,
+  getGitLabUrl,
+  initGitLab,
+  isGitLabEnabled,
+  resetGitLab,
+  verifyGitLabWebhook,
+} from "./auth";
+export { handleIssue, handleMergeRequest, handleNote, handlePipeline } from "./handlers";
+export { addGitLabNoteReaction, addGitLabReaction, postGitLabComment } from "./reactions";
+export type {
+  GitLabWebhookEvent,
+  IssueEvent,
+  MergeRequestEvent,
+  NoteEvent,
+  PipelineEvent,
+} from "./types";

package/src/gitlab/reactions.ts

@@ -0,0 +1,104 @@
+/**
+ * GitLab API interactions — reactions (award emoji) and comments (notes).
+ * Uses raw fetch, no SDK.
+ */
+
+import { getGitLabToken, getGitLabUrl } from "./auth";
+
+function headers(): Record<string, string> {
+  const token = getGitLabToken();
+  if (!token) throw new Error("[GitLab] No API token configured");
+  return {
+    "PRIVATE-TOKEN": token,
+    "Content-Type": "application/json",
+  };
+}
+
+function apiBase(): string {
+  return `${getGitLabUrl()}/api/v4`;
+}
+
+/** Encode a project path (e.g. "group/project") for use in API URLs. */
+function encodeProject(pathWithNamespace: string): string {
+  return encodeURIComponent(pathWithNamespace);
+}
+
+/**
+ * Add an award emoji (reaction) to a merge request or issue.
+ * GitLab calls reactions "award emoji".
+ * @see https://docs.gitlab.com/ee/api/award_emoji.html
+ */
+export async function addGitLabReaction(
+  project: string,
+  entityType: "issue" | "mr",
+  iid: number,
+  emojiName: string,
+): Promise<void> {
+  const resourceType = entityType === "mr" ? "merge_requests" : "issues";
+  const url = `${apiBase()}/projects/${encodeProject(project)}/${resourceType}/${iid}/award_emoji`;
+  try {
+    const resp = await fetch(url, {
+      method: "POST",
+      headers: headers(),
+      body: JSON.stringify({ name: emojiName }),
+    });
+    if (!resp.ok) {
+      console.error(`[GitLab] Failed to add reaction: ${resp.status} ${await resp.text()}`);
+    }
+  } catch (err) {
+    console.error(`[GitLab] Error adding reaction:`, err);
+  }
+}
+
+/**
+ * Add an award emoji (reaction) to a specific note (comment).
+ * @see https://docs.gitlab.com/ee/api/award_emoji.html#award-a-new-emoji-on-a-comment
+ */
+export async function addGitLabNoteReaction(
+  project: string,
+  entityType: "issue" | "mr",
+  iid: number,
+  noteId: number,
+  emojiName: string,
+): Promise<void> {
+  const resourceType = entityType === "mr" ? "merge_requests" : "issues";
+  const url = `${apiBase()}/projects/${encodeProject(project)}/${resourceType}/${iid}/notes/${noteId}/award_emoji`;
+  try {
+    const resp = await fetch(url, {
+      method: "POST",
+      headers: headers(),
+      body: JSON.stringify({ name: emojiName }),
+    });
+    if (!resp.ok) {
+      console.error(`[GitLab] Failed to add note reaction: ${resp.status} ${await resp.text()}`);
+    }
+  } catch (err) {
+    console.error(`[GitLab] Error adding note reaction:`, err);
+  }
+}
+
+/**
+ * Post a comment (note) on a merge request or issue.
+ * @see https://docs.gitlab.com/ee/api/notes.html
+ */
+export async function postGitLabComment(
+  project: string,
+  entityType: "issue" | "mr",
+  iid: number,
+  body: string,
+): Promise<void> {
+  const resourceType = entityType === "mr" ? "merge_requests" : "issues";
+  const url = `${apiBase()}/projects/${encodeProject(project)}/${resourceType}/${iid}/notes`;
+  try {
+    const resp = await fetch(url, {
+      method: "POST",
+      headers: headers(),
+      body: JSON.stringify({ body }),
+    });
+    if (!resp.ok) {
+      console.error(`[GitLab] Failed to post comment: ${resp.status} ${await resp.text()}`);
+    }
+  } catch (err) {
+    console.error(`[GitLab] Error posting comment:`, err);
+  }
+}

package/src/gitlab/templates.ts

@@ -0,0 +1,140 @@
+/**
+ * GitLab event prompt template definitions.
+ *
+ * Each template is registered at module load time via registerTemplate().
+ * Handlers import this module for the side-effect of registration.
+ *
+ * Template text uses {{variable}} syntax. The resolver interpolates variables
+ * and expands {{@template[id]}} references before returning the final text.
+ */
+
+import { registerTemplate } from "../prompts/registry";
+
+// ============================================================================
+// Common building blocks (GitLab-specific)
+// ============================================================================
+
+registerTemplate({
+  eventType: "common.delegation_instruction.gitlab",
+  header: "",
+  defaultBody:
+    "\n\n**Delegation instruction:** As the lead agent, analyze this and decide whether to handle it yourself or delegate to a worker agent. Use `send-task` to delegate with clear instructions.",
+  variables: [],
+  category: "common",
+});
+
+registerTemplate({
+  eventType: "common.command_suggestions.gitlab_mr",
+  header: "",
+  defaultBody:
+    "Suggested commands: `/review-pr` to review the MR, `/implement-issue` to implement changes.",
+  variables: [],
+  category: "common",
+});
+
+registerTemplate({
+  eventType: "common.command_suggestions.gitlab_issue",
+  header: "",
+  defaultBody: "Suggested commands: `/implement-issue` to implement, `/close-issue` to close.",
+  variables: [],
+  category: "common",
+});
+
+// ============================================================================
+// Merge Request events
+// ============================================================================
+
+registerTemplate({
+  eventType: "gitlab.merge_request.opened",
+  header: "[GitLab MR #{{mr_iid}}] {{mr_title}}",
+  defaultBody: `Repo: {{repo}}
+Author: @{{username}}
+Branch: {{source_branch}} \u2192 {{target_branch}}
+URL: {{mr_url}}
+
+{{context_section}}{{@template[common.command_suggestions.gitlab_mr]}}{{@template[common.delegation_instruction.gitlab]}}`,
+  variables: [
+    { name: "mr_iid", description: "Merge request IID" },
+    { name: "mr_title", description: "Merge request title" },
+    { name: "repo", description: "Project path with namespace" },
+    { name: "username", description: "Event sender username" },
+    { name: "source_branch", description: "Source branch name" },
+    { name: "target_branch", description: "Target branch name" },
+    { name: "mr_url", description: "Merge request URL" },
+    { name: "context_section", description: "Context section with description or empty string" },
+  ],
+  category: "event",
+});
+
+// ============================================================================
+// Issue events
+// ============================================================================
+
+registerTemplate({
+  eventType: "gitlab.issue.assigned",
+  header: "[GitLab Issue #{{issue_iid}}] {{issue_title}}",
+  defaultBody: `Repo: {{repo}}
+Author: @{{username}}
+URL: {{issue_url}}
+
+{{context_section}}{{@template[common.command_suggestions.gitlab_issue]}}{{@template[common.delegation_instruction.gitlab]}}`,
+  variables: [
+    { name: "issue_iid", description: "Issue IID" },
+    { name: "issue_title", description: "Issue title" },
+    { name: "repo", description: "Project path with namespace" },
+    { name: "username", description: "Event sender username" },
+    { name: "issue_url", description: "Issue URL" },
+    { name: "context_section", description: "Context section with description or empty string" },
+  ],
+  category: "event",
+});
+
+// ============================================================================
+// Comment (Note) events
+// ============================================================================
+
+registerTemplate({
+  eventType: "gitlab.comment.mentioned",
+  header: "[GitLab Comment on {{entity_label}}] @{{username}} mentioned bot",
+  defaultBody: `Repo: {{repo}}
+URL: {{target_url}}
+
+Comment:
+{{context}}{{existing_task_note}}{{@template[common.delegation_instruction.gitlab]}}`,
+  variables: [
+    { name: "entity_label", description: "Entity label (e.g. 'MR #1' or 'Issue #2')" },
+    { name: "username", description: "Comment author username" },
+    { name: "repo", description: "Project path with namespace" },
+    { name: "target_url", description: "Target entity URL" },
+    { name: "context", description: "Extracted mention context from comment" },
+    {
+      name: "existing_task_note",
+      description: "Note about existing active task, or empty string",
+    },
+  ],
+  category: "event",
+});
+
+// ============================================================================
+// Pipeline events
+// ============================================================================
+
+registerTemplate({
+  eventType: "gitlab.pipeline.failed",
+  header: "[GitLab CI Failed] Pipeline #{{pipeline_id}} failed for MR #{{mr_iid}}",
+  defaultBody: `Repo: {{repo}}
+MR: {{mr_title}}
+URL: {{mr_url}}
+Branch: {{source_branch}}
+
+The CI pipeline has failed. Please investigate and fix the issues.{{@template[common.delegation_instruction.gitlab]}}`,
+  variables: [
+    { name: "pipeline_id", description: "Pipeline ID" },
+    { name: "mr_iid", description: "Merge request IID" },
+    { name: "repo", description: "Project path with namespace" },
+    { name: "mr_title", description: "Merge request title" },
+    { name: "mr_url", description: "Merge request URL" },
+    { name: "source_branch", description: "Source branch name" },
+  ],
+  category: "event",
+});