forked from kubernetes/kops
/
memfs.go
127 lines (105 loc) · 2.3 KB
/
memfs.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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
package vfs
import (
"os"
"path"
"strings"
)
type MemFSPath struct {
context *MemFSContext
location string
contents []byte
children map[string]*MemFSPath
}
var _ Path = &MemFSPath{}
type MemFSContext struct {
clusterReadable bool
root *MemFSPath
}
func NewMemFSContext() *MemFSContext {
c := &MemFSContext{}
c.root = &MemFSPath{
context: c,
location: "",
}
return c
}
// MarkClusterReadable pretends the current memfscontext is cluster readable; this is useful for tests
func (c *MemFSContext) MarkClusterReadable() {
c.clusterReadable = true
}
func (c *MemFSPath) IsClusterReadable() bool {
return c.context.clusterReadable
}
var _ HasClusterReadable = &MemFSPath{}
func NewMemFSPath(context *MemFSContext, location string) *MemFSPath {
return context.root.Join(location).(*MemFSPath)
}
func (p *MemFSPath) Join(relativePath ...string) Path {
joined := path.Join(relativePath...)
tokens := strings.Split(joined, "/")
current := p
for _, token := range tokens {
if current.children == nil {
current.children = make(map[string]*MemFSPath)
}
child := current.children[token]
if child == nil {
child = &MemFSPath{
context: p.context,
location: path.Join(current.location, token),
}
current.children[token] = child
}
current = child
}
return current
}
func (p *MemFSPath) WriteFile(data []byte) error {
p.contents = data
return nil
}
func (p *MemFSPath) CreateFile(data []byte) error {
// Check if exists
if p.contents != nil {
return os.ErrExist
}
return p.WriteFile(data)
}
func (p *MemFSPath) ReadFile() ([]byte, error) {
if p.contents == nil {
return nil, os.ErrNotExist
}
// TODO: Copy?
return p.contents, nil
}
func (p *MemFSPath) ReadDir() ([]Path, error) {
var paths []Path
for _, f := range p.children {
paths = append(paths, f)
}
return paths, nil
}
func (p *MemFSPath) ReadTree() ([]Path, error) {
var paths []Path
p.readTree(&paths)
return paths, nil
}
func (p *MemFSPath) readTree(dest *[]Path) {
for _, f := range p.children {
*dest = append(*dest, f)
f.readTree(dest)
}
}
func (p *MemFSPath) Base() string {
return path.Base(p.location)
}
func (p *MemFSPath) Path() string {
return p.location
}
func (p *MemFSPath) String() string {
return p.Path()
}
func (p *MemFSPath) Remove() error {
p.contents = nil
return nil
}