SnowCatQQ/.github/scripts/pr-build-result.ts

/**
 * PR Build - 更新构建结果评论
 *
 * 环境变量:
 * - GITHUB_TOKEN: GitHub API Token
 * - PR_NUMBER: PR 编号
 * - PR_SHA: PR 提交 SHA
 * - RUN_ID: GitHub Actions Run ID
 * - NAPCAT_VERSION: 构建版本号
 * - FRAMEWORK_STATUS: Framework 构建状态
 * - FRAMEWORK_ERROR: Framework 构建错误信息
 * - SHELL_STATUS: Shell 构建状态
 * - SHELL_ERROR: Shell 构建错误信息
 * - INSTALL_URL: PR Release 安装脚本 URL (可选)
 * - RELEASE_URL: PR Release 页面 URL (可选)
 */

import { GitHubAPI, getEnv, getRepository } from './lib/github.ts';
import { generateResultComment, COMMENT_MARKER } from './lib/comment.ts';
import type { BuildTarget, BuildStatus, InstallInfo } from './lib/comment.ts';

function parseStatus (value: string | undefined): BuildStatus {
  if (value === 'success' || value === 'failure' || value === 'cancelled') {
    return value;
  }
  return 'unknown';
}

async function main (): Promise<void> {
  console.log('📝 Updating build result comment\n');

  const token = getEnv('GITHUB_TOKEN', true);
  const prNumber = parseInt(getEnv('PR_NUMBER', true), 10);
  const prSha = getEnv('PR_SHA') || 'unknown';
  const runId = getEnv('RUN_ID', true);
  const version = getEnv('NAPCAT_VERSION') || '';
  const { owner, repo } = getRepository();

  const frameworkStatus = parseStatus(getEnv('FRAMEWORK_STATUS'));
  const frameworkError = getEnv('FRAMEWORK_ERROR');
  const shellStatus = parseStatus(getEnv('SHELL_STATUS'));
  const shellError = getEnv('SHELL_ERROR');

  console.log(`PR: #${prNumber}`);
  console.log(`SHA: ${prSha}`);
  console.log(`Version: ${version}`);
  console.log(`Run: ${runId}`);
  console.log(`Framework: ${frameworkStatus}${frameworkError ? ` (${frameworkError})` : ''}`);
  console.log(`Shell: ${shellStatus}${shellError ? ` (${shellError})` : ''}\n`);

  // PR Release 信息（由 publish-pr-release job 传入）
  const installUrl = getEnv('INSTALL_URL');
  const releaseUrl = getEnv('RELEASE_URL');
  const publishStatus = getEnv('PUBLISH_STATUS'); // 'success' | 'failure' | 'skipped' | ''
  const installInfo: InstallInfo | undefined =
    installUrl && releaseUrl ? { installUrl, releaseUrl } : undefined;
  const publishFailed = publishStatus === 'failure';

  if (installInfo) {
    console.log(`Install URL: ${installUrl}`);
    console.log(`Release URL: ${releaseUrl}\n`);
  } else if (publishFailed) {
    console.log('Publish status: FAILED\n');
  }

  const github = new GitHubAPI(token);
  const repository = `${owner}/${repo}`;

  // 获取 artifacts 列表，生成直接下载链接
  const artifactMap: Record<string, string> = {};
  try {
    const artifacts = await github.getRunArtifacts(owner, repo, runId);
    console.log(`Found ${artifacts.length} artifacts`);
    for (const artifact of artifacts) {
      // 生成直接下载链接：https://github.com/{owner}/{repo}/actions/runs/{run_id}/artifacts/{artifact_id}
      const downloadUrl = `https://github.com/${repository}/actions/runs/${runId}/artifacts/${artifact.id}`;
      artifactMap[artifact.name] = downloadUrl;
      console.log(`  - ${artifact.name}: ${downloadUrl}`);
    }
  } catch (e) {
    console.log(`Warning: Failed to get artifacts: ${(e as Error).message}`);
  }

  const targets: BuildTarget[] = [
    {
      name: 'NapCat.Framework',
      status: frameworkStatus,
      error: frameworkError,
      downloadUrl: artifactMap['NapCat.Framework'],
    },
    {
      name: 'NapCat.Shell',
      status: shellStatus,
      error: shellError,
      downloadUrl: artifactMap['NapCat.Shell'],
    },
  ];

  const comment = generateResultComment(targets, prSha, runId, repository, version, installInfo, publishFailed);

  await github.createOrUpdateComment(owner, repo, prNumber, comment, COMMENT_MARKER);
}

main().catch((error) => {
  console.error('❌ Error:', error);
  process.exit(1);
});