@zjex/git-workflow 0.2.5 → 0.2.7

package/src/ai-service.ts

@@ -0,0 +1,350 @@
+import { execOutput } from "./utils.js";
+import type { GwConfig } from "./config.js";
+
+interface AIProvider {
+  name: string;
+  endpoint: string;
+  defaultModel: string;
+  free: boolean;
+  needsKey: boolean;
+}
+
+const AI_PROVIDERS: Record<string, AIProvider> = {
+  github: {
+    name: "GitHub Models",
+    endpoint: "https://models.github.ai/inference/chat/completions",
+    defaultModel: "gpt-4o-mini",
+    free: true,
+    needsKey: true,
+  },
+  groq: {
+    name: "Groq",
+    endpoint: "https://api.groq.com/openai/v1/chat/completions",
+    defaultModel: "llama-3.1-8b-instant",
+    free: true,
+    needsKey: true,
+  },
+  openai: {
+    name: "OpenAI",
+    endpoint: "https://api.openai.com/v1/chat/completions",
+    defaultModel: "gpt-4o-mini",
+    free: false,
+    needsKey: true,
+  },
+  claude: {
+    name: "Claude",
+    endpoint: "https://api.anthropic.com/v1/messages",
+    defaultModel: "claude-3-haiku-20240307",
+    free: false,
+    needsKey: true,
+  },
+  ollama: {
+    name: "Ollama",
+    endpoint: "http://localhost:11434/api/generate",
+    defaultModel: "qwen2.5-coder:7b",
+    free: true,
+    needsKey: false,
+  },
+};
+
+/**
+ * 获取 git diff 内容
+ */
+function getGitDiff(): string {
+  try {
+    // 获取已暂存的更改
+    const diff = execOutput("git diff --cached");
+    if (!diff) {
+      // 如果没有暂存的更改，获取所有更改
+      return execOutput("git diff");
+    }
+    return diff;
+  } catch {
+    return "";
+  }
+}
+
+/**
+ * 构建 AI prompt
+ */
+function buildPrompt(diff: string, language: string): string {
+  const isZh = language === "zh-CN";
+
+  const systemPrompt = isZh
+    ? `你是一个专业的 Git commit message 生成助手。请根据提供的 git diff 生成符合 Conventional Commits 规范的 commit message。
+
+规则：
+1. 格式：<type>(<scope>): <subject>
+2. type 必须是以下之一：feat, fix, docs, style, refactor, perf, test, build, ci, chore, revert
+3. scope 是可选的，表示影响范围
+4. subject 用中文描述，简洁明了，不超过 50 字
+5. 只返回 commit message，不要有其他解释
+
+示例：
+- feat(auth): 添加用户登录功能
+- fix(api): 修复数据获取失败的问题
+- docs(readme): 更新安装说明`
+    : `You are a professional Git commit message generator. Generate a commit message following Conventional Commits specification based on the provided git diff.
+
+Rules:
+1. Format: <type>(<scope>): <subject>
+2. type must be one of: feat, fix, docs, style, refactor, perf, test, build, ci, chore, revert
+3. scope is optional, indicates the affected area
+4. subject should be concise, no more than 50 characters
+5. Return only the commit message, no explanations
+
+Examples:
+- feat(auth): add user login functionality
+- fix(api): resolve data fetching failure
+- docs(readme): update installation guide`;
+
+  const userPrompt = isZh
+    ? `请根据以下 git diff 生成 commit message：\n\n${diff}`
+    : `Generate a commit message based on the following git diff:\n\n${diff}`;
+
+  return `${systemPrompt}\n\n${userPrompt}`;
+}
+
+/**
+ * 调用 GitHub Models API
+ */
+async function callGitHubAPI(
+  prompt: string,
+  apiKey: string,
+  model: string,
+  maxTokens: number
+): Promise<string> {
+  const response = await fetch(AI_PROVIDERS.github.endpoint, {
+    method: "POST",
+    headers: {
+      Authorization: `Bearer ${apiKey}`,
+      "Content-Type": "application/json",
+    },
+    body: JSON.stringify({
+      model,
+      messages: [{ role: "user", content: prompt }],
+      max_tokens: maxTokens,
+      temperature: 0.3,
+    }),
+  });
+
+  if (!response.ok) {
+    const error = await response.text();
+    throw new Error(`GitHub Models API 错误: ${response.status} ${error}`);
+  }
+
+  const data = await response.json();
+  return data.choices[0]?.message?.content?.trim() || "";
+}
+
+/**
+ * 调用 Groq API
+ */
+async function callGroqAPI(
+  prompt: string,
+  apiKey: string,
+  model: string,
+  maxTokens: number
+): Promise<string> {
+  const response = await fetch(AI_PROVIDERS.groq.endpoint, {
+    method: "POST",
+    headers: {
+      Authorization: `Bearer ${apiKey}`,
+      "Content-Type": "application/json",
+    },
+    body: JSON.stringify({
+      model,
+      messages: [{ role: "user", content: prompt }],
+      max_tokens: maxTokens,
+      temperature: 0.3,
+    }),
+  });
+
+  if (!response.ok) {
+    const error = await response.text();
+    throw new Error(`Groq API 错误: ${response.status} ${error}`);
+  }
+
+  const data = await response.json();
+  return data.choices[0]?.message?.content?.trim() || "";
+}
+
+/**
+ * 调用 OpenAI API
+ */
+async function callOpenAIAPI(
+  prompt: string,
+  apiKey: string,
+  model: string,
+  maxTokens: number
+): Promise<string> {
+  const response = await fetch(AI_PROVIDERS.openai.endpoint, {
+    method: "POST",
+    headers: {
+      Authorization: `Bearer ${apiKey}`,
+      "Content-Type": "application/json",
+    },
+    body: JSON.stringify({
+      model,
+      messages: [{ role: "user", content: prompt }],
+      max_tokens: maxTokens,
+      temperature: 0.3,
+    }),
+  });
+
+  if (!response.ok) {
+    const error = await response.text();
+    throw new Error(`OpenAI API 错误: ${response.status} ${error}`);
+  }
+
+  const data = await response.json();
+  return data.choices[0]?.message?.content?.trim() || "";
+}
+
+/**
+ * 调用 Claude API
+ */
+async function callClaudeAPI(
+  prompt: string,
+  apiKey: string,
+  model: string,
+  maxTokens: number
+): Promise<string> {
+  const response = await fetch(AI_PROVIDERS.claude.endpoint, {
+    method: "POST",
+    headers: {
+      "x-api-key": apiKey,
+      "anthropic-version": "2023-06-01",
+      "Content-Type": "application/json",
+    },
+    body: JSON.stringify({
+      model,
+      messages: [{ role: "user", content: prompt }],
+      max_tokens: maxTokens,
+      temperature: 0.3,
+    }),
+  });
+
+  if (!response.ok) {
+    const error = await response.text();
+    throw new Error(`Claude API 错误: ${response.status} ${error}`);
+  }
+
+  const data = await response.json();
+  return data.content[0]?.text?.trim() || "";
+}
+
+/**
+ * 调用 Ollama API
+ */
+async function callOllamaAPI(
+  prompt: string,
+  model: string,
+  maxTokens: number
+): Promise<string> {
+  try {
+    const response = await fetch(AI_PROVIDERS.ollama.endpoint, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({
+        model,
+        prompt,
+        stream: false,
+        options: {
+          num_predict: maxTokens,
+          temperature: 0.3,
+        },
+      }),
+    });
+
+    if (!response.ok) {
+      throw new Error(`Ollama 未运行或模型未安装`);
+    }
+
+    const data = await response.json();
+    return data.response?.trim() || "";
+  } catch (error) {
+    throw new Error(
+      `Ollama 连接失败。请确保：\n1. 已安装 Ollama (https://ollama.com)\n2. 运行 'ollama serve'\n3. 下载模型 'ollama pull ${model}'`
+    );
+  }
+}
+
+/**
+ * 生成 AI commit message
+ */
+export async function generateAICommitMessage(
+  config: GwConfig
+): Promise<string> {
+  const aiConfig = config.aiCommit || {};
+  const provider = aiConfig.provider || "groq";
+  const language = aiConfig.language || "zh-CN";
+  const maxTokens = aiConfig.maxTokens || 200;
+
+  // 获取 git diff
+  const diff = getGitDiff();
+  if (!diff) {
+    throw new Error("没有检测到代码更改");
+  }
+
+  // 限制 diff 长度，避免超过 token 限制
+  const maxDiffLength = 4000;
+  const truncatedDiff =
+    diff.length > maxDiffLength ? diff.slice(0, maxDiffLength) + "\n..." : diff;
+
+  // 构建 prompt
+  const prompt = buildPrompt(truncatedDiff, language);
+
+  // 根据提供商调用对应的 API
+  const providerInfo = AI_PROVIDERS[provider];
+  if (!providerInfo) {
+    throw new Error(`不支持的 AI 提供商: ${provider}`);
+  }
+
+  const model = aiConfig.model || providerInfo.defaultModel;
+
+  // 获取 API key
+  const apiKey = aiConfig.apiKey || "";
+
+  if (providerInfo.needsKey && !apiKey) {
+    throw new Error(
+      `${providerInfo.name} 需要 API key。请运行 'gw init' 配置 AI commit，或在 .gwrc.json 中设置 aiCommit.apiKey`
+    );
+  }
+
+  // 调用 API
+  switch (provider) {
+    case "github":
+      return await callGitHubAPI(prompt, apiKey, model, maxTokens);
+    case "groq":
+      return await callGroqAPI(prompt, apiKey, model, maxTokens);
+    case "openai":
+      return await callOpenAIAPI(prompt, apiKey, model, maxTokens);
+    case "claude":
+      return await callClaudeAPI(prompt, apiKey, model, maxTokens);
+    case "ollama":
+      return await callOllamaAPI(prompt, model, maxTokens);
+    default:
+      throw new Error(`不支持的 AI 提供商: ${provider}`);
+  }
+}
+
+/**
+ * 检查 AI commit 是否可用
+ */
+export function isAICommitAvailable(config: GwConfig): boolean {
+  const aiConfig = config.aiCommit;
+  if (!aiConfig) return true; // 默认启用
+
+  // 如果明确禁用，返回 false
+  if (aiConfig.enabled === false) return false;
+
+  return true;
+}
+
+/**
+ * 获取 AI 提供商信息
+ */
+export function getProviderInfo(provider: string): AIProvider | null {
+  return AI_PROVIDERS[provider] || null;
+}

