Hydro/packages/hydrooj/src/lib/testdataConfig.ts

import readYamlCases from '@hydrooj/utils/lib/cases';
import { load } from 'js-yaml';
import type { ProblemConfig } from '../interface';

interface ParseResult {
    count: number,
    memoryMax: number,
    memoryMin: number,
    timeMax: number,
    timeMin: number,
}

export async function parseConfig(config: string | ProblemConfig = {}) {
    const result: ParseResult = {
        count: 0,
        memoryMin: Number.MAX_SAFE_INTEGER,
        memoryMax: 0,
        timeMin: Number.MAX_SAFE_INTEGER,
        timeMax: 0,
    };
    let cfg: ProblemConfig = {};
    if (typeof config === 'string' && config.length) {
        // TODO should validate here?
        cfg = await readYamlCases(load(config) as Record<string, any>);
    } else if (typeof config === 'object') cfg = await readYamlCases(config);
    if (cfg.subtasks.length) {
        for (const subtask of cfg.subtasks) {
            result.memoryMax = Math.max(result.memoryMax, subtask.memory);
            result.memoryMin = Math.min(result.memoryMin, subtask.memory);
            result.timeMax = Math.max(result.timeMax, subtask.time);
            result.timeMin = Math.min(result.timeMin, subtask.time);
        }
    } else if (cfg.time || cfg.memory) {
        if (cfg.time) result.timeMax = result.timeMin = cfg.time as unknown as number;
        if (cfg.memory) result.memoryMax = result.memoryMin = cfg.memory as unknown as number;
    }
    if (result.memoryMax < result.memoryMin) {
        result.memoryMax = result.memoryMin = 256;
    }
    if (result.timeMax < result.timeMin) {
        result.timeMax = result.timeMin = 1000;
    }
    return result;
}

global.Hydro.lib.testdataConfig = { parseConfig };