UNPKG

@electron/asar

Version:

Creating Electron app packages

162 lines 6.64 kB
import path from 'node:path'; import { wrappedFs as fs } from './wrapped-fs.js'; import { Pickle } from './pickle.js'; import { Filesystem } from './filesystem.js'; import stream from 'node:stream/promises'; let filesystemCache = Object.create(null); async function copyFile(dest, src, filename) { const srcFile = path.join(src, filename); const targetFile = path.join(dest, filename); const [content, stats] = await Promise.all([ fs.readFile(srcFile), fs.stat(srcFile), fs.mkdirp(path.dirname(targetFile)), ]); return fs.writeFile(targetFile, content, { mode: stats.mode }); } async function streamTransformedFile(stream, outStream) { return new Promise((resolve, reject) => { stream.pipe(outStream, { end: false }); stream.on('error', reject); stream.on('end', () => resolve()); }); } const writeFileListToStream = async function (dest, filesystem, out, lists, metadata) { const { files, links } = lists; for (const file of files) { if (file.unpack) { // the file should not be packed into archive const filename = path.relative(filesystem.getRootPath(), file.filename); await copyFile(`${dest}.unpacked`, filesystem.getRootPath(), filename); } else { const transformed = metadata[file.filename].transformed; const stream = fs.createReadStream(transformed ? transformed.path : file.filename); await streamTransformedFile(stream, out); } } for (const file of links.filter((f) => f.unpack)) { // the symlink needs to be recreated outside in .unpacked const filename = path.relative(filesystem.getRootPath(), file.filename); const link = await fs.readlink(file.filename); await createSymlink(dest, filename, link); } return out.end(); }; export async function writeFilesystem(dest, filesystem, lists, metadata) { const out = await createFilesystemWriteStream(filesystem, dest); return writeFileListToStream(dest, filesystem, out, lists, metadata); } export async function streamFilesystem(dest, filesystem, lists) { const out = await createFilesystemWriteStream(filesystem, dest); const { files, links } = lists; for await (const file of files) { // the file should not be packed into archive if (file.unpack) { const targetFile = path.join(`${dest}.unpacked`, file.filename); await fs.mkdirp(path.dirname(targetFile)); const writeStream = fs.createWriteStream(targetFile, { mode: file.mode }); await stream.pipeline(file.streamGenerator(), writeStream); } else { await streamTransformedFile(file.streamGenerator(), out); } } for (const file of links.filter((f) => f.unpack && f.link)) { // the symlink needs to be recreated outside in .unpacked await createSymlink(dest, file.filename, file.link); } return out.end(); } export function readArchiveHeaderSync(archivePath) { const fd = fs.openSync(archivePath, 'r'); let size; let headerBuf; try { const sizeBuf = Buffer.alloc(8); if (fs.readSync(fd, sizeBuf, 0, 8, null) !== 8) { throw new Error('Unable to read header size'); } const sizePickle = Pickle.createFromBuffer(sizeBuf); size = sizePickle.createIterator().readUInt32(); headerBuf = Buffer.alloc(size); if (fs.readSync(fd, headerBuf, 0, size, null) !== size) { throw new Error('Unable to read header'); } } finally { fs.closeSync(fd); } const headerPickle = Pickle.createFromBuffer(headerBuf); const header = headerPickle.createIterator().readString(); return { headerString: header, header: JSON.parse(header), headerSize: size }; } export function readFilesystemSync(archivePath) { if (!filesystemCache[archivePath]) { const header = readArchiveHeaderSync(archivePath); const filesystem = new Filesystem(archivePath); filesystem.setHeader(header.header, header.headerSize); filesystemCache[archivePath] = filesystem; } return filesystemCache[archivePath]; } export function uncacheFilesystem(archivePath) { if (filesystemCache[archivePath]) { filesystemCache[archivePath] = undefined; return true; } return false; } export function uncacheAll() { filesystemCache = {}; } export function readFileSync(filesystem, filename, info) { let buffer = Buffer.alloc(info.size); if (info.size <= 0) { return buffer; } if (info.unpacked) { // it's an unpacked file, copy it. buffer = fs.readFileSync(path.join(`${filesystem.getRootPath()}.unpacked`, filename)); } else { // Node throws an exception when reading 0 bytes into a 0-size buffer, // so we short-circuit the read in this case. const fd = fs.openSync(filesystem.getRootPath(), 'r'); try { const offset = 8 + filesystem.getHeaderSize() + parseInt(info.offset); fs.readSync(fd, buffer, 0, info.size, offset); } finally { fs.closeSync(fd); } } return buffer; } async function createFilesystemWriteStream(filesystem, dest) { const headerPickle = Pickle.createEmpty(); headerPickle.writeString(JSON.stringify(filesystem.getHeader())); const headerBuf = headerPickle.toBuffer(); const sizePickle = Pickle.createEmpty(); sizePickle.writeUInt32(headerBuf.length); const sizeBuf = sizePickle.toBuffer(); const out = fs.createWriteStream(dest); await new Promise((resolve, reject) => { out.on('error', reject); out.write(sizeBuf); return out.write(headerBuf, () => resolve()); }); return out; } async function createSymlink(dest, filepath, link) { // if symlink is within subdirectories, then we need to recreate dir structure await fs.mkdirp(path.join(`${dest}.unpacked`, path.dirname(filepath))); // create symlink within unpacked dir await fs.symlink(link, path.join(`${dest}.unpacked`, filepath)).catch(async (error) => { if (error.code === 'EPERM' && error.syscall === 'symlink') { throw new Error('Could not create symlinks for unpacked assets. On Windows, consider activating Developer Mode to allow non-admin users to create symlinks by following the instructions at https://docs.microsoft.com/en-us/windows/apps/get-started/enable-your-device-for-development.'); } throw error; }); } //# sourceMappingURL=disk.js.map