@vividcodeai/embeddedcowork 0.0.21 → 0.0.23

package/dist/workspaces/git-status.js

@@ -0,0 +1,323 @@
+import { spawn } from "child_process";
+import { readFile } from "fs/promises";
+import path from "path";
+import { normalizeGitWorktreeRelativePath } from "./git-mutations";
+async function readFileAsDiffText(filePath) {
+    return readFile(filePath, "utf-8");
+}
+async function readGitBlobAsDiffText(resultPromise, missingOk = false) {
+    const result = await resultPromise;
+    if (!result.ok) {
+        return decodeGitShowResult(result, missingOk);
+    }
+    return result.stdout;
+}
+function runGit(args, cwd, acceptedExitCodes = [0]) {
+    return new Promise((resolve) => {
+        const child = spawn("git", args, { cwd, stdio: ["ignore", "pipe", "pipe"] });
+        let stdout = "";
+        let stderr = "";
+        child.stdout?.on("data", (chunk) => {
+            stdout += chunk.toString();
+        });
+        child.stderr?.on("data", (chunk) => {
+            stderr += chunk.toString();
+        });
+        child.once("error", (error) => {
+            resolve({ ok: false, error, stdout, stderr });
+        });
+        child.once("close", (code) => {
+            if (acceptedExitCodes.includes(code ?? 0)) {
+                resolve({ ok: true, stdout });
+            }
+            else {
+                const error = new Error(stderr.trim() || `git ${args.join(" ")} failed with code ${code}`);
+                resolve({ ok: false, error, stdout, stderr });
+            }
+        });
+    });
+}
+function ensureEntry(map, path) {
+    const existing = map.get(path);
+    if (existing)
+        return existing;
+    const next = {
+        path,
+        originalPath: null,
+        stagedStatus: null,
+        stagedAdditions: 0,
+        stagedDeletions: 0,
+        unstagedStatus: null,
+        unstagedAdditions: 0,
+        unstagedDeletions: 0,
+    };
+    map.set(path, next);
+    return next;
+}
+function normalizeGitStatusPath(value) {
+    return value.trim().replace(/\\+/g, "/");
+}
+function parseGitChangeKind(code) {
+    const normalized = code.trim().toUpperCase();
+    if (!normalized)
+        return null;
+    if (normalized === "A")
+        return "added";
+    if (normalized === "M")
+        return "modified";
+    if (normalized === "D")
+        return "deleted";
+    if (normalized.startsWith("R"))
+        return "renamed";
+    if (normalized.startsWith("C"))
+        return "copied";
+    if (normalized === "U")
+        return "unmerged";
+    return null;
+}
+function applyNameStatusOutput(map, output, target) {
+    const tokens = output.split("\0");
+    let index = 0;
+    while (index < tokens.length) {
+        const record = tokens[index++] ?? "";
+        if (!record)
+            continue;
+        const parts = record.split("\t");
+        const statusCode = parseGitChangeKind(parts[0] ?? "");
+        if (!statusCode)
+            continue;
+        const inlinePath = parts.slice(1).join("\t");
+        const firstPath = inlinePath || tokens[index++] || "";
+        const secondPath = statusCode === "renamed" || statusCode === "copied" ? tokens[index++] || "" : "";
+        const path = statusCode === "renamed" || statusCode === "copied" ? secondPath || firstPath : firstPath;
+        const normalizedPath = normalizeGitStatusPath(path);
+        if (!normalizedPath)
+            continue;
+        const entry = ensureEntry(map, normalizedPath);
+        entry[target] = statusCode;
+        if (statusCode === "renamed" || statusCode === "copied") {
+            const originalPath = normalizeGitStatusPath(firstPath);
+            entry.originalPath = originalPath || entry.originalPath || null;
+        }
+    }
+}
+function applyUntrackedOutput(map, output) {
+    for (const rawLine of output.split(/\r?\n/)) {
+        const path = normalizeGitStatusPath(rawLine);
+        if (!path)
+            continue;
+        ensureEntry(map, path).unstagedStatus = "untracked";
+    }
+}
+function parseSingleNumstat(output) {
+    for (const rawLine of output.split(/\r?\n/)) {
+        const line = rawLine.trim();
+        if (!line)
+            continue;
+        const parts = rawLine.split("\t");
+        const isBinary = parts[0] === "-" || parts[1] === "-";
+        return {
+            additions: isBinary ? 0 : Number.parseInt(parts[0] ?? "0", 10) || 0,
+            deletions: isBinary ? 0 : Number.parseInt(parts[1] ?? "0", 10) || 0,
+            isBinary,
+            found: true,
+        };
+    }
+    return { additions: 0, deletions: 0, isBinary: false, found: false };
+}
+async function getUntrackedFileNumstat(workspaceFolder, relativePath) {
+    const absolutePath = path.join(workspaceFolder, relativePath);
+    const result = await runGit(["diff", "--numstat", "--no-index", "--", "/dev/null", absolutePath], workspaceFolder, [0, 1]);
+    if (!result.ok) {
+        throw result.error;
+    }
+    const parsed = parseSingleNumstat(result.stdout);
+    return { additions: parsed.additions, deletions: parsed.deletions };
+}
+async function applyUntrackedFileStats(map, workspaceFolder) {
+    const pending = Array.from(map.values())
+        .filter((entry) => entry.unstagedStatus === "untracked")
+        .map(async (entry) => {
+        try {
+            const stats = await getUntrackedFileNumstat(workspaceFolder, entry.path);
+            entry.unstagedAdditions = stats.additions;
+            entry.unstagedDeletions = stats.deletions;
+        }
+        catch {
+            entry.unstagedAdditions = 0;
+            entry.unstagedDeletions = 0;
+        }
+    });
+    await Promise.all(pending);
+}
+function applyNumstatOutput(map, output, target) {
+    const tokens = output.split("\0");
+    let index = 0;
+    while (index < tokens.length) {
+        const record = tokens[index++] ?? "";
+        if (!record)
+            continue;
+        const parts = record.split("\t");
+        if (parts.length < 3)
+            continue;
+        const additions = parts[0] === "-" ? 0 : Number.parseInt(parts[0] ?? "0", 10);
+        const deletions = parts[1] === "-" ? 0 : Number.parseInt(parts[1] ?? "0", 10);
+        const inlinePath = parts.slice(2).join("\t");
+        const isRenameLike = inlinePath === "";
+        const originalPath = isRenameLike ? normalizeGitStatusPath(tokens[index++] ?? "") : null;
+        const normalizedPath = normalizeGitStatusPath(isRenameLike ? tokens[index++] ?? "" : inlinePath);
+        if (!normalizedPath)
+            continue;
+        const entry = ensureEntry(map, normalizedPath);
+        if (originalPath) {
+            entry.originalPath = originalPath;
+        }
+        if (target === "staged") {
+            entry.stagedAdditions = Number.isFinite(additions) ? additions : 0;
+            entry.stagedDeletions = Number.isFinite(deletions) ? deletions : 0;
+        }
+        else {
+            entry.unstagedAdditions = Number.isFinite(additions) ? additions : 0;
+            entry.unstagedDeletions = Number.isFinite(deletions) ? deletions : 0;
+        }
+    }
+}
+export async function getWorktreeGitStatus(params) {
+    const { workspaceFolder, logger } = params;
+    const [stagedResult, unstagedResult, untrackedResult, stagedNumstatResult, unstagedNumstatResult] = await Promise.all([
+        runGit(["diff", "--name-status", "-z", "--cached", "--find-renames", "--find-copies"], workspaceFolder),
+        runGit(["diff", "--name-status", "-z", "--find-renames", "--find-copies"], workspaceFolder),
+        runGit(["ls-files", "--others", "--exclude-standard"], workspaceFolder),
+        runGit(["diff", "--numstat", "-z", "--cached", "--find-renames", "--find-copies"], workspaceFolder),
+        runGit(["diff", "--numstat", "-z", "--find-renames", "--find-copies"], workspaceFolder),
+    ]);
+    for (const result of [stagedResult, unstagedResult, untrackedResult, stagedNumstatResult, unstagedNumstatResult]) {
+        if (!result.ok) {
+            logger?.warn?.({ workspaceFolder, err: result.error }, "Failed to read git status for worktree");
+            throw result.error;
+        }
+    }
+    const stagedOutput = stagedResult.stdout;
+    const unstagedOutput = unstagedResult.stdout;
+    const untrackedOutput = untrackedResult.stdout;
+    const stagedNumstatOutput = stagedNumstatResult.stdout;
+    const unstagedNumstatOutput = unstagedNumstatResult.stdout;
+    const entries = new Map();
+    applyNameStatusOutput(entries, stagedOutput, "stagedStatus");
+    applyNameStatusOutput(entries, unstagedOutput, "unstagedStatus");
+    applyUntrackedOutput(entries, untrackedOutput);
+    applyNumstatOutput(entries, stagedNumstatOutput, "staged");
+    applyNumstatOutput(entries, unstagedNumstatOutput, "unstaged");
+    await applyUntrackedFileStats(entries, workspaceFolder);
+    return Array.from(entries.values()).sort((a, b) => a.path.localeCompare(b.path));
+}
+function decodeGitShowResult(result, missingOk = false) {
+    if (result.ok)
+        return result.stdout;
+    const message = result.stderr?.trim() || result.error.message || "";
+    if (missingOk &&
+        (message.includes("exists on disk, but not in") ||
+            message.includes("Path '") ||
+            message.includes("does not exist") ||
+            message.includes("unknown revision or path not in the working tree"))) {
+        return "";
+    }
+    throw result.error;
+}
+async function readGitIndexBlob(workspaceFolder, normalizedPath) {
+    return runGit(["cat-file", "-p", `:${normalizedPath}`], workspaceFolder);
+}
+async function getTrackedDiffMetadata(params) {
+    const args = ["diff", "--numstat"];
+    if (params.scope === "staged") {
+        args.push("--cached");
+    }
+    args.push("--find-renames", "--find-copies", "--");
+    args.push(params.normalizedPath);
+    if (params.normalizedOriginalPath && params.normalizedOriginalPath !== params.normalizedPath) {
+        args.push(params.normalizedOriginalPath);
+    }
+    const result = await runGit(args, params.workspaceFolder);
+    if (!result.ok) {
+        throw result.error;
+    }
+    const parsed = parseSingleNumstat(result.stdout);
+    return { isBinary: parsed.isBinary, found: parsed.found };
+}
+async function getUntrackedDiffMetadata(params) {
+    const absolutePath = path.join(params.workspaceFolder, params.normalizedPath);
+    const result = await runGit(["diff", "--numstat", "--no-index", "--", "/dev/null", absolutePath], params.workspaceFolder, [0, 1]);
+    if (!result.ok) {
+        throw result.error;
+    }
+    return { isBinary: parseSingleNumstat(result.stdout).isBinary };
+}
+async function resolveUnstagedBeforePath(params) {
+    const currentPathResult = await readGitIndexBlob(params.workspaceFolder, params.normalizedPath);
+    if (currentPathResult.ok || !params.normalizedOriginalPath || params.normalizedOriginalPath === params.normalizedPath) {
+        return currentPathResult;
+    }
+    return readGitIndexBlob(params.workspaceFolder, params.normalizedOriginalPath);
+}
+export async function getWorktreeGitDiff(params) {
+    const normalizedPath = normalizeGitWorktreeRelativePath(params.path);
+    const normalizedOriginalPath = params.originalPath ? normalizeGitWorktreeRelativePath(params.originalPath) : null;
+    const trackedMetadata = await getTrackedDiffMetadata({
+        workspaceFolder: params.workspaceFolder,
+        scope: params.scope,
+        normalizedPath,
+        normalizedOriginalPath,
+    });
+    const diffMetadata = params.scope === "unstaged" && !trackedMetadata.found
+        ? await getUntrackedDiffMetadata({
+            workspaceFolder: params.workspaceFolder,
+            normalizedPath,
+        })
+        : trackedMetadata;
+    if (diffMetadata.isBinary) {
+        return {
+            path: normalizedPath,
+            originalPath: normalizedOriginalPath,
+            scope: params.scope,
+            before: "",
+            after: "",
+            isBinary: true,
+        };
+    }
+    if (params.scope === "staged") {
+        const [beforeResult, afterResult] = await Promise.all([
+            readGitBlobAsDiffText(runGit(["show", `HEAD:${normalizedOriginalPath ?? normalizedPath}`], params.workspaceFolder), true),
+            readGitBlobAsDiffText(readGitIndexBlob(params.workspaceFolder, normalizedPath), true),
+        ]);
+        return {
+            path: normalizedPath,
+            originalPath: normalizedOriginalPath,
+            scope: params.scope,
+            before: beforeResult,
+            after: afterResult,
+            isBinary: false,
+        };
+    }
+    const indexResult = await resolveUnstagedBeforePath({
+        workspaceFolder: params.workspaceFolder,
+        normalizedPath,
+        normalizedOriginalPath,
+    });
+    const beforeResult = await readGitBlobAsDiffText(Promise.resolve(indexResult), true);
+    let after = beforeResult;
+    const fsPath = path.join(params.workspaceFolder, normalizedPath);
+    try {
+        after = await readFileAsDiffText(fsPath);
+    }
+    catch {
+        after = "";
+    }
+    return {
+        path: normalizedPath,
+        originalPath: normalizedOriginalPath,
+        scope: params.scope,
+        before: beforeResult,
+        after,
+        isBinary: false,
+    };
+}

