-
Notifications
You must be signed in to change notification settings - Fork 18
/
fs.go
88 lines (70 loc) · 2.2 KB
/
fs.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
package builtin
import (
"context"
"errors"
"github.com/cirruslabs/cirrus-cli/pkg/larker/fs"
"go.starlark.net/starlark"
"os"
)
func FS(ctx context.Context, fs fs.FileSystem) starlark.StringDict {
return starlark.StringDict{
"exists": exists(ctx, fs),
"read": read(ctx, fs),
"readdir": readdir(ctx, fs),
}
}
func exists(ctx context.Context, fs fs.FileSystem) starlark.Value {
const funcName = "exists"
return starlark.NewBuiltin(funcName, func(thread *starlark.Thread, fn *starlark.Builtin, args starlark.Tuple, kwargs []starlark.Tuple) (starlark.Value, error) {
var path string
if err := starlark.UnpackPositionalArgs(funcName, args, kwargs, 1, &path); err != nil {
return nil, err
}
_, err := fs.Stat(ctx, path)
if err != nil {
if errors.Is(err, os.ErrNotExist) {
return starlark.Bool(false), nil
}
return nil, err
}
return starlark.Bool(true), nil
})
}
func read(ctx context.Context, fs fs.FileSystem) starlark.Value {
const funcName = "read"
return starlark.NewBuiltin(funcName, func(thread *starlark.Thread, fn *starlark.Builtin, args starlark.Tuple, kwargs []starlark.Tuple) (starlark.Value, error) {
var path string
if err := starlark.UnpackPositionalArgs(funcName, args, kwargs, 1, &path); err != nil {
return nil, err
}
fileBytes, err := fs.Get(ctx, path)
if err != nil {
if errors.Is(err, os.ErrNotExist) {
return starlark.None, nil
}
return nil, err
}
return starlark.String(fileBytes), nil
})
}
func readdir(ctx context.Context, fs fs.FileSystem) starlark.Value {
const funcName = "readdir"
return starlark.NewBuiltin(funcName, func(thread *starlark.Thread, fn *starlark.Builtin, args starlark.Tuple, kwargs []starlark.Tuple) (starlark.Value, error) {
var path string
if err := starlark.UnpackPositionalArgs(funcName, args, kwargs, 1, &path); err != nil {
return nil, err
}
entries, err := fs.ReadDir(ctx, path)
if err != nil {
if errors.Is(err, os.ErrNotExist) {
return starlark.None, nil
}
return nil, err
}
var starlarkEntries []starlark.Value
for _, entry := range entries {
starlarkEntries = append(starlarkEntries, starlark.String(entry))
}
return starlark.NewList(starlarkEntries), nil
})
}