@yuandc/aica 0.1.1 → 0.1.2

package/dist/core/file-transfer-worker.js

@@ -1,169 +1 @@
-import fs from "node:fs";
-import os from "node:os";
-import path from "node:path";
-import { Readable } from "node:stream";
-import { acaServerRequest, acaServerStreamRequest } from "./aca-server-client.js";
-import { loadAcaConfig } from "./aca-config.js";
-import { getAcaHome } from "./paths.js";
-import { startLongPollWorker } from "./long-poll-worker.js";
-const LONG_POLL_WAIT_MS = 25_000;
-export function startFileTransferWorkerLoop(logger) {
-    return startLongPollWorker({
-        run: async () => {
-            const config = loadAcaConfig();
-            if (!config.token)
-                throw new Error("Worker 尚未配置认证令牌");
-            const response = await acaServerRequest("GET", `/api/client/file-requests/claim?machineId=${encodeURIComponent(config.machineId)}&waitMs=${LONG_POLL_WAIT_MS}`);
-            if (!response.item)
-                return;
-            try {
-                await streamFileRequest(response.item, config, logger);
-            }
-            catch (error) {
-                const message = error instanceof Error ? error.message : String(error);
-                logger.warn(`file request failed request=${response.item.requestId}: ${message}`);
-                try {
-                    await failFileRequest(response.item.requestId, message);
-                }
-                catch (failError) {
-                    logger.warn(`file request fail callback failed request=${response.item.requestId}: ${failError instanceof Error ? failError.message : String(failError)}`);
-                }
-            }
-        },
-        onError: (error, retryMs) => logger.warn(`file transfer worker failed, retry in ${retryMs}ms: ${error instanceof Error ? error.message : String(error)}`)
-    });
-}
-async function failFileRequest(requestId, message) {
-    await acaServerRequest("POST", `/api/client/file-requests/${encodeURIComponent(requestId)}/fail`, { message });
-}
-async function streamFileRequest(request, config, logger) {
-    const project = config.projects.find((item) => item.projectId === request.projectId);
-    const projectRoot = expandProjectRoot(request.projectRoot || project?.rootPath || "");
-    if (!projectRoot)
-        throw new Error(`file request ${request.requestId} project root is unknown`);
-    const purpose = request.purpose === "temp-image" ? "temp-image" : "project-file";
-    const resolved = purpose === "temp-image"
-        ? resolveTempImageFile(request.path)
-        : resolveProjectFile(projectRoot, request.path);
-    const stat = fs.statSync(resolved);
-    if (!stat.isFile())
-        throw new Error(`file request ${request.requestId} target is not a file`);
-    if (stat.size > request.maxBytes)
-        throw new Error(`file request ${request.requestId} target is too large (${stat.size} bytes)`);
-    const mimeType = purpose === "temp-image" ? detectSafeTempImageMime(resolved) : mimeForPath(resolved);
-    logger.info(`streaming file request=${request.requestId} path=${resolved}`);
-    await acaServerStreamRequest("POST", `/api/client/file-requests/${encodeURIComponent(request.requestId)}/stream`, Readable.toWeb(fs.createReadStream(resolved)), {
-        "content-type": "application/octet-stream",
-        "x-aca-file-name": encodeURIComponent(path.basename(resolved)),
-        "x-aca-file-size": String(stat.size),
-        "x-aca-file-mime": mimeType
-    });
-}
-function resolveProjectFile(projectRoot, requestedPath) {
-    const root = fs.realpathSync(expandProjectRoot(projectRoot));
-    const candidate = path.isAbsolute(requestedPath)
-        ? requestedPath
-        : path.join(root, requestedPath);
-    const target = fs.realpathSync(candidate);
-    const relative = path.relative(root, target);
-    if (relative === "" || relative.startsWith("..") || path.isAbsolute(relative)) {
-        throw new Error("requested file is outside project root");
-    }
-    return target;
-}
-export function resolveTempImageFile(requestedPath) {
-    const candidate = localPathFromUri(requestedPath);
-    if (!path.isAbsolute(candidate))
-        throw new Error("temporary image path must be absolute");
-    const sourceStat = fs.lstatSync(candidate);
-    if (sourceStat.isSymbolicLink())
-        throw new Error("temporary image cannot be a symbolic link");
-    if (!sourceStat.isFile())
-        throw new Error("temporary image target is not a regular file");
-    const tempRoot = fs.realpathSync(os.tmpdir());
-    const target = fs.realpathSync(candidate);
-    const relative = path.relative(tempRoot, target);
-    if (relative === "" || relative.startsWith("..") || path.isAbsolute(relative)) {
-        throw new Error("temporary image is outside the system temp directory");
-    }
-    return target;
-}
-export function detectSafeTempImageMime(filePath) {
-    const extension = path.extname(filePath).toLowerCase();
-    const header = Buffer.alloc(16);
-    const file = fs.openSync(filePath, "r");
-    let bytesRead = 0;
-    try {
-        bytesRead = fs.readSync(file, header, 0, header.length, 0);
-    }
-    finally {
-        fs.closeSync(file);
-    }
-    const bytes = header.subarray(0, bytesRead);
-    if (extension === ".png" && bytes.length >= 8 && bytes.subarray(0, 8).equals(Buffer.from([0x89, 0x50, 0x4e, 0x47, 0x0d, 0x0a, 0x1a, 0x0a]))) {
-        return "image/png";
-    }
-    if ([".jpg", ".jpeg"].includes(extension) && bytes.length >= 3 && bytes[0] === 0xff && bytes[1] === 0xd8 && bytes[2] === 0xff) {
-        return "image/jpeg";
-    }
-    if (extension === ".gif" && bytes.length >= 6 && ["GIF87a", "GIF89a"].includes(bytes.subarray(0, 6).toString("ascii"))) {
-        return "image/gif";
-    }
-    if (extension === ".webp" && bytes.length >= 12 && bytes.subarray(0, 4).toString("ascii") === "RIFF" && bytes.subarray(8, 12).toString("ascii") === "WEBP") {
-        return "image/webp";
-    }
-    throw new Error("temporary file is not a supported raster image");
-}
-function localPathFromUri(value) {
-    const trimmed = String(value || "").trim();
-    if (!trimmed.startsWith("file://"))
-        return trimmed;
-    try {
-        return decodeURIComponent(new URL(trimmed).pathname);
-    }
-    catch {
-        throw new Error("temporary image path is invalid");
-    }
-}
-function expandProjectRoot(value) {
-    if (!value)
-        return "";
-    if (value === "~")
-        return path.dirname(getAcaHome());
-    if (value.startsWith("~/"))
-        return path.join(path.dirname(getAcaHome()), value.slice(2));
-    return value;
-}
-function mimeForPath(filePath) {
-    switch (path.extname(filePath).toLowerCase()) {
-        case ".pdf":
-            return "application/pdf";
-        case ".md":
-        case ".markdown":
-            return "text/markdown; charset=utf-8";
-        case ".txt":
-        case ".log":
-            return "text/plain; charset=utf-8";
-        case ".html":
-            return "text/html; charset=utf-8";
-        case ".json":
-            return "application/json; charset=utf-8";
-        case ".png":
-            return "image/png";
-        case ".jpg":
-        case ".jpeg":
-            return "image/jpeg";
-        case ".gif":
-            return "image/gif";
-        case ".webp":
-            return "image/webp";
-        case ".zip":
-            return "application/zip";
-        case ".docx":
-            return "application/vnd.openxmlformats-officedocument.wordprocessingml.document";
-        case ".xlsx":
-            return "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet";
-        default:
-            return "application/octet-stream";
-    }
-}
+import s from"node:fs";import g from"node:os";import i from"node:path";import{Readable as h}from"node:stream";import{acaServerRequest as p,acaServerStreamRequest as w}from"./aca-server-client.js";import{loadAcaConfig as y}from"./aca-config.js";import{getAcaHome as f}from"./paths.js";import{startLongPollWorker as x}from"./long-poll-worker.js";const S=25e3;function P(e){return x({run:async()=>{const t=y();if(!t.token)throw new Error("Worker 尚未配置认证令牌");const a=await p("GET",`/api/client/file-requests/claim?machineId=${encodeURIComponent(t.machineId)}&waitMs=${S}`);if(a.item)try{await I(a.item,t,e)}catch(n){const o=n instanceof Error?n.message:String(n);e.warn(`file request failed request=${a.item.requestId}: ${o}`);try{await b(a.item.requestId,o)}catch(r){e.warn(`file request fail callback failed request=${a.item.requestId}: ${r instanceof Error?r.message:String(r)}`)}}},onError:(t,a)=>e.warn(`file transfer worker failed, retry in ${a}ms: ${t instanceof Error?t.message:String(t)}`)})}async function b(e,t){await p("POST",`/api/client/file-requests/${encodeURIComponent(e)}/fail`,{message:t})}async function I(e,t,a){const n=t.projects.find(d=>d.projectId===e.projectId),o=l(e.projectRoot||n?.rootPath||"");if(!o)throw new Error(`file request ${e.requestId} project root is unknown`);const r=e.purpose==="temp-image"?"temp-image":"project-file",c=r==="temp-image"?E(e.path):j(o,e.path),m=s.statSync(c);if(!m.isFile())throw new Error(`file request ${e.requestId} target is not a file`);if(m.size>e.maxBytes)throw new Error(`file request ${e.requestId} target is too large (${m.size} bytes)`);const u=r==="temp-image"?R(c):q(c);a.info(`streaming file request=${e.requestId} path=${c}`),await w("POST",`/api/client/file-requests/${encodeURIComponent(e.requestId)}/stream`,h.toWeb(s.createReadStream(c)),{"content-type":"application/octet-stream","x-aca-file-name":encodeURIComponent(i.basename(c)),"x-aca-file-size":String(m.size),"x-aca-file-mime":u})}function j(e,t){const a=s.realpathSync(l(e)),n=i.isAbsolute(t)?t:i.join(a,t),o=s.realpathSync(n),r=i.relative(a,o);if(r===""||r.startsWith("..")||i.isAbsolute(r))throw new Error("requested file is outside project root");return o}function E(e){const t=$(e);if(!i.isAbsolute(t))throw new Error("temporary image path must be absolute");const a=s.lstatSync(t);if(a.isSymbolicLink())throw new Error("temporary image cannot be a symbolic link");if(!a.isFile())throw new Error("temporary image target is not a regular file");const n=s.realpathSync(g.tmpdir()),o=s.realpathSync(t),r=i.relative(n,o);if(r===""||r.startsWith("..")||i.isAbsolute(r))throw new Error("temporary image is outside the system temp directory");return o}function R(e){const t=i.extname(e).toLowerCase(),a=Buffer.alloc(16),n=s.openSync(e,"r");let o=0;try{o=s.readSync(n,a,0,a.length,0)}finally{s.closeSync(n)}const r=a.subarray(0,o);if(t===".png"&&r.length>=8&&r.subarray(0,8).equals(Buffer.from([137,80,78,71,13,10,26,10])))return"image/png";if([".jpg",".jpeg"].includes(t)&&r.length>=3&&r[0]===255&&r[1]===216&&r[2]===255)return"image/jpeg";if(t===".gif"&&r.length>=6&&["GIF87a","GIF89a"].includes(r.subarray(0,6).toString("ascii")))return"image/gif";if(t===".webp"&&r.length>=12&&r.subarray(0,4).toString("ascii")==="RIFF"&&r.subarray(8,12).toString("ascii")==="WEBP")return"image/webp";throw new Error("temporary file is not a supported raster image")}function $(e){const t=String(e||"").trim();if(!t.startsWith("file://"))return t;try{return decodeURIComponent(new URL(t).pathname)}catch{throw new Error("temporary image path is invalid")}}function l(e){return e?e==="~"?i.dirname(f()):e.startsWith("~/")?i.join(i.dirname(f()),e.slice(2)):e:""}function q(e){switch(i.extname(e).toLowerCase()){case".pdf":return"application/pdf";case".md":case".markdown":return"text/markdown; charset=utf-8";case".txt":case".log":return"text/plain; charset=utf-8";case".html":return"text/html; charset=utf-8";case".json":return"application/json; charset=utf-8";case".png":return"image/png";case".jpg":case".jpeg":return"image/jpeg";case".gif":return"image/gif";case".webp":return"image/webp";case".zip":return"application/zip";case".docx":return"application/vnd.openxmlformats-officedocument.wordprocessingml.document";case".xlsx":return"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet";default:return"application/octet-stream"}}export{R as detectSafeTempImageMime,E as resolveTempImageFile,P as startFileTransferWorkerLoop};

