mihomo/constant/path.go

package constant

import (
	"os"
	P "path"
	"path/filepath"
)

const Name = "clash"

// Path is used to get the configuration path
var Path = func() *path {
	homeDir, err := os.UserHomeDir()
	if err != nil {
		homeDir, _ = os.Getwd()
	}

	homeDir = P.Join(homeDir, ".config", Name)
	return &path{homeDir: homeDir, configFile: "config.yaml"}
}()

type path struct {
	homeDir    string
	configFile string
	scriptDir  string
}

// SetHomeDir is used to set the configuration path
func SetHomeDir(root string) {
	Path.homeDir = root
}

// SetConfig is used to set the configuration file
func SetConfig(file string) {
	Path.configFile = file
}

func (p *path) HomeDir() string {
	return p.homeDir
}

func (p *path) Config() string {
	return p.configFile
}

// Resolve return a absolute path or a relative path with homedir
func (p *path) Resolve(path string) string {
	if !filepath.IsAbs(path) {
		return filepath.Join(p.HomeDir(), path)
	}

	return path
}

func (p *path) MMDB() string {
	return P.Join(p.homeDir, "Country.mmdb")
}

func (p *path) Cache() string {
	return P.Join(p.homeDir, ".cache")
}

func (p *path) GeoIP() string {
	return P.Join(p.homeDir, "geoip.dat")
}

func (p *path) GeoSite() string {
	return P.Join(p.homeDir, "geosite.dat")
}

func (p *path) ScriptDir() string {
	if len(p.scriptDir) != 0 {
		return p.scriptDir
	}
	if dir, err := os.MkdirTemp("", Name+"-"); err == nil {
		p.scriptDir = dir
	} else {
		p.scriptDir = P.Join(os.TempDir(), Name)
		os.MkdirAll(p.scriptDir, 0644)
	}
	return p.scriptDir
}

func (p *path) Script() string {
	return P.Join(p.ScriptDir(), "clash_script.py")
}

func (p *path) GetAssetLocation(file string) string {
	return P.Join(p.homeDir, file)
}