XProxy/next/assets/assets.go

package assets

import (
	"XProxy/next/logger"
	"os"
	"time"
)

// saveAsset is used to write to a local file and specify its last modify
// time. If the file exists, it will be replaced.
func saveAsset(file string, content []byte, date *time.Time) error {
	fp, err := os.OpenFile(file, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
	if err != nil {
		logger.Errorf("Failed to open file `%s` -> %v", file, err)
		return err
	}
	defer fp.Close()

	_, err = fp.Write(content)
	if err != nil {
		logger.Errorf("Failed to save file `%s` -> %v", file, err)
		return err
	}

	if date != nil { // change local file last modify time
		if err := os.Chtimes(file, *date, *date); err != nil {
			logger.Warnf("Failed to change asset modification time")
		} else {
			logger.Debugf("Change `%s` modification time to `%v`", file, date)
		}
	}
	return nil
}

// updateRemoteAsset will download remote asset via the optional proxy and
// save them locally. Local files will be overwritten if they exist.
func updateRemoteAsset(file string, url string, proxy string) error {
	logger.Debugf("Start downloading remote asset `%s` to `%s`", url, file)
	asset, date, err := downloadAsset(url, proxy)
	if err != nil {
		logger.Errorf("Failed to download remote asset `%s`", url)
		return err
	}
	if asset, err = tryExtract(asset); err != nil {
		return err
	}
	if err := saveAsset(file, asset, date); err != nil {
		return err
	}
	logger.Infof("Successfully obtained remote asset `%s`", file)
	return nil
}

// updateLocalAsset will extract local asset and save them locally. If the
// local file already exists, it will be skipped.
func updateLocalAsset(file string, src string) error {
	_, err := os.Stat(file)
	if err == nil {
		logger.Debugf("Local asset `%s` already exist", file)
		return nil // skip local asset extract
	}

	var date = time.Now()
	if stat, err := os.Stat(src); err == nil {
		date = stat.ModTime() // using last modify time of src file
	}

	logger.Debugf("Start extracting local asset `%s`", file)
	asset, err := os.ReadFile(src)
	if err != nil {
		logger.Errorf("Failed to read local asset -> %v", err)
		return err
	}
	if asset, err = tryExtract(asset); err != nil {
		return err
	}
	if err := saveAsset(file, asset, &date); err != nil {
		return err
	}
	logger.Infof("Successfully extracted local asset `%s`", file)
	return nil
}

func Demo() {

	//updateRemoteAsset("geosite.dat", "http://cdn.dnomd343.top/v2ray-rules-dat/geosite.dat.xz", "")
	//updateRemoteAsset("geosite.dat", "http://cdn.dnomd343.top/v2ray-rules-dat/geosite.dat.xz", "socks5://192.168.2.2:1084")
	updateLocalAsset("geosite.dat", "geosite.dat.xz")

}