import * as strings from '@angular-devkit/core/src/utils/strings';
import { chain, DirEntry, Rule, Tree } from '@angular-devkit/schematics';

import { addProviderToNgModule } from '../../ast/ast-wrappers';
import { modifySourceFile } from '../../rules/modify-source-file.rule';
import { processTemplates } from '../../rules/process-templates.rule';
import { findModuleFilenameInTree, getSubDirFileEntry } from '../../rules/tree-helpers';
import { Folders } from '../../types/folders/folders.enum';
import { findParentModuleName } from '../../types/path-options/path-options.functions';
import {
  getContainingFolderPath,
  validateRegularSchema
} from '../../types/schema-options/schema-options.functions';
import { SchemaOptions } from '../../types/schema-options/schema-options.interface';

function findAppStateTypesFile(directory: DirEntry): string {
  const appStateInterfaceFileEntry = getSubDirFileEntry(
    directory,
    ['types', 'app-state'],
    'app-state.interface.ts'
  );

  if (appStateInterfaceFileEntry) {
    return appStateInterfaceFileEntry.path.replace(/\.ts$/gi, '').slice(1);
  }

  return directory.parent ? findAppStateTypesFile(directory.parent) : null;
}

export default function(options: SchemaOptions): Rule {
  validateRegularSchema(options);

  options.path = getContainingFolderPath(options.path, Folders.Store);

  return chain([
    (tree: Tree) => {
      return processTemplates(
        {
          ...options,
          appStateInterfacePath: findAppStateTypesFile(tree.getDir(options.path)),
          parentName: findParentModuleName(options)
        },
        options.path
      );
    },
    modifySourceFile(
      tree => findModuleFilenameInTree(tree, options),
      (sourceFile, moduleFilename) =>
        addProviderToNgModule(
          sourceFile,
          moduleFilename,
          strings.classify(`${options.name}Store`),
          `.${Folders.Store}/${strings.dasherize(options.name)}.store`
        )
    )
  ]);
}