如何使用Claude、GitHub Actions以及JavaScript构建一个安全的人工智能公关审查工具

main().catch((error) => {
console.error(error);
process.exit(1);
});


从这一刻起，这个项目才真正具备了生产环境适用性。

我们不再只是简单地说“Claude已经给出了回应”，而是要验证这些回应在结构上是否合法有效。

在本地测试审核功能

在将任何代码连接到 GitHub 之前，我们应该先在终端中测试这个审核工具。

创建一个包含漏洞的文件，例如 vulnerable.js，内容如下：

app.get("/user", async (req, res) => {
    const result = await db.query(
        `SELECT * FROM users WHERE id = ${req.query.id}`;
    );
    res.json(result.rows);
});

这是一个典型的 SQL 注入漏洞，因为用户输入被直接插入到了 SQL 查询语句中。

现在再创建一个没有漏洞的文件，例如 safe.js：

export function add(a, b) {
    return a + b;
}

然后使用这个审核工具来测试这两个文件。

运行并验证本地 CLI

CLI 可用于进行本地测试。它允许你将差分文件或文件内容输入到相同的审核逻辑中，而 GitHub Actions 之后也会使用同样的逻辑。

执行以下命令：

cat vulnerable.js | node index.js

如果你的配置正确，终端中应该会显示 JSON 格式的响应结果。

你也可以测试那个没有漏洞的文件：

cat safe.js | node index.js

在正常的情况下，包含漏洞的代码应该会返回 “fail”，而正常的文件则应该会返回 “pass” 或者根据评估结果给出相应的建议。

你还可以用实际的差分文件来进行测试：

cat pr.diff | node index.js

如果差分文件中同时包含了不安全的代码和用于注入提示信息的代码，Claude 应该能够检测到这两点。我已经在 GitHub 仓库中上传了一个 示例差分文件，供你进行测试。

提示：在进行 GitHub Actions 测试之前，使用本地 CLI 进行调试是最快的方式，这样可以确保模型提示信息、数据结构验证、修改逻辑以及输出处理功能都能正常工作。

将相同的逻辑应用到 GitHub Actions 中

下一步就是让这个审核工具在 GitHub Actions 中也能发挥作用。

GitHub 会自动设置一个名为 GITHUB ACTIONS 的环境变量。当脚本在 GitHub Action 中运行时，这个变量的值会是 "true"。

因此我们可以根据不同的环境来选择输入来源：

const isGitHubAction = process.env.GITHUB_actions === "true";
const diffText = isGitHubAction
    ? process.env.PRDIFF
    : fs.readFileSync(0, "utf8");

现在我们的应用程序已经支持这两种输入方式了：

• 通过标准输入读取本地 CLI 提供的代码

• 通过 PR_diff 变量自动获取差分文件内容

这意味着我们不需要使用两种不同的审核系统，一种代码处理方式就足够了。

使用 Octokit 发布 PR 评论

在GitHub Actions中运行脚本时，仅仅将日志信息以JSON格式输出到控制台是远远不够的。我们希望能够直接在Pull Request上发布一些易于阅读的Markdown格式评论。

安装并验证 Octokit

Octokit 是 GitHub 提供的 JavaScript SDK。我们使用它来与 GitHub API 进行交互，并根据我们的工作流程生成 PR 评论。

如果您还没有安装它，请现在就进行安装：

npm install @octokit/rest

验证安装是否成功：

npm list @octokit/rest

您应该会在依赖项列表中看到这个包。

现在创建文件 postPRComment.js：

import { Octokit } from "@octokit/rest";

export async function postPRComment(reviewResult) {
    const token = process.env.GITHUB_TOKEN;
    const repo = process.env.REPO;
    const prNumber = Number(process.env.PR_NUMBER);

    if (!token || !repo || !prNumber) {
        throw new Error("缺少 GITHUB_TOKEN、REPO 或 PR_NUMBER");
    }

    const [owner, repoName] = repo.split("/");
    const octokit = new Octokit({ auth: token });

    const body = toMarkdown(reviewResult);

    await octokit.issues.createComment({
        owner,
        repo: repoName,
        issue_number: prNumber,
        body,
    });
}

我们还需要函数 toMarkdown()。

创建文件 to-markdown.js：

