import { ABI, ContractRegister } from '@neo-one/client';
import { tsUtils } from '@neo-one/ts-utils';
import _ from 'lodash';
import { RawSourceMap } from 'source-map';
import ts from 'typescript';
import { compile, WithLinked } from './compile';
import { createContextForPath, updateContext } from './createContext';
import { transpile } from './transpile';
import { normalizePath } from './utils';

export interface CompileContractOptions extends WithLinked {
  readonly filePath: string;
}

export interface CompileContractResult {
  readonly abi: ABI;
  readonly diagnostics: ReadonlyArray<ts.Diagnostic>;
  readonly contract: ContractRegister;
  readonly sourceMap: Promise<RawSourceMap>;
}

export const compileContract = ({
  filePath: filePathIn,
  linked: linkedIn = {},
}: CompileContractOptions): CompileContractResult => {
  const filePath = normalizePath(filePathIn);
  const linked = _.fromPairs(Object.entries(linkedIn).map(([key, value]) => [normalizePath(key), value]));
  const transpileContext = createContextForPath(filePath);
  const transpileResult = transpile({
    sourceFile: tsUtils.file.getSourceFileOrThrow(transpileContext.program, filePath),
    context: transpileContext,
  });
  const context =
    transpileResult === undefined
      ? transpileContext
      : updateContext(transpileContext, { [filePath]: transpileResult.text });

  const { abi, sourceMap: finalSourceMap, contract } = compile({
    sourceFile: tsUtils.file.getSourceFileOrThrow(context.program, filePath),
    context,
    linked,
    sourceMaps: transpileResult === undefined ? {} : { [filePath]: transpileResult.sourceMap },
  });

  return {
    diagnostics: context.diagnostics,
    sourceMap: finalSourceMap,
    abi,
    contract,
  };
};