import {Compiler as TraceurCompiler} from 'traceur/src/Compiler';
import {ClassTransformer} from './dart_class_transformer';
import {DartTreeWriter} from './dart_writer';
import {CollectingErrorReporter} from 'traceur/src/util/CollectingErrorReporter';
import {Parser} from './parser';
import {SourceFile} from 'traceur/src/syntax/SourceFile';
import {
  options as traceurOptions
} from 'traceur/src/Options';

export class Compiler extends TraceurCompiler {
  constructor(options, sourceRoot) {
    super(options, sourceRoot);
  }
  compile(source, filename) {
    var parsed = this.parse(source, filename || '<unknown_file>');
    if (this.options_.outputLanguage === 'dart') {
      return this.writeDart(this.transformDart(parsed), filename);
    } else {
      return this.write(this.transform(parsed));
    }
  }
  transformDart(tree) {
    var transformer = new ClassTransformer();
    return transformer.transformAny(tree);
  }
  writeDart(tree, filename) {
    var writer = new DartTreeWriter();
    writer.visitAny(tree);
    // TODO: Do this in the writer...
    var library = 'library '+this.resolveModuleName(filename).replace(/\//g, '.')+';\n';
    return library + writer.toString();
  }
  // Copy of the original method to use our custom Parser
  parse(content, sourceName) {
    if (!content) {
      throw new Error('Compiler: no content to compile.');
    } else if (!sourceName) {
      throw new Error('Compiler: no source name for content.');
    }

    this.sourceMapGenerator_ = null;
    // Here we mutate the global/module options object to be used in parsing.
    traceurOptions.setFromObject(this.options_);

    var errorReporter = new CollectingErrorReporter();
    sourceName = this.sourceName(sourceName);
    var sourceFile = new SourceFile(sourceName, content);
    var parser = new Parser(sourceFile, errorReporter);
    var tree =
        this.options_.script ? parser.parseScript() : parser.parseModule();
    this.throwIfErrors(errorReporter);
    return tree;
  }
}