60 lines
1.9 KiB
JavaScript
60 lines
1.9 KiB
JavaScript
|
import { copyFile, mkdir, readdir, readFile, writeFile } from "node:fs/promises";
|
|||
|
|
import path from "node:path";
|
|||
|
|
|
|||
|
|
const sourceRoot = process.argv[2];
|
|||
|
|
const targetRoot = process.env.UPLOAD_DIR || process.argv[3];
|
|||
|
|
if (!sourceRoot || !targetRoot) {
|
|||
|
|
throw new Error(
|
|||
|
|
"用法：UPLOAD_DIR=/data/koc/uploads npm run storage:import -- /path/to/r2-export",
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const contentTypes = {
|
|||
|
|
".avif": "image/avif",
|
|||
|
|
".gif": "image/gif",
|
|||
|
|
".jpeg": "image/jpeg",
|
|||
|
|
".jpg": "image/jpeg",
|
|||
|
|
".png": "image/png",
|
|||
|
|
".webp": "image/webp",
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
let copied = 0;
|
|||
|
|
async function walk(directory) {
|
|||
|
|
const entries = await readdir(directory, { withFileTypes: true });
|
|||
|
|
for (const entry of entries) {
|
|||
|
|
const absolute = path.join(directory, entry.name);
|
|||
|
|
if (entry.isDirectory()) {
|
|||
|
|
await walk(absolute);
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
if (entry.name.endsWith(".metadata.json")) continue;
|
|||
|
|
const relative = path.relative(sourceRoot, absolute);
|
|||
|
|
if (relative.startsWith("..") || path.isAbsolute(relative)) {
|
|||
|
|
throw new Error("导入目录越界");
|
|||
|
|
}
|
|||
|
|
const destination = path.join(targetRoot, relative);
|
|||
|
|
await mkdir(path.dirname(destination), { recursive: true });
|
|||
|
|
await copyFile(absolute, destination);
|
|||
|
|
const sourceMetadata = `${absolute}.metadata.json`;
|
|||
|
|
const metadataDestination = `${destination}.metadata.json`;
|
|||
|
|
let metadata;
|
|||
|
|
try {
|
|||
|
|
metadata = JSON.parse(await readFile(sourceMetadata, "utf8"));
|
|||
|
|
} catch {
|
|||
|
|
metadata = {
|
|||
|
|
contentType:
|
|||
|
|
contentTypes[path.extname(entry.name).toLowerCase()] ||
|
|||
|
|
"application/octet-stream",
|
|||
|
|
customMetadata: { source: "r2-export" },
|
|||
|
|
uploadedAt: new Date().toISOString(),
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
await writeFile(metadataDestination, JSON.stringify(metadata), "utf8");
|
|||
|
|
copied += 1;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
await mkdir(targetRoot, { recursive: true });
|
|||
|
|
await walk(sourceRoot);
|
|||
|
|
console.info(`imported objects: ${copied}`);
|