123 lines
2.6 KiB
Go
123 lines
2.6 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
"github.com/makeworld-the-better-one/amfora/cache"
|
|
homedir "github.com/mitchellh/go-homedir"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var amforaAppData string // Where amfora files are stored on Windows - cached here
|
|
|
|
func configDir() string {
|
|
home, err := homedir.Dir()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if runtime.GOOS == "windows" {
|
|
return amforaAppData
|
|
}
|
|
|
|
// Unix / POSIX system
|
|
return filepath.Join(home, ".config", "amfora")
|
|
}
|
|
|
|
func configPath() string {
|
|
return filepath.Join(configDir(), "config.toml")
|
|
}
|
|
|
|
var TofuStore = viper.New()
|
|
|
|
func tofuDBDir() string {
|
|
home, err := homedir.Dir()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
// Windows just stores it in APPDATA along with other stuff
|
|
if runtime.GOOS == "windows" {
|
|
return amforaAppData
|
|
}
|
|
// XDG cache dir on POSIX systems
|
|
return filepath.Join(home, ".cache", "amfora")
|
|
}
|
|
|
|
func tofuDBPath() string {
|
|
return filepath.Join(tofuDBDir(), "tofu.toml")
|
|
}
|
|
|
|
func Init() error {
|
|
home, err := homedir.Dir()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if runtime.GOOS == "windows" {
|
|
appdata, ok := os.LookupEnv("APPDATA")
|
|
if ok {
|
|
amforaAppData = filepath.Join(appdata, "amfora")
|
|
} else {
|
|
amforaAppData = filepath.Join(home, filepath.FromSlash("AppData/Roaming/amfora/"))
|
|
}
|
|
}
|
|
|
|
err = os.MkdirAll(configDir(), 0755)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
f, err := os.OpenFile(configPath(), os.O_RDWR|os.O_CREATE|os.O_EXCL, 0666)
|
|
if err == nil {
|
|
// Config file doesn't exist yet, write the default one
|
|
_, err = f.Write(defaultConf)
|
|
if err != nil {
|
|
f.Close()
|
|
return err
|
|
}
|
|
f.Close()
|
|
}
|
|
|
|
err = os.MkdirAll(tofuDBDir(), 0755)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
os.OpenFile(tofuDBPath(), os.O_RDWR|os.O_CREATE|os.O_EXCL, 0666)
|
|
|
|
TofuStore.SetConfigFile(tofuDBPath())
|
|
TofuStore.SetConfigType("toml")
|
|
err = TofuStore.ReadInConfig()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
viper.SetDefault("a-general.home", "gemini.circumlunar.space")
|
|
viper.SetDefault("a-general.http", "default")
|
|
viper.SetDefault("a-general.search", "gus.guru/search")
|
|
viper.SetDefault("a-general.color", true)
|
|
viper.SetDefault("a-general.bullets", true)
|
|
viper.SetDefault("a-general.wrap_width", 100)
|
|
viper.SetDefault("cache.max_size", 0)
|
|
viper.SetDefault("cache.max_pages", 20)
|
|
|
|
viper.SetConfigFile(configPath())
|
|
viper.SetConfigType("toml")
|
|
err = viper.ReadInConfig()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Setup cache from config
|
|
cache.SetMaxSize(viper.GetInt("cache.max_size"))
|
|
cache.SetMaxPages(viper.GetInt("cache.max_pages"))
|
|
|
|
return nil
|
|
}
|
|
|
|
func GetWrapWidth() int {
|
|
i := viper.GetInt("a-general.wrap_width")
|
|
if i <= 0 {
|
|
return 100 // The default
|
|
}
|
|
return i
|
|
}
|