mirror of
https://github.com/safedep/vet.git
synced 2025-12-10 12:07:30 -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>
89 lines
1.6 KiB
Go
89 lines
1.6 KiB
Go
package nodes
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
sitter "github.com/smacker/go-tree-sitter"
|
|
)
|
|
|
|
type CSTFunctionNode struct {
|
|
cst *CST
|
|
|
|
// The function declaration
|
|
declaration *sitter.Node
|
|
|
|
// Container node, such as a class or a module
|
|
container *sitter.Node
|
|
|
|
// Name of the function
|
|
name *sitter.Node
|
|
|
|
// Arguments of the function
|
|
args *sitter.Node
|
|
|
|
// Body of the function
|
|
body *sitter.Node
|
|
}
|
|
|
|
func NewCSTFunctionNode(cst *CST) *CSTFunctionNode {
|
|
return &CSTFunctionNode{cst: cst}
|
|
}
|
|
|
|
func (n *CSTFunctionNode) WithDeclaration(declaration *sitter.Node) *CSTFunctionNode {
|
|
n.declaration = declaration
|
|
return n
|
|
}
|
|
|
|
func (n *CSTFunctionNode) WithContainer(container *sitter.Node) *CSTFunctionNode {
|
|
n.container = container
|
|
return n
|
|
}
|
|
|
|
func (n *CSTFunctionNode) WithName(name *sitter.Node) *CSTFunctionNode {
|
|
n.name = name
|
|
return n
|
|
}
|
|
|
|
func (n *CSTFunctionNode) WithArgs(args *sitter.Node) *CSTFunctionNode {
|
|
n.args = args
|
|
return n
|
|
}
|
|
|
|
func (n *CSTFunctionNode) WithBody(body *sitter.Node) *CSTFunctionNode {
|
|
n.body = body
|
|
return n
|
|
}
|
|
|
|
func (n *CSTFunctionNode) Declaration() *sitter.Node {
|
|
return n.declaration
|
|
}
|
|
|
|
func (n *CSTFunctionNode) ContainerNode() *sitter.Node {
|
|
return n.container
|
|
}
|
|
|
|
func (n *CSTFunctionNode) NameNode() *sitter.Node {
|
|
return n.name
|
|
}
|
|
|
|
// Human readable Id for use in graph query
|
|
func (n CSTFunctionNode) Id() string {
|
|
return fmt.Sprintf("%s/%s", n.Container(), n.Name())
|
|
}
|
|
|
|
func (n CSTFunctionNode) Name() string {
|
|
if n.name != nil {
|
|
return n.name.Content(n.cst.code)
|
|
}
|
|
|
|
return ""
|
|
}
|
|
|
|
func (n CSTFunctionNode) Container() string {
|
|
if n.container != nil {
|
|
return n.container.Content(n.cst.code)
|
|
}
|
|
|
|
return ""
|
|
}
|