@ridit/lens 0.2.4 → 0.2.6

package/src/commands/commit.tsx

@@ -0,0 +1,686 @@
+import React, { useState, useEffect } from "react";
+import { Box, Text, useInput } from "ink";
+import TextInput from "ink-text-input";
+import { execSync } from "child_process";
+import { existsSync } from "fs";
+import path from "path";
+import figures from "figures";
+import { ACCENT, GREEN, RED, CYAN } from "../colors";
+import { ProviderPicker } from "../components/provider/ProviderPicker";
+import { callChat } from "../utils/chat";
+import { useThinkingPhrase } from "../utils/thinking";
+import type { Provider } from "../types/config";
+import type { Message } from "../types/chat";
+
+function gitRun(cmd: string, cwd: string): { ok: boolean; out: string } {
+  try {
+    const out = execSync(cmd, {
+      cwd,
+      encoding: "utf-8",
+      stdio: ["pipe", "pipe", "pipe"],
+      timeout: 30_000,
+    }).trim();
+    return { ok: true, out };
+  } catch (e: any) {
+    const msg =
+      [e.stdout, e.stderr].filter(Boolean).join("\n").trim() || e.message;
+    return { ok: false, out: msg };
+  }
+}
+
+function gitCommit(message: string, cwd: string): { ok: boolean; out: string } {
+  const { execFileSync } =
+    require("child_process") as typeof import("child_process");
+
+  const paragraphs = message
+    .split(/\n\n+/)
+    .map((p) => p.trim())
+    .filter(Boolean);
+
+  const mArgs: string[] = [];
+  for (const p of paragraphs) {
+    mArgs.push("-m", p);
+  }
+
+  try {
+    const out = execFileSync("git", ["commit", ...mArgs], {
+      cwd,
+      encoding: "utf-8",
+      stdio: ["pipe", "pipe", "pipe"],
+      timeout: 30_000,
+    }).trim();
+    return { ok: true, out };
+  } catch (e: any) {
+    const msg =
+      [e.stdout, e.stderr].filter(Boolean).join("\n").trim() || e.message;
+    return { ok: false, out: msg };
+  }
+}
+
+function stageFiles(
+  files: string[],
+  cwd: string,
+): { ok: boolean; out: string } {
+  if (files.length === 0) return gitRun("git add -A", cwd);
+  const paths = files.map((f) => `"${f}"`).join(" ");
+  return gitRun(`git add -- ${paths}`, cwd);
+}
+
+function getStagedDiff(cwd: string): string {
+  return gitRun("git diff --staged", cwd).out;
+}
+
+function getFileDiff(files: string[], cwd: string): string {
+  if (files.length === 0) {
+    const tracked = gitRun("git diff HEAD", cwd).out;
+    const untracked = gitRun("git ls-files --others --exclude-standard", cwd)
+      .out.split("\n")
+      .filter(Boolean)
+      .slice(0, 10)
+      .map((f) => `=== new file: ${f} ===`)
+      .join("\n");
+    return [tracked, untracked].filter(Boolean).join("\n\n");
+  }
+  const paths = files.map((f) => `"${f}"`).join(" ");
+  return gitRun(`git diff HEAD -- ${paths}`, cwd).out;
+}
+
+function hasStagedChanges(cwd: string): boolean {
+  return !gitRun("git diff --staged --quiet", cwd).ok;
+}
+
+function hasAnyChanges(cwd: string): boolean {
+  return gitRun("git status --porcelain", cwd).out.trim().length > 0;
+}
+
+function validateFiles(
+  files: string[],
+  cwd: string,
+): { missing: string[]; valid: string[] } {
+  const missing: string[] = [];
+  const valid: string[] = [];
+  for (const f of files) {
+    const abs = path.isAbsolute(f) ? f : path.join(cwd, f);
+    if (existsSync(abs)) {
+      valid.push(path.relative(cwd, abs).replace(/\\/g, "/"));
+    } else {
+      missing.push(f);
+    }
+  }
+  return { missing, valid };
+}
+
+function detectSplitOpportunity(diff: string): string[] {
+  const fileMatches = [...diff.matchAll(/^diff --git a\/.+ b\/(.+)$/gm)];
+  const files = fileMatches.map((m) => m[1]!);
+  if (files.length <= 3) return [];
+
+  const groups = new Map<string, string[]>();
+  for (const f of files) {
+    const parts = f.split("/");
+    const group =
+      parts[0] === "src" && parts.length > 1 ? parts[1]! : parts[0]!;
+    if (!groups.has(group)) groups.set(group, []);
+    groups.get(group)!.push(f);
+  }
+
+  const meaningful = [...groups.entries()].filter(([, fs]) => fs.length >= 2);
+  return meaningful.length >= 2
+    ? meaningful.map(([g, fs]) => `${g}/ (${fs.length} files)`)
+    : [];
+}
+
+const SYSTEM_PROMPT = `You are an expert at writing conventional commit messages.
+Given a git diff, analyze the changes and write a single commit message.
+
+Rules:
+- Use conventional commits format: type(scope): description
+- Types: feat, fix, refactor, perf, docs, style, test, chore, ci, build
+- First line: max 72 chars, imperative mood (add, fix, update — not added/fixed)
+- Only add a body (bullet points) if the change is complex or touches multiple unrelated areas
+- If the change is small or obvious from the subject line, output ONLY the first line — no body
+- When a body is needed: add a blank line then 2–4 bullets max, only for non-obvious details
+- Bullet format: "- <what changed and why>"
+- Skip bullets that just restate the subject line or describe trivial version bumps
+- Be specific — mention file names, feature names, component names
+- No markdown, no backticks, no code blocks
+- Output ONLY the commit message, nothing else
+
+Examples of good short commits:
+chore: bump version to 0.1.6
+fix(parser): handle null input in parseResponse
+docs: update README installation steps
+
+Examples of when to add a body:
+feat(chat): add persistent memory across sessions
+
+- store memories in ~/.lens/memories per repo
+- inject memory summary into system prompt on load
+- expose /memory commands for manual management`;
+
+async function generateCommitMessage(
+  provider: Provider,
+  diff: string,
+): Promise<string> {
+  const msgs: Message[] = [
+    {
+      role: "user",
+      content: `Write a conventional commit message for this diff:\n\n${diff.slice(0, 8000)}`,
+      type: "text",
+    },
+  ];
+  const raw = await callChat(provider, SYSTEM_PROMPT, msgs);
+  return typeof raw === "string" ? raw.trim() : "chore: update files";
+}
+
+function trunc(s: string, n: number) {
+  return s.length > n ? s.slice(0, n - 1) + "…" : s;
+}
+
+type Phase =
+  | { type: "checking" }
+  | { type: "no-changes" }
+  | { type: "no-staged"; hasUnstaged: boolean; files: string[] }
+  | { type: "staging"; files: string[] }
+  | { type: "generating" }
+  | { type: "preview"; message: string; splitGroups: string[]; diff: string }
+  | { type: "editing"; message: string; diff: string }
+  | { type: "committing"; message: string }
+  | { type: "pushing"; message: string; hash: string }
+  | { type: "done"; message: string; hash: string; pushed: boolean }
+  | { type: "preview-only"; message: string }
+  | { type: "error"; message: string };
+
+function CommitRunner({
+  cwd,
+  provider,
+  files,
+  auto,
+  preview,
+  push,
+  confirm,
+}: {
+  cwd: string;
+  provider: Provider;
+  files: string[];
+  auto: boolean;
+  preview: boolean;
+  push: boolean;
+  confirm: boolean;
+}) {
+  const [phase, setPhase] = useState<Phase>({ type: "checking" });
+  const phraseText = useThinkingPhrase(
+    phase.type === "generating",
+    "commit",
+    2800,
+  );
+
+  useEffect(() => {
+    (async () => {
+      if (!gitRun("git rev-parse --git-dir", cwd).ok) {
+        setPhase({ type: "error", message: "not a git repository" });
+        return;
+      }
+
+      if (files.length > 0) {
+        const { missing, valid } = validateFiles(files, cwd);
+        if (missing.length > 0) {
+          setPhase({
+            type: "error",
+            message: `file${missing.length > 1 ? "s" : ""} not found:\n${missing.map((f) => `  ${f}`).join("\n")}`,
+          });
+          return;
+        }
+        setPhase({ type: "staging", files: valid });
+        const r = stageFiles(valid, cwd);
+        if (!r.ok) {
+          setPhase({ type: "error", message: `staging failed: ${r.out}` });
+          return;
+        }
+      } else if (auto) {
+        if (!hasAnyChanges(cwd)) {
+          setPhase({ type: "no-changes" });
+          return;
+        }
+        setPhase({ type: "staging", files: [] });
+        gitRun("git add -A", cwd);
+      }
+
+      if (!hasStagedChanges(cwd)) {
+        const unstaged = hasAnyChanges(cwd);
+        setPhase({ type: "no-staged", hasUnstaged: unstaged, files });
+        return;
+      }
+
+      const diff = getStagedDiff(cwd) || getFileDiff(files, cwd);
+      if (!diff.trim()) {
+        setPhase({ type: "no-changes" });
+        return;
+      }
+
+      setPhase({ type: "generating" });
+
+      const commitAndMaybePush = (message: string) => {
+        setPhase({ type: "committing", message });
+        const r = gitCommit(message, cwd);
+        if (!r.ok) {
+          setPhase({ type: "error", message: r.out });
+          return false;
+        }
+        const hash = gitRun("git rev-parse --short HEAD", cwd).out || "?";
+        if (push) {
+          setPhase({ type: "pushing", message, hash });
+          const pr = gitRun("git push", cwd);
+          if (!pr.ok) {
+            setPhase({ type: "error", message: `push failed: ${pr.out}` });
+            return false;
+          }
+        }
+        setPhase({ type: "done", message, hash, pushed: push });
+        return true;
+      };
+
+      try {
+        const message = await generateCommitMessage(provider, diff);
+        const splitGroups = detectSplitOpportunity(diff);
+
+        if (preview) {
+          setPhase({ type: "preview-only", message });
+          return;
+        }
+
+        if (auto && files.length === 0 && !confirm) {
+          commitAndMaybePush(message);
+          return;
+        }
+
+        setPhase({ type: "preview", message, splitGroups, diff });
+      } catch (e: any) {
+        setPhase({
+          type: "error",
+          message: `AI error: ${e.message ?? String(e)}`,
+        });
+      }
+    })();
+  }, []);
+
+  useInput((inp, key) => {
+    if (phase.type === "preview") {
+      if (inp === "y" || inp === "Y" || key.return) {
+        const message = phase.message;
+        setPhase({ type: "committing", message });
+        const r = gitCommit(message, cwd);
+        if (!r.ok) {
+          setPhase({ type: "error", message: r.out });
+          return;
+        }
+        const hash = gitRun("git rev-parse --short HEAD", cwd).out || "?";
+        if (push) {
+          setPhase({ type: "pushing", message, hash });
+          const pr = gitRun("git push", cwd);
+          if (!pr.ok) {
+            setPhase({ type: "error", message: `push failed: ${pr.out}` });
+            return;
+          }
+        }
+        setPhase({ type: "done", message, hash, pushed: push });
+        return;
+      }
+      if (inp === "e" || inp === "E") {
+        setPhase({ type: "editing", message: phase.message, diff: phase.diff });
+        return;
+      }
+      if (inp === "n" || inp === "N" || key.escape) {
+        process.exit(0);
+      }
+    }
+
+    if (phase.type === "editing" && key.escape) {
+      setPhase((prev) =>
+        prev.type === "editing"
+          ? {
+              type: "preview",
+              message: prev.message,
+              splitGroups: [],
+              diff: prev.diff,
+            }
+          : prev,
+      );
+    }
+
+    if (
+      (phase.type === "done" ||
+        phase.type === "no-changes" ||
+        phase.type === "no-staged" ||
+        phase.type === "preview-only" ||
+        phase.type === "error") &&
+      (key.return || key.escape || inp === "q")
+    ) {
+      process.exit(0);
+    }
+  });
+
+  const w = process.stdout.columns ?? 80;
+  const div = "─".repeat(w);
+
+  return (
+    <Box flexDirection="column" paddingY={1}>
+      <Box gap={2} marginBottom={1}>
+        <Text color={ACCENT} bold>
+          ◈ COMMIT
+        </Text>
+        <Text color="gray" dimColor>
+          {cwd}
+        </Text>
+        {files.length > 0 && (
+          <Text color={CYAN} dimColor>
+            {files.length} file{files.length !== 1 ? "s" : ""}
+          </Text>
+        )}
+      </Box>
+
+      {files.length > 0 && (
+        <Box flexDirection="column" marginBottom={1}>
+          {files.map((f, i) => (
+            <Box key={i} gap={1}>
+              <Text color="gray" dimColor>
+                {"  ·"}
+              </Text>
+              <Text color="white">{f}</Text>
+            </Box>
+          ))}
+        </Box>
+      )}
+
+      <Text color="gray" dimColor>
+        {div}
+      </Text>
+
+      {phase.type === "checking" && (
+        <Box gap={1} marginTop={1}>
+          <Text color={ACCENT}>*</Text>
+          <Text color="gray" dimColor>
+            checking changes…
+          </Text>
+        </Box>
+      )}
+
+      {phase.type === "staging" && (
+        <Box gap={1} marginTop={1}>
+          <Text color={ACCENT}>*</Text>
+          <Text color="gray" dimColor>
+            {phase.files.length > 0
+              ? `staging ${phase.files.length} file${phase.files.length !== 1 ? "s" : ""}…`
+              : "staging all changes…"}
+          </Text>
+        </Box>
+      )}
+
+      {phase.type === "no-changes" && (
+        <Box flexDirection="column" marginTop={1} gap={1}>
+          <Box gap={1}>
+            <Text color="yellow">{figures.warning}</Text>
+            <Text color="white">nothing to commit — working tree is clean</Text>
+          </Box>
+        </Box>
+      )}
+
+      {phase.type === "no-staged" && (
+        <Box flexDirection="column" marginTop={1} gap={1}>
+          <Box gap={1}>
+            <Text color="yellow">{figures.warning}</Text>
+            <Text color="white">no staged changes found</Text>
+          </Box>
+          {phase.hasUnstaged && (
+            <Box flexDirection="column" marginLeft={2} gap={1}>
+              <Text color="gray" dimColor>
+                you have unstaged changes. try:
+              </Text>
+              {phase.files.length > 0 ? (
+                <Text color="gray" dimColor>
+                  {"  "}
+                  <Text color={ACCENT}>
+                    lens commit {phase.files.join(" ")}
+                  </Text>
+                  {"  "}(stages and commits those files)
+                </Text>
+              ) : (
+                <Text color="gray" dimColor>
+                  {"  "}
+                  <Text color={ACCENT}>git add {"<files>"}</Text>
+                  {"  "}or{"  "}
+                  <Text color={ACCENT}>lens commit --auto</Text>
+                </Text>
+              )}
+            </Box>
+          )}
+        </Box>
+      )}
+
+      {phase.type === "generating" && (
+        <Box gap={1} marginTop={1}>
+          <Text color={ACCENT}>●</Text>
+          <Text color="gray">{phraseText}</Text>
+        </Box>
+      )}
+
+      {phase.type === "preview" && (
+        <Box flexDirection="column" marginTop={1} gap={1}>
+          <Text color={ACCENT} bold>
+            GENERATED MESSAGE
+          </Text>
+          <Box
+            flexDirection="column"
+            marginLeft={2}
+            marginTop={1}
+            marginBottom={1}
+          >
+            {phase.message.split("\n").map((line, i) => (
+              <Text key={i} color={i === 0 ? "white" : "gray"} bold={i === 0}>
+                {line || " "}
+              </Text>
+            ))}
+          </Box>
+          {phase.splitGroups.length > 0 && (
+            <Box flexDirection="column" marginLeft={2} marginBottom={1}>
+              <Text color="yellow" dimColor>
+                ⚡ large diff — consider splitting into{" "}
+                {phase.splitGroups.length} commits:
+              </Text>
+              {phase.splitGroups.map((g, i) => (
+                <Text key={i} color="gray" dimColor>
+                  {"  · "}
+                  {g}
+                </Text>
+              ))}
+            </Box>
+          )}
+          <Text color="gray" dimColor>
+            {div}
+          </Text>
+          <Box gap={3} marginTop={1}>
+            <Text color={GREEN}>y/enter commit</Text>
+            <Text color={CYAN}>e edit</Text>
+            <Text color="gray" dimColor>
+              n/esc cancel
+            </Text>
+          </Box>
+        </Box>
+      )}
+
+      {phase.type === "editing" && (
+        <Box flexDirection="column" marginTop={1} gap={1}>
+          <Text color={ACCENT} bold>
+            EDIT MESSAGE
+          </Text>
+          <Box marginLeft={2} marginTop={1} flexDirection="column" gap={1}>
+            <TextInput
+              value={phase.message}
+              onChange={(msg) =>
+                setPhase((prev) =>
+                  prev.type === "editing" ? { ...prev, message: msg } : prev,
+                )
+              }
+              onSubmit={(msg) =>
+                setPhase((prev) =>
+                  prev.type === "editing"
+                    ? {
+                        type: "preview",
+                        message: msg,
+                        splitGroups: [],
+                        diff: prev.diff,
+                      }
+                    : prev,
+                )
+              }
+            />
+            <Text color="gray" dimColor>
+              enter confirm · esc back
+            </Text>
+          </Box>
+        </Box>
+      )}
+
+      {phase.type === "committing" && (
+        <Box gap={1} marginTop={1}>
+          <Text color={ACCENT}>*</Text>
+          <Text color="gray" dimColor>
+            committing…
+          </Text>
+        </Box>
+      )}
+
+      {phase.type === "pushing" && (
+        <Box flexDirection="column" marginTop={1} gap={1}>
+          <Box gap={2}>
+            <Text color={GREEN}>{figures.tick}</Text>
+            <Text color={ACCENT}>{phase.hash}</Text>
+            <Text color="white">
+              {trunc(phase.message.split("\n")[0]!, 65)}
+            </Text>
+          </Box>
+          <Box gap={1} marginLeft={2}>
+            <Text color={ACCENT}>*</Text>
+            <Text color="gray" dimColor>
+              pushing…
+            </Text>
+          </Box>
+        </Box>
+      )}
+
+      {phase.type === "done" && (
+        <Box flexDirection="column" marginTop={1} gap={1}>
+          <Box gap={2}>
+            <Text color={GREEN}>{figures.tick}</Text>
+            <Text color={ACCENT}>{phase.hash}</Text>
+            <Text color="white" bold>
+              {trunc(phase.message.split("\n")[0]!, 65)}
+            </Text>
+          </Box>
+          {phase.message
+            .split("\n")
+            .slice(2)
+            .filter(Boolean)
+            .map((line, i) => (
+              <Text key={i} color="gray" dimColor>
+                {line}
+              </Text>
+            ))}
+          {phase.pushed && (
+            <Box gap={2} marginTop={1}>
+              <Text color={GREEN}>{figures.tick}</Text>
+              <Text color="gray" dimColor>
+                pushed to remote
+              </Text>
+            </Box>
+          )}
+          <Text color="gray" dimColor>
+            press any key to exit
+          </Text>
+        </Box>
+      )}
+
+      {phase.type === "preview-only" && (
+        <Box flexDirection="column" marginTop={1} gap={1}>
+          <Text color={ACCENT} bold>
+            GENERATED MESSAGE
+          </Text>
+          <Box flexDirection="column" marginLeft={2} marginTop={1}>
+            {phase.message.split("\n").map((line, i) => (
+              <Text key={i} color={i === 0 ? "white" : "gray"} bold={i === 0}>
+                {line || " "}
+              </Text>
+            ))}
+          </Box>
+          <Text color="gray" dimColor>
+            (preview only — not committed)
+          </Text>
+        </Box>
+      )}
+
+      {phase.type === "error" && (
+        <Box flexDirection="column" marginTop={1} gap={1}>
+          <Box gap={1}>
+            <Text color={RED}>{figures.cross}</Text>
+            <Text color="white">{phase.message.split("\n")[0]}</Text>
+          </Box>
+          {phase.message
+            .split("\n")
+            .slice(1)
+            .map((line, i) => (
+              <Text key={i} color="gray" dimColor>
+                {line}
+              </Text>
+            ))}
+        </Box>
+      )}
+    </Box>
+  );
+}
+
+interface Props {
+  path: string;
+  files: string[];
+  auto: boolean;
+  preview: boolean;
+  push: boolean;
+  confirm: boolean;
+}
+
+export function CommitCommand({
+  path: inputPath,
+  files,
+  auto,
+  preview,
+  push,
+  confirm,
+}: Props) {
+  const cwd = path.resolve(inputPath);
+  const [provider, setProvider] = useState<Provider | null>(null);
+
+  if (!existsSync(cwd)) {
+    return (
+      <Box marginTop={1}>
+        <Text color={RED}>
+          {figures.cross} path not found: {cwd}
+        </Text>
+      </Box>
+    );
+  }
+
+  if (!provider) {
+    return <ProviderPicker onDone={setProvider} />;
+  }
+
+  return (
+    <CommitRunner
+      cwd={cwd}
+      provider={provider}
+      files={files}
+      auto={auto}
+      preview={preview}
+      push={push}
+      confirm={confirm}
+    />
+  );
+}