package/dist/core/fs.js

@@ -1,28 +1,2 @@
-import fs from "node:fs";
-import path from "node:path";
-export function ensureDir(dir) {
-    fs.mkdirSync(dir, { recursive: true });
-}
-export function writeJsonFile(file, value) {
-    ensureDir(path.dirname(file));
-    fs.writeFileSync(file, `${JSON.stringify(value, null, 2)}\n`, "utf8");
-}
-export function readJsonFile(file) {
-    try {
-        return JSON.parse(fs.readFileSync(file, "utf8"));
-    }
-    catch (error) {
-        if (error.code === "ENOENT")
-            return null;
-        throw error;
-    }
-}
-export function removeFileIfExists(file) {
-    try {
-        fs.unlinkSync(file);
-    }
-    catch (error) {
-        if (error.code !== "ENOENT")
-            throw error;
-    }
-}
+import t from"node:fs";import n from"node:path";function i(r){t.mkdirSync(r,{recursive:!0})}function u(r,e){i(n.dirname(r)),t.writeFileSync(r,`${JSON.stringify(e,null,2)}
+`,"utf8")}function f(r){try{return JSON.parse(t.readFileSync(r,"utf8"))}catch(e){if(e.code==="ENOENT")return null;throw e}}function l(r){try{t.unlinkSync(r)}catch(e){if(e.code!=="ENOENT")throw e}}export{i as ensureDir,f as readJsonFile,l as removeFileIfExists,u as writeJsonFile};