ginka-generator/data/src/review.ts

39 lines
1.2 KiB
TypeScript

import { readFile, writeFile } from 'fs-extra';
import { chooseFrom, DatasetMergable, mergeDataset } from './utils';
const [target, ...review] = process.argv.slice(2);
const n = getNum();
function getNum() {
const last = review.at(-1);
if (!last) return 1000;
else {
const n = parseInt(last);
if (!n) return 1000;
else {
review.pop();
return n;
}
}
}
(async () => {
const datas = await Promise.all(
review.map(async v => {
const file = await readFile(v, 'utf-8');
return JSON.parse(file) as DatasetMergable<any>;
})
);
const targetFile = await readFile(target, 'utf-8');
const targetData = JSON.parse(targetFile) as DatasetMergable<any>;
const merged = mergeDataset(true, ...datas);
const keys = Object.keys(merged.data);
const toReview = chooseFrom(keys, n);
const reviewData: DatasetMergable<any> = {
datasetId: Math.floor(Math.random() * 1e12),
data: Object.fromEntries(toReview.map(v => [v, merged.data[v]]))
};
const reviewed = mergeDataset(false, targetData, reviewData);
await writeFile(target, JSON.stringify(reviewed), 'utf-8');
})();