EdgeAPI/internal/installers/deploy_manager.go

package installers

import (
	"github.com/iwind/TeaGo/Tea"
	"github.com/iwind/TeaGo/files"
	stringutil "github.com/iwind/TeaGo/utils/string"
	"regexp"
)

var SharedDeployManager = NewDeployManager()

// DeployManager 节点部署文件管理器
// 如果节点部署文件有变化，需要重启API节点以便于生效
type DeployManager struct {
	dir         string
	nodeFiles   []*DeployFile
	nsNodeFiles []*DeployFile
}

// NewDeployManager 获取新节点部署文件管理器
func NewDeployManager() *DeployManager {
	var manager = &DeployManager{
		dir: Tea.Root + "/deploy",
	}
	manager.LoadNodeFiles()
	manager.LoadNSNodeFiles()
	return manager
}

// LoadNodeFiles 加载所有边缘节点文件
func (this *DeployManager) LoadNodeFiles() []*DeployFile {
	if len(this.nodeFiles) > 0 {
		return this.nodeFiles
	}

	keyMap := map[string]*DeployFile{} // key => File

	reg := regexp.MustCompile(`^edge-node-(\w+)-(\w+)-v([0-9.]+)\.zip$`)
	for _, file := range files.NewFile(this.dir).List() {
		name := file.Name()
		if !reg.MatchString(name) {
			continue
		}
		matches := reg.FindStringSubmatch(name)
		osName := matches[1]
		arch := matches[2]
		version := matches[3]

		key := osName + "_" + arch
		oldFile, ok := keyMap[key]
		if ok && stringutil.VersionCompare(oldFile.Version, version) > 0 {
			continue
		}
		keyMap[key] = &DeployFile{
			OS:      osName,
			Arch:    arch,
			Version: version,
			Path:    file.Path(),
		}
	}

	result := []*DeployFile{}
	for _, v := range keyMap {
		result = append(result, v)
	}

	this.nodeFiles = result

	return result
}

// FindNodeFile 查找特别平台的节点文件
func (this *DeployManager) FindNodeFile(os string, arch string) *DeployFile {
	for _, file := range this.LoadNodeFiles() {
		if file.OS == os && file.Arch == arch {
			return file
		}
	}
	return nil
}

// LoadNSNodeFiles 加载所有NS节点安装文件
func (this *DeployManager) LoadNSNodeFiles() []*DeployFile {
	if len(this.nsNodeFiles) > 0 {
		return this.nsNodeFiles
	}

	keyMap := map[string]*DeployFile{} // key => File

	reg := regexp.MustCompile(`^edge-dns-(\w+)-(\w+)-v([0-9.]+)\.zip$`)
	for _, file := range files.NewFile(this.dir).List() {
		name := file.Name()
		if !reg.MatchString(name) {
			continue
		}
		matches := reg.FindStringSubmatch(name)
		osName := matches[1]
		arch := matches[2]
		version := matches[3]

		key := osName + "_" + arch
		oldFile, ok := keyMap[key]
		if ok && stringutil.VersionCompare(oldFile.Version, version) > 0 {
			continue
		}
		keyMap[key] = &DeployFile{
			OS:      osName,
			Arch:    arch,
			Version: version,
			Path:    file.Path(),
		}
	}

	result := []*DeployFile{}
	for _, v := range keyMap {
		result = append(result, v)
	}

	this.nsNodeFiles = result

	return result
}

// FindNSNodeFile 查找特别平台的NS节点安装文件
func (this *DeployManager) FindNSNodeFile(os string, arch string) *DeployFile {
	for _, file := range this.LoadNSNodeFiles() {
		if file.OS == os && file.Arch == arch {
			return file
		}
	}
	return nil
}