strip-beats/config/config.go

package config

import (
	"errors"
	"github.com/BurntSushi/toml"
	"github.com/adrg/xdg"
	"github.com/kirsle/configdir"
	"os"
	"path/filepath"
)

type Config struct {
	Source           string            `toml:"source"`
	SourceExtensions []string          `toml:"source_ext"`
	Recurse          bool              `toml:"recurse"`
	SavePath         string            `toml:"save_path"`
	MusicPath        string            `toml:"music_path"`
	Codec            string            `toml:"codec"`
	FfEncoder        string            `toml:"ff_encoder"`
	AllowedCodecs    []string          `toml:"allowed_codec"`
	CodecExt         map[string]string `toml:"codec_ext"`
}

var config Config

func newConfig() Config {
	return Config{
		Source:           xdg.UserDirs.Download,
		SourceExtensions: []string{"webm", "mp4"},
		Recurse:          true,
		SavePath:         filepath.Join(xdg.UserDirs.Documents, "StripBeats"),
		MusicPath:        xdg.UserDirs.Music,
		Codec:            "opus",
		FfEncoder:        "libopus",
		AllowedCodecs:    []string{"aac", "opus"},
		CodecExt:         map[string]string{"aac": "m4a", "opus": "opus"},
	}
}

func init() {
	config = newConfig()
	configPath := configdir.LocalConfig("strip-beats")
	if err := configdir.MakePath(configPath); err != nil {
		panic(err)
	}

	configFile := filepath.Join(configPath, "profile.toml")
	if !exists(configFile) {
		file, err := os.Create(configFile)
		if err != nil {
			panic(err)
		}
		err = toml.NewEncoder(file).Encode(config)
		if err != nil {
			panic(err)
		}
		file.Close()
	} else {
		if _, err := toml.DecodeFile(configFile, &config); err != nil {
			panic(err)
		}
	}
}

func GetConfig() Config {
	return config
}

func exists(path string) bool {
	_, err := os.Stat(path)
	return !errors.Is(err, os.ErrNotExist)
}