-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
79 lines (71 loc) · 1.71 KB
/
main.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 main
import (
"flag"
"fmt"
"io/ioutil"
"path/filepath"
"github.com/honeystats/ssh/files"
"github.com/sirupsen/logrus"
)
func realPathToDir(startPath string, firstPass bool) (*files.FilesystemDir, error) {
root := &files.FilesystemDir{
Subdirs: []*files.FilesystemDir{},
Files: []*files.FilesystemFile{},
}
if !firstPass {
root.Name = filepath.Base(startPath)
}
dirFiles, err := ioutil.ReadDir(startPath)
if err != nil {
return nil, err
}
for _, file := range dirFiles {
if file.IsDir() {
path := filepath.Join(startPath, file.Name())
var res *files.FilesystemDir
res, err = realPathToDir(path, false)
if err != nil {
return nil, err
}
root.Subdirs = append(root.Subdirs, res)
} else {
filePath := filepath.Join(startPath, file.Name())
content, err := ioutil.ReadFile(filePath)
if err != nil {
return nil, err
}
newFile := &files.FilesystemFile{
Name: file.Name(),
Content: string(content),
}
root.Files = append(root.Files, newFile)
}
}
return root, nil
}
func realPathToConfig(path string) (*files.FilesystemConfig, error) {
rootDir, err := realPathToDir(path, true)
if err != nil {
return nil, err
}
return &files.FilesystemConfig{
Root: rootDir,
}, nil
}
var pathToLookup string
func main() {
flag.StringVar(&pathToLookup, "source-path", "", "path from which to generate the YAML file")
flag.Parse()
if pathToLookup == "" {
logrus.Fatalln("Missing source-path arg.")
}
conf, err := realPathToConfig(pathToLookup)
if err != nil {
logrus.WithError(err).Fatalln("error getting info for dir")
}
str, err := conf.ToString()
if err != nil {
logrus.WithError(err).Fatalln("error marshalling dir to string")
}
fmt.Println(str)
}