import { chain, Rule, noop, Tree, applyToSubtree } from '@angular-devkit/schematics';
import {
  addModuleImportToModule,
  buildComponent,
  findModuleFromOptions,
} from '@angular/cdk/schematics';

import * as figlet from 'figlet';

import { Schema } from './schema';

/**
 * Scaffolds a new sidebar component.
 * Internally it bootstraps the base component schematic
 */
export default function(options: Schema): Rule {
  console.log(`\n${figlet.textSync('ng g sidebar')}\n`);
  const rules: Rule[] = [
    buildComponent({ ...options }),
    options['skipImport'] ? noop() : addComponentModulesToModule(options)
  ];

  const wd = options.workingDirectory;
  return wd ? applyToSubtree(wd, rules) : chain(rules);
}

/**
 * Adds the required modules to the relative module.
 */
function addComponentModulesToModule(options: Schema) {
  return (host: Tree) => {
    const modulePath = findModuleFromOptions(host, options) || null;
    if (modulePath) {
      addModuleImportToModule(host, modulePath, 'SidebarModule', 'primeng/sidebar');
      addModuleImportToModule(host, modulePath, 'ButtonModule', 'primeng/button');
    }
    return host;
  };
}