Hydro/packages/ui-default/components/problemconfig/reducer/config.ts

import yaml from 'js-yaml';
import type { ProblemConfigFile } from 'hydrooj/src/interface';
import { cloneDeep } from 'lodash';

export default function reducer(state = { type: 'default' } as ProblemConfigFile, action): ProblemConfigFile {
  switch (action.type) {
  case 'CONFIG_LOAD_FULFILLED': {
    // TODO set yaml schema
    return { ...state, ...yaml.load(action.payload.config) as object };
  }
  case 'CONFIG_FORM_UPDATE': {
    const next = { ...state, [action.key]: action.value };
    if (action.key === 'score' && action.value) next.score = +next.score;
    if (!action.value || (typeof action.value === 'object' && !action.value.join(''))) delete next[action.key];
    return next;
  }
  case 'CONFIG_CODE_UPDATE': {
    try {
      return { ...state, ...yaml.load(action.payload) as object };
    } catch {
      return state;
    }
  }
  case 'CONFIG_AUTOCASES_UPDATE': {
    const next = { ...state };
    const autocases = action.value;
    if (autocases.subtasks.length === 0) next.cases = [];
    else if (autocases.subtasks.length === 1) {
      next.score = autocases.subtasks[0].score;
      next.cases = [];
      autocases.subtasks[0].cases.map((i) => next.cases.push({
        time: autocases.subtasks[0].time, memory: autocases.subtasks[0].memory, input: i.input, output: i.output,
      }));
    } else {
      next.subtasks = autocases.subtasks.map((subtask) => (
        { ...subtask, ...{ cases: subtask.cases.map((i) => ({ input: i.input, output: i.output })) } }));
    }
    return next;
  }
  case 'CONFIG_SUBTASKS_SWITCH': {
    const next = { ...state };
    next.subtasks = next.cases.map((k, v) => ({
      id: v, time: k.time, memory: k.memory, cases: [{ input: k.input, output: k.output }],
    }));
    if (!next.subtasks.length) next.subtasks.push({ id: 0 });
    delete next.cases;
    return next;
  }
  case 'CONFIG_CASES_UPDATE': {
    if (['time', 'memory'].includes(action.casesKey) && action.value !== '') action.value = +action.value;
    if (action.key === 'cases-add') return { ...state, cases: [...state.cases, action.value] };
    if (action.key === 'cases-delete') {
      return { ...state, cases: state.cases.filter((k, v) => v !== action.value) };
    }
    const cases = cloneDeep(state.cases);
    cases[action.casesId][action.casesKey] = action.value;
    if (action.value === '') delete cases[action.casesId][action.casesKey];
    return { ...state, cases };
  }
  case 'CONFIG_SUBTASK_UPDATE': {
    const subtasks = cloneDeep(state.subtasks);
    const subsubtasks = cloneDeep(state.subtasks[action.id]);
    if (action.value !== ''
     && (['time', 'memory'].includes(action.casesKey) || ['time', 'memory', 'score', 'id'].includes(action.key))
    ) action.value = +action.value;
    if (action.key === 'if' && action.value.join('') !== '') action.value = action.value.map((i) => +i);
    if (action.key.split('-')[0] === 'cases') {
      if (action.key === 'cases-add') subsubtasks.cases.push(action.value);
      else if (action.key === 'cases-edit') subsubtasks.cases[action.casesId][action.casesKey] = action.value;
      else if (action.key === 'cases-delete') {
        subsubtasks.cases = subsubtasks.cases.filter((k, v) => v !== action.value);
      }
    } else if (action.key === 'add') subtasks.push({ time: 1000, memory: 256, cases: [] });
    else if (action.key === 'delete') delete subtasks[action.key];
    else {
      if (action.value === '' || (action.key === 'if' && action.value.join('') === '')) delete subsubtasks[action.key];
      else subsubtasks[action.key] = action.value;
    }
    if (action.key !== 'delete') subtasks[action.id] = subsubtasks;
    return { ...state, subtasks };
  }
  default:
    return state;
  }
}