feat(omo-codex): batch 35 (4 files)

2026-05-30 19:12:06 +09:00
parent 77d3e32d5c
commit c576eae679
4 changed files with 773 additions and 0 deletions
@@ -0,0 +1,195 @@
+import { spawn } from "node:child_process";
+import { existsSync } from "node:fs";
+import { createRequire } from "node:module";
+import { dirname, join } from "node:path";
+
+import type { CommentCheckerHookInput } from "./core.js";
+
+export type ProcessResult = {
+	exitCode: number | null;
+	stdout: string;
+	stderr: string;
+};
+
+export const MAX_PROCESS_OUTPUT_BYTES = 64 * 1024;
+
+export type ProcessExecutor = (command: string, args: string[], stdin: string) => Promise<ProcessResult>;
+
+export type RunCommentCheckerOptions = {
+	binaryPath?: string;
+	customPrompt?: string;
+	resolveBinary?: () => string | undefined;
+	executor?: ProcessExecutor;
+};
+
+export type CommentCheckerRunResult = {
+	status: "pass" | "warning" | "error" | "missing";
+	message: string;
+	binaryPath?: string;
+	exitCode?: number | null;
+	stdout?: string;
+	stderr?: string;
+};
+
+export type CommentCheckerRunner = (input: CommentCheckerHookInput) => Promise<CommentCheckerRunResult>;
+
+export async function runCommentChecker(
+	input: CommentCheckerHookInput,
+	options: RunCommentCheckerOptions = {},
+): Promise<CommentCheckerRunResult> {
+	const binaryPath =
+		options.binaryPath ?? (options.resolveBinary ? options.resolveBinary() : resolveCommentCheckerBinary());
+	if (!binaryPath) {
+		return {
+			status: "missing",
+			message: "comment-checker binary not found. Run npm install for the codex-comment-checker plugin.",
+		};
+	}
+
+	const args = ["check"];
+	if (options.customPrompt) {
+		args.push("--prompt", options.customPrompt);
+	}
+
+	const executor = options.executor ?? spawnProcess;
+	const result = await executor(binaryPath, args, JSON.stringify(input));
+	const message = result.stderr || result.stdout;
+	if (result.exitCode === 0) {
+		return {
+			status: "pass",
+			message: "",
+			binaryPath,
+			exitCode: result.exitCode,
+			stdout: result.stdout,
+			stderr: result.stderr,
+		};
+	}
+	if (result.exitCode === 2) {
+		return {
+			status: "warning",
+			message,
+			binaryPath,
+			exitCode: result.exitCode,
+			stdout: result.stdout,
+			stderr: result.stderr,
+		};
+	}
+	return {
+		status: "error",
+		message,
+		binaryPath,
+		exitCode: result.exitCode,
+		stdout: result.stdout,
+		stderr: result.stderr,
+	};
+}
+
+export function resolveCommentCheckerBinary(): string | undefined {
+	const binaryName = process.platform === "win32" ? "comment-checker.exe" : "comment-checker";
+	const fromPackageApi = resolvePackageApiBinary();
+	if (fromPackageApi) return fromPackageApi;
+	const fromPackage = resolvePackageBinary(binaryName);
+	if (fromPackage) return fromPackage;
+	return undefined;
+}
+
+function resolvePackageApiBinary(): string | undefined {
+	try {
+		const require = createRequire(import.meta.url);
+		const packageExports: unknown = require("@code-yeongyu/comment-checker");
+		if (!isCommentCheckerPackage(packageExports)) return undefined;
+		const binaryPath = packageExports.getBinaryPath();
+		return existsSync(binaryPath) ? binaryPath : undefined;
+	} catch {
+		return undefined;
+	}
+}
+
+function resolvePackageBinary(binaryName: string): string | undefined {
+	try {
+		const require = createRequire(import.meta.url);
+		const packagePath = require.resolve("@code-yeongyu/comment-checker/package.json");
+		const binaryPath = join(dirname(packagePath), "bin", binaryName);
+		return existsSync(binaryPath) ? binaryPath : undefined;
+	} catch {
+		return undefined;
+	}
+}
+
+function isCommentCheckerPackage(value: unknown): value is { getBinaryPath: () => string } {
+	return isRecord(value) && typeof value["getBinaryPath"] === "function";
+}
+
+function isRecord(value: unknown): value is Record<string, unknown> {
+	return typeof value === "object" && value !== null;
+}
+
+interface OutputAccumulator {
+	text: string;
+	bytes: number;
+	truncated: boolean;
+}
+
+function appendOutput(output: OutputAccumulator, chunk: string, maxOutputBytes: number): void {
+	if (output.truncated) return;
+
+	const remainingBytes = maxOutputBytes - output.bytes;
+	const chunkBytes = Buffer.byteLength(chunk, "utf8");
+	if (chunkBytes <= remainingBytes) {
+		output.text += chunk;
+		output.bytes += chunkBytes;
+		return;
+	}
+
+	if (remainingBytes > 0) {
+		output.text += Buffer.from(chunk, "utf8").subarray(0, remainingBytes).toString("utf8");
+		output.bytes += remainingBytes;
+	}
+	output.truncated = true;
+}
+
+function formatOutput(output: OutputAccumulator, streamName: "stdout" | "stderr", maxOutputBytes: number): string {
+	if (!output.truncated) return output.text;
+	return `${output.text}\n[${streamName} truncated after ${maxOutputBytes} bytes]`;
+}
+
+export function spawnProcess(
+	command: string,
+	args: string[],
+	stdin: string,
+	maxOutputBytes: number = MAX_PROCESS_OUTPUT_BYTES,
+): Promise<ProcessResult> {
+	return new Promise((resolve) => {
+		const outputByteLimit = Number.isFinite(maxOutputBytes) && maxOutputBytes > 0 ? Math.floor(maxOutputBytes) : 0;
+		const proc = spawn(command, args, {
+			stdio: ["pipe", "pipe", "pipe"],
+		});
+		const stdout: OutputAccumulator = { text: "", bytes: 0, truncated: false };
+		const stderr: OutputAccumulator = { text: "", bytes: 0, truncated: false };
+
+		proc.stdout.setEncoding("utf-8");
+		proc.stderr.setEncoding("utf-8");
+		proc.stdout.on("data", (chunk: string) => {
+			appendOutput(stdout, chunk, outputByteLimit);
+		});
+		proc.stderr.on("data", (chunk: string) => {
+			appendOutput(stderr, chunk, outputByteLimit);
+		});
+		proc.once("error", (error) => {
+			appendOutput(stderr, error.message, outputByteLimit);
+			resolve({
+				exitCode: null,
+				stdout: formatOutput(stdout, "stdout", outputByteLimit),
+				stderr: formatOutput(stderr, "stderr", outputByteLimit),
+			});
+		});
+		proc.once("close", (exitCode) => {
+			resolve({
+				exitCode,
+				stdout: formatOutput(stdout, "stdout", outputByteLimit),
+				stderr: formatOutput(stderr, "stderr", outputByteLimit),
+			});
+		});
+		proc.stdin.end(stdin);
+	});
+}