Files
fiveq-website-2026/docs/scripts/product-sites.mjs
T
2026-08-24 15:37:08 +08:00

116 lines
3.1 KiB
JavaScript

import fs from 'node:fs';
import path from 'node:path';
import { fileURLToPath } from 'node:url';
export const PRODUCT_DIRS = ['smartbuddy', 'smartinput', 'smarthanzi', 'massiveedit', 'html'];
const SKIP_NAMES = new Set([
'.DS_Store',
'.texpadtmp',
'smartbuddy_readme',
]);
const SKIP_EXT = new Set([
'.tex',
'.aux',
'.log',
'.toc',
'.out',
'.synctex.gz',
]);
const MIME = {
'.html': 'text/html; charset=utf-8',
'.css': 'text/css; charset=utf-8',
'.js': 'text/javascript; charset=utf-8',
'.mjs': 'text/javascript; charset=utf-8',
'.json': 'application/json; charset=utf-8',
'.svg': 'image/svg+xml',
'.png': 'image/png',
'.jpg': 'image/jpeg',
'.jpeg': 'image/jpeg',
'.webp': 'image/webp',
'.gif': 'image/gif',
'.ico': 'image/x-icon',
'.woff2': 'font/woff2',
'.txt': 'text/plain; charset=utf-8',
'.md': 'text/plain; charset=utf-8',
'.apk': 'application/vnd.android.package-archive',
'.deb': 'application/vnd.debian.binary-package',
'.pkg': 'application/octet-stream',
'.exe': 'application/octet-stream',
'.zip': 'application/zip',
};
export function shouldCopy(src) {
const base = path.basename(src);
if (SKIP_NAMES.has(base)) return false;
if (base.startsWith('.')) return false;
if (SKIP_EXT.has(path.extname(base))) return false;
return true;
}
export function copyProductSites(destRoot, sourceRoot = process.cwd()) {
for (const dir of PRODUCT_DIRS) {
const from = path.join(sourceRoot, dir);
if (!fs.existsSync(from)) continue;
fs.cpSync(from, path.join(destRoot, dir), {
recursive: true,
filter: shouldCopy,
});
}
}
function resolveProductFile(root, urlPath) {
const clean = urlPath.split('?')[0];
const product = PRODUCT_DIRS.find(
(name) => clean === `/${name}` || clean === `/${name}/` || clean.startsWith(`/${name}/`),
);
if (!product) return null;
let rel = clean.slice(product.length + 2);
if (!rel || rel.endsWith('/')) rel = `${rel}index.html`;
const file = path.join(root, product, rel);
if (fs.existsSync(file) && fs.statSync(file).isFile()) return file;
const asIndex = path.join(root, product, rel, 'index.html');
if (fs.existsSync(asIndex) && fs.statSync(asIndex).isFile()) return asIndex;
return null;
}
function serveProduct(root, req, res, next) {
const file = resolveProductFile(root, req.url ?? '');
if (!file) return next();
res.statusCode = 200;
res.setHeader('Content-Type', MIME[path.extname(file).toLowerCase()] ?? 'application/octet-stream');
fs.createReadStream(file).pipe(res);
}
export function productSites() {
const root = process.cwd();
return {
name: 'product-sites',
hooks: {
'astro:config:setup'({ updateConfig }) {
updateConfig({
vite: {
plugins: [
{
name: 'serve-product-sites',
configureServer(server) {
server.middlewares.use((req, res, next) => serveProduct(root, req, res, next));
},
},
],
},
});
},
'astro:build:done'({ dir }) {
copyProductSites(fileURLToPath(dir));
},
},
};
}