import type { Rule, SchematicContext, Tree } from '@angular-devkit/schematics';
import {
  apply,
  MergeStrategy,
  mergeWith,
  move,
  template,
  url,
} from '@angular-devkit/schematics';
import { normalize } from '@angular-devkit/core';
import { resolve } from 'path';

interface Options {
  directory: string;
  serviceName: string;
}

export function main(options: Options): Rule {
  return (_tree: Tree, context: SchematicContext) => {
    context.logger.info('Creating Lambda Quickstart...');

    const templateSource = apply(url(resolve('.', 'files')), [
      template({ ...options }),
      move(normalize(options.directory)),
      renameFile(options),
    ]);

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

function renameFile(options: Options): Rule {
  return (tree: Tree) => {
    const normalizedPath = normalize(options.directory);
    tree.rename(
      resolve(normalizedPath, 'nvmrc'),
      resolve(normalizedPath, '.nvmrc'),
    );
    tree.rename(
      resolve(normalizedPath, 'env.sample'),
      resolve(normalizedPath, '.env.sample'),
    );
    return tree;
  };
}