newTaskLinkResolver.ts

import type { NewTaskLinkPayload } from "@posthog/shared";
import { ANALYTICS_EVENTS } from "@posthog/shared/analytics-events";
import { inject, injectable } from "inversify";
import {
  GITHUB_ISSUE_CLIENT,
  type GitHubIssueClient,
  NEW_TASK_LINK_RESOLVER,
  type NewTaskLinkResolution,
} from "./identifiers";

export { NEW_TASK_LINK_RESOLVER };

@injectable()
export class NewTaskLinkResolver {
  constructor(
    @inject(GITHUB_ISSUE_CLIENT)
    private readonly github: GitHubIssueClient,
  ) {}

  async resolve(payload: NewTaskLinkPayload): Promise<NewTaskLinkResolution> {
    switch (payload.action) {
      case "new":
        return this.resolveNew(payload);
      case "plan":
        return this.resolvePlan(payload);
      case "issue":
        return this.resolveIssue(payload);
    }
  }

  private resolveNew(
    payload: Extract<NewTaskLinkPayload, { action: "new" }>,
  ): NewTaskLinkResolution {
    return {
      kind: "navigate",
      navigation: {
        initialPrompt: payload.prompt,
        initialCloudRepository: payload.repo,
        initialModel: payload.model,
        initialMode: payload.mode,
      },
      analytics: {
        event: ANALYTICS_EVENTS.DEEP_LINK_NEW_TASK,
        properties: {
          has_prompt: !!payload.prompt,
          has_repo: !!payload.repo,
          mode: payload.mode,
          model: payload.model,
          source: payload.source,
          issue_identifier: payload.issueIdentifier,
          prompt_length_chars: payload.prompt?.length,
        },
      },
    };
  }

  private resolvePlan(
    payload: Extract<NewTaskLinkPayload, { action: "plan" }>,
  ): NewTaskLinkResolution {
    return {
      kind: "navigate",
      navigation: {
        initialPrompt: payload.plan,
        initialCloudRepository: payload.repo,
        initialModel: payload.model,
        initialMode: payload.mode,
      },
      analytics: {
        event: ANALYTICS_EVENTS.DEEP_LINK_PLAN,
        properties: {
          has_repo: !!payload.repo,
          mode: payload.mode,
          model: payload.model,
          source: payload.source,
          plan_length_chars: payload.plan.length,
        },
      },
    };
  }

  private async resolveIssue(
    payload: Extract<NewTaskLinkPayload, { action: "issue" }>,
  ): Promise<NewTaskLinkResolution> {
    let issue: Awaited<ReturnType<GitHubIssueClient["getGithubIssue"]>>;
    try {
      issue = await this.github.getGithubIssue(
        payload.owner,
        payload.issueRepo,
        payload.issueNumber,
      );
    } catch (error) {
      const message = error instanceof Error ? error.message : String(error);
      return {
        kind: "fetch_failed",
        title: "Failed to fetch GitHub issue",
        description: message,
        analytics: {
          event: ANALYTICS_EVENTS.DEEP_LINK_ISSUE_FAILED,
          properties: {
            owner: payload.owner,
            repo: payload.issueRepo,
            issue_number: payload.issueNumber,
            reason: "fetch_failed",
            error_message: message,
          },
        },
      };
    }

    if (!issue) {
      return {
        kind: "not_found",
        title: "GitHub issue not found",
        description: `${payload.owner}/${payload.issueRepo}#${payload.issueNumber} could not be opened.`,
        analytics: {
          event: ANALYTICS_EVENTS.DEEP_LINK_ISSUE_FAILED,
          properties: {
            owner: payload.owner,
            repo: payload.issueRepo,
            issue_number: payload.issueNumber,
            reason: "not_found",
          },
        },
      };
    }

    const labelsText =
      issue.labels.length > 0 ? `\nLabels: ${issue.labels.join(", ")}` : "";
    const prompt = `GitHub Issue: ${issue.title}\n${issue.url}${labelsText}`;
    const cloudRepo = payload.repo ?? `${payload.owner}/${payload.issueRepo}`;

    return {
      kind: "navigate",
      navigation: {
        initialPrompt: prompt,
        initialCloudRepository: cloudRepo,
        initialModel: payload.model,
        initialMode: payload.mode,
      },
      analytics: {
        event: ANALYTICS_EVENTS.DEEP_LINK_ISSUE,
        properties: {
          owner: payload.owner,
          repo: payload.issueRepo,
          issue_number: payload.issueNumber,
          mode: payload.mode,
          model: payload.model,
          source: payload.source,
        },
      },
    };
  }
}