155 lines
2.9 KiB
Go
155 lines
2.9 KiB
Go
package constant
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
P "path"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
const Name = "clash"
|
|
|
|
var (
|
|
GeositeName = "GeoSite.dat"
|
|
GeoipName = "GeoIP.dat"
|
|
)
|
|
|
|
// 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 {
|
|
files, err := ioutil.ReadDir(p.homeDir)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
for _, fi := range files {
|
|
if fi.IsDir() {
|
|
// 目录则直接跳过
|
|
continue
|
|
} else {
|
|
if strings.EqualFold(fi.Name(), "Country.mmdb") {
|
|
GeoipName = fi.Name()
|
|
return P.Join(p.homeDir, fi.Name())
|
|
}
|
|
}
|
|
}
|
|
return P.Join(p.homeDir, "Country.mmdb")
|
|
}
|
|
|
|
func (p *path) OldCache() string {
|
|
return P.Join(p.homeDir, ".cache")
|
|
}
|
|
|
|
func (p *path) Cache() string {
|
|
return P.Join(p.homeDir, "cache.db")
|
|
}
|
|
|
|
func (p *path) GeoIP() string {
|
|
files, err := ioutil.ReadDir(p.homeDir)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
for _, fi := range files {
|
|
if fi.IsDir() {
|
|
// 目录则直接跳过
|
|
continue
|
|
} else {
|
|
if strings.EqualFold(fi.Name(), "GeoIP.dat") {
|
|
GeoipName = fi.Name()
|
|
return P.Join(p.homeDir, fi.Name())
|
|
}
|
|
}
|
|
}
|
|
return P.Join(p.homeDir, "GeoIP.dat")
|
|
}
|
|
|
|
func (p *path) GeoSite() string {
|
|
files, err := ioutil.ReadDir(p.homeDir)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
for _, fi := range files {
|
|
if fi.IsDir() {
|
|
// 目录则直接跳过
|
|
continue
|
|
} else {
|
|
if strings.EqualFold(fi.Name(), "GeoSite.dat") {
|
|
GeositeName = fi.Name()
|
|
return P.Join(p.homeDir, fi.Name())
|
|
}
|
|
}
|
|
}
|
|
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, 0o644)
|
|
}
|
|
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)
|
|
}
|
|
|
|
func (p *path) GetExecutableFullPath() string {
|
|
exePath, err := os.Executable()
|
|
if err != nil {
|
|
return "clash"
|
|
}
|
|
res, _ := filepath.EvalSymlinks(exePath)
|
|
return res
|
|
}
|