Merge pull request #982 from Kyoungwoong/feature/embedded-engine

refac[vibe]: Analysis-Engine 의존성 제거

Author: Kyoungwoong

packages/mcp/package.json

@@ -15,14 +15,18 @@
 	"author": "",
 	"license": "ISC",
 	"dependencies": {
-		"@githru-vscode-ext/analysis-engine": "^0.7.2",
 		"@modelcontextprotocol/sdk": "^1.12.1",
-		"@octokit/rest": "^22.0.0",
+		"@octokit/core": "^4.0.4",
+		"@octokit/plugin-retry": "^3.0.9",
+		"@octokit/plugin-throttling": "^4.1.0",
+		"@octokit/rest": "^19.0.3",
+		"@octokit/types": "^13.5.0",
 		"cors": "^2.8.5",
 		"d3": "^7.8.5",
 		"dayjs": "^1.11.18",
 		"express": "^5.1.0",
-		"puppeteer": "^21.0.0",
+		"puppeteer": "^24.15.0",
+		"glob": "^10.0.0",
 		"react": "^18.2.0",
 		"react-dom": "^18.2.0",
 		"sharp": "^0.34.4",
@@ -31,6 +35,7 @@
 	"devDependencies": {
 		"@smithery/cli": "^1.2.4",
 		"@types/d3": "^7.4.3",
+		"@types/node": "^24.7.1",
 		"@types/react": "^18.2.0",
 		"@types/react-dom": "^18.2.0",
 		"copyfiles": "^2.4.1",

packages/mcp/src/common/parser.ts

@@ -0,0 +1,152 @@
+import { getCommitMessageType } from '../engine/commit.js';
+import type { CommitRaw, DifferenceStatistic } from "./types.js";
+
+interface ParsedRefs {
+  branches: string[];
+  tags: string[];
+}
+
+interface MessageAndDiffStats {
+  message: string;
+  diffStats: DifferenceStatistic;
+}
+
+const EOL_REGEX = /\r?\n/;
+const COMMIT_SEPARATOR = new RegExp(`${EOL_REGEX.source}{4}`);
+const INDENTATION = "    ";
+
+export function splitLogIntoCommits(log: string): string[] {
+  return log.substring(2).split(COMMIT_SEPARATOR);
+}
+
+export function extractCommitData(commit: string) {
+  const [
+    id,
+    parents,
+    refs,
+    authorName,
+    authorEmail,
+    authorDate,
+    committerName,
+    committerEmail,
+    committerDate,
+    ...messageAndDiffStats
+  ] = commit.split(EOL_REGEX);
+
+  return {
+    id,
+    parents,
+    refs,
+    authorName,
+    authorEmail,
+    authorDate,
+    committerName,
+    committerEmail,
+    committerDate,
+    messageAndDiffStats,
+  };
+}
+
+export function extractBranchesAndTags(refs: string): ParsedRefs {
+  if (!refs) return { branches: [], tags: [] };
+
+  const refsArray = refs.replace(" -> ", ", ").split(", ");
+  return refsArray.reduce<ParsedRefs>(
+    (acc, ref) => {
+      if (ref === "") return acc;
+      if (ref.startsWith("tag: ")) {
+        acc.tags.push(ref.replace("tag: ", ""));
+      } else {
+        acc.branches.push(ref);
+      }
+      return acc;
+    },
+    { branches: [], tags: [] }
+  );
+}
+
+export function processDiffStatLine(line: string, diffStats: DifferenceStatistic) {
+  const [insertions, deletions, path] = line.split("\t");
+  const numberedInsertions = insertions === "-" ? 0 : Number(insertions);
+  const numberedDeletions = deletions === "-" ? 0 : Number(deletions);
+
+  diffStats.totalInsertionCount += numberedInsertions;
+  diffStats.totalDeletionCount += numberedDeletions;
+  diffStats.fileDictionary[path] = {
+    insertionCount: numberedInsertions,
+    deletionCount: numberedDeletions,
+  };
+}
+
+export function extractMessageAndDiffStats(messageAndDiffStats: string[]): MessageAndDiffStats {
+  let messageSubject = "";
+  let messageBody = "";
+  const diffStats: DifferenceStatistic = {
+    totalInsertionCount: 0,
+    totalDeletionCount: 0,
+    fileDictionary: {},
+  };
+
+  for (const [idx, line] of messageAndDiffStats.entries()) {
+    if (idx === 0) {
+      messageSubject = line;
+      continue;
+    }
+
+    if (line.startsWith(INDENTATION)) {
+      messageBody += idx === 1 ? line.trim() : `\n${line.trim()}`;
+      continue;
+    }
+
+    if (line === "") continue;
+
+    processDiffStatLine(line, diffStats);
+  }
+
+  const message = messageBody === "" ? messageSubject : `${messageSubject}\n${messageBody}`;
+  return { message, diffStats };
+}
+
+export function createCommitRaw(
+  commitIdx: number,
+  commitData: ReturnType<typeof extractCommitData>,
+  parsedRefs: ParsedRefs,
+  parsedMessage: ReturnType<typeof extractMessageAndDiffStats>
+): CommitRaw {
+  const { id, parents, authorName, authorEmail, authorDate, committerName, committerEmail, committerDate } = commitData;
+  const { branches, tags } = parsedRefs;
+  const { message, diffStats } = parsedMessage;
+
+  return {
+    sequence: commitIdx,
+    id,
+    parents: parents.length === 0 ? [] : parents.split(" "),
+    branches,
+    tags,
+    author: {
+      name: authorName,
+      email: authorEmail,
+    },
+    authorDate: new Date(authorDate),
+    committer: {
+      name: committerName,
+      email: committerEmail,
+    },
+    committerDate: new Date(committerDate),
+    message,
+    commitMessageType: getCommitMessageType(message),
+    differenceStatistic: diffStats,
+  };
+}
+
+export default function getCommitRaws(log: string): CommitRaw[] {
+  if (!log) return [];
+
+  const commits = splitLogIntoCommits(log);
+  return commits.map((commit, idx) => {
+    const commitData = extractCommitData(commit);
+    const parsedRefs = extractBranchesAndTags(commitData.refs);
+    const parsedMessage = extractMessageAndDiffStats(commitData.messageAndDiffStats);
+    return createCommitRaw(idx, commitData, parsedRefs, parsedMessage);
+  });
+}

packages/mcp/src/common/queue.ts

@@ -0,0 +1,30 @@
+export default class Queue<T> {
+  private queue: Array<T> = [];
+
+  private readonly compareFn: (a: T, b: T) => number;
+
+  constructor(compareFn: (a: T, b: T) => number) {
+    this.compareFn = compareFn;
+  }
+
+  push(node: T): void {
+    this.queue.push(node);
+    this.queue.sort(this.compareFn);
+  }
+
+  pop(): T | undefined {
+    return this.queue.shift();
+  }
+
+  isEmpty(): boolean {
+    return this.queue.length === 0;
+  }
+
+  pushFront(node: T): void {
+    this.queue.unshift(node);
+  }
+
+  pushBack(node: T): void {
+    this.queue.push(node);
+  }
+}

packages/mcp/src/common/summary.ts

@@ -0,0 +1,56 @@
+import type { CommitRaw } from "./types.js";
+
+const apiKey = process.env.GEMENI_API_KEY || '';
+const apiUrl = "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash-latest:generateContent?key=";
+
+export async function getSummary(csmNodes: CommitRaw[]) {
+  const commitMessages = csmNodes.map((csmNode) => csmNode.message.split('\n')[0]).join(', ');
+
+  try {
+    const response = await fetch(apiUrl + apiKey, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+      },
+      body: JSON.stringify({
+        contents: [{parts: [{text: `${prompt} \n${commitMessages}`}]}],
+      }),
+    });
+
+    if (!response.ok) {
+      throw new Error(`Error: ${response.status} ${response.statusText}`);
+    }
+
+    const data = await response.json();
+    return data.candidates[0].content.parts[0].text.trim();
+  } catch (error) {
+    console.error("Error fetching summary:", error);
+    return undefined;
+  }
+}
+
+const prompt = `Proceed with the task of summarising the contents of the commit message provided.
+
+Procedure:
+1. Separate the commits based on , .
+2. Extract only the commits given, excluding the merge commits.
+3. Summarise the commits based on the most common words. Keep the shape of your commit message.
+
+Example Merge commits:
+- Merge pull request #633 from HIITMEMARIO/main
+- Merge branch ‘githru:main’ into main
+
+Rules:
+- Summarize in 3 to 5 lines.
+- Combine similar or overlapping content. (e.g. feat: add button, feat: add button to header -> feat: add button)
+- Include prefixes if present (e.g. feat, fix, refactor)
+- Please preserve the stylistic style of the commit.
+
+Output format:
+‘’
+- {prefix (if any)}:{Commit summary1}
+- {prefix (if any)}:{Commit summary2}
+- {prefix (if any)}:{commit summary3}
+‘’
+
+Commits:`

