import chalk from "chalk";
import * as path from "path";
import * as fs from "fs-extra";
import * as lambda from "@aws-cdk/aws-lambda";
import { bundle as bundlePython } from "./python/bundling";
import { FunctionBundlePythonProps } from "../Function";
import { addExtensionToHandler, getHandlerFullPosixPath } from "./builder";

interface BuilderProps {
  readonly srcPath: string;
  readonly handler: string;
  readonly runtime: lambda.Runtime;
  readonly bundle: FunctionBundlePythonProps;
}

interface BuilderOutput {
  readonly outCode: lambda.Code;
  readonly outHandler: string;
}

// Do not re-bundle dependencies for the same srcPath
const existingBundlesBySrcPath: { [srcPath: string]: lambda.Code } = {};

export function builder(builderProps: BuilderProps): BuilderOutput {
  const { bundle, runtime, srcPath, handler } = builderProps;
  const handlerPosixPath = getHandlerFullPosixPath(srcPath, handler);

  console.log(chalk.grey(`Building Lambda function ${handlerPosixPath}`));

  // Check entry path exists
  const entryPath = path.join(srcPath, addExtensionToHandler(handler, ".py"));
  if (!fs.existsSync(path.join(entryPath))) {
    throw new Error(`Cannot find a handler file at "${entryPath}".`);
  }

  // Bundle dependency with code
  let outCode;
  if (existingBundlesBySrcPath[srcPath]) {
    outCode = existingBundlesBySrcPath[srcPath];
  } else {
    console.log(
      chalk.grey(`Bundling dependencies for ${srcPath} in Docker...`)
    );
    outCode = bundlePython({
      installCommands: bundle && bundle.installCommands,
      runtime,
      entry: srcPath,
      outputPathSuffix: ".",
    });
    existingBundlesBySrcPath[srcPath] = outCode;
  }

  return {
    outCode,
    outHandler: handler,
  };
}