ele-HeiXiu/scripts/upload-oss.mjs

// ============================================================
// 上传安装包到阿里云 OSS
//
// 用法：
//   node scripts/upload-oss.mjs <platform> <edition> [version]
//
// 环境变量（使用 Node.js 直传时需要）：
//   OSS_ENDPOINT        — OSS endpoint（如 oss-cn-hangzhou.aliyuncs.com）
//   OSS_BUCKET          — Bucket 名称
//   OSS_ACCESS_KEY_ID     — AccessKey ID
//   OSS_ACCESS_KEY_SECRET — AccessKey Secret
//   OSS_BASE_PATH         — Bucket 内基路径，默认 releases
//
// 不传 OSS 环境变量时 → 打印手动上传命令（ossutil / 控制台）
//
// 示例：
//   # 仅打印上传命令
//   node scripts/upload-oss.mjs win32 personal 0.1.0
//
//   # 直接上传
//   OSS_BUCKET=heixiu OSS_ENDPOINT=oss-cn-hangzhou.aliyuncs.com \
//   OSS_ACCESS_KEY_ID=xxx OSS_ACCESS_KEY_SECRET=xxx \
//   node scripts/upload-oss.mjs win32 personal 0.1.0
// ============================================================

import {readFileSync, readdirSync} from 'node:fs';
import {join, dirname} from 'node:path';
import {fileURLToPath} from 'node:url';
import {createHmac} from 'node:crypto';

const __dirname = dirname(fileURLToPath(import.meta.url));
const ROOT = join(__dirname, '..');

// ---------- 参数 ----------

const platform = process.argv[2];
const edition = process.argv[3];

if (!platform || !edition) {
    console.error('用法: node scripts/upload-oss.mjs <platform> <edition> [version]');
    console.error('示例: node scripts/upload-oss.mjs win32 personal 0.1.0');
    process.exit(1);
}

const pkg = JSON.parse(readFileSync(join(ROOT, 'package.json'), 'utf-8'));
const version = process.argv[4] || pkg.version;

// ---------- 查找安装包 ----------

const releaseDir = join(ROOT, 'release', version);

let installerName;
try {
    const all = readdirSync(releaseDir);
    const exts = platform === 'win32' ? ['.exe'] : platform === 'darwin' ? ['.dmg'] : ['.AppImage'];
    installerName = all.find((f) => exts.some((e) => f.endsWith(e)));
    if (!installerName) console.error(`错误: ${releaseDir} — 未找到安装包`);
} catch (err) {
    console.error(`错误: ${releaseDir} — ${err.message}`);
    process.exit(1);
}

const installerPath = join(releaseDir, installerName);

// ---------- OSS 配置 ----------

const {
    OSS_ENDPOINT,
    OSS_BUCKET,
    OSS_ACCESS_KEY_ID,
    OSS_ACCESS_KEY_SECRET,
} = process.env;

const basePath = (process.env.OSS_BASE_PATH || 'releases').replace(/^\/+|\/+$/g, '');
const objectKey = `${basePath}/${version}/${installerName}`;

// ---------- 模式 A：无 OSS 配置 → 打印手动命令 ----------

if (!OSS_ENDPOINT || !OSS_BUCKET || !OSS_ACCESS_KEY_ID) {
    const exampleBucket = OSS_BUCKET || '<bucket>';
    const exampleEndpoint = OSS_ENDPOINT || '<endpoint>';
    const downloadUrl = `https://${exampleBucket}.${exampleEndpoint}/${objectKey}`;
    
    console.log('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━');
    console.log('📦 上传安装包（手动）');
    console.log('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━');
    console.log(`文件: ${installerPath}`);
    console.log(`大小: ${(readFileSync(installerPath).length / 1024 / 1024).toFixed(2)} MB\n`);
    
    console.log('方式 1 — ossutil CLI（推荐）：');
    console.log(`  ossutil cp "${installerPath}" oss://${exampleBucket}/${objectKey}\n`);
    
    console.log('方式 2 — 阿里云控制台：');
    console.log(`  登录 OSS 控制台 → ${exampleBucket} → 上传 → 目标路径: ${objectKey}\n`);
    
    console.log('上传后执行以下命令，用实际 URL 重新生成 info：');
    console.log(`  node scripts/generate-update-info.mjs ${platform} ${edition} ${version} --url "${downloadUrl}"\n`);
    
    console.log('或直接发布到后端 API：');
    console.log(`  node scripts/publish-release.mjs ${platform} ${edition} ${version} --url "${downloadUrl}"`);
    console.log('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━');
    process.exit(0);
}

// ---------- 模式 B：Node.js 直传 OSS ----------

const fileBuffer = readFileSync(installerPath);
const contentType = platform === 'win32'
    ? 'application/vnd.microsoft.portable-executable'
    : 'application/octet-stream';

const date = new Date().toUTCString();

// 阿里云 OSS Authorization 签名
const stringToSign = [
    'PUT',
    '',
    contentType,
    date,
    `/${OSS_BUCKET}/${objectKey}`,
].join('\n');

const signature = createHmac('sha1', OSS_ACCESS_KEY_SECRET)
    .update(stringToSign)
    .digest('base64');

const authHeader = `OSS ${OSS_ACCESS_KEY_ID}:${signature}`;
const downloadUrl = `https://${OSS_BUCKET}.${OSS_ENDPOINT}/${objectKey}`;

console.log(`上传中: ${installerName} (${(fileBuffer.length / 1024 / 1024).toFixed(2)} MB)`);
console.log(`目标: oss://${OSS_BUCKET}/${objectKey}`);

try {
    const response = await fetch(downloadUrl, {
        method: 'PUT',
        headers: {
            'Content-Type': contentType,
            'Content-Length': String(fileBuffer.length),
            'Date': date,
            'Authorization': authHeader,
        },
        body: fileBuffer,
    });
    
    if (response.ok) {
        console.log(`✅ 上传成功`);
        console.log(`下载 URL: ${downloadUrl}\n`);
        console.log('接下来：');
        console.log(`  # 重新生成 update-info.json：`);
        console.log(`  node scripts/generate-update-info.mjs ${platform} ${edition} ${version} --url "${downloadUrl}"`);
        console.log(`\n  # 或直接发布到后端：`);
        console.log(`  node scripts/publish-release.mjs ${platform} ${edition} ${version} --url "${downloadUrl}"`);
    } else {
        const body = await response.text();
        console.error(`❌ 上传失败 HTTP ${response.status}: ${body}`);
        process.exit(1);
    }
} catch (err) {
    console.error(`❌ 上传失败: ${err.message}`);
    process.exit(1);
}