import { Rule, SchematicContext, Tree, chain, schematic } from '@angular-devkit/schematics';
import { AngularFieldType, IModel } from '@vmfvmf/ywtc-lib';
import { FrontEndGenerator } from '../../../shared/generator';
import { getPrjSrd } from '../../../router.gen';

export function gen(model: IModel): Rule {
  
  const fixDateStr = model.fields
  .filter(field => field.angularType === AngularFieldType.DATE)
  .map(field => `t.${field.name} = new Date((t.${field.name}) + ' 00:00:00');`)
  .join('\n') + '\n';

  return (tree: Tree, _context: SchematicContext) => {   
    let schematics: Rule[] = [
      new FrontEndGenerator(__filename).gen({...model, ...getPrjSrd(), fixDateStr})
    ];
    return chain(schematics)(tree, _context);
  }
}