package/dist/workspaces/git-worktrees.js

@@ -0,0 +1,216 @@
+import path from "path";
+import { spawn } from "child_process";
+import { promises as fsp } from "fs";
+function isGitUnavailableResult(result) {
+    return !result.ok && result.error?.code === "ENOENT";
+}
+function runGit(args, cwd) {
+    return new Promise((resolve) => {
+        const child = spawn("git", args, { cwd, stdio: ["ignore", "pipe", "pipe"] });
+        let stdout = "";
+        let stderr = "";
+        child.stdout?.on("data", (chunk) => {
+            stdout += chunk.toString();
+        });
+        child.stderr?.on("data", (chunk) => {
+            stderr += chunk.toString();
+        });
+        child.once("error", (error) => {
+            resolve({ ok: false, error, stdout, stderr });
+        });
+        child.once("close", (code) => {
+            if (code === 0) {
+                resolve({ ok: true, stdout });
+            }
+            else {
+                const error = new Error(stderr.trim() || `git ${args.join(" ")} failed with code ${code}`);
+                resolve({ ok: false, error, stdout, stderr });
+            }
+        });
+    });
+}
+export async function resolveRepoRoot(folder, logger) {
+    const result = await runGit(["rev-parse", "--show-toplevel"], folder);
+    if (isGitUnavailableResult(result)) {
+        throw new Error("Git is not installed or not available in PATH");
+    }
+    if (!result.ok) {
+        logger?.debug?.({ folder, err: result.error }, "Folder is not a Git repository; using workspace folder as root");
+        return { repoRoot: folder, isGitRepo: false };
+    }
+    const repoRoot = result.stdout.trim();
+    if (!repoRoot) {
+        return { repoRoot: folder, isGitRepo: false };
+    }
+    return { repoRoot, isGitRepo: true };
+}
+export async function isGitAvailable(folder) {
+    const result = await runGit(["--version"], folder);
+    return result.ok || !isGitUnavailableResult(result);
+}
+function parseWorktreePorcelain(output) {
+    const records = [];
+    const lines = output.split(/\r?\n/);
+    let current = {};
+    const flush = () => {
+        if (current.worktree) {
+            records.push({ worktree: current.worktree, branch: current.branch });
+        }
+        current = {};
+    };
+    for (const line of lines) {
+        const trimmed = line.trim();
+        if (!trimmed) {
+            flush();
+            continue;
+        }
+        const [key, ...rest] = trimmed.split(" ");
+        const value = rest.join(" ").trim();
+        if (key === "worktree") {
+            current.worktree = value;
+        }
+        else if (key === "branch") {
+            // branch is like refs/heads/foo
+            current.branch = value.replace(/^refs\/heads\//, "");
+        }
+        else if (key === "HEAD") {
+            current.head = value;
+        }
+        else if (key === "detached") {
+            current.detached = true;
+        }
+    }
+    flush();
+    return records;
+}
+export async function listWorktrees(params) {
+    const { repoRoot, workspaceFolder, logger } = params;
+    const result = await runGit(["worktree", "list", "--porcelain"], workspaceFolder);
+    if (!result.ok) {
+        const rootDescriptor = { slug: "root", directory: repoRoot, kind: "root" };
+        logger?.debug?.({ repoRoot, err: result.error }, "Failed to list git worktrees; returning root only");
+        return [rootDescriptor];
+    }
+    const records = parseWorktreePorcelain(result.stdout);
+    const rootRecord = records.find((record) => path.resolve(record.worktree) === path.resolve(repoRoot));
+    const rootDescriptor = {
+        slug: "root",
+        directory: repoRoot,
+        kind: "root",
+        branch: rootRecord?.branch,
+    };
+    const worktrees = [rootDescriptor];
+    const seen = new Set(["root"]);
+    const normalizeSlug = (record) => {
+        const branch = (record.branch ?? "").trim();
+        if (branch) {
+            return branch;
+        }
+        const head = (record.head ?? "").trim();
+        if (head && /^[0-9a-f]{7,40}$/i.test(head)) {
+            return `detached-${head.slice(0, 7)}`;
+        }
+        // Fallback: stable-ish identifier derived from directory basename.
+        const base = path.basename(record.worktree || "");
+        return base ? `worktree-${base}` : "worktree";
+    };
+    for (const record of records) {
+        const abs = record.worktree;
+        if (!abs || typeof abs !== "string")
+            continue;
+        // Skip the root record (we always expose it as slug="root").
+        if (path.resolve(abs) === path.resolve(repoRoot)) {
+            continue;
+        }
+        const slug = normalizeSlug(record);
+        if (!slug || slug === "root") {
+            continue;
+        }
+        if (seen.has(slug)) {
+            continue;
+        }
+        seen.add(slug);
+        worktrees.push({ slug, directory: abs, kind: "worktree", branch: record.branch });
+    }
+    return worktrees;
+}
+export function isValidWorktreeSlug(slug) {
+    if (!slug)
+        return false;
+    const trimmed = slug.trim();
+    if (!trimmed)
+        return false;
+    if (trimmed.length > 200)
+        return false;
+    // Disallow control characters; allow branch-like slugs including '/'.
+    if (/[\x00-\x1F\x7F]/.test(trimmed))
+        return false;
+    return true;
+}
+export async function createManagedWorktree(params) {
+    const { repoRoot, workspaceFolder, logger } = params;
+    const branch = params.slug.trim();
+    if (!branch || branch === "root" || !isValidWorktreeSlug(branch)) {
+        throw new Error("Invalid worktree slug");
+    }
+    const sanitizeDirName = (input) => {
+        const normalized = input
+            .trim()
+            .replace(/[\\/]+/g, "-")
+            .replace(/\s+/g, "-")
+            .replace(/[^a-zA-Z0-9_.-]+/g, "-")
+            .replace(/-{2,}/g, "-")
+            .replace(/^-+|-+$/g, "");
+        return normalized || "worktree";
+    };
+    const worktreesDir = path.join(repoRoot, ".embeddedcowork", "worktrees");
+    const targetDir = path.join(worktreesDir, sanitizeDirName(branch));
+    await fsp.mkdir(worktreesDir, { recursive: true });
+    try {
+        const stat = await fsp.stat(targetDir);
+        if (stat.isDirectory()) {
+            throw new Error("Worktree directory already exists");
+        }
+    }
+    catch (error) {
+        const code = error.code;
+        if (code !== "ENOENT") {
+            throw error;
+        }
+    }
+    logger?.debug?.({ slug: branch, branch, targetDir }, "Creating managed git worktree");
+    // Prefer creating a new branch from HEAD.
+    const first = await runGit(["worktree", "add", "-b", branch, targetDir, "HEAD"], workspaceFolder);
+    if (first.ok) {
+        return { slug: branch, directory: targetDir, branch };
+    }
+    const message = first.stderr?.toLowerCase() ?? first.error.message.toLowerCase();
+    if (message.includes("already exists")) {
+        // If the branch already exists, add worktree for that branch.
+        const second = await runGit(["worktree", "add", targetDir, branch], workspaceFolder);
+        if (second.ok) {
+            return { slug: branch, directory: targetDir, branch };
+        }
+        throw second.error;
+    }
+    throw first.error;
+}
+export async function removeWorktree(params) {
+    const { workspaceFolder, logger } = params;
+    const directory = (params.directory ?? "").trim();
+    if (!directory) {
+        throw new Error("Invalid worktree directory");
+    }
+    logger?.debug?.({ directory, force: Boolean(params.force) }, "Removing git worktree");
+    const args = ["worktree", "remove"];
+    if (params.force) {
+        args.push("--force");
+    }
+    args.push(directory);
+    const result = await runGit(args, workspaceFolder);
+    if (!result.ok) {
+        throw result.error;
+    }
+    // Best-effort cleanup of stale metadata.
+    await runGit(["worktree", "prune"], workspaceFolder).catch(() => undefined);
+}