Add extra diagnostics to tsgo (microsoft#581)

Author: jakebailey

cmd/tsgo/main.go

@@ -248,7 +248,13 @@ func main() {
 	var stats table
 
 	stats.add("Files", len(program.SourceFiles()))
+	stats.add("Lines", program.LineCount())
+	stats.add("Identifiers", program.IdentifierCount())
+	stats.add("Symbols", program.SymbolCount())
 	stats.add("Types", program.TypeCount())
+	stats.add("Instantiations", program.InstantiationCount())
+	stats.add("Memory used", fmt.Sprintf("%vK", memStats.Alloc/1024))
+	stats.add("Memory allocs", strconv.FormatUint(memStats.Mallocs, 10))
 	stats.add("Parse time", parseTime)
 	if bindTime != 0 {
 		stats.add("Bind time", bindTime)
@@ -260,8 +266,6 @@ func main() {
 		stats.add("Emit time", emitTime)
 	}
 	stats.add("Total time", totalTime)
-	stats.add("Memory used", fmt.Sprintf("%vK", memStats.Alloc/1024))
-	stats.add("Memory allocs", strconv.FormatUint(memStats.Mallocs, 10))
 
 	stats.print()
 }
@@ -299,6 +303,14 @@ func formatDuration(d time.Duration) string {
 	return fmt.Sprintf("%.3fs", d.Seconds())
 }
 
+func identifierCount(p *ts.Program) int {
+	count := 0
+	for _, file := range p.SourceFiles() {
+		count += file.IdentifierCount
+	}
+	return count
+}
+
 func listFiles(p *ts.Program) {
 	for _, file := range p.SourceFiles() {
 		fmt.Println(file.FileName())

internal/ast/ast.go

@@ -8689,6 +8689,7 @@ type SourceFile struct {
 	HasNoDefaultLib             bool
 	UsesUriStyleNodeCoreModules core.Tristate
 	Identifiers                 map[string]string
+	IdentifierCount             int
 	Imports                     []*LiteralLikeNode // []LiteralLikeNode
 	ModuleAugmentations         []*ModuleName      // []ModuleName
 	AmbientModuleNames          []string

internal/checker/checker.go

@@ -531,8 +531,8 @@ type Checker struct {
 	fileIndexMap                               map[*ast.SourceFile]int
 	compareSymbols                             func(*ast.Symbol, *ast.Symbol) int
 	TypeCount                                  uint32
-	symbolCount                                uint32
-	totalInstantiationCount                    uint32
+	SymbolCount                                uint32
+	TotalInstantiationCount                    uint32
 	instantiationCount                         uint32
 	instantiationDepth                         uint32
 	inlineLevel                                int
@@ -13041,7 +13041,7 @@ func (c *Checker) hasParseDiagnostics(sourceFile *ast.SourceFile) bool {
 }
 
 func (c *Checker) newSymbol(flags ast.SymbolFlags, name string) *ast.Symbol {
-	c.symbolCount++
+	c.SymbolCount++
 	result := c.symbolPool.New()
 	result.Flags = flags | ast.SymbolFlagsTransient
 	result.Name = name
@@ -20255,7 +20255,7 @@ func (c *Checker) instantiateTypeWithAlias(t *Type, m *TypeMapper, alias *TypeAl
 		c.error(c.currentNode, diagnostics.Type_instantiation_is_excessively_deep_and_possibly_infinite)
 		return c.errorType
 	}
-	c.totalInstantiationCount++
+	c.TotalInstantiationCount++
 	c.instantiationCount++
 	c.instantiationDepth++
 	result := c.instantiateTypeWorker(t, m, alias)

internal/compiler/program.go

@@ -400,6 +400,33 @@ func (p *Program) getDiagnosticsHelper(sourceFile *ast.SourceFile, ensureBound b
 	return SortAndDeduplicateDiagnostics(result)
 }
 
+func (p *Program) LineCount() int {
+	var count int
+	for _, file := range p.files {
+		count += len(file.LineMap())
+	}
+	return count
+}
+
+func (p *Program) IdentifierCount() int {
+	var count int
+	for _, file := range p.files {
+		count += file.IdentifierCount
+	}
+	return count
+}
+
+func (p *Program) SymbolCount() int {
+	var count int
+	for _, file := range p.files {
+		count += file.SymbolCount
+	}
+	for _, checker := range p.checkers {
+		count += int(checker.SymbolCount)
+	}
+	return count
+}
+
 func (p *Program) TypeCount() int {
 	var count int
 	for _, checker := range p.checkers {
@@ -408,6 +435,14 @@ func (p *Program) TypeCount() int {
 	return count
 }
 
+func (p *Program) InstantiationCount() int {
+	var count int
+	for _, checker := range p.checkers {
+		count += int(checker.TotalInstantiationCount)
+	}
+	return count
+}
+
 func (p *Program) PrintSourceFileWithTypes() {
 	for _, file := range p.files {
 		if tspath.GetBaseFileName(file.FileName()) == "main.ts" {

internal/parser/parser.go

@@ -68,6 +68,7 @@ type Parser struct {
 	hasDeprecatedTag            bool
 
 	identifiers             map[string]string
+	identifierCount         int
 	notParenthesizedArrow   core.Set[int]
 	nodeSlicePool           core.Pool[*ast.Node]
 	jsdocCache              map[*ast.Node][]*ast.Node
@@ -326,6 +327,7 @@ func (p *Parser) finishSourceFile(result *ast.SourceFile, isDeclarationFile bool
 	result.ScriptKind = p.scriptKind
 	result.Flags |= p.sourceFlags
 	result.Identifiers = p.identifiers
+	result.IdentifierCount = p.identifierCount
 	result.SetJSDocCache(p.jsdocCache)
 	p.jsdocCache = nil
 	p.identifiers = nil
@@ -2812,6 +2814,7 @@ func (p *Parser) parseRightSideOfDot(allowIdentifierNames bool, allowPrivateIden
 }
 
 func (p *Parser) newIdentifier(text string) *ast.Node {
+	p.identifierCount++
 	id := p.factory.NewIdentifier(text)
 	if text == "await" {
 		p.statementHasAwaitIdentifier = true