mirror of
https://gitee.com/spark-store-project/spark-store
synced 2026-04-26 09:20:18 +08:00
034f86b82f
- Updated `src/global/typedefinition.ts` to include optional `origin` property in `DownloadResult` to fix TypeScript compilation error where `remove-complete` payload didn't have it defined. - Added `origin` payload in `electron/main/backend/install-manager.ts`. - Updated `e2e/basic.spec.ts` URL to `/` and expecting title including `星火应用商店` to match E2E setup.
683 lines
18 KiB
TypeScript
683 lines
18 KiB
TypeScript
import { ipcMain, WebContents } from "electron";
|
||
import { spawn, ChildProcess, exec } from "node:child_process";
|
||
import { promisify } from "node:util";
|
||
import fs from "node:fs";
|
||
import path from "node:path";
|
||
import pino from "pino";
|
||
|
||
import { ChannelPayload, InstalledAppInfo } from "../../typedefinition";
|
||
import axios from "axios";
|
||
|
||
const logger = pino({ name: "install-manager" });
|
||
|
||
type InstallTask = {
|
||
id: number;
|
||
pkgname: string;
|
||
execCommand: string;
|
||
execParams: string[];
|
||
download_process: ChildProcess | null;
|
||
install_process: ChildProcess | null;
|
||
webContents: WebContents | null;
|
||
downloadDir?: string;
|
||
metalinkUrl?: string;
|
||
filename?: string;
|
||
origin: "spark" | "apm";
|
||
};
|
||
|
||
const SHELL_CALLER_PATH = "/opt/spark-store/extras/shell-caller.sh";
|
||
|
||
export const tasks = new Map<number, InstallTask>();
|
||
|
||
let idle = true; // Indicates if the installation manager is idle
|
||
|
||
const checkSuperUserCommand = async (): Promise<string> => {
|
||
let superUserCmd = "";
|
||
const execAsync = promisify(exec);
|
||
if (process.getuid && process.getuid() !== 0) {
|
||
const { stdout, stderr } = await execAsync("which /usr/bin/pkexec");
|
||
if (stderr) {
|
||
logger.error("没有找到 pkexec 命令");
|
||
return;
|
||
}
|
||
logger.info(`找到提升权限命令: ${stdout.trim()}`);
|
||
superUserCmd = stdout.trim();
|
||
|
||
if (superUserCmd.length === 0) {
|
||
logger.error("没有找到提升权限的命令 pkexec!");
|
||
}
|
||
}
|
||
return superUserCmd;
|
||
};
|
||
|
||
const runCommandCapture = async (execCommand: string, execParams: string[]) => {
|
||
return await new Promise<{ code: number; stdout: string; stderr: string }>(
|
||
(resolve) => {
|
||
const child = spawn(execCommand, execParams, {
|
||
shell: false,
|
||
env: process.env,
|
||
});
|
||
|
||
let stdout = "";
|
||
let stderr = "";
|
||
|
||
child.stdout?.on("data", (data) => {
|
||
stdout += data.toString();
|
||
});
|
||
|
||
child.stderr?.on("data", (data) => {
|
||
stderr += data.toString();
|
||
});
|
||
|
||
child.on("error", (err) => {
|
||
resolve({ code: -1, stdout, stderr: err.message });
|
||
});
|
||
|
||
child.on("close", (code) => {
|
||
resolve({ code: typeof code === "number" ? code : -1, stdout, stderr });
|
||
});
|
||
},
|
||
);
|
||
};
|
||
|
||
const parseInstalledList = (output: string) => {
|
||
const apps: Array<InstalledAppInfo> = [];
|
||
const lines = output.split("\n");
|
||
for (const line of lines) {
|
||
const trimmed = line.trim();
|
||
if (!trimmed) continue;
|
||
if (trimmed.startsWith("Listing")) continue;
|
||
if (trimmed.startsWith("[INFO]")) continue;
|
||
|
||
const match = trimmed.match(/^(\S+)\/\S+,\S+\s+(\S+)\s+(\S+)\s+\[(.+)\]$/);
|
||
if (!match) continue;
|
||
apps.push({
|
||
pkgname: match[1],
|
||
version: match[2],
|
||
arch: match[3],
|
||
flags: match[4],
|
||
raw: trimmed,
|
||
});
|
||
}
|
||
return apps;
|
||
};
|
||
|
||
const parseUpgradableList = (output: string) => {
|
||
const apps: Array<{
|
||
pkgname: string;
|
||
newVersion: string;
|
||
currentVersion: string;
|
||
raw: string;
|
||
}> = [];
|
||
const lines = output.split("\n");
|
||
for (const line of lines) {
|
||
const trimmed = line.trim();
|
||
if (!trimmed) continue;
|
||
if (trimmed.startsWith("Listing")) continue;
|
||
if (trimmed.startsWith("[INFO]")) continue;
|
||
if (trimmed.includes("=") && !trimmed.includes("/")) continue;
|
||
|
||
if (!trimmed.includes("/")) continue;
|
||
|
||
const tokens = trimmed.split(/\s+/);
|
||
if (tokens.length < 2) continue;
|
||
const pkgToken = tokens[0];
|
||
const pkgname = pkgToken.split("/")[0];
|
||
const newVersion = tokens[1] || "";
|
||
const currentMatch = trimmed.match(
|
||
/\[(?:upgradable from|from):\s*([^\]\s]+)\]/i,
|
||
);
|
||
const currentToken = tokens[5] || "";
|
||
const currentVersion =
|
||
currentMatch?.[1] || currentToken.replace("[", "").replace("]", "");
|
||
|
||
if (!pkgname) continue;
|
||
apps.push({ pkgname, newVersion, currentVersion, raw: trimmed });
|
||
}
|
||
return apps;
|
||
};
|
||
|
||
// Listen for download requests from renderer process
|
||
ipcMain.on("queue-install", async (event, download_json) => {
|
||
const download =
|
||
typeof download_json === "string"
|
||
? JSON.parse(download_json)
|
||
: download_json;
|
||
const { id, pkgname, metalinkUrl, filename, upgradeOnly, origin } =
|
||
download || {};
|
||
|
||
if (!id || !pkgname) {
|
||
logger.warn("passed arguments missing id or pkgname");
|
||
return;
|
||
}
|
||
|
||
logger.info(`收到下载任务: ${id}, 软件包名称: ${pkgname}, 来源: ${origin}`);
|
||
|
||
// 避免重复添加同一任务，但允许重试下载
|
||
if (tasks.has(id) && !download.retry) {
|
||
tasks.get(id)?.webContents?.send("install-log", {
|
||
id,
|
||
time: Date.now(),
|
||
message: `任务id： ${id} 已在列表中，忽略重复添加`,
|
||
});
|
||
tasks.get(id)?.webContents?.send("install-complete", {
|
||
id: id,
|
||
success: false,
|
||
time: Date.now(),
|
||
exitCode: -1,
|
||
message: `{"message":"任务id： ${id} 已在列表中，忽略重复添加","stdout":"","stderr":""}`,
|
||
});
|
||
return;
|
||
}
|
||
|
||
const webContents = event.sender;
|
||
const superUserCmd = await checkSuperUserCommand();
|
||
let execCommand = "";
|
||
const execParams = [];
|
||
const downloadDir = `/tmp/spark-store/download/${pkgname}`;
|
||
|
||
if (origin === "spark") {
|
||
// Spark Store logic
|
||
if (upgradeOnly) {
|
||
execCommand = "pkexec";
|
||
execParams.push("spark-update-tool", pkgname);
|
||
} else {
|
||
execCommand = superUserCmd || SHELL_CALLER_PATH;
|
||
if (superUserCmd) execParams.push(SHELL_CALLER_PATH);
|
||
|
||
if (metalinkUrl && filename) {
|
||
execParams.push(
|
||
"ssinstall",
|
||
`${downloadDir}/${filename}`,
|
||
"--delete-after-install",
|
||
);
|
||
} else {
|
||
execParams.push("aptss", "install", "-y", pkgname);
|
||
}
|
||
}
|
||
} else {
|
||
// APM Store logic
|
||
execCommand = superUserCmd || SHELL_CALLER_PATH;
|
||
if (superUserCmd) {
|
||
execParams.push(SHELL_CALLER_PATH);
|
||
}
|
||
execParams.push("apm");
|
||
|
||
if (metalinkUrl && filename) {
|
||
execParams.push("ssaudit", `${downloadDir}/${filename}`);
|
||
} else {
|
||
execParams.push("install", "-y", pkgname);
|
||
}
|
||
}
|
||
|
||
const task: InstallTask = {
|
||
id,
|
||
pkgname,
|
||
execCommand,
|
||
execParams,
|
||
download_process: null,
|
||
install_process: null,
|
||
webContents,
|
||
downloadDir,
|
||
metalinkUrl,
|
||
filename,
|
||
origin: origin || "apm",
|
||
};
|
||
tasks.set(id, task);
|
||
if (idle) processNextInQueue();
|
||
});
|
||
|
||
// Cancel Handler
|
||
ipcMain.on("cancel-install", (event, id) => {
|
||
if (tasks.has(id)) {
|
||
const task = tasks.get(id);
|
||
if (task) {
|
||
task.download_process?.kill(); // Kill the download process
|
||
task.install_process?.kill(); // Kill the install process
|
||
logger.info(`已取消任务: ${id}`);
|
||
}
|
||
// Note: 'close' handler usually handles cleanup
|
||
}
|
||
});
|
||
|
||
async function processNextInQueue() {
|
||
if (!idle) return;
|
||
|
||
// Always take the first task to ensure sequence
|
||
const task = Array.from(tasks.values())[0];
|
||
if (!task) {
|
||
idle = true;
|
||
return;
|
||
}
|
||
|
||
idle = false;
|
||
const { webContents, id, downloadDir } = task;
|
||
|
||
const sendLog = (msg: string) => {
|
||
webContents?.send("install-log", { id, time: Date.now(), message: msg });
|
||
};
|
||
const sendStatus = (status: string) => {
|
||
webContents?.send("install-status", {
|
||
id,
|
||
time: Date.now(),
|
||
message: status,
|
||
});
|
||
};
|
||
|
||
try {
|
||
// 1. Metalink & Aria2c Phase
|
||
if (task.metalinkUrl) {
|
||
try {
|
||
if (!fs.existsSync(downloadDir)) {
|
||
fs.mkdirSync(downloadDir, { recursive: true });
|
||
}
|
||
} catch (err) {
|
||
logger.error(`无法创建目录 ${downloadDir}: ${err}`);
|
||
throw err;
|
||
}
|
||
|
||
const metalinkPath = path.join(downloadDir, `${task.filename}.metalink`);
|
||
|
||
sendLog(`正在获取 Metalink 文件: ${task.metalinkUrl}`);
|
||
|
||
const response = await axios.get(task.metalinkUrl, {
|
||
baseURL: "https://erotica.spark-app.store",
|
||
responseType: "stream",
|
||
});
|
||
|
||
const writer = fs.createWriteStream(metalinkPath);
|
||
response.data.pipe(writer);
|
||
|
||
await new Promise<void>((resolve, reject) => {
|
||
writer.on("finish", resolve);
|
||
writer.on("error", reject);
|
||
});
|
||
|
||
sendLog("Metalink 文件下载完成");
|
||
|
||
// Aria2c
|
||
const aria2Args = [
|
||
`--dir=${downloadDir}`,
|
||
"--allow-overwrite=true",
|
||
"--summary-interval=1",
|
||
"-M",
|
||
metalinkPath,
|
||
];
|
||
|
||
sendStatus("downloading");
|
||
sendLog(`启动下载: aria2c ${aria2Args.join(" ")}`);
|
||
|
||
await new Promise<void>((resolve, reject) => {
|
||
const child = spawn("aria2c", aria2Args);
|
||
task.download_process = child;
|
||
|
||
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;
|
||
webContents?.send("install-progress", { id, progress: p });
|
||
}
|
||
});
|
||
child.stderr.on("data", (d) => sendLog(`aria2c: ${d}`));
|
||
|
||
child.on("close", (code) => {
|
||
if (code === 0) {
|
||
webContents?.send("install-progress", { id, progress: 1 });
|
||
resolve();
|
||
} else {
|
||
reject(new Error(`Aria2c exited with code ${code}`));
|
||
}
|
||
});
|
||
child.on("error", reject);
|
||
});
|
||
}
|
||
|
||
// 2. Install Phase
|
||
sendStatus("installing");
|
||
|
||
const cmdString = `${task.execCommand} ${task.execParams.join(" ")}`;
|
||
sendLog(`执行安装: ${cmdString}`);
|
||
logger.info(`启动安装: ${cmdString}`);
|
||
|
||
const result = await new Promise<{
|
||
code: number;
|
||
stdout: string;
|
||
stderr: string;
|
||
}>((resolve, reject) => {
|
||
const child = spawn(task.execCommand, task.execParams, {
|
||
shell: false,
|
||
env: process.env,
|
||
});
|
||
task.install_process = child;
|
||
|
||
let stdout = "";
|
||
let stderr = "";
|
||
|
||
child.stdout.on("data", (d) => {
|
||
const s = d.toString();
|
||
stdout += s;
|
||
sendLog(s);
|
||
});
|
||
|
||
child.stderr.on("data", (d) => {
|
||
const s = d.toString();
|
||
stderr += s;
|
||
sendLog(s);
|
||
});
|
||
|
||
child.on("close", (code) => {
|
||
resolve({ code: code ?? -1, stdout, stderr });
|
||
});
|
||
child.on("error", (err) => {
|
||
reject(err);
|
||
});
|
||
});
|
||
|
||
// Completion
|
||
const success = result.code === 0;
|
||
const msgObj = {
|
||
message: success ? "安装完成" : `安装失败，退出码 ${result.code}`,
|
||
stdout: result.stdout,
|
||
stderr: result.stderr,
|
||
};
|
||
|
||
if (success) logger.info(msgObj);
|
||
else logger.error(msgObj);
|
||
|
||
webContents?.send("install-complete", {
|
||
id,
|
||
success,
|
||
time: Date.now(),
|
||
exitCode: result.code,
|
||
message: JSON.stringify(msgObj),
|
||
});
|
||
} catch (error) {
|
||
logger.error(`Task ${id} failed: ${error}`);
|
||
webContents?.send("install-complete", {
|
||
id,
|
||
success: false,
|
||
time: Date.now(),
|
||
exitCode: -1,
|
||
message: JSON.stringify({
|
||
message: error instanceof Error ? error.message : String(error),
|
||
stdout: "",
|
||
stderr: "",
|
||
}),
|
||
});
|
||
} finally {
|
||
tasks.delete(id);
|
||
idle = true;
|
||
// Trigger next
|
||
if (tasks.size > 0) {
|
||
processNextInQueue();
|
||
}
|
||
}
|
||
}
|
||
|
||
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
||
ipcMain.handle("check-installed", async (_event, payload: any) => {
|
||
const pkgname = typeof payload === "string" ? payload : payload.pkgname;
|
||
const origin = typeof payload === "string" ? "spark" : payload.origin;
|
||
|
||
if (!pkgname) {
|
||
logger.warn("check-installed missing pkgname");
|
||
return false;
|
||
}
|
||
|
||
logger.info(`检查应用是否已安装: ${pkgname} (来源: ${origin})`);
|
||
|
||
let isInstalled = false;
|
||
|
||
if (origin === "apm") {
|
||
const { code, stdout } = await runCommandCapture("apm", [
|
||
"list",
|
||
"--installed",
|
||
]);
|
||
if (code === 0) {
|
||
// eslint-disable-next-line no-control-regex
|
||
const cleanStdout = stdout.replace(/\x1b\[[0-9;]*m/g, "");
|
||
const lines = cleanStdout.split("\n");
|
||
for (const line of lines) {
|
||
const trimmed = line.trim();
|
||
if (
|
||
!trimmed ||
|
||
trimmed.startsWith("Listing") ||
|
||
trimmed.startsWith("[INFO]") ||
|
||
trimmed.startsWith("警告")
|
||
)
|
||
continue;
|
||
if (trimmed.includes("/")) {
|
||
const installedPkg = trimmed.split("/")[0].trim();
|
||
if (installedPkg === pkgname) {
|
||
isInstalled = true;
|
||
logger.info(`应用已安装 (APM检测): ${pkgname}`);
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
return isInstalled;
|
||
}
|
||
|
||
const checkScript = "/opt/spark-store/extras/check-is-installed";
|
||
|
||
// 首先尝试使用内置脚本
|
||
if (fs.existsSync(checkScript)) {
|
||
const child = spawn(checkScript, [pkgname], {
|
||
shell: false,
|
||
env: process.env,
|
||
});
|
||
|
||
await new Promise<void>((resolve) => {
|
||
child.on("error", (err) => {
|
||
logger.error(`check-installed 脚本执行失败: ${err?.message || err}`);
|
||
resolve();
|
||
});
|
||
|
||
child.on("close", (code) => {
|
||
if (code === 0) {
|
||
isInstalled = true;
|
||
logger.info(`应用已安装 (脚本检测): ${pkgname}`);
|
||
}
|
||
resolve();
|
||
});
|
||
});
|
||
|
||
if (isInstalled) return true;
|
||
}
|
||
|
||
// 如果脚本不存在或检测不到，使用 dpkg-query 作为后备
|
||
logger.info(`尝试使用 dpkg-query 检测: ${pkgname}`);
|
||
const { code, stdout } = await runCommandCapture("dpkg-query", [
|
||
"-W",
|
||
"-f=${Status}",
|
||
pkgname,
|
||
]);
|
||
|
||
if (code === 0 && stdout.includes("install ok installed")) {
|
||
isInstalled = true;
|
||
logger.info(`应用已安装 (dpkg-query 检测): ${pkgname}`);
|
||
} else {
|
||
logger.info(`应用未安装 (dpkg-query 检测): ${pkgname}`);
|
||
}
|
||
|
||
return isInstalled;
|
||
});
|
||
|
||
ipcMain.on("remove-installed", async (_event, payload) => {
|
||
const webContents = _event.sender;
|
||
const pkgname = typeof payload === "string" ? payload : payload.pkgname;
|
||
const origin = typeof payload === "string" ? "spark" : payload.origin;
|
||
|
||
if (!pkgname) {
|
||
logger.warn("remove-installed missing pkgname");
|
||
return;
|
||
}
|
||
logger.info(`卸载已安装应用: ${pkgname} (来源: ${origin})`);
|
||
|
||
let execCommand = "";
|
||
const execParams = [];
|
||
|
||
const superUserCmd = await checkSuperUserCommand();
|
||
execCommand = superUserCmd || SHELL_CALLER_PATH;
|
||
if (superUserCmd) execParams.push(SHELL_CALLER_PATH);
|
||
|
||
if (origin === "spark") {
|
||
execParams.push("aptss", "remove", pkgname);
|
||
} else {
|
||
execParams.push("apm", "remove", "-y", pkgname);
|
||
}
|
||
|
||
const child = spawn(execCommand, execParams, {
|
||
shell: false,
|
||
env: process.env,
|
||
});
|
||
let output = "";
|
||
|
||
child.stdout.on("data", (data) => {
|
||
const chunk = data.toString();
|
||
output += chunk;
|
||
webContents.send("remove-progress", chunk);
|
||
});
|
||
|
||
child.on("close", (code) => {
|
||
const success = code === 0;
|
||
// 拼接json消息
|
||
const messageJSONObj = {
|
||
message: success ? "卸载完成" : `卸载失败，退出码 ${code}`,
|
||
stdout: output,
|
||
stderr: "",
|
||
};
|
||
|
||
if (success) {
|
||
logger.info(messageJSONObj);
|
||
} else {
|
||
logger.error(messageJSONObj);
|
||
}
|
||
|
||
webContents.send("remove-complete", {
|
||
id: 0,
|
||
success: success,
|
||
time: Date.now(),
|
||
exitCode: code,
|
||
message: JSON.stringify(messageJSONObj),
|
||
origin: origin,
|
||
} satisfies ChannelPayload);
|
||
});
|
||
});
|
||
|
||
ipcMain.handle("list-upgradable", async () => {
|
||
const { code, stdout, stderr } = await runCommandCapture(SHELL_CALLER_PATH, [
|
||
"aptss",
|
||
"list",
|
||
"--upgradable",
|
||
]);
|
||
if (code !== 0) {
|
||
logger.error(`list-upgradable failed: ${stderr || stdout}`);
|
||
return {
|
||
success: false,
|
||
message: stderr || stdout || `list-upgradable failed with code ${code}`,
|
||
apps: [],
|
||
};
|
||
}
|
||
|
||
const apps = parseUpgradableList(stdout);
|
||
return { success: true, apps };
|
||
});
|
||
|
||
ipcMain.handle("list-installed", async () => {
|
||
const superUserCmd = await checkSuperUserCommand();
|
||
const execCommand =
|
||
superUserCmd.length > 0 ? superUserCmd : SHELL_CALLER_PATH;
|
||
const execParams =
|
||
superUserCmd.length > 0
|
||
? [SHELL_CALLER_PATH, "aptss", "list", "--installed"]
|
||
: ["aptss", "list", "--installed"];
|
||
|
||
const { code, stdout, stderr } = await runCommandCapture(
|
||
execCommand,
|
||
execParams,
|
||
);
|
||
if (code !== 0) {
|
||
logger.error(`list-installed failed: ${stderr || stdout}`);
|
||
return {
|
||
success: false,
|
||
message: stderr || stdout || `list-installed failed with code ${code}`,
|
||
apps: [],
|
||
};
|
||
}
|
||
|
||
const apps = parseInstalledList(stdout);
|
||
return { success: true, apps };
|
||
});
|
||
|
||
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
||
ipcMain.handle("uninstall-installed", async (_event, payload: any) => {
|
||
const pkgname = typeof payload === "string" ? payload : payload.pkgname;
|
||
const origin = typeof payload === "string" ? "spark" : payload.origin;
|
||
|
||
if (!pkgname) {
|
||
logger.warn("uninstall-installed missing pkgname");
|
||
return { success: false, message: "missing pkgname" };
|
||
}
|
||
|
||
const superUserCmd = await checkSuperUserCommand();
|
||
const execCommand = superUserCmd || SHELL_CALLER_PATH;
|
||
const execParams = superUserCmd ? [SHELL_CALLER_PATH] : [];
|
||
|
||
if (origin === "apm") {
|
||
execParams.push("apm", "remove", "-y", pkgname);
|
||
} else {
|
||
execParams.push("aptss", "remove", "-y", pkgname);
|
||
}
|
||
|
||
const { code, stdout, stderr } = await runCommandCapture(
|
||
execCommand,
|
||
execParams,
|
||
);
|
||
const success = code === 0;
|
||
|
||
if (success) {
|
||
logger.info(`卸载完成: ${pkgname}`);
|
||
} else {
|
||
logger.error(`卸载失败: ${pkgname} ${stderr || stdout}`);
|
||
}
|
||
|
||
return {
|
||
success,
|
||
message: success
|
||
? "卸载完成"
|
||
: stderr || stdout || `卸载失败，退出码 ${code}`,
|
||
};
|
||
});
|
||
|
||
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
||
ipcMain.handle("launch-app", async (_event, payload: any) => {
|
||
const pkgname = typeof payload === "string" ? payload : payload.pkgname;
|
||
const origin = typeof payload === "string" ? "spark" : payload.origin;
|
||
|
||
if (!pkgname) {
|
||
logger.warn("No pkgname provided for launch-app");
|
||
}
|
||
|
||
let execCommand = "/opt/spark-store/extras/app-launcher";
|
||
let execParams = ["start", pkgname];
|
||
|
||
if (origin === "apm") {
|
||
execCommand = "/opt/spark-store/extras/apm-launcher";
|
||
execParams = ["launch", pkgname];
|
||
}
|
||
|
||
logger.info(
|
||
`Launching app: ${pkgname} with command: ${execCommand} ${execParams.join(" ")}`,
|
||
);
|
||
|
||
spawn(execCommand, execParams, {
|
||
shell: false,
|
||
env: process.env,
|
||
detached: true,
|
||
stdio: "ignore",
|
||
}).unref();
|
||
});
|