export function toMarkdown(reviewResult) {
    const { verdict, summary, findings } = reviewResult;

    let output = `## AI PR Review\n\n`;
    output += `**评审结果:** ${verdict}\n\n`;
    output += `**总结:** ${summary}\n\n`;

    if (!findings.length) {
        output += `未发现任何问题。\n`;
        return output;
    }

    output += `### 问题详情\n\n`;

    for (const finding of findings) {
        output += `- **${finding.title}**\n`;
        output += `  - 严重程度: ${finding.severity}\n`;
        output += `  - 文件路径: ${finding.file_path}\n`;
        output += `  - 出现行号: ${finding.line_number}\n`;
        output += `  - 问题描述: ${finding.summary}\n`;
        output += `  - 证据信息: ${finding.evidence}\n`;
        output += `  - 建议措施: ${finding.recommendations}\n\n`;
    }

    return output;
}

现在更新文件 index.js，使其在通过 Actions 运行时能够将结果发布到 GitHub 上：

import fs from "fs";
import { reviewCode } from "./review.js";
import { reviewJsonSchema, reviewSchema } from "./schema.js";
import { redactSecrets } from "./redact-secrets.js";
import { failClosedResult } from "./fail-closed-result.js";
import { postPRComment } from "./postPRComment.js";

async function main() {
    const isGitHubAction = process.env.GITHUB ACTIONS === "true";

    const diffText = isGitHubAction
        ? process.env.PRDIFF
        : fs.readFileSync(0, "utf8");

    if (!diffText) {
        console.error("未提供差异文本");
        process.exit(1);
    }

    const redactedDiff = redactSecrets(diffText);
    const limitedDiff = redactedDiff.slice(0, 4000);

    const result = await reviewCode(limitedDiff, reviewJsonSchema);

    let validated;

    try {
        const rawJson = JSON.parse(result.content[0].text);
        validated = reviewSchema.parse(rawJson);
    } catch (error) {
        validated = failClosedResult(error);
    }

    if (isGitHubAction) {
        await postPRComment(validated);
    } else {
        console.log(JSON.stringify(validated, null, 2));
    }
}

main().catch((error) => {
    console.error(error);
    process.exit(1);
});

创建 GitHub Actions 工作流

现在请创建文件 .github/workflows/review.yml。

GitHub Actions 是一种自动化工具，它会监听 Pull Request 事件，并在 GitHub 提供的运行环境中执行相应的审核流程。

安装并验证 GitHub Actions 的支持功能

对于 GitHub Actions 本身来说，不需要在本地进行任何安装操作，但您需要将工作流文件放置在正确的路径中，然后将其推送到 GitHub 上。

所需文件夹结构如下：

mkdir -p .github/workflows

将仓库推送到 GitHub 后，您可以通过打开 GitHub 的 “Actions” 选项卡来验证工作流是否正常运行。一旦 YAML 文件格式正确，工作流的名称就会显示在那里。

以下是具体的工作流配置：

name: 安全 AI Pull Request 审核工具

on:
    pull_request:
        types: [opened, synchronize, reopened]

permissions:
    contents: read
    pull-requests: write

jobs:
    review:
        runs-on: ubuntu-latest

        env:
            ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
            GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
            REPO: ${{ github.repository }}
            PR_NUMBER: ${{ github.event.pull_request.number }}

        steps:
            - name: 检出代码
              uses: actions/checkout@v4

            - name: 安装 Node.js 环境
              uses: actions/setup-node@v4
              with:
                  node-version: 24

            - name: 安装依赖包
              run: npm install

            - name: 下载 Pull Request 的差异文件
              run: |
                  curl -L \
                    -H "Authorization: Bearer $GITHUB_TOKEN" \
                    -H "Accept: application/vnd.github.v3.diff" \
                    "https://api.github.com/repos/\(REPO/pulls/\)PR_NUMBER" \
                    -o pr.diff

            - name: 保存差异文件
              run: |
                  {
                    echo "PR_DIFF<>> $GITHUB_ENV

            - name: 运行审核脚本
              run: node index.js

每一步的具体功能如下：

1. 检出代码：将您的仓库代码下载到运行环境中。

2. 安装 Node.js 环境：为后续步骤准备 Node.js 运行环境。

3. 安装依赖包：安装项目中所需的 npm 包。

4. 下载差异文件：使用 GitHub API 下载 Pull Request 的差异内容。

5. 保存差异文件：将差异信息保存到指定的文件中。

6. 运行审核脚本

   :执行您的 index.js 脚本，开始进行审核流程。

git init
git remote add origin 
git add .
git commit -m "初始提交"
git push origin main

git checkout -b staging