goodiffer 1.0.0 → 1.0.1

package/src/services/git.js

@@ -1,136 +1,235 @@
-import { simpleGit } from 'simple-git';
-import logger from '../utils/logger.js';
+import simpleGit from 'simple-git';
+import path from 'path';
 
-const git = simpleGit();
+export class GitService {
+  constructor(basePath = process.cwd()) {
+    this.git = simpleGit(basePath);
+    this.basePath = basePath;
+  }
 
-export async function isGitRepo() {
-  try {
-    await git.revparse(['--git-dir']);
-    return true;
-  } catch {
-    return false;
+  async isGitRepo() {
+    try {
+      await this.git.status();
+      return true;
+    } catch {
+      return false;
+    }
   }
-}
 
-export async function getLastCommitInfo() {
-  try {
-    const log = await git.log({ maxCount: 1 });
-    if (log.latest) {
-      return {
-        sha: log.latest.hash,
-        message: log.latest.message,
-        author: log.latest.author_name,
-        date: log.latest.date
-      };
+  async getLastCommitInfo() {
+    const log = await this.git.log({ maxCount: 1 });
+    if (!log.latest) {
+      throw new Error('无法获取 commit 信息');
     }
-    return null;
-  } catch (error) {
-    logger.error(`获取 commit 信息失败: ${error.message}`);
-    return null;
+    return {
+      sha: log.latest.hash,
+      message: log.latest.message
+    };
   }
-}
 
-export async function getCommitInfo(sha) {
-  try {
-    const log = await git.log({ from: sha, to: sha, maxCount: 1 });
-    if (log.latest) {
+  async getCommitInfo(sha) {
+    // 使用 raw 命令获取特定 commit 信息
+    try {
+      const result = await this.git.raw(['log', '-1', '--format=%H%n%s', sha]);
+      const lines = result.trim().split('\n');
+      if (lines.length < 2) {
+        throw new Error(`无法获取 commit ${sha} 的信息`);
+      }
       return {
-        sha: log.latest.hash,
-        message: log.latest.message,
-        author: log.latest.author_name,
-        date: log.latest.date
+        sha: lines[0],
+        message: lines.slice(1).join('\n')
       };
+    } catch (error) {
+      throw new Error(`无法获取 commit ${sha} 的信息`);
     }
-    return null;
-  } catch (error) {
-    logger.error(`获取 commit ${sha} 信息失败: ${error.message}`);
-    return null;
   }
-}
 
-export async function getDiff(options = {}) {
-  try {
-    const { staged, commit, from, to } = options;
+  async getLastCommitDiff() {
+    const diff = await this.git.diff(['HEAD~1', 'HEAD']);
+    return diff;
+  }
 
-    if (staged) {
-      // 暂存区 diff
-      return await git.diff(['--cached']);
-    }
+  async getCommitDiff(sha) {
+    const diff = await this.git.diff([`${sha}~1`, sha]);
+    return diff;
+  }
 
-    if (commit) {
-      // 指定 commit 的 diff
-      return await git.diff([`${commit}^`, commit]);
-    }
+  async getStagedDiff() {
+    const diff = await this.git.diff(['--cached']);
+    return diff;
+  }
 
-    if (from && to) {
-      // commit 范围
-      return await git.diff([from, to]);
-    }
+  async getRangeDiff(from, to) {
+    const diff = await this.git.diff([from, to]);
+    return diff;
+  }
+
+  async getChangedFiles() {
+    const status = await this.git.status();
+    return {
+      staged: status.staged,
+      modified: status.modified,
+      created: status.created,
+      deleted: status.deleted
+    };
+  }
 
-    // 默认: 最近一次 commit 的 diff
-    const log = await git.log({ maxCount: 1 });
-    if (log.latest) {
-      return await git.diff([`${log.latest.hash}^`, log.latest.hash]);
+  // 获取 commit 的 author 信息
+  async getCommitAuthor(sha) {
+    if (!sha || sha === 'staged') {
+      // 对于暂存区，使用当前用户配置
+      const config = await this.git.raw(['config', 'user.name']);
+      const email = await this.git.raw(['config', 'user.email']);
+      return {
+        name: config.trim(),
+        email: email.trim(),
+        date: new Date().toISOString()
+      };
     }
 
-    return '';
-  } catch (error) {
-    // 如果是首次 commit，没有父 commit
-    if (error.message.includes('unknown revision')) {
-      try {
-        // 尝试获取首次 commit 的全部内容
-        return await git.diff(['--cached', 'HEAD']);
-      } catch {
-        return await git.diff(['HEAD']);
+    // 使用 raw 命令获取特定 commit 的作者信息
+    try {
+      const result = await this.git.raw([
+        'log', '-1',
+        '--format=%an%n%ae%n%aI',
+        sha
+      ]);
+      const lines = result.trim().split('\n');
+      if (lines.length < 3) {
+        throw new Error(`无法获取 commit ${sha} 的作者信息`);
       }
+      return {
+        name: lines[0],
+        email: lines[1],
+        date: lines[2]
+      };
+    } catch (error) {
+      throw new Error(`无法获取 commit ${sha} 的作者信息`);
     }
-    logger.error(`获取 diff 失败: ${error.message}`);
-    return '';
   }
-}
 
-export async function getChangedFiles(options = {}) {
-  try {
-    const { staged, commit, from, to } = options;
+  // 获取最近 commit 的 author 信息
+  async getLastCommitAuthor() {
+    const log = await this.git.log({ maxCount: 1 });
+    if (!log.latest) {
+      throw new Error('无法获取 commit 作者信息');
+    }
+    return {
+      name: log.latest.author_name,
+      email: log.latest.author_email,
+      date: log.latest.date
+    };
+  }
 
-    if (staged) {
-      const status = await git.status();
-      return status.staged;
+  // 获取项目名称 (从 remote 或目录名)
+  async getProjectName() {
+    try {
+      const remotes = await this.git.getRemotes(true);
+      const origin = remotes.find(r => r.name === 'origin');
+      if (origin && origin.refs && origin.refs.fetch) {
+        // 从 URL 提取项目名
+        const url = origin.refs.fetch;
+        const match = url.match(/\/([^\/]+?)(?:\.git)?$/);
+        if (match) return match[1];
+      }
+    } catch {
+      // 忽略错误
     }
+    // 回退到目录名
+    return path.basename(this.basePath);
+  }
+
+  // 获取 diff 统计
+  async getDiffStats(from, to) {
+    try {
+      let args = [];
+      if (from && to) {
+        args = [from, to];
+      } else if (from) {
+        args = [from];
+      } else {
+        args = ['HEAD~1', 'HEAD'];
+      }
 
-    if (commit) {
-      const result = await git.diff(['--name-only', `${commit}^`, commit]);
-      return result.trim().split('\n').filter(Boolean);
+      const summary = await this.git.diffSummary(args);
+      return {
+        filesChanged: summary.changed || 0,
+        insertions: summary.insertions || 0,
+        deletions: summary.deletions || 0
+      };
+    } catch {
+      return {
+        filesChanged: 0,
+        insertions: 0,
+        deletions: 0
+      };
     }
+  }
 
-    if (from && to) {
-      const result = await git.diff(['--name-only', from, to]);
-      return result.trim().split('\n').filter(Boolean);
+  // 获取暂存区的 diff 统计
+  async getStagedDiffStats() {
+    try {
+      const summary = await this.git.diffSummary(['--cached']);
+      return {
+        filesChanged: summary.changed || 0,
+        insertions: summary.insertions || 0,
+        deletions: summary.deletions || 0
+      };
+    } catch {
+      return {
+        filesChanged: 0,
+        insertions: 0,
+        deletions: 0
+      };
     }
+  }
 
-    // 默认: 最近一次 commit
-    const log = await git.log({ maxCount: 1 });
-    if (log.latest) {
-      const result = await git.diff(['--name-only', `${log.latest.hash}^`, log.latest.hash]);
-      return result.trim().split('\n').filter(Boolean);
+  // 获取当前分支
+  async getCurrentBranch() {
+    try {
+      const branch = await this.git.branch();
+      return branch.current;
+    } catch {
+      return 'unknown';
     }
+  }
 
-    return [];
-  } catch (error) {
-    logger.error(`获取变更文件列表失败: ${error.message}`);
-    return [];
+  // 获取最近 n 条 commits (按时间从新到旧排序)
+  async getRecentCommits(count) {
+    const log = await this.git.log({ maxCount: count });
+    return log.all.map(commit => ({
+      sha: commit.hash,
+      message: commit.message,
+      author: {
+        name: commit.author_name,
+        email: commit.author_email,
+        date: commit.date
+      }
+    }));
   }
-}
 
-export async function getStagedDiff() {
-  return getDiff({ staged: true });
+  // 获取指定范围的 commits (从第 start 条到第 end 条，1-based，按时间从新到旧)
+  async getCommitRange(start, end) {
+    // start 和 end 是 1-based 索引
+    // 例如 start=2, end=5 表示第 2、3、4、5 条 commit
+    const skip = start - 1;
+    const count = end - start + 1;
+
+    const log = await this.git.log({
+      maxCount: count,
+      '--skip': skip
+    });
+
+    return log.all.map(commit => ({
+      sha: commit.hash,
+      message: commit.message,
+      author: {
+        name: commit.author_name,
+        email: commit.author_email,
+        date: commit.date
+      }
+    }));
+  }
 }
 
-export default {
-  isGitRepo,
-  getLastCommitInfo,
-  getCommitInfo,
-  getDiff,
-  getChangedFiles,
-  getStagedDiff
-};
+export default GitService;

package/src/services/report-generator.js

@@ -0,0 +1,298 @@
+import fs from 'fs';
+import path from 'path';
+import dayjs from 'dayjs';
+import { getDatabase, getConfigDir } from './database.js';
+import { AIClient } from './ai-client.js';
+import { getConfig } from '../utils/config-store.js';
+import { buildProjectReportPrompt, buildDeveloperReportPrompt } from '../prompts/report-prompt.js';
+
+export class ReportGenerator {
+  constructor() {
+    this.db = getDatabase();
+    this.config = getConfig();
+    this.aiClient = new AIClient(this.config);
+  }
+
+  // 生成项目报告
+  async generateProjectReport(projectName, options = {}, onProgress) {
+    // 获取项目
+    const project = this.db.getProject(projectName);
+    if (!project) {
+      throw new Error(`项目 "${projectName}" 不存在`);
+    }
+
+    // 构建日期范围
+    const dateRange = this.buildDateRange(options);
+
+    // 收集数据
+    if (onProgress) onProgress('正在收集项目数据...');
+    const data = await this.gatherProjectData(project, dateRange);
+
+    // 构建提示词
+    if (onProgress) onProgress('正在生成报告...');
+    const prompt = buildProjectReportPrompt(data);
+
+    // 调用 AI 生成 HTML
+    const html = await this.aiClient.analyze(prompt);
+
+    // 保存报告
+    const outputPath = this.saveReport(html, projectName, options.output);
+
+    return outputPath;
+  }
+
+  // 生成开发者报告
+  async generateDeveloperReport(developerEmail, options = {}, onProgress) {
+    // 获取开发者
+    const developer = this.db.getDeveloper(developerEmail);
+    if (!developer) {
+      throw new Error(`开发者 "${developerEmail}" 不存在`);
+    }
+
+    // 构建日期范围
+    const dateRange = this.buildDateRange(options);
+
+    // 收集数据
+    if (onProgress) onProgress('正在收集开发者数据...');
+    const data = await this.gatherDeveloperData(developer, dateRange);
+
+    // 构建提示词
+    if (onProgress) onProgress('正在生成报告...');
+    const prompt = buildDeveloperReportPrompt(data);
+
+    // 调用 AI 生成 HTML
+    const html = await this.aiClient.analyze(prompt);
+
+    // 保存报告
+    const safeName = developerEmail.replace(/[@.]/g, '_');
+    const outputPath = this.saveReport(html, `developer_${safeName}`, options.output);
+
+    return outputPath;
+  }
+
+  // 构建日期范围
+  buildDateRange(options) {
+    const dateRange = {
+      start: options.since ? dayjs(options.since).format('YYYY-MM-DD') : dayjs().subtract(30, 'day').format('YYYY-MM-DD'),
+      end: options.until ? dayjs(options.until).format('YYYY-MM-DD') : dayjs().format('YYYY-MM-DD')
+    };
+
+    return {
+      ...dateRange,
+      since: options.since ? dayjs(options.since).startOf('day').toISOString() : null,
+      until: options.until ? dayjs(options.until).endOf('day').toISOString() : null
+    };
+  }
+
+  // 收集项目数据
+  async gatherProjectData(project, dateRange) {
+    const filters = {
+      projectId: project.id,
+      since: dateRange.since,
+      until: dateRange.until,
+      limit: 100
+    };
+
+    // 获取统计数据
+    const stats = this.db.getProjectStats(project.id, {
+      since: dateRange.since,
+      until: dateRange.until
+    });
+
+    // 获取开发者统计
+    const developerStatsRaw = this.db.getDeveloperStatsByProject(project.id, {
+      since: dateRange.since,
+      until: dateRange.until
+    });
+
+    // 获取最近的 reviews
+    const recentReviews = this.db.queryReviews({ ...filters, limit: 20 });
+
+    // 处理开发者统计
+    const developerStats = developerStatsRaw.map(dev => {
+      const topIssues = this.db.getDeveloperTopIssues(dev.id, 3);
+      const matchRate = dev.total_reviews > 0
+        ? Math.round((dev.commit_match_count / dev.total_reviews) * 100)
+        : 0;
+
+      return {
+        displayName: dev.display_name,
+        email: dev.git_email,
+        team: dev.team,
+        totalReviews: dev.total_reviews,
+        insertions: dev.total_insertions || 0,
+        deletions: dev.total_deletions || 0,
+        errors: dev.total_errors || 0,
+        warnings: dev.total_warnings || 0,
+        commitMatchRate: matchRate,
+        topIssues: topIssues.map(i => ({
+          level: i.level,
+          file: i.file,
+          description: i.description
+        })),
+        topRisks: [] // 可以扩展添加风险统计
+      };
+    });
+
+    return {
+      project: {
+        name: project.name,
+        path: project.path
+      },
+      dateRange: {
+        start: dateRange.start,
+        end: dateRange.end
+      },
+      stats: {
+        totalReviews: stats.total_reviews || 0,
+        totalInsertions: stats.total_insertions || 0,
+        totalDeletions: stats.total_deletions || 0,
+        totalErrors: stats.total_errors || 0,
+        totalWarnings: stats.total_warnings || 0,
+        totalInfos: stats.total_infos || 0,
+        totalRisks: stats.total_risks || 0
+      },
+      developers: developerStatsRaw.map(d => ({
+        displayName: d.display_name,
+        email: d.git_email
+      })),
+      developerStats,
+      recentReviews: recentReviews.map(r => ({
+        commitSha: r.commit_sha,
+        commitMessage: r.commit_message,
+        commitDate: dayjs(r.commit_date).format('YYYY-MM-DD HH:mm'),
+        developerName: r.developer_name,
+        summary: r.summary,
+        errorCount: r.error_count,
+        warningCount: r.warning_count
+      }))
+    };
+  }
+
+  // 收集开发者数据
+  async gatherDeveloperData(developer, dateRange) {
+    const filters = {
+      developerId: developer.id,
+      since: dateRange.since,
+      until: dateRange.until,
+      limit: 100
+    };
+
+    // 获取统计数据
+    const stats = this.db.getDeveloperStats(developer.id, {
+      since: dateRange.since,
+      until: dateRange.until
+    });
+
+    // 获取 reviews
+    const reviews = this.db.queryReviews(filters);
+
+    // 获取问题分布
+    const issueDistribution = this.db.getIssueTypeDistribution({
+      developerId: developer.id,
+      since: dateRange.since,
+      until: dateRange.until
+    });
+
+    // 获取典型问题
+    const topIssues = this.db.getDeveloperTopIssues(developer.id, 5);
+
+    // 按项目分组统计
+    const projectStats = {};
+    reviews.forEach(r => {
+      if (!projectStats[r.project_name]) {
+        projectStats[r.project_name] = { name: r.project_name, commits: 0 };
+      }
+      projectStats[r.project_name].commits++;
+    });
+
+    // 获取关联风险
+    const topRisks = [];
+    for (const review of reviews.slice(0, 10)) {
+      const risks = this.db.db.prepare('SELECT * FROM association_risks WHERE review_id = ? LIMIT 2').all(review.id);
+      risks.forEach(risk => {
+        topRisks.push({
+          changedFile: risk.changed_file,
+          relatedFiles: risk.related_files,
+          risk: risk.risk
+        });
+      });
+    }
+
+    const matchRate = stats.total_reviews > 0
+      ? Math.round((stats.commit_match_count / stats.total_reviews) * 100)
+      : 0;
+
+    return {
+      developer: {
+        displayName: developer.display_name,
+        email: developer.git_email,
+        team: developer.team
+      },
+      dateRange: {
+        start: dateRange.start,
+        end: dateRange.end
+      },
+      projects: Object.values(projectStats),
+      stats: {
+        totalReviews: stats.total_reviews || 0,
+        totalInsertions: stats.total_insertions || 0,
+        totalDeletions: stats.total_deletions || 0,
+        totalErrors: stats.total_errors || 0,
+        totalWarnings: stats.total_warnings || 0,
+        commitMatchRate: matchRate
+      },
+      issueDistribution: issueDistribution.map(i => ({
+        type: i.type || 'unknown',
+        level: i.level,
+        count: i.count
+      })),
+      topIssues: topIssues.map(i => ({
+        level: i.level,
+        type: i.type,
+        file: i.file,
+        line: i.line,
+        description: i.description,
+        suggestion: i.suggestion
+      })),
+      topRisks: topRisks.slice(0, 5),
+      reviews: reviews.map(r => ({
+        projectName: r.project_name,
+        commitSha: r.commit_sha,
+        commitMessage: r.commit_message,
+        commitDate: dayjs(r.commit_date).format('YYYY-MM-DD HH:mm'),
+        summary: r.summary,
+        errorCount: r.error_count,
+        warningCount: r.warning_count,
+        infoCount: r.info_count
+      }))
+    };
+  }
+
+  // 保存报告
+  saveReport(html, name, customOutput) {
+    let outputPath;
+
+    if (customOutput) {
+      outputPath = path.resolve(customOutput);
+    } else {
+      const reportsDir = path.join(getConfigDir(), 'reports');
+      if (!fs.existsSync(reportsDir)) {
+        fs.mkdirSync(reportsDir, { recursive: true });
+      }
+      const filename = `${name}_${dayjs().format('YYYY-MM-DD_HH-mm')}.html`;
+      outputPath = path.join(reportsDir, filename);
+    }
+
+    // 确保目录存在
+    const dir = path.dirname(outputPath);
+    if (!fs.existsSync(dir)) {
+      fs.mkdirSync(dir, { recursive: true });
+    }
+
+    fs.writeFileSync(outputPath, html, 'utf-8');
+    return outputPath;
+  }
+}
+
+export default ReportGenerator;