packages/mcp/src/common/types.ts

@@ -1,3 +1,5 @@
+import type { RestEndpointMethodTypes } from "@octokit/rest";
+
 export interface GitHubRepoInfo {
   owner: string;
   repo: string;
@@ -78,6 +80,26 @@ export interface CommitNode {
   commit: CommitRaw;
 }
 
+export type CommitDict = Map<string, CommitNode>;
+
+export const CommitMessageTypeList = [
+  "build",
+  "chore",
+  "ci",
+  "docs",
+  "feat",
+  "fix",
+  "pert",
+  "refactor",
+  "revert",
+  "style",
+  "test",
+];
+
+const COMMIT_MESSAGE_TYPE = [...CommitMessageTypeList] as const;
+
+export type CommitMessageType = (typeof COMMIT_MESSAGE_TYPE)[number];
+
 export interface CSMNode {
   base: CommitNode;
   source: CommitNode[];
@@ -173,3 +195,16 @@ export interface AuthorWorkPatternPayload {
   typeMix: Array<{ label: string; value: number }>;
   locale?: "en" | "ko";
 }
+
+export interface Stem {
+  nodes: CommitNode[];
+}
+
+export type StemDict = Map<string, Stem>;
+
+export interface PullRequest {
+  detail: RestEndpointMethodTypes["pulls"]["get"]["response"];
+  commitDetails: RestEndpointMethodTypes["pulls"]["listCommits"]["response"];
+}
+
+export type PullRequestDict = Map<string, PullRequest>;

packages/mcp/src/engine/Engine.ts

@@ -0,0 +1,74 @@
+import "reflect-metadata";
+
+import PluginOctokit from "./pluginOctokit.js";
+import getCommitRaws from "../common/parser.js";
+import { buildCommitDict } from "./commit.js";
+import { buildStemDict } from "./stem.js";
+import { buildCSMDict } from "./csm.js";
+import { getSummary } from "../common/summary.js";
+
+type AnalysisEngineArgs = {
+  isDebugMode?: boolean;
+  gitLog: string;
+  owner: string;
+  repo: string;
+  baseBranchName: string;
+  auth?: string;
+};
+
+export class AnalysisEngine {
+  private gitLog!: string;
+
+  private isDebugMode?: boolean;
+
+  private octokit!: PluginOctokit;
+
+  private baseBranchName!: string;
+
+  constructor(args: AnalysisEngineArgs) {
+    this.insertArgs(args);
+  }
+
+  private insertArgs = (args: AnalysisEngineArgs) => {
+    const { isDebugMode, gitLog, owner, repo, auth, baseBranchName } = args;
+    this.gitLog = gitLog;
+    this.baseBranchName = baseBranchName;
+    this.isDebugMode = isDebugMode;
+    this.octokit = new PluginOctokit({
+      owner,
+      repo,
+      options: {
+        auth,
+      },
+    });
+  };
+
+  public analyzeGit = async () => {
+    let isPRSuccess = true;
+
+    const commitRaws = getCommitRaws(this.gitLog);
+
+    const commitDict = buildCommitDict(commitRaws);
+
+    const pullRequests = await this.octokit
+      .getPullRequests()
+      .catch((err) => {
+        console.error(err);
+        isPRSuccess = false;
+        return [];
+      })
+      .then((pullRequests) => {
+        return pullRequests;
+      });
+
+    const stemDict = buildStemDict(commitDict, this.baseBranchName);
+    const csmDict = buildCSMDict(commitDict, stemDict, this.baseBranchName, pullRequests);
+    const nodes = stemDict.get(this.baseBranchName)?.nodes?.map(({ commit }) => commit);
+    const geminiCommitSummary = await getSummary(nodes ? nodes?.slice(-10) : []);
+
+    return {
+      isPRSuccess,
+      csmDict,
+    };
+  }
+}