wehub-resource-sync
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
48 lines
1.4 KiB
Go
48 lines
1.4 KiB
Go
package languages
|
|
|
|
import (
|
|
"regexp"
|
|
|
|
matlabforest "github.com/alexaandru/go-sitter-forest/matlab"
|
|
|
|
"github.com/zzet/gortex/internal/graph"
|
|
"github.com/zzet/gortex/internal/parser"
|
|
"github.com/zzet/gortex/internal/parser/forest"
|
|
)
|
|
|
|
// MATLAB migration: forest ships tags.scm AND the walker handles
|
|
// class_definition / function_definition via the generic suffix
|
|
// matcher. MATLAB's `import pkg.X` idiom stays regex.
|
|
var matlabImportRe = regexp.MustCompile(`(?m)^\s*import\s+(\w+(?:\.\w+)*)`)
|
|
|
|
type MatlabExtractor struct {
|
|
forest *forest.Extractor
|
|
}
|
|
|
|
func NewMatlabExtractor() *MatlabExtractor {
|
|
return &MatlabExtractor{
|
|
forest: forest.New("matlab", []string{".mlx"}, matlabforest.GetLanguage, matlabforest.GetQuery),
|
|
}
|
|
}
|
|
|
|
func (e *MatlabExtractor) Language() string { return "matlab" }
|
|
func (e *MatlabExtractor) Extensions() []string { return []string{".mlx"} }
|
|
|
|
func (e *MatlabExtractor) Extract(filePath string, src []byte) (*parser.ExtractionResult, error) {
|
|
res, err := e.forest.Extract(filePath, src)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, m := range matlabImportRe.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 = (*MatlabExtractor)(nil)
|