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;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|