import { Rule } from '@angular-devkit/schematics';
import { IEnumDef } from '@vmfvmf/ywtc-lib';
import { BackEndGenerator } from '../../../shared/generator';
import { getPrjSrd, setEnumDef } from '../../../router.gen';

export function gen(enumDef: IEnumDef): Rule {
  setEnumDef(enumDef);
  const allOps = enumDef.options.map(opt => `${opt.value}`).join(',\n\t');
  return new BackEndGenerator(__filename).gen({...enumDef, ...getPrjSrd(), allOps});
}