Skip to content

Commit e034b67

Browse files
committed
Export Node interface instead of struct.
1 parent ae677e9 commit e034b67

File tree

1 file changed

+16
-10
lines changed

1 file changed

+16
-10
lines changed

node/root.go

+16-10
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,16 @@ import (
77
"io/ioutil"
88
)
99

10+
// Node - File or directory helper methods
11+
type Node interface {
12+
Name() string
13+
Size() int
14+
}
15+
16+
type node struct {
17+
file os.FileInfo
18+
}
19+
1020
// Fetch - Fetch nodes in currently directory
1121
func Fetch() []Node {
1222
files, err := ioutil.ReadDir(".")
@@ -15,24 +25,20 @@ func Fetch() []Node {
1525
nodes := make([]Node, len(files))
1626

1727
for i:=0; i<len(files); i++ {
18-
nodes[i] = Node{files[i]}
28+
nodes[i] = node{files[i]}
1929
}
2030

2131
return nodes
2232
}
2333

24-
type Node struct {
25-
file os.FileInfo
26-
}
27-
28-
func (node Node) Name() string {
29-
name := node.file.Name()
30-
if node.file.IsDir() {
34+
func (n node) Name() string {
35+
name := n.file.Name()
36+
if n.file.IsDir() {
3137
return fmt.Sprintf(" %s/ ", name)
3238
}
3339
return fmt.Sprintf(" %s ", name)
3440
}
3541

36-
func (node Node) Size() int {
37-
return len([]rune(node.Name()))
42+
func (n node) Size() int {
43+
return len([]rune(n.Name()))
3844
}

0 commit comments

Comments
 (0)