208 lines
7.3 KiB
JavaScript
208 lines
7.3 KiB
JavaScript
import axios from 'axios';
|
||
import https from 'https';
|
||
import { GITEA_TOKEN, GITEA_COMMENT_TOKEN, GITEA_SERVER_URL, GITEA_REPOSITORY, GITEA_SKIP_TLS_VERIFY, PR_NUMBER, PR_HEAD_SHA, PR_HEAD_BRANCH } from './config.js';
|
||
import { line, warn } from './log.js';
|
||
|
||
const httpsAgent = GITEA_SKIP_TLS_VERIFY ? new https.Agent({ rejectUnauthorized: false }) : undefined;
|
||
const headers = (token = GITEA_TOKEN) => ({ Authorization: `token ${token}`, 'Content-Type': 'application/json' });
|
||
const api = (path) => `${GITEA_SERVER_URL.replace(/\/$/, '')}/api/v1${path}`;
|
||
|
||
function extractCommitMessage(payload) {
|
||
return payload?.message
|
||
|| payload?.commit?.message
|
||
|| payload?.commit?.commit?.message
|
||
|| '';
|
||
}
|
||
|
||
export function getBotReviewOutcome(message) {
|
||
const match = String(message || '').match(/\[ai-review-bot\](?:\[(success|failure)\])?/i);
|
||
return match?.[1]?.toLowerCase() || 'unknown';
|
||
}
|
||
|
||
/**
|
||
* 取得 PR 的 Git Diff 內容，已自動排除 .gitea/ 資料夾。
|
||
*/
|
||
export async function getPRDiff() {
|
||
const resp = await axios.get(api(`/repos/${GITEA_REPOSITORY}/pulls/${PR_NUMBER}.diff`), { headers: headers(), timeout: 60000, httpsAgent });
|
||
return filterDiff(resp.data, [
|
||
'.gitea/',
|
||
'.github/',
|
||
'README.md',
|
||
'TODO.md',
|
||
]);
|
||
}
|
||
|
||
export async function getCommitMessageBySha(sha) {
|
||
if (!sha) return '';
|
||
try {
|
||
const resp = await axios.get(api(`/repos/${GITEA_REPOSITORY}/git/commits/${encodeURIComponent(sha)}`), {
|
||
headers: headers(),
|
||
timeout: 30000,
|
||
httpsAgent,
|
||
});
|
||
return extractCommitMessage(resp.data);
|
||
} catch (e) {
|
||
warn(`取得 commit 訊息失敗: sha=${sha} error=${e.message}`);
|
||
return '';
|
||
}
|
||
}
|
||
|
||
export async function getBranchHeadCommitMessage(branch = PR_HEAD_BRANCH) {
|
||
if (!branch) return '';
|
||
try {
|
||
const resp = await axios.get(api(`/repos/${GITEA_REPOSITORY}/branches/${encodeURIComponent(branch)}`), {
|
||
headers: headers(),
|
||
timeout: 30000,
|
||
httpsAgent,
|
||
});
|
||
const sha = resp.data?.commit?.id || resp.data?.commit?.sha || '';
|
||
return await getCommitMessageBySha(sha);
|
||
} catch (e) {
|
||
warn(`取得分支 head 訊息失敗: branch=${branch} error=${e.message}`);
|
||
return '';
|
||
}
|
||
}
|
||
|
||
/** 檢查 PR head（commit sha 或分支 head）的訊息是否帶 [ai-review-bot] 標記，是則代表本次是自動提交、應跳過審查。 */
|
||
export async function shouldSkipBotCommit({ sha = PR_HEAD_SHA || process.env.GITHUB_SHA, branch = PR_HEAD_BRANCH } = {}) {
|
||
const shaMessage = await getCommitMessageBySha(sha);
|
||
if (sha && shaMessage.includes('[ai-review-bot]')) return true;
|
||
|
||
const branchMessage = await getBranchHeadCommitMessage(branch);
|
||
if (branch && branchMessage.includes('[ai-review-bot]')) return true;
|
||
|
||
return false;
|
||
}
|
||
|
||
/**
|
||
* 過濾 diff 內容，移除路徑符合 excludePrefixes 的區塊。
|
||
* 每個區塊以 "diff --git a/<prefix>" 開頭判斷，使用 startsWith 精確比對前綴。
|
||
*/
|
||
export function filterDiff(diff, excludePrefixes) {
|
||
return diff.split(/(?=^diff --git )/m)
|
||
.filter(block => !excludePrefixes.some(p => {
|
||
const prefix = `diff --git a/${p}`;
|
||
const singleFile = `diff --git a/${p} b/${p}`;
|
||
return block.startsWith(prefix) || block.startsWith(singleFile);
|
||
}))
|
||
.join('');
|
||
}
|
||
|
||
export async function postComment(body) {
|
||
const resp = await axios.post(
|
||
api(`/repos/${GITEA_REPOSITORY}/issues/${PR_NUMBER}/comments`),
|
||
{ body },
|
||
{ headers: headers(GITEA_COMMENT_TOKEN || GITEA_TOKEN), timeout: 30000, httpsAgent },
|
||
);
|
||
return resp.data;
|
||
}
|
||
|
||
/**
|
||
* 在 PR 指定檔案的指定行數發布行內 review comment（標註程式碼位置）。
|
||
* 透過 Gitea 的 pull reviews API，以 new_position 對應新版檔案的行號。
|
||
* 若該行不在 diff 範圍內，Gitea 會回傳錯誤，由呼叫端決定是否降級為一般 comment。
|
||
*/
|
||
export async function postPullReviewComment({ path: filePath, line, body }) {
|
||
const resp = await axios.post(
|
||
api(`/repos/${GITEA_REPOSITORY}/pulls/${PR_NUMBER}/reviews`),
|
||
{
|
||
commit_id: PR_HEAD_SHA || undefined,
|
||
event: 'COMMENT',
|
||
body: '',
|
||
comments: [{ path: filePath, body, new_position: line }],
|
||
},
|
||
{ headers: headers(GITEA_COMMENT_TOKEN || GITEA_TOKEN), timeout: 30000, httpsAgent },
|
||
);
|
||
return resp.data;
|
||
}
|
||
|
||
/**
|
||
* 建立一個 PR review，本文放統計摘要，comments 放多筆行內 review comments。
|
||
*/
|
||
export async function postPullReview({ body, comments = [] }) {
|
||
const resp = await axios.post(
|
||
api(`/repos/${GITEA_REPOSITORY}/pulls/${PR_NUMBER}/reviews`),
|
||
{
|
||
commit_id: PR_HEAD_SHA || undefined,
|
||
event: 'COMMENT',
|
||
body,
|
||
comments,
|
||
},
|
||
{ headers: headers(GITEA_COMMENT_TOKEN || GITEA_TOKEN), timeout: 30000, httpsAgent },
|
||
);
|
||
return resp.data;
|
||
}
|
||
|
||
/**
|
||
* 取得 PR 上所有的 review（每個 review 可含多個行內 comment）。
|
||
*/
|
||
export async function listPullReviews() {
|
||
const resp = await axios.get(
|
||
api(`/repos/${GITEA_REPOSITORY}/pulls/${PR_NUMBER}/reviews`),
|
||
{ headers: headers(), timeout: 30000, httpsAgent },
|
||
);
|
||
return Array.isArray(resp.data) ? resp.data : [];
|
||
}
|
||
|
||
/**
|
||
* 取得單一 review 底下的所有行內 comment。
|
||
*/
|
||
export async function getPullReviewComments(reviewId) {
|
||
const resp = await axios.get(
|
||
api(`/repos/${GITEA_REPOSITORY}/pulls/${PR_NUMBER}/reviews/${reviewId}/comments`),
|
||
{ headers: headers(), timeout: 30000, httpsAgent },
|
||
);
|
||
return Array.isArray(resp.data) ? resp.data : [];
|
||
}
|
||
|
||
/**
|
||
* 取得 PR 上所有 review 的行內 comment，展平成單一陣列。
|
||
* 單一 review 取 comment 失敗時記錄警告並略過，不中斷整體流程。
|
||
*/
|
||
export async function listAllReviewComments() {
|
||
const reviews = await listPullReviews();
|
||
const all = [];
|
||
for (const review of reviews) {
|
||
if (!review?.id) continue;
|
||
try {
|
||
all.push(...await getPullReviewComments(review.id));
|
||
} catch (e) {
|
||
warn(`取得 review #${review.id} 的 comments 失敗（略過）: ${e.message}`);
|
||
}
|
||
}
|
||
line(`取得 PR review comments: reviews=${reviews.length} comments=${all.length}`);
|
||
return all;
|
||
}
|
||
|
||
/**
|
||
* 解決（resolve）一個 review comment 所屬的對話。
|
||
* 對應 Gitea 官方 API：POST /repos/{repo}/pulls/comments/{id}/resolve。
|
||
*/
|
||
export async function resolvePullReviewComment(commentId) {
|
||
const resp = await axios.post(
|
||
api(`/repos/${GITEA_REPOSITORY}/pulls/comments/${commentId}/resolve`),
|
||
{},
|
||
{ headers: headers(GITEA_COMMENT_TOKEN || GITEA_TOKEN), timeout: 30000, httpsAgent },
|
||
);
|
||
return resp.data;
|
||
}
|
||
|
||
/**
|
||
* 取得指定 ref（預設 PR head）下某檔案的最新文字內容；
|
||
* Gitea contents API 回傳 base64，這裡解碼成字串。檔案不存在或非文字時回傳空字串。
|
||
*/
|
||
export async function getFileContentAtRef(filePath, ref = PR_HEAD_SHA || PR_HEAD_BRANCH) {
|
||
try {
|
||
const resp = await axios.get(
|
||
api(`/repos/${GITEA_REPOSITORY}/contents/${encodeURIComponent(filePath).replace(/%2F/g, '/')}`),
|
||
{ headers: headers(), params: ref ? { ref } : undefined, timeout: 30000, httpsAgent },
|
||
);
|
||
const { content, encoding } = resp.data || {};
|
||
if (typeof content !== 'string') return '';
|
||
return encoding === 'base64' ? Buffer.from(content, 'base64').toString('utf8') : content;
|
||
} catch (e) {
|
||
warn(`取得檔案內容失敗（視為空）: ${filePath}@${ref || 'head'} error=${e.message}`);
|
||
return '';
|
||
}
|
||
}
|