115 lines
2.7 KiB
Go
115 lines
2.7 KiB
Go
package config
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"sync"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// dirPath indicates the path to the config. This variable is created when
|
|
// __init is called.
|
|
var dirPath string
|
|
|
|
// Singleton to initialize the config directories once.
|
|
var __initonce sync.Once
|
|
|
|
func __init() {
|
|
// Load the config dir:
|
|
d, err := os.UserConfigDir()
|
|
if err != nil {
|
|
log.Fatalln("Failed to get config dir:", err)
|
|
}
|
|
|
|
// Fill Path:
|
|
dirPath = filepath.Join(d, "cchat-gtk")
|
|
|
|
// Ensure it exists:
|
|
if err := os.Mkdir(dirPath, 0755|os.ModeDir); err != nil && !os.IsExist(err) {
|
|
log.Fatalln("Failed to make config dir:", err)
|
|
}
|
|
}
|
|
|
|
// PrettyMarshal pretty marshals v into dst as formatted JSON.
|
|
func PrettyMarshal(dst io.Writer, v interface{}) error {
|
|
enc := json.NewEncoder(dst)
|
|
enc.SetIndent("", "\t")
|
|
return enc.Encode(v)
|
|
}
|
|
|
|
// DirPath returns the config directory.
|
|
func DirPath() string {
|
|
// Ensure that files and folders are initialized.
|
|
__initonce.Do(__init)
|
|
|
|
return dirPath
|
|
}
|
|
|
|
// SaveToFile saves the given bytes into the given filename. The filename will
|
|
// be prepended with the config directory.
|
|
func SaveToFile(file string, v []byte) error {
|
|
file = filepath.Join(DirPath(), file)
|
|
|
|
f, err := os.OpenFile(file, os.O_CREATE|os.O_WRONLY|os.O_SYNC|os.O_TRUNC, 0644)
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to open file")
|
|
}
|
|
defer f.Close()
|
|
|
|
if _, err := f.Write(v); err != nil {
|
|
return errors.Wrap(err, "failed to write")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// MarshalToFile marshals the given interface into the given filename. The
|
|
// filename will be prepended with the config directory.
|
|
func MarshalToFile(file string, from interface{}) error {
|
|
file = filepath.Join(DirPath(), file)
|
|
|
|
dir := filepath.Dir(file)
|
|
if err := os.MkdirAll(dir, os.ModePerm); err != nil {
|
|
return errors.Wrap(err, "failed to create config dir")
|
|
}
|
|
|
|
f, err := os.OpenFile(file, os.O_CREATE|os.O_WRONLY|os.O_SYNC|os.O_TRUNC, 0644)
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to open file")
|
|
}
|
|
defer f.Close()
|
|
|
|
if err := PrettyMarshal(f, from); err != nil {
|
|
return errors.Wrap(err, "failed to marshal given struct")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// UnmarshalFromFile unmarshals the given filename to the given interface. The
|
|
// filename will be prepended with the config directory. IsNotExist errors are
|
|
// ignored.
|
|
func UnmarshalFromFile(file string, to interface{}) error {
|
|
file = filepath.Join(DirPath(), file)
|
|
|
|
f, err := os.OpenFile(file, os.O_RDONLY, 0644)
|
|
if err != nil {
|
|
// Ignore does not exist error, leave struct as it is.
|
|
if os.IsNotExist(err) {
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
|
|
if err := json.NewDecoder(f).Decode(to); err != nil {
|
|
return errors.Wrap(err, "Failed to unmarshal to given struct")
|
|
}
|
|
|
|
return nil
|
|
}
|