zzet--gortex/internal/contracts/parse_helper.go

package contracts

import (
	"github.com/zzet/gortex/internal/parser"
	gosrc "github.com/zzet/gortex/internal/parser/tsitter/golang"
)

// ParseTreeForLang produces a fresh ParseTree for src using the
// grammar registered for lang. Returns nil if the language doesn't
// have a grammar Phase 1 supports — currently only "go". Used by the
// indexer's incremental re-walk path (extractContracts) which doesn't
// have a tree from the language extractor.
//
// The caller must Release() the returned tree when done.
func ParseTreeForLang(lang string, src []byte) *parser.ParseTree {
	if lang != "go" || len(src) == 0 {
		return nil
	}
	tree, err := parser.ParseFile(src, gosrc.GetLanguage())
	if err != nil || tree == nil {
		return nil
	}
	return parser.NewParseTree(tree, src, "go")
}