package/src/commands/commit.ts

@@ -1,8 +1,9 @@
 import { execSync } from "child_process";
-import { select, input, confirm, checkbox } from "@inquirer/prompts";
+import { select, input, checkbox } from "@inquirer/prompts";
 import ora from "ora";
 import { colors, theme, execOutput, divider } from "../utils.js";
 import { getConfig } from "../config.js";
+import { generateAICommitMessage, isAICommitAvailable } from "../ai-service.js";
 
 // Conventional Commits 类型 + Gitmoji
 const DEFAULT_COMMIT_TYPES = [
@@ -140,6 +141,119 @@ export async function commit(): Promise<void> {
     divider();
   }
 
+  // 询问用户选择手动还是 AI 生成
+  const aiAvailable = isAICommitAvailable(config);
+  let commitMode: "ai" | "manual" = "manual";
+
+  if (aiAvailable) {
+    commitMode = await select({
+      message: "选择 commit 方式:",
+      choices: [
+        {
+          name: "🤖 AI 自动生成 commit message",
+          value: "ai",
+          description: "使用 AI 分析代码变更自动生成",
+        },
+        {
+          name: "✍️  手动编写 commit message",
+          value: "manual",
+          description: "传统的交互式输入方式",
+        },
+      ],
+      theme,
+    });
+  }
+
+  let message: string;
+
+  if (commitMode === "ai") {
+    // AI 生成模式
+    const spinner = ora("AI 正在分析代码变更...").start();
+
+    try {
+      const aiMessage = await generateAICommitMessage(config);
+      spinner.succeed("AI 生成完成");
+
+      console.log("");
+      console.log("AI 生成的 commit message:");
+      console.log(colors.green(aiMessage));
+      divider();
+
+      const useAI = await select({
+        message: "使用这个 commit message?",
+        choices: [
+          { name: "✅ 使用", value: true },
+          { name: "❌ 不使用，切换到手动模式", value: false },
+        ],
+        theme,
+      });
+
+      if (useAI) {
+        message = aiMessage;
+      } else {
+        spinner.info("切换到手动模式");
+        commitMode = "manual";
+      }
+    } catch (error) {
+      spinner.fail("AI 生成失败");
+      console.log(
+        colors.red(error instanceof Error ? error.message : String(error))
+      );
+      console.log(colors.yellow("\n切换到手动模式..."));
+      divider();
+      commitMode = "manual";
+    }
+  }
+
+  if (commitMode === "manual") {
+    // 手动输入模式（原有逻辑）
+    message = await buildManualCommitMessage(config);
+  }
+
+  divider();
+  console.log("提交信息预览:");
+  console.log(colors.green(message));
+  divider();
+
+  const shouldCommit = await select({
+    message: "确认提交?",
+    choices: [
+      { name: "✅ 确认提交", value: true },
+      { name: "❌ 取消", value: false },
+    ],
+    theme,
+  });
+
+  if (!shouldCommit) {
+    console.log(colors.yellow("已取消"));
+    return;
+  }
+
+  const spinner = ora("正在提交...").start();
+
+  try {
+    // 使用 -m 参数，需要转义引号
+    const escapedMessage = message.replace(/"/g, '\\"');
+    execSync(`git commit -m "${escapedMessage}"`, { stdio: "pipe" });
+    spinner.succeed("提交成功");
+
+    // 显示提交信息
+    const commitHash = execOutput("git rev-parse --short HEAD");
+    console.log(colors.dim(`commit: ${commitHash}`));
+  } catch (error) {
+    spinner.fail("提交失败");
+    if (error instanceof Error) {
+      console.log(colors.red(error.message));
+    }
+  }
+}
+
+/**
+ * 手动构建 commit message
+ */
+async function buildManualCommitMessage(
+  config: ReturnType<typeof getConfig>
+): Promise<string> {
   // 获取提交类型（支持自定义 emoji）
   const commitTypes = getCommitTypes(config);
 
@@ -177,9 +291,12 @@ export async function commit(): Promise<void> {
   });
 
   // 是否有破坏性变更
-  const hasBreaking = await confirm({
+  const hasBreaking = await select({
     message: "是否包含破坏性变更 (BREAKING CHANGE)?",
-    default: false,
+    choices: [
+      { name: "否", value: false },
+      { name: "是", value: true },
+    ],
     theme,
   });
 
@@ -227,37 +344,5 @@ export async function commit(): Promise<void> {
     }
   }
 
-  divider();
-  console.log("提交信息预览:");
-  console.log(colors.green(message));
-  divider();
-
-  const shouldCommit = await confirm({
-    message: "确认提交?",
-    default: true,
-    theme,
-  });
-
-  if (!shouldCommit) {
-    console.log(colors.yellow("已取消"));
-    return;
-  }
-
-  const spinner = ora("正在提交...").start();
-
-  try {
-    // 使用 -m 参数，需要转义引号
-    const escapedMessage = message.replace(/"/g, '\\"');
-    execSync(`git commit -m "${escapedMessage}"`, { stdio: "pipe" });
-    spinner.succeed("提交成功");
-
-    // 显示提交信息
-    const commitHash = execOutput("git rev-parse --short HEAD");
-    console.log(colors.dim(`commit: ${commitHash}`));
-  } catch (error) {
-    spinner.fail("提交失败");
-    if (error instanceof Error) {
-      console.log(colors.red(error.message));
-    }
-  }
+  return message;
 }