import * as strings from '@angular-devkit/core/src/utils/strings';
import * as typescript from 'typescript';

import { Folders } from '../../types/folders/folders.enum';
import { addImportStatementToFile } from '../ast-wrappers';
import { SourceFileModification } from '../source-file-modification.interface';

export function addStateMemberToParentStateInterface(
  sourceFile: typescript.SourceFile,
  folderType: Folders,
  name: string
): SourceFileModification[] {
  const appStateInterface = sourceFile.statements.find(
    statement =>
      statement.kind === typescript.SyntaxKind.InterfaceDeclaration &&
      (statement as typescript.InterfaceDeclaration).name.getText().endsWith('State')
  ) as typescript.InterfaceDeclaration;

  if (appStateInterface) {
    return [
      addImportStatementToFile(
        sourceFile,
        `${strings.classify(name)}State`,
        `../..${folderType}/${strings.dasherize(name)}/types/${strings.dasherize(
          name
        )}-state/${strings.dasherize(name)}-state.interface`
      ),
      {
        index: appStateInterface.members.pos,
        toAdd: `${strings.camelize(name)}State: ${strings.classify(name)}State;`
      }
    ];
  }

  return [];
}