105 lines
3.5 KiB
TypeScript
105 lines
3.5 KiB
TypeScript
import { api } from "@/db/supabase";
|
||
import { CodeAnalysisEngine } from "@/services/codeAnalysis";
|
||
import { unzipSync, strFromU8 } from "fflate";
|
||
|
||
const TEXT_EXTENSIONS = [
|
||
".js", ".ts", ".tsx", ".jsx", ".py", ".java", ".go", ".rs", ".cpp", ".c", ".h", ".cs", ".php", ".rb", ".kt", ".swift", ".sql", ".sh", ".json", ".yml", ".yaml", ".md"
|
||
];
|
||
|
||
const MAX_FILE_SIZE_BYTES = 200 * 1024;
|
||
|
||
function isTextFile(path: string): boolean {
|
||
const lower = path.toLowerCase();
|
||
return TEXT_EXTENSIONS.some(ext => lower.endsWith(ext));
|
||
}
|
||
|
||
export async function runZipRepositoryAudit(params: {
|
||
projectId: string;
|
||
repoUrl: string; // https://github.com/owner/repo
|
||
branch?: string;
|
||
}) {
|
||
const branch = params.branch || "main";
|
||
const task = await api.createAuditTask({
|
||
project_id: params.projectId,
|
||
task_type: "repository",
|
||
branch_name: branch,
|
||
exclude_patterns: [],
|
||
scan_config: {},
|
||
created_by: null
|
||
} as any);
|
||
|
||
try {
|
||
const m = params.repoUrl.match(/github\.com\/(.+?)\/(.+?)(?:\.git)?$/i);
|
||
if (!m) throw new Error("仅支持 GitHub 仓库 URL，例如 https://github.com/owner/repo");
|
||
const owner = m[1];
|
||
const repo = m[2];
|
||
|
||
// GitHub 提供仓库 zip 下载（无需 token）
|
||
const zipUrl = `https://codeload.github.com/${owner}/${repo}/zip/refs/heads/${encodeURIComponent(branch)}`;
|
||
const res = await fetch(zipUrl);
|
||
if (!res.ok) throw new Error(`下载仓库压缩包失败: ${res.status}`);
|
||
const buf = new Uint8Array(await res.arrayBuffer());
|
||
const files = unzipSync(buf);
|
||
|
||
let totalFiles = 0;
|
||
let totalLines = 0;
|
||
let createdIssues = 0;
|
||
|
||
const rootPrefix = `${repo}-${branch}/`;
|
||
|
||
for (const name of Object.keys(files)) {
|
||
if (!name.startsWith(rootPrefix)) continue;
|
||
const relPath = name.slice(rootPrefix.length);
|
||
if (!relPath || relPath.endsWith("/")) continue; // 目录
|
||
if (!isTextFile(relPath)) continue;
|
||
|
||
const fileData = files[name];
|
||
if (!fileData || fileData.length > MAX_FILE_SIZE_BYTES) continue;
|
||
const content = strFromU8(fileData);
|
||
totalFiles += 1;
|
||
totalLines += content.split(/\r?\n/).length;
|
||
|
||
const ext = relPath.split(".").pop() || "";
|
||
const language = ext.toLowerCase();
|
||
const analysis = await CodeAnalysisEngine.analyzeCode(content, language);
|
||
const issues = analysis.issues || [];
|
||
createdIssues += issues.length;
|
||
|
||
for (const issue of issues) {
|
||
await api.createAuditIssue({
|
||
task_id: (task as any).id,
|
||
file_path: relPath,
|
||
line_number: issue.line || null,
|
||
column_number: issue.column || null,
|
||
issue_type: issue.type || "maintainability",
|
||
severity: issue.severity || "low",
|
||
title: issue.title || "Issue",
|
||
description: issue.description || null,
|
||
suggestion: issue.suggestion || null,
|
||
code_snippet: issue.code_snippet || null,
|
||
ai_explanation: issue.xai ? JSON.stringify(issue.xai) : (issue.ai_explanation || null),
|
||
status: "open",
|
||
resolved_by: null,
|
||
resolved_at: null
|
||
} as any);
|
||
}
|
||
}
|
||
|
||
await api.updateAuditTask((task as any).id, {
|
||
status: "completed",
|
||
total_files: totalFiles,
|
||
scanned_files: totalFiles,
|
||
total_lines: totalLines,
|
||
issues_count: createdIssues,
|
||
quality_score: 0
|
||
} as any);
|
||
|
||
return (task as any).id as string;
|
||
} catch (e) {
|
||
await api.updateAuditTask((task as any).id, { status: "failed" } as any);
|
||
throw e;
|
||
}
|
||
}
|
||
|
||
|