broccoli-typescript-compiler/src/compiler.ts

import * as ts from "typescript";
import ConfigParser from "./compiler/config-parser";
import createCompilerHost from "./compiler/create-compiler-host";
import Input from "./compiler/input-io";
import OutputPatcher from "./compiler/output-patcher";
import PathResolver from "./compiler/path-resolver";
import SourceCache from "./compiler/source-cache";
import {
  normalizePath,
  relativePathWithin,
  toAbsolutePath,
} from "./fs/path-utils";
import { heimdall } from "./helpers";
import {
  AbsolutePath,
  DiagnosticsHandler,
  NormalizedOptions,
} from "./interfaces";

export default class Compiler {
  private resolver: PathResolver;
  private workingPath: AbsolutePath;
  private rootPath: AbsolutePath;
  private buildPath: AbsolutePath | undefined;
  private input: Input;
  private configParser: ConfigParser;
  private sourceCache: SourceCache | undefined;
  private output: OutputPatcher;
  private program: ts.Program | undefined;

  constructor(
    public inputPath: AbsolutePath,
    public outputPath: AbsolutePath,
    public options: NormalizedOptions,
    private diagnosticsHandler: DiagnosticsHandler
  ) {
    const workingPath = (this.workingPath = options.workingPath);
    const rootPath = (this.rootPath = options.rootPath);
    this.buildPath = options.buildPath;
    const resolver = (this.resolver = new PathResolver(rootPath, inputPath));
    const input = (this.input = new Input(resolver));
    this.configParser = new ConfigParser(
      options.projectPath,
      options.rawConfig,
      options.configFileName,
      options.compilerOptions,
      workingPath,
      input
    );
    this.output = new OutputPatcher(outputPath);
  }

  public compile() {
    const config = this.parseConfig();

    const sourceCache = this.getSourceCache(config.options);

    const program = this.createProgram(config, sourceCache);

    this.emitDiagnostics(program);

    sourceCache.releaseUnusedSourceFiles(program);

    this.emitProgram(program, this.resolveBuildPath(config.options));

    this.patchOutput();

    this.resetCaches();
  }

  protected parseConfig() {
    const token = heimdall.start("TypeScript:parseConfig");
    const config = this.configParser.parseConfig();
    heimdall.stop(token);
    return config;
  }

  protected getSourceCache(options: ts.CompilerOptions) {
    let sourceCache = this.sourceCache;
    if (sourceCache === undefined) {
      sourceCache = this.sourceCache = new SourceCache(this.resolver, options);
    } else {
      sourceCache.updateOptions(options);
    }
    return sourceCache;
  }

  protected createProgram(
    config: ts.ParsedCommandLine,
    sourceCache: SourceCache
  ): ts.Program {
    const token = heimdall.start("TypeScript:createProgram");

    const host = createCompilerHost(
      this.workingPath,
      this.input,
      sourceCache,
      config.options
    );

    const oldProgram = this.program;
    const program = ts.createProgram(
      config.fileNames,
      config.options,
      host,
      oldProgram
    );
    this.program = program;

    heimdall.stop(token);
    return program;
  }

  protected emitDiagnostics(program: ts.Program) {
    // this is where bindings are resolved and typechecking is done
    const token = heimdall.start("TypeScript:emitDiagnostics");
    const diagnostics = ts.getPreEmitDiagnostics(program);
    heimdall.stop(token);
    this.diagnosticsHandler.check(diagnostics);
  }

  protected resolveBuildPath(options: ts.CompilerOptions): AbsolutePath {
    if (this.buildPath !== undefined) {
      return this.buildPath;
    }
    if (options.outDir !== undefined) {
      return normalizePath(options.outDir) as AbsolutePath;
    }
    return this.rootPath;
  }

  protected emitProgram(program: ts.Program, buildPath: AbsolutePath) {
    const token = heimdall.start("TypeScript:emitProgram");
    const { output } = this;

    const emitResult = program.emit(
      undefined,
      (fileName: string, data: string) => {
        /* tslint:disable:no-console */
        // the fileName is absolute but not normalized if outDir is not normalized
        const relativePath = relativePathWithin(
          buildPath,
          toAbsolutePath(fileName, this.workingPath)
        );
        if (relativePath) {
          output.add(relativePath, data);
        }
      }
    );
    heimdall.stop(token);
    this.diagnosticsHandler.check(emitResult.diagnostics);
  }

  protected patchOutput() {
    const token = heimdall.start("TypeScript:patchOutput");
    this.output.patch();
    heimdall.stop(token);
  }

  protected resetCaches() {
    this.resolver.reset();
    this.input.reset();
  }
}