angular-crud/src/crud-module/index.ts

Generate CRUD applications with Angular, the CLI, and Schematics

import {
  apply,
  MergeStrategy,
  mergeWith,
  move,
  Rule,
  SchematicsException,
  template,
  Tree,
  url
} from '@angular-devkit/schematics';
import { strings as stringUtils } from '@angular-devkit/core';

import * as JSON5 from 'json5';
import * as crudModelUtils from '../utils/crud-model-utils'

import { CrudOptions } from './schema';
import { CrudModel } from './model';
import { capitalize } from '@angular-devkit/core/src/utils/strings';
import { getWorkspace } from '@schematics/angular/utility/workspace';
import { addModuleImportToModule } from '@angular/cdk/schematics';

export const BOOTSTRAP = 'bootstrap';
export const MATERIAL = 'material';
export const PAPER_DASHBOARD = 'paper-dashboard';

function getFramework(host: Tree): string {
  let possibleFiles = ['/package.json'];
  const path = possibleFiles.filter(path => host.exists(path))[0];

  const configBuffer = host.read(path);
  if (configBuffer === null) {
    throw new SchematicsException(`Could not find (${path})`);
  } else {
    const content = JSON.parse(configBuffer.toString());
    if (content.dependencies['bootstrap']) {
      return BOOTSTRAP;
    } else if (content.dependencies['@angular/material']) {
      return MATERIAL;
    } else {
      return PAPER_DASHBOARD;
    }
  }
}

export function generate(options: CrudOptions): Rule {

  return async (host: Tree) => {
    // allow passing the CSS framework in (for testing)
    let cssFramework = options.style;

    // if no CSS framework defined, try to detect it
    // defaults to paper-dashboard if nothing found (for backward compatibility)
    if (!cssFramework) {
      cssFramework = getFramework(host);
    }

    const workspace = await getWorkspace(host);
    if (!options.project) {
      options.project = workspace.projects.keys().next().value;
    }
    const project = workspace.projects.get(options.project);
    const appPath = `${project?.sourceRoot}/app`;

    const modelFile = `${appPath}/${options.name}/${options.model}`;
    const modelBuffer = host.read(modelFile);

    if (modelBuffer === null) {
      throw new SchematicsException(`Model file ${options.model} does not exist.`);
    }

    const modelJson = modelBuffer.toString('utf-8');
    const model = JSON5.parse(modelJson) as CrudModel;

    // add imports to app.module.ts
    addModuleImportToModule(host,
      `${appPath}/app.module.ts`,
      `${capitalize(model.entity)}Module`,
      `./${options.name}/${model.entity}.module`);

    const templateSource = apply(url(`./files/${cssFramework}`), [
      template({
        ...stringUtils,
        ...options,
        ...crudModelUtils as any,
        model
      }),
      move(`${appPath}/${options.name}`),
    ]);

    return mergeWith(templateSource, MergeStrategy.Overwrite);
  }
}