mirror of
https://gitee.com/spark-store-project/spark-store
synced 2026-04-30 03:10:16 +08:00
369 lines
9.7 KiB
TypeScript
369 lines
9.7 KiB
TypeScript
/**
|
||
* 共享的安装/下载逻辑
|
||
* 被 install-manager.ts 和 update-center 共同使用
|
||
*/
|
||
import { spawn, ChildProcess } from "node:child_process";
|
||
import { createWriteStream } from "node:fs";
|
||
import * as fs from "node:fs";
|
||
import * as path from "node:path";
|
||
import axios from "axios";
|
||
import pino from "pino";
|
||
|
||
const logger = pino({ name: "shared-installer" });
|
||
|
||
export const SHELL_CALLER_PATH = "/opt/spark-store/extras/shell-caller.sh";
|
||
|
||
export interface DownloadOptions {
|
||
pkgname: string;
|
||
metalinkUrl: string;
|
||
filename: string;
|
||
downloadDir: string;
|
||
onLog?: (msg: string) => void;
|
||
onProgress?: (progress: number) => void;
|
||
onStatus?: (status: string) => void;
|
||
signal?: AbortSignal;
|
||
}
|
||
|
||
export interface DownloadResult {
|
||
filePath: string;
|
||
downloadDir: string;
|
||
}
|
||
|
||
/**
|
||
* 下载 metalink 文件并使用 aria2c 下载 deb 包
|
||
* 与 install-manager.ts 中的下载逻辑保持一致
|
||
*/
|
||
export const downloadPackage = async ({
|
||
pkgname,
|
||
metalinkUrl,
|
||
filename,
|
||
downloadDir,
|
||
onLog,
|
||
onProgress,
|
||
onStatus,
|
||
signal,
|
||
}: DownloadOptions): Promise<DownloadResult> => {
|
||
// 1. 创建下载目录
|
||
try {
|
||
if (!fs.existsSync(downloadDir)) {
|
||
fs.mkdirSync(downloadDir, { recursive: true });
|
||
}
|
||
} catch (err) {
|
||
logger.error(`无法创建目录 ${downloadDir}: ${err}`);
|
||
throw err;
|
||
}
|
||
|
||
const metalinkPath = path.join(downloadDir, `${filename}.metalink`);
|
||
|
||
onLog?.(`正在获取 Metalink 文件: ${metalinkUrl}`);
|
||
|
||
// 2. 下载 metalink 文件
|
||
const response = await axios.get(metalinkUrl, {
|
||
baseURL: "https://erotica.spark-app.store",
|
||
responseType: "stream",
|
||
});
|
||
|
||
const writer = createWriteStream(metalinkPath);
|
||
response.data.pipe(writer);
|
||
|
||
await new Promise<void>((resolve, reject) => {
|
||
writer.on("finish", resolve);
|
||
writer.on("error", reject);
|
||
});
|
||
|
||
onLog?.("Metalink 文件下载完成");
|
||
|
||
// 3. 清理下载目录中的旧文件（保留 .metalink 文件）
|
||
const existingFiles = fs.readdirSync(downloadDir);
|
||
for (const file of existingFiles) {
|
||
if (file.endsWith(".metalink")) continue;
|
||
const filePath = path.join(downloadDir, file);
|
||
try {
|
||
fs.unlinkSync(filePath);
|
||
onLog?.(`已清理旧文件: ${file}`);
|
||
} catch (err) {
|
||
logger.warn(`清理文件失败 ${filePath}: ${err}`);
|
||
}
|
||
}
|
||
|
||
// 4. 使用 aria2c 下载 deb 文件
|
||
const aria2Args = [
|
||
`--dir=${downloadDir}`,
|
||
"--allow-overwrite=true",
|
||
"--summary-interval=1",
|
||
"--connect-timeout=10",
|
||
"--timeout=15",
|
||
"--max-tries=3",
|
||
"--retry-wait=5",
|
||
"--max-concurrent-downloads=4",
|
||
"--min-split-size=1M",
|
||
"--lowest-speed-limit=1K",
|
||
"--auto-file-renaming=false",
|
||
"-M",
|
||
metalinkPath,
|
||
];
|
||
|
||
onStatus?.("downloading");
|
||
|
||
// 下载重试逻辑：每次超时时间递增，最多3次
|
||
const timeoutList = [3000, 5000, 15000];
|
||
let retryCount = 0;
|
||
let downloadSuccess = false;
|
||
|
||
while (retryCount < timeoutList.length && !downloadSuccess) {
|
||
const currentTimeout = timeoutList[retryCount];
|
||
|
||
if (retryCount > 0) {
|
||
onLog?.(`第 ${retryCount} 次重试下载...`);
|
||
onProgress?.(0);
|
||
// 重试前清理旧文件
|
||
const retryFiles = fs.readdirSync(downloadDir);
|
||
for (const file of retryFiles) {
|
||
if (file.endsWith(".metalink")) continue;
|
||
const filePath = path.join(downloadDir, file);
|
||
try {
|
||
fs.unlinkSync(filePath);
|
||
} catch (cleanErr) {
|
||
logger.warn(`重试清理文件失败 ${filePath}: ${cleanErr}`);
|
||
}
|
||
}
|
||
}
|
||
|
||
try {
|
||
await new Promise<void>((resolve, reject) => {
|
||
onLog?.(`启动下载: aria2c ${aria2Args.join(" ")}`);
|
||
const child = spawn("aria2c", aria2Args);
|
||
|
||
let lastProgressTime = Date.now();
|
||
let lastProgress = 0;
|
||
const progressCheckInterval = 1000;
|
||
|
||
// 设置超时检测定时器
|
||
const timeoutChecker = setInterval(() => {
|
||
const now = Date.now();
|
||
// 只在进度为0时检查超时
|
||
if (lastProgress === 0 && now - lastProgressTime > currentTimeout) {
|
||
clearInterval(timeoutChecker);
|
||
child.kill();
|
||
reject(new Error(`下载卡在0%超过 ${currentTimeout / 1000} 秒`));
|
||
}
|
||
}, progressCheckInterval);
|
||
|
||
child.stdout.on("data", (data) => {
|
||
const str = data.toString();
|
||
// Match ( 12%) or (12%)
|
||
const match = str.match(/[0-9]+(\.[0-9]+)?%/g);
|
||
if (match) {
|
||
const p = parseFloat(match.at(-1)) / 100;
|
||
if (p > lastProgress) {
|
||
lastProgress = p;
|
||
lastProgressTime = Date.now();
|
||
}
|
||
onProgress?.(p);
|
||
}
|
||
});
|
||
child.stderr.on("data", (d) => onLog?.(`aria2c: ${d}`));
|
||
|
||
// 处理取消信号
|
||
const abortHandler = () => {
|
||
clearInterval(timeoutChecker);
|
||
child.kill();
|
||
reject(new Error("下载已取消"));
|
||
};
|
||
|
||
signal?.addEventListener("abort", abortHandler, { once: true });
|
||
|
||
child.on("close", (code) => {
|
||
clearInterval(timeoutChecker);
|
||
signal?.removeEventListener("abort", abortHandler);
|
||
|
||
if (code === 0) {
|
||
onProgress?.(1);
|
||
resolve();
|
||
} else {
|
||
reject(new Error(`Aria2c exited with code ${code}`));
|
||
}
|
||
});
|
||
child.on("error", (err) => {
|
||
clearInterval(timeoutChecker);
|
||
signal?.removeEventListener("abort", abortHandler);
|
||
reject(err);
|
||
});
|
||
});
|
||
|
||
// 检查是否已取消
|
||
if (signal?.aborted) {
|
||
throw new Error("下载已取消");
|
||
}
|
||
downloadSuccess = true;
|
||
} catch (err) {
|
||
retryCount++;
|
||
if (retryCount >= timeoutList.length) {
|
||
throw new Error(`下载失败，已重试 ${timeoutList.length} 次: ${err}`);
|
||
}
|
||
onLog?.(`下载失败，准备重试 (${retryCount}/${timeoutList.length})`);
|
||
// 等待2秒后重试
|
||
await new Promise((r) => setTimeout(r, 2000));
|
||
}
|
||
}
|
||
|
||
const filePath = path.join(downloadDir, filename);
|
||
return { filePath, downloadDir };
|
||
};
|
||
|
||
export interface InstallOptions {
|
||
pkgname: string;
|
||
filePath: string;
|
||
origin: "spark" | "apm";
|
||
superUserCmd?: string;
|
||
onLog?: (msg: string) => void;
|
||
signal?: AbortSignal;
|
||
}
|
||
|
||
/**
|
||
* 安装已下载的包
|
||
* 与 install-manager.ts 中的安装逻辑保持一致
|
||
*/
|
||
export const installPackage = async ({
|
||
pkgname,
|
||
filePath,
|
||
origin,
|
||
superUserCmd,
|
||
onLog,
|
||
signal,
|
||
}: InstallOptions): Promise<void> => {
|
||
// 构建安装命令
|
||
let execCommand = "";
|
||
const execParams: string[] = [];
|
||
|
||
if (origin === "spark") {
|
||
execCommand = superUserCmd || SHELL_CALLER_PATH;
|
||
if (superUserCmd) execParams.push(SHELL_CALLER_PATH);
|
||
execParams.push(
|
||
"ssinstall",
|
||
filePath,
|
||
"--delete-after-install",
|
||
"--no-create-desktop-entry",
|
||
"--native"
|
||
);
|
||
} else {
|
||
// APM
|
||
execCommand = superUserCmd || SHELL_CALLER_PATH;
|
||
if (superUserCmd) {
|
||
execParams.push(SHELL_CALLER_PATH);
|
||
}
|
||
execParams.push("apm", "ssinstall", filePath);
|
||
}
|
||
|
||
const cmdString = `${execCommand} ${execParams.join(" ")}`;
|
||
onLog?.(`执行安装: ${cmdString}`);
|
||
logger.info(`启动安装: ${cmdString}`);
|
||
|
||
return new Promise<void>((resolve, reject) => {
|
||
const child = spawn(execCommand, execParams, {
|
||
shell: false,
|
||
env: process.env,
|
||
});
|
||
|
||
let stdout = "";
|
||
let stderr = "";
|
||
let logBuffer = "";
|
||
let logBufferTimer: NodeJS.Timeout | null = null;
|
||
const LOG_FLUSH_MS = 100;
|
||
|
||
const flushLogBuffer = () => {
|
||
if (logBuffer.length > 0) {
|
||
onLog?.(logBuffer);
|
||
logBuffer = "";
|
||
}
|
||
logBufferTimer = null;
|
||
};
|
||
|
||
const bufferedSendLog = (message: string) => {
|
||
logBuffer += message;
|
||
if (!logBufferTimer) {
|
||
logBufferTimer = setTimeout(flushLogBuffer, LOG_FLUSH_MS);
|
||
}
|
||
};
|
||
|
||
// 处理取消信号
|
||
const abortHandler = () => {
|
||
child.kill();
|
||
reject(new Error("安装已取消"));
|
||
};
|
||
|
||
signal?.addEventListener("abort", abortHandler, { once: true });
|
||
|
||
child.stdout?.on("data", (data) => {
|
||
stdout += data.toString();
|
||
bufferedSendLog(data.toString());
|
||
});
|
||
|
||
child.stderr?.on("data", (data) => {
|
||
stderr += data.toString();
|
||
bufferedSendLog(data.toString());
|
||
});
|
||
|
||
child.on("error", (err) => {
|
||
signal?.removeEventListener("abort", abortHandler);
|
||
if (logBufferTimer) clearTimeout(logBufferTimer);
|
||
flushLogBuffer();
|
||
reject(err);
|
||
});
|
||
|
||
child.on("close", (code) => {
|
||
signal?.removeEventListener("abort", abortHandler);
|
||
if (logBufferTimer) clearTimeout(logBufferTimer);
|
||
flushLogBuffer();
|
||
|
||
if (code === 0) {
|
||
resolve();
|
||
} else {
|
||
reject(new Error(`安装失败，退出码: ${code}`));
|
||
}
|
||
});
|
||
});
|
||
};
|
||
|
||
/**
|
||
* 检查是否有 apm 命令
|
||
*/
|
||
export const checkApmAvailable = async (): Promise<boolean> => {
|
||
return new Promise((resolve) => {
|
||
const child = spawn("which", ["apm"]);
|
||
let stdout = "";
|
||
child.stdout?.on("data", (data) => {
|
||
stdout += data.toString();
|
||
});
|
||
child.on("close", (code) => {
|
||
resolve(code === 0 && stdout.trim().length > 0);
|
||
});
|
||
child.on("error", () => {
|
||
resolve(false);
|
||
});
|
||
});
|
||
};
|
||
|
||
/**
|
||
* 检查提权命令
|
||
*/
|
||
export const checkSuperUserCommand = async (): Promise<string> => {
|
||
return new Promise((resolve) => {
|
||
const child = spawn("which", ["/usr/bin/pkexec"]);
|
||
let stdout = "";
|
||
child.stdout?.on("data", (data) => {
|
||
stdout += data.toString();
|
||
});
|
||
child.on("close", (code) => {
|
||
if (code === 0) {
|
||
resolve(stdout.trim());
|
||
} else {
|
||
resolve("");
|
||
}
|
||
});
|
||
child.on("error", () => {
|
||
resolve("");
|
||
});
|
||
});
|
||
};
|