-
Notifications
You must be signed in to change notification settings - Fork 18
/
utils.go
79 lines (73 loc) · 1.42 KB
/
utils.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
package dirtree
import (
"fmt"
"io"
"path"
"strings"
"github.com/mandelsoft/vfs/pkg/vfs"
)
func createFile(d *DirNode, p string, mode vfs.FileMode, size int64, r io.Reader) (*FileNode, error) {
dir := path.Dir(p)
name := path.Base(path.Clean(p))
d, err := lookupDir(d, dir, true)
if err != nil {
return nil, err
}
n, err := NewFileNode(d.ctx, mode, size, r)
if err == nil {
d.AddNode(name, n)
}
return n, err
}
func createLink(d *DirNode, p string, target string) (*FileNode, error) {
dir := path.Dir(p)
name := path.Base(path.Clean(p))
d, err := lookupDir(d, dir, true)
if err != nil {
return nil, err
}
n, err := NewLinkNode(d.ctx, target)
if err == nil {
d.AddNode(name, n)
}
return n, err
}
func lookupFile(d *DirNode, p string) Node {
p = path.Clean(p)
if path.IsAbs(p) {
p = p[1:]
}
comps := strings.Split(p, "/")
for i, c := range comps {
n := d.content[c]
if n == nil || i == len(comps)-1 {
return n
}
if dd, ok := n.(*DirNode); ok {
d = dd
}
}
return nil
}
func lookupDir(d *DirNode, p string, create bool) (*DirNode, error) {
p = path.Clean(p)
if path.IsAbs(p) {
p = p[1:]
}
if p == "." {
return d, nil
}
for _, c := range strings.Split(p, "/") {
n := d.content[c]
if n == nil {
n = NewDirNode(d.ctx)
d.content[c] = n
}
if dd, ok := n.(*DirNode); ok {
d = dd
} else {
return nil, fmt.Errorf("component %q in %q is no dir", c, p)
}
}
return d, nil
}