105 lines
2.3 KiB
Go
105 lines
2.3 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"errors"
|
||
|
|
"os"
|
||
|
|
"path"
|
||
|
|
"slices"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/charmbracelet/log"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Config struct {
|
||
|
|
UserName string
|
||
|
|
UserID string
|
||
|
|
ClientID string
|
||
|
|
ClientSecret string
|
||
|
|
}
|
||
|
|
|
||
|
|
// Finds a valid configuration file if it exists. If nil is returned one
|
||
|
|
// should be created.
|
||
|
|
func find_config() (string, error) {
|
||
|
|
// Paths where a tremble.conf file might reside
|
||
|
|
// TODO: Check XDG_CONFIG_DIRS last in case XDG_CONFIG_HOME isn't set.
|
||
|
|
var paths []string = []string{
|
||
|
|
".",
|
||
|
|
"$XDG_CONFIG_HOME/tremble",
|
||
|
|
"$HOME/.tremble",
|
||
|
|
"$HOME/.config/tremble",
|
||
|
|
}
|
||
|
|
|
||
|
|
if configDirs, isSet := os.LookupEnv("XDG_CONFIG_DIRS"); isSet {
|
||
|
|
paths = slices.Concat(paths, strings.Split(configDirs, ":"))
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, p := range paths {
|
||
|
|
dir := os.ExpandEnv(p)
|
||
|
|
pstr := path.Join(dir, "tremble.conf")
|
||
|
|
|
||
|
|
_, err := os.Lstat(pstr)
|
||
|
|
if err != nil {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
return pstr, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
return "", errors.New("No config file found.")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Finds a configuration file and processes it if one exists.
|
||
|
|
func read_config() (*Config, error) {
|
||
|
|
pstr, err := find_config()
|
||
|
|
if err != nil {
|
||
|
|
log.Info(err)
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
cdata, err := os.ReadFile(pstr)
|
||
|
|
if err != nil {
|
||
|
|
log.Info(err)
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Parse the file and return.
|
||
|
|
var confOut Config
|
||
|
|
err = json.Unmarshal(cdata, &confOut)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return &confOut, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Writes configuration to the tremble config file. Currently this overwrites
|
||
|
|
// the entire thing if found, or creates a new one if not.
|
||
|
|
func write_config(config *Config) error {
|
||
|
|
var pstr string
|
||
|
|
|
||
|
|
// XXX: Maybe check to make sure this isn't some substantial error.
|
||
|
|
// XXX: Some duplication here.
|
||
|
|
pstr, err := find_config()
|
||
|
|
if err != nil {
|
||
|
|
dir := os.ExpandEnv("$XDG_CONFIG_HOME/tremble")
|
||
|
|
if err := os.MkdirAll(dir, 0755); err != nil {
|
||
|
|
log.Fatalf("Couldnt create directory '%v': %v\n", dir, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
pstr = path.Join(dir, "tremble.conf")
|
||
|
|
}
|
||
|
|
|
||
|
|
// TODO: Make this fatal in the caller, or... not. Just not here.
|
||
|
|
cdata, err := json.Marshal(config)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("Couldn't write config %v: %v\n", config, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
err = os.WriteFile(pstr, cdata, 0755)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("Couldn't write config %v: %v\n", config, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|