one
Version:
One is a new React Framework that makes Vite serve both native and web.
75 lines (74 loc) • 2.63 kB
JavaScript
import { existsSync } from "fs";
import { dirname, relative, resolve } from "path";
import { processImageMeta } from "../../image/getImageData.native.js";
var IMAGEDATA_SUFFIX = "?imagedata";
var VIRTUAL_PREFIX = "\0imagedata:";
function imageDataPlugin() {
var publicDir;
var root;
function getSrcPath(filePath) {
return filePath.startsWith(publicDir) ? "/" + relative(publicDir, filePath) : "/" + relative(root, filePath);
}
function isPathWithinBounds(filePath, allowedDir) {
var resolved = resolve(filePath);
var allowed = resolve(allowedDir);
return resolved.startsWith(allowed + "/") || resolved === allowed;
}
function createImageDataExport(src) {
var width = arguments.length > 1 && arguments[1] !== void 0 ? arguments[1] : 0,
height = arguments.length > 2 && arguments[2] !== void 0 ? arguments[2] : 0,
blurDataURL = arguments.length > 3 && arguments[3] !== void 0 ? arguments[3] : "";
return `export default ${JSON.stringify({
src,
width,
height,
blurDataURL
})}`;
}
return {
name: "one:imagedata",
enforce: "pre",
configResolved(resolvedConfig) {
publicDir = resolvedConfig.publicDir;
root = resolvedConfig.root;
},
async resolveId(id, importer) {
if (!id.endsWith(IMAGEDATA_SUFFIX)) return null;
var cleanId = id.slice(0, -IMAGEDATA_SUFFIX.length);
var filePath;
var allowedDir;
if (cleanId.startsWith("/")) {
filePath = resolve(publicDir, cleanId.slice(1));
allowedDir = publicDir;
} else if (importer) {
filePath = resolve(dirname(importer.replace(VIRTUAL_PREFIX, "")), cleanId);
allowedDir = root;
} else {
filePath = resolve(root, cleanId);
allowedDir = root;
}
if (!isPathWithinBounds(filePath, allowedDir)) {
console.warn(`[one] ?imagedata: Path traversal blocked: ${cleanId}`);
return null;
}
if (!existsSync(filePath)) {
console.warn(`[one] ?imagedata: File not found: ${filePath}`);
return null;
}
return VIRTUAL_PREFIX + filePath;
},
async load(id) {
if (!id.startsWith(VIRTUAL_PREFIX)) return null;
var filePath = id.slice(VIRTUAL_PREFIX.length);
var src = getSrcPath(filePath);
this.addWatchFile(filePath);
var meta = await processImageMeta(filePath);
if (!meta) {
return createImageDataExport(src);
}
return createImageDataExport(src, meta.width, meta.height, meta.blurDataURL);
}
};
}
export { imageDataPlugin };
//# sourceMappingURL=imageDataPlugin.native.js.map