Add FileName and DirName to Node

m-mizutani · Sep 19, 2022 · 20c43f3 · 20c43f3
1 parent 470d7f6
commit 20c43f3
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/model.go b/model.go
@@ -3,11 +3,15 @@ package goast
 import (
 	"go/ast"
 	"go/token"
+	"path/filepath"
 	"reflect"
 )
 
 type Node struct {
-	Path string
+	Path     string
+	FileName string
+	DirName  string
+
 	Node ast.Node
 	Kind string
 
@@ -16,7 +20,10 @@ type Node struct {
 
 func newNode(fpath string, node ast.Node, fset *token.FileSet) *Node {
 	return &Node{
-		Path:    fpath,
+		Path:     fpath,
+		DirName:  filepath.Dir(fpath),
+		FileName: filepath.Base(fpath),
+
 		Node:    node,
 		Kind:    reflect.ValueOf(node).Elem().Type().Name(),
 		fileSet: fset,