diff --git a/main.go b/main.go index 64d1a10..91051f7 100644 --- a/main.go +++ b/main.go @@ -2,15 +2,16 @@ package main import ( "fmt" + toml "github.com/pelletier/go-toml/v2" "io/ioutil" "os" "path/filepath" - toml "github.com/pelletier/go-toml/v2" ) type Config struct { - Address string - Port int + Address string + Port int + ScanDirectory string //Age int //Cats []string //Pi float64 @@ -18,29 +19,30 @@ type Config struct { //DOB time.Time // requires `import time` } - -func get_config() (Config, error) { - config := Config{ - Address: "127.0.0.1", - Port: 32520, - } - // Find config path +func get_config_dir() string { + // Find config directory // Linux (XDG base directory specification compliant) xdg_config_home := os.Getenv("XDG_CONFIG_HOME") - var config_dir string - if xdg_config_home != "" { - config_dir = filepath.Join(xdg_config_home, "ikinuki") + return filepath.Join(xdg_config_home, "ikinuki") } else { home := os.Getenv("HOME") - config_dir = filepath.Join(home, ".config", "ikinuki") + return filepath.Join(home, ".config", "ikinuki") + } +} + +func get_config(config_dir string) (*Config, error) { + config := Config{ + Address: "127.0.0.1", + Port: 32520, + ScanDirectory: "FILL IN", } err := os.MkdirAll(config_dir, 0755) if err != nil { - return config, err + return &config, err } config_path := filepath.Join(config_dir, "server.toml") @@ -48,48 +50,54 @@ func get_config() (Config, error) { fmt.Printf("Config File Dir: %v\n", config_path) // open file, creating it if empty - config_file, err := os.OpenFile(config_path, os.O_RDWR | os.O_CREATE, 0755) + config_file, err := os.OpenFile(config_path, os.O_RDWR|os.O_CREATE, 0755) defer config_file.Close() if err != nil { - return config, err + return &config, err } fi, err := config_file.Stat() if err != nil { - return config, err + return &config, err } if fi.Size() == 0 { // if file is empty, write default config data, err := toml.Marshal(config) if err != nil { - return config, err + return &config, err } config_file.Write(data) - return config, nil + // return nil as the user must edit the config file + return nil, nil } else { // try to parse config file data, err := ioutil.ReadAll(config_file) if err != nil { - return config, err + return &config, err } err = toml.Unmarshal(data, &config) if err != nil { - return config, err + return &config, err } - return config, nil + return &config, nil } } func main() { - config, err := get_config() + + config_dir := get_config_dir() + + config, err := get_config(config_dir) if err != nil { panic(err) } - - fmt.Printf("Config: %v\n", config) - + if config == nil { + fmt.Printf("Created config file at %v, please edit and launch again\n", config_dir) + } else { + fmt.Printf("Config: %v\n", config) + } }