mirror of
https://github.com/safedep/vet.git
synced 2025-12-10 00:22:08 -06:00
e6f6288701
feat: Building code graph Refactor to support import processing Handle relative import name fixup Add docs for code analysis framework Update docs to include additional examples feat: Function call graph Update code graph to link function decl and calls Include call node in function calls feat: Flatten vulnerabilities in CSV reporter refactor: Maintain separation of concerns for code analysis framework refactor: Separate storage entities in its own package feat: Add callback support in code graph builder docs: Fix code analysis framework docs Signed-off-by: abhisek <abhisek.datta@gmail.com>
81 lines
2.3 KiB
Go
81 lines
2.3 KiB
Go
package code
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
sitter "github.com/smacker/go-tree-sitter"
|
|
)
|
|
|
|
type tsQueryMatchHandler func(*sitter.QueryMatch, *sitter.Query, bool) error
|
|
|
|
func TSExecQuery(query string, lang *sitter.Language, source []byte,
|
|
node *sitter.Node, handler tsQueryMatchHandler) error {
|
|
tsQuery, err := sitter.NewQuery([]byte(query), lang)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
tsQueryCursor := sitter.NewQueryCursor()
|
|
tsQueryCursor.Exec(tsQuery, node)
|
|
|
|
for {
|
|
match, ok := tsQueryCursor.NextMatch()
|
|
if !ok {
|
|
break
|
|
}
|
|
|
|
match = tsQueryCursor.FilterPredicates(match, source)
|
|
|
|
if len(match.Captures) == 0 {
|
|
continue
|
|
}
|
|
|
|
if err := handler(match, tsQuery, ok); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Maps a file path to a module name. This is an important operation
|
|
// because it serves as the bridge between FS and Language domain
|
|
func LangMapFileToModule(file SourceFile, repo SourceRepository, lang SourceLanguage, includeImports bool) (string, error) {
|
|
// Get the relative path of the file in the repository because most language
|
|
// runtimes (module loaders) identify modules by relative paths
|
|
relSourceFilePath, err := repo.GetRelativePath(file.Path, includeImports)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to get relative path: %w", err)
|
|
}
|
|
|
|
// Use the language adapter to translate the relative path to a module name
|
|
// which is language specific
|
|
moduleName, err := lang.ResolveImportNameFromPath(relSourceFilePath)
|
|
if err != nil {
|
|
return "", fmt.Errorf("failed to resolve import name from path: %w", err)
|
|
}
|
|
|
|
return moduleName, nil
|
|
}
|
|
|
|
// Maps a module name back to a file path that must exist in the repository
|
|
func LangMapModuleToFile(moduleName string, currentFile SourceFile,
|
|
repo SourceRepository, lang SourceLanguage, includeImports bool) (SourceFile, error) {
|
|
// Use the language adapter to get possible relative paths for the module name
|
|
relPaths, err := lang.ResolveImportPathsFromName(currentFile, moduleName, includeImports)
|
|
if err != nil {
|
|
return SourceFile{}, fmt.Errorf("failed to resolve import paths from name: %w", err)
|
|
}
|
|
|
|
for _, relPath := range relPaths {
|
|
sf, err := repo.GetSourceFileByPath(relPath, includeImports)
|
|
if err != nil {
|
|
continue
|
|
} else {
|
|
return sf, nil
|
|
}
|
|
}
|
|
|
|
return SourceFile{}, fmt.Errorf("no source file found for module: %s", moduleName)
|
|
}
|