Add emitter scaffolding

Author: rbuckton

cmd/tsgo/main.go

@@ -26,6 +26,7 @@ var printTypes = false
 var pretty = true
 var listFiles = false
 var pprofDir = ""
+var outDir = ""
 
 func printDiagnostic(d *ast.Diagnostic, level int, comparePathOptions tspath.ComparePathsOptions) {
 	file := d.File()
@@ -57,10 +58,16 @@ func main() {
 	flag.BoolVar(&pretty, "pretty", true, "Get prettier errors")
 	flag.BoolVar(&listFiles, "listfiles", false, "List files in the program")
 	flag.StringVar(&pprofDir, "pprofdir", "", "Generate pprof CPU/memory profiles to the given directory")
+	flag.StringVar(&outDir, "outDir", "", "Emit to the given directory")
 	flag.Parse()
 
 	rootPath := flag.Arg(0)
-	compilerOptions := &core.CompilerOptions{Strict: core.TSTrue, Target: core.ScriptTargetESNext, ModuleKind: core.ModuleKindNodeNext}
+	compilerOptions := &core.CompilerOptions{Strict: core.TSTrue, Target: core.ScriptTargetESNext, ModuleKind: core.ModuleKindNodeNext, NoEmit: core.TSTrue}
+	if len(outDir) > 0 {
+		compilerOptions.NoEmit = core.TSFalse
+		compilerOptions.OutDir = outDir
+	}
+
 	currentDirectory, err := os.Getwd()
 	if err != nil {
 		fmt.Fprintf(os.Stderr, "Error getting current directory: %v\n", err)
@@ -99,6 +106,13 @@ func main() {
 	}
 	compileTime := time.Since(startTime)
 
+	startTime = time.Now()
+	if len(outDir) > 0 {
+		result := program.Emit(&ts.EmitOptions{})
+		diagnostics = append(diagnostics, result.Diagnostics...)
+	}
+	emitTime := time.Since(startTime)
+
 	var memStats runtime.MemStats
 	runtime.GC()
 	runtime.GC()
@@ -133,6 +147,7 @@ func main() {
 	fmt.Printf("Files:         %v\n", len(program.SourceFiles()))
 	fmt.Printf("Types:         %v\n", program.TypeCount())
 	fmt.Printf("Compile time:  %v\n", compileTime)
+	fmt.Printf("Emit time:     %v\n", emitTime)
 	fmt.Printf("Memory used:   %vK\n", memStats.Alloc/1024)
 }
 

internal/compiler/emitHost.go

@@ -0,0 +1,67 @@
+package compiler
+
+import (
+	"sync"
+
+	"github.com/microsoft/typescript-go/internal/ast"
+	"github.com/microsoft/typescript-go/internal/core"
+)
+
+type WriteFileData struct {
+	SourceMapUrlPos int
+	// BuildInfo BuildInfo
+	Diagnostics      []*ast.Diagnostic
+	DiffersOnlyInMap bool
+	SkippedDtsWrite  bool
+}
+
+// NOTE: EmitHost operations must be thread-safe
+type EmitHost interface {
+	Options() *core.CompilerOptions
+	SourceFiles() []*ast.SourceFile
+	UseCaseSensitiveFileNames() bool
+	GetCurrentDirectory() string
+	CommonSourceDirectory() string
+	IsEmitBlocked(file string) bool
+	WriteFile(fileName string, text string, writeByteOrderMark bool, relatedSourceFiles []*ast.SourceFile, data *WriteFileData) error
+}
+
+var _ EmitHost = (*emitHost)(nil)
+
+// NOTE: emitHost operations must be thread-safe
+type emitHost struct {
+	program *Program
+
+	commonSourceDirectory      string
+	commonSourceDirectoryMutex sync.Mutex
+}
+
+func (host *emitHost) Options() *core.CompilerOptions { return host.program.Options() }
+func (host *emitHost) SourceFiles() []*ast.SourceFile { return host.program.SourceFiles() }
+func (host *emitHost) GetCurrentDirectory() string    { return host.program.host.GetCurrentDirectory() }
+func (host *emitHost) CommonSourceDirectory() string {
+	if len(host.commonSourceDirectory) > 0 {
+		return host.commonSourceDirectory
+	}
+
+	host.commonSourceDirectoryMutex.Lock()
+	defer host.commonSourceDirectoryMutex.Unlock()
+
+	// double-check after lock
+	if len(host.commonSourceDirectory) > 0 {
+		return host.commonSourceDirectory
+	}
+
+	host.commonSourceDirectory = host.program.CommonSourceDirectory()
+	return host.commonSourceDirectory
+}
+func (host *emitHost) UseCaseSensitiveFileNames() bool {
+	return host.program.host.FS().UseCaseSensitiveFileNames()
+}
+func (host *emitHost) IsEmitBlocked(file string) bool {
+	// !!!
+	return false
+}
+func (host *emitHost) WriteFile(fileName string, text string, writeByteOrderMark bool, _ []*ast.SourceFile, _ *WriteFileData) error {
+	return host.program.host.FS().WriteFile(fileName, text, writeByteOrderMark)
+}