mihomo/config/update_ui.go

146 lines
2.9 KiB
Go
Raw Permalink Normal View History

package config
import (
"archive/zip"
"errors"
"fmt"
"io"
"os"
"path"
"path/filepath"
"strings"
"sync"
2023-09-20 14:23:58 +08:00
2023-11-03 21:01:45 +08:00
C "github.com/metacubex/mihomo/constant"
)
2023-09-18 19:21:30 +08:00
var (
ExternalUIURL string
ExternalUIPath string
ExternalUIFolder string
ExternalUIName string
)
var (
2023-09-24 19:00:51 +08:00
ErrIncompleteConf = errors.New("ExternalUI configure incomplete")
)
var xdMutex sync.Mutex
2023-09-18 19:21:30 +08:00
func UpdateUI() error {
xdMutex.Lock()
defer xdMutex.Unlock()
err := prepare()
if err != nil {
return err
2023-09-18 19:21:30 +08:00
}
data, err := downloadForBytes(ExternalUIURL)
if err != nil {
2023-09-18 19:21:30 +08:00
return fmt.Errorf("can't download file: %w", err)
}
2023-09-20 14:23:58 +08:00
saved := path.Join(C.Path.HomeDir(), "download.zip")
if err = saveFile(data, saved); err != nil {
2023-09-18 19:21:30 +08:00
return fmt.Errorf("can't save zip file: %w", err)
}
defer os.Remove(saved)
2023-09-20 14:23:58 +08:00
err = cleanup(ExternalUIFolder)
if err != nil {
if !os.IsNotExist(err) {
return fmt.Errorf("cleanup exist file error: %w", err)
}
}
unzipFolder, err := unzip(saved, C.Path.HomeDir())
if err != nil {
2023-09-18 19:21:30 +08:00
return fmt.Errorf("can't extract zip file: %w", err)
}
2023-09-18 19:21:30 +08:00
err = os.Rename(unzipFolder, ExternalUIFolder)
if err != nil {
return fmt.Errorf("can't rename folder: %w", err)
}
return nil
}
func prepare() error {
if ExternalUIPath == "" || ExternalUIURL == "" {
2023-09-24 19:00:51 +08:00
return ErrIncompleteConf
}
if ExternalUIName != "" {
ExternalUIFolder = filepath.Clean(path.Join(ExternalUIPath, ExternalUIName))
if _, err := os.Stat(ExternalUIPath); os.IsNotExist(err) {
if err := os.MkdirAll(ExternalUIPath, os.ModePerm); err != nil {
return err
}
}
} else {
ExternalUIFolder = ExternalUIPath
}
return nil
}
2023-09-18 19:21:30 +08:00
func unzip(src, dest string) (string, error) {
r, err := zip.OpenReader(src)
if err != nil {
2023-09-18 19:21:30 +08:00
return "", err
}
defer r.Close()
2023-09-18 19:21:30 +08:00
var extractedFolder string
for _, f := range r.File {
fpath := filepath.Join(dest, f.Name)
if !strings.HasPrefix(fpath, filepath.Clean(dest)+string(os.PathSeparator)) {
2023-09-18 19:21:30 +08:00
return "", fmt.Errorf("invalid file path: %s", fpath)
}
if f.FileInfo().IsDir() {
os.MkdirAll(fpath, os.ModePerm)
continue
}
if err = os.MkdirAll(filepath.Dir(fpath), os.ModePerm); err != nil {
2023-09-18 19:21:30 +08:00
return "", err
}
outFile, err := os.OpenFile(fpath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
if err != nil {
2023-09-18 19:21:30 +08:00
return "", err
}
rc, err := f.Open()
if err != nil {
2023-09-18 19:21:30 +08:00
return "", err
}
_, err = io.Copy(outFile, rc)
outFile.Close()
rc.Close()
if err != nil {
2023-09-18 19:21:30 +08:00
return "", err
}
if extractedFolder == "" {
extractedFolder = filepath.Dir(fpath)
}
}
2023-09-18 19:21:30 +08:00
return extractedFolder, nil
}
func cleanup(root string) error {
2023-09-18 19:21:30 +08:00
if _, err := os.Stat(root); os.IsNotExist(err) {
return nil
}
return filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
if err != nil {
return err
}
if info.IsDir() {
if err := os.RemoveAll(path); err != nil {
return err
}
} else {
if err := os.Remove(path); err != nil {
return err
}
}
return nil
})
}