diff options
author | Flu0r1ne <flur01ne@flu0r1ne.net> | 2021-08-03 02:02:40 -0500 |
---|---|---|
committer | Flu0r1ne <flur01ne@flu0r1ne.net> | 2021-08-03 02:02:40 -0500 |
commit | a0b020a78eb0b33965c59460fc093c6959216e44 (patch) | |
tree | 3b2722388ea6b312e927b6066b0831524ba4ed6b /fs.go | |
download | planr-a0b020a78eb0b33965c59460fc093c6959216e44.tar.xz planr-a0b020a78eb0b33965c59460fc093c6959216e44.zip |
Initial commit with basic build structure
Diffstat (limited to 'fs.go')
-rw-r--r-- | fs.go | 170 |
1 files changed, 170 insertions, 0 deletions
@@ -0,0 +1,170 @@ +package planr + +import ( + "log" + "os" + "io" + "path" + "path/filepath" + // "fmt" +) + +/* CONFIG DIRECTORY */ + +// Consumes path +// Returns true if traversal should halt +type traversalFunc func(path string) bool; + +// Traverse up until the root is reached +// Calls traverseFunc each iteration +// Returns true if prematurely stopped +func traverseUp(reference string, shouldStop traversalFunc) bool { + cursor := reference + + for !shouldStop(cursor) { + if filepath.ToSlash(cursor) == "/" { + return false + } + cursor = filepath.Join(cursor, "..") + } + + return true +} + +func directoryExists(path string) bool { + info, err := os.Stat(path) + + if err != nil { + if !os.IsNotExist(err) { + log.Fatal(err) + } + + return false; + } + + return info.IsDir() +} + +// Find the configuration directory +// Uses: +// 1. PlANR_DIRECTORY env if set +// 2. planr +// 3. .planr +func configDir() string { + + // Return environmental override if set + if dir, isSet := os.LookupEnv("PLANR_DIRECTORY"); isSet { + + if !directoryExists(dir) { + log.Fatalf("Cannot find planr directory %s", dir); + } + + return dir; + } + + cwd, err := os.Getwd() + + if err != nil { + log.Fatal(err) + } + + var rubricDir string + + rubric_search_dirs := [2]string{ + "planr", + ".planr", + } + + found := traverseUp(cwd, func (path string) bool { + + for _, dir := range rubric_search_dirs { + rubricDir = filepath.Join(path, dir) + + if directoryExists(rubricDir) { + return true + } + } + + return false + }); + + if !found { + log.Fatal("Could not find planr directory"); + } + + return rubricDir +} + +// Find rubric directory at PLANR_DIR/rubric +func RubricDir() string { + rubricDir := path.Join(configDir(), "rubric"); + + if !directoryExists(rubricDir) { + log.Fatal("Could not find the rubric directory inside of planr") + } + + return rubricDir +} + +// Collects the units from the configuration tree +// TODO: Cleanup +func collectUnits( + name string, + defaults *Defaults, + cfgs []AdapterConfig, + units *[]TestCase, +) { + fp, err := os.Open(name) + if err != nil { + log.Fatal(err) + } + + // Process defaults for this directory if a defaults.toml is found + defaultsPath := path.Join(name, "defaults.toml") + if info, err := os.Stat(defaultsPath); err == nil && !info.IsDir() { + d := DecodeDefaults(defaultsPath, cfgs) + + // inherit the properties not defined in this defaults + if defaults != nil { + d.Inherit(defaults) + } + + defaults = &d + } + + // Read the entries in this directory + for { + dirs, err := fp.ReadDir(100) + if err == io.EOF { + break + } else if err != nil { + log.Fatal(err) + } + + + for _, ent := range dirs { + child := path.Join(name, ent.Name()) + nm := ent.Name() + + if ent.IsDir() { + collectUnits(child, defaults, cfgs, units) + } else { + if nm == "defaults.toml" { + continue + } + + // Decode a unit + config := DecodeConfig(child, cfgs) + config.Inherit(*defaults) + + tc := TestCase { + Path: child, + Cname: nm, + Config: config, + } + + *units = append(*units, tc) + } + } + } +} |