a06f331eb8
CI / benchmark (push) Has been skipped
install-script / posix-syntax (push) Successful in 6m1s
CI / build-onnx (push) Failing after 6m43s
init-smoke / dry-run (push) Failing after 15m57s
security / govulncheck (push) Has been cancelled
security / trivy-fs (push) Has been cancelled
CI / test (1.26, ubuntu-latest) (push) Has been cancelled
Scorecard supply-chain security / Scorecard analysis (push) Has been cancelled
CI / test (1.26, macos-latest) (push) Has been cancelled
CI / build-windows (push) Has been cancelled
CI / lint (push) Has been cancelled
install-script / powershell-syntax (push) Has been cancelled
install-script / install (macos-14) (push) Has been cancelled
install-script / install (ubuntu-latest) (push) Has been cancelled
61 lines
1.9 KiB
Go
61 lines
1.9 KiB
Go
package languages
|
|
|
|
import (
|
|
"regexp"
|
|
|
|
odinforest "github.com/alexaandru/go-sitter-forest/odin"
|
|
|
|
"github.com/zzet/gortex/internal/graph"
|
|
"github.com/zzet/gortex/internal/parser"
|
|
"github.com/zzet/gortex/internal/parser/forest"
|
|
)
|
|
|
|
// Odin migration: forest's walker (per-language map for
|
|
// `procedure_declaration` / `struct_declaration` /
|
|
// `package_declaration`) catches definitions. `import` and
|
|
// `foreign import` edges stay regex — they have grammar nodes but
|
|
// tags.scm doesn't tag them as graph imports.
|
|
var (
|
|
odinImportRe = regexp.MustCompile(`(?m)^\s*import\s+(?:(\w+)\s+)?"([^"]+)"`)
|
|
odinForeignImportRe = regexp.MustCompile(`(?m)^\s*foreign\s+import\s+\w+\s+"([^"]+)"`)
|
|
)
|
|
|
|
type OdinExtractor struct {
|
|
forest *forest.Extractor
|
|
}
|
|
|
|
func NewOdinExtractor() *OdinExtractor {
|
|
return &OdinExtractor{
|
|
forest: forest.New("odin", []string{".odin"}, odinforest.GetLanguage, odinforest.GetQuery),
|
|
}
|
|
}
|
|
|
|
func (e *OdinExtractor) Language() string { return "odin" }
|
|
func (e *OdinExtractor) Extensions() []string { return []string{".odin"} }
|
|
|
|
func (e *OdinExtractor) Extract(filePath string, src []byte) (*parser.ExtractionResult, error) {
|
|
res, err := e.forest.Extract(filePath, src)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, m := range odinImportRe.FindAllSubmatchIndex(src, -1) {
|
|
mod := string(src[m[4]:m[5]])
|
|
line := lineAt(src, m[0])
|
|
res.Edges = append(res.Edges, &graph.Edge{
|
|
From: filePath, To: "unresolved::import::" + mod,
|
|
Kind: graph.EdgeImports, FilePath: filePath, Line: line,
|
|
})
|
|
}
|
|
for _, m := range odinForeignImportRe.FindAllSubmatchIndex(src, -1) {
|
|
mod := string(src[m[2]:m[3]])
|
|
line := lineAt(src, m[0])
|
|
res.Edges = append(res.Edges, &graph.Edge{
|
|
From: filePath, To: "unresolved::import::" + mod,
|
|
Kind: graph.EdgeImports, FilePath: filePath, Line: line,
|
|
})
|
|
}
|
|
return res, nil
|
|
}
|
|
|
|
var _ parser.Extractor = (*OdinExtractor)(nil)
|