Class FpsProblemImportHandler

packages/fps-importer/index.ts:14–118

};

class FpsProblemImportHandler extends Handler {
    async get() {
        this.response.template = 'problem_import_fps.html';
    }

    async run(domainId: string, result: any) {
        if (!result?.fps) throw new BadRequestError('Selected file is not a valid FPS problemset.');
        for (const p of result.fps.item) {
            const markdown = [p.description?.[0], p.input?.[0], p.output?.[0], p.hint?.[0]].some((i) => i?.includes('[md]'));
            let content = buildContent({
                description: p.description?.[0],
                input: p.input?.[0],
                output: p.output?.[0],
                samples: p.sample_input?.map((input, i) => [input, p.sample_output[i]]),
                hint: p.hint?.[0],
                source: p.source?.join(' '),
            }, 'html', (s) => this.translate(s)).replace(/<math xm<x>lns=/g, '<math xmlns=').replace(/\[\/?md\]/g, '');
            const config: ProblemConfigFile = {
                time: p.time_limit[0]._ + p.time_limit[0].$.unit,
                memory: p.memory_limit[0]._ + p.memory_limit[0].$.unit,
            };
            if (p.remote_oj?.[0]) {
                config.type = ProblemType.Remote;
                config.subType = knownRemoteMapping[p.remote_oj[0]] || p.remote_oj[0];
                config.target = p.remote_id[0];
            }
            const title = decodeHTML(p.title.join(' '));
            const tags = _.filter(p.source, (i: string) => i.trim()).flatMap((i) => i.split(' ')).filter((i) => i);
            const pid = await ProblemModel.add(domainId, null, title, content, this.user._id, tags);
            const tasks: Promise<any>[] = [ProblemModel.addTestdata(domainId, pid, 'config.yaml', Buffer.from(yaml.dump(config)))];
            if (!markdown) tasks.push(ProblemModel.edit(domainId, pid, { html: true }));
            const addTestdata = (node: any, index: string, ext: string) => {
                if (!node || !['object', 'string'].includes(typeof node)) return; // Ignore file not exist
                const id = node.$?.name || `${index}`;
                // `filename` attribute introduced by winterant/OnlineJudge
                // PLEASE, respect the spec
                const filename = node.$?.filename || `${id}.${ext}`;
                const c = node._ || (typeof node === 'string' ? node : '');
                tasks.push(ProblemModel.addTestdata(domainId, pid, filename, Buffer.from(c)));
            };
            if (p.test_output) {
                for (let i = 0; i < p.test_input.length; i++) {
                    addTestdata(p.test_input[i], `${i + 1}`, 'in');
                    addTestdata(p.test_output[i], `${i + 1}`, 'out');
                }
            } else if (p.test_input) {
                // Some version of hustoj exports only test_input section
                for (let i = 0; i < p.test_input.length / 2; i++) {
                    addTestdata(p.test_input[2 * i], `${i + 1}`, 'in');
                    addTestdata(p.test_input[2 * i + 1], `${i + 1}`, 'out');
                }
            }
            if (p.img?.length) {
                for (const img of p.img) {
                    const filename = randomstring(8) + img.src[0].split('/').pop().split('.').pop();
                    tasks.push(ProblemModel.addAdditionalFile(domainId, pid, filename, Buffer.from(img.base64[0], 'base64')));
                    content = content.replace(img.src[0], `file://${filename}`);
                }