Concurrent program with module resolution (#107)

Co-authored-by: Eric Cornelson <[email protected]>

Author: EricCornelson

cmd/tsgo/main.go

@@ -24,6 +24,7 @@ var singleThreaded = false
 var parseAndBindOnly = false
 var printTypes = false
 var pretty = true
+var listFiles = false
 var pprofDir = ""
 
 func printDiagnostic(d *ast.Diagnostic, level int, comparePathOptions tspath.ComparePathsOptions) {
@@ -54,6 +55,7 @@ func main() {
 	flag.BoolVar(&parseAndBindOnly, "p", false, "Parse and bind only")
 	flag.BoolVar(&printTypes, "t", false, "Print types defined in main.ts")
 	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.Parse()
 
@@ -66,7 +68,7 @@ func main() {
 	}
 	fs := vfs.FromOS()
 	useCaseSensitiveFileNames := fs.UseCaseSensitiveFileNames()
-	host := ts.NewCompilerHost(compilerOptions, singleThreaded, currentDirectory, fs)
+	host := ts.NewCompilerHost(compilerOptions, currentDirectory, fs)
 
 	normalizedRootPath := tspath.ResolvePath(currentDirectory, rootPath)
 	if !fs.DirectoryExists(normalizedRootPath) {
@@ -122,6 +124,12 @@ func main() {
 		}
 	}
 
+	if listFiles {
+		for _, file := range program.SourceFiles() {
+			fmt.Println(file.FileName())
+		}
+	}
+
 	fmt.Printf("Files:         %v\n", len(program.SourceFiles()))
 	fmt.Printf("Types:         %v\n", program.TypeCount())
 	fmt.Printf("Compile time:  %v\n", compileTime)

internal/ast/ast.go

@@ -5436,6 +5436,10 @@ type SourceFile struct {
 	AmbientModuleNames          []string
 	HasNoDefaultLib             bool
 	jsdocCache                  map[*Node][]*Node
+	Pragmas                     []Pragma
+	ReferencedFiles             []*FileReference
+	TypeReferenceDirectives     []*FileReference
+	LibReferenceDirectives      []*FileReference
 }
 
 func (f *NodeFactory) NewSourceFile(text string, fileName string, statements *NodeList) *Node {
@@ -5482,3 +5486,61 @@ func (node *SourceFile) ForEachChild(v Visitor) bool {
 func IsSourceFile(node *Node) bool {
 	return node.Kind == KindSourceFile
 }
+
+type CommentRange struct {
+	core.TextRange
+	HasTrailingNewLine bool
+	Kind               Kind
+}
+
+func NewCommentRange(kind Kind, pos int, end int, hasTrailingNewLine bool) CommentRange {
+	return CommentRange{
+		TextRange:          core.NewTextRange(pos, end),
+		HasTrailingNewLine: hasTrailingNewLine,
+		Kind:               kind,
+	}
+}
+
+type FileReference struct {
+	core.TextRange
+	FileName       string
+	ResolutionMode core.ResolutionMode
+	Preserve       bool
+}
+
+type PragmaArgument struct {
+	core.TextRange
+	Name  string
+	Value string
+}
+
+type Pragma struct {
+	Name      string
+	Args      map[string]PragmaArgument
+	ArgsRange CommentRange
+}
+
+type PragmaKindFlags = uint8
+
+const (
+	PragmaKindFlagsNone PragmaKindFlags = iota
+	PragmaKindTripleSlashXML
+	PragmaKindSingleLine
+	PragmaKindMultiLine
+	PragmaKindAll     = PragmaKindTripleSlashXML | PragmaKindSingleLine | PragmaKindMultiLine
+	PragmaKindDefault = PragmaKindAll
+)
+
+type PragmaArgumentSpecification struct {
+	Name        string
+	Optional    bool
+	CaptureSpan bool
+}
+type PragmaSpecification struct {
+	Args []PragmaArgumentSpecification
+	Kind PragmaKindFlags
+}
+
+func (spec *PragmaSpecification) IsTripleSlash() bool {
+	return (spec.Kind & PragmaKindTripleSlashXML) > 0
+}

internal/ast/kind.go

@@ -7,6 +7,8 @@ type Kind int16
 const (
 	KindUnknown Kind = iota
 	KindEndOfFile
+	KindSingleLineCommentTrivia
+	KindMultiLineCommentTrivia
 	KindConflictMarkerTrivia
 	KindNonTextFileMarkerTrivia
 	KindNumericLiteral
@@ -420,4 +422,5 @@ const (
 	KindLastJSDocTagNode        = KindJSDocImportTag
 	KindFirstContextualKeyword  = KindAbstractKeyword
 	KindLastContextualKeyword   = KindOfKeyword
+	KindComment                 = KindSingleLineCommentTrivia | KindMultiLineCommentTrivia
 )