clanker-agent/packages/coding-agent/test/package-manager-ssh.test.ts

import { mkdirSync, rmSync } from "node:fs";
import { tmpdir } from "node:os";
import { join } from "node:path";
import { afterEach, beforeEach, describe, expect, it } from "vitest";
import { DefaultPackageManager } from "../src/core/package-manager.js";
import { SettingsManager } from "../src/core/settings-manager.js";

describe("Package Manager git source parsing", () => {
  let tempDir: string;
  let agentDir: string;
  let settingsManager: SettingsManager;
  let packageManager: DefaultPackageManager;

  beforeEach(() => {
    tempDir = join(
      tmpdir(),
      `pm-ssh-test-${Date.now()}-${Math.random().toString(36).slice(2)}`,
    );
    mkdirSync(tempDir, { recursive: true });
    agentDir = join(tempDir, "agent");
    mkdirSync(agentDir, { recursive: true });

    settingsManager = SettingsManager.inMemory();
    packageManager = new DefaultPackageManager({
      cwd: tempDir,
      agentDir,
      settingsManager,
    });
  });

  afterEach(() => {
    rmSync(tempDir, { recursive: true, force: true });
  });

  describe("protocol URLs without git: prefix", () => {
    it("should parse https:// URL", () => {
      const parsed = (packageManager as any).parseSource(
        "https://github.com/user/repo",
      );
      expect(parsed.type).toBe("git");
      expect(parsed.host).toBe("github.com");
      expect(parsed.path).toBe("user/repo");
    });

    it("should parse ssh:// URL", () => {
      const parsed = (packageManager as any).parseSource(
        "ssh://git@github.com/user/repo",
      );
      expect(parsed.type).toBe("git");
      expect(parsed.host).toBe("github.com");
      expect(parsed.path).toBe("user/repo");
      expect(parsed.repo).toBe("ssh://git@github.com/user/repo");
    });
  });

  describe("shorthand URLs with git: prefix", () => {
    it("should parse git@host:path format", () => {
      const parsed = (packageManager as any).parseSource(
        "git:git@github.com:user/repo",
      );
      expect(parsed.type).toBe("git");
      expect(parsed.host).toBe("github.com");
      expect(parsed.path).toBe("user/repo");
      expect(parsed.repo).toBe("git@github.com:user/repo");
      expect(parsed.pinned).toBe(false);
    });

    it("should parse host/path shorthand", () => {
      const parsed = (packageManager as any).parseSource(
        "git:github.com/user/repo",
      );
      expect(parsed.type).toBe("git");
      expect(parsed.host).toBe("github.com");
      expect(parsed.path).toBe("user/repo");
    });

    it("should parse shorthand with ref", () => {
      const parsed = (packageManager as any).parseSource(
        "git:git@github.com:user/repo@v1.0.0",
      );
      expect(parsed.type).toBe("git");
      expect(parsed.ref).toBe("v1.0.0");
      expect(parsed.pinned).toBe(true);
    });
  });

  describe("unsupported without git: prefix", () => {
    it("should treat git@host:path as local without git: prefix", () => {
      const parsed = (packageManager as any).parseSource(
        "git@github.com:user/repo",
      );
      expect(parsed.type).toBe("local");
    });

    it("should treat host/path shorthand as local without git: prefix", () => {
      const parsed = (packageManager as any).parseSource(
        "github.com/user/repo",
      );
      expect(parsed.type).toBe("local");
    });
  });

  describe("identity normalization", () => {
    it("should normalize protocol and shorthand-prefixed URLs to same identity", () => {
      const prefixed = (packageManager as any).getPackageIdentity(
        "git:git@github.com:user/repo",
      );
      const https = (packageManager as any).getPackageIdentity(
        "https://github.com/user/repo",
      );
      const ssh = (packageManager as any).getPackageIdentity(
        "ssh://git@github.com/user/repo",
      );

      expect(prefixed).toBe("git:github.com/user/repo");
      expect(prefixed).toBe(https);
      expect(prefixed).toBe(ssh);
    });
  });
});