waf-platform/EdgeAPI/internal/installers/installer_ns_node_plus.go

//go:build plus

package installers

import (
	"bytes"
	"errors"
	"fmt"
	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
	"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
	"os"
	"path/filepath"
	"regexp"
)

type NSNodeInstaller struct {
	BaseInstaller
}

func (this *NSNodeInstaller) Install(dir string, params interface{}, installStatus *models.NodeInstallStatus) error {
	if params == nil {
		return errors.New("'params' required for node installation")
	}
	nodeParams, ok := params.(*NodeParams)
	if !ok {
		return errors.New("'params' should be *NodeParams")
	}
	err := nodeParams.Validate()
	if err != nil {
		return fmt.Errorf("params validation: %w", err)
	}

	// 检查目标目录是否存在
	_, err = this.client.Stat(dir)
	if err != nil {
		err = this.client.MkdirAll(dir)
		if err != nil {
			installStatus.ErrorCode = "CREATE_ROOT_DIRECTORY_FAILED"
			return fmt.Errorf("create directory  '%s' failed: %w", dir, err)
		}
	}

	// 安装助手
	env, err := this.InstallHelper(dir, nodeconfigs.NodeRoleDNS)
	if err != nil {
		installStatus.ErrorCode = "INSTALL_HELPER_FAILED"
		return err
	}

	// 上传安装文件
	filePrefix := "edge-dns-" + env.OS + "-" + env.Arch
	zipFile, err := this.LookupLatestInstaller(filePrefix)
	if err != nil {
		return err
	}
	if len(zipFile) == 0 {
		return errors.New("can not find installer file for " + env.OS + "/" + env.Arch)
	}
	var targetZip = ""
	var firstCopyErr error
	var zipName = filepath.Base(zipFile)
	for _, candidateTargetZip := range []string{
		dir + "/" + zipName,
		this.client.UserHome() + "/" + zipName,
		"/tmp/" + zipName,
	} {
		err = this.client.Copy(zipFile, candidateTargetZip, 0777)
		if err != nil {
			if firstCopyErr == nil {
				firstCopyErr = err
			}
		} else {
			err = nil
			firstCopyErr = nil
			targetZip = candidateTargetZip
			break
		}
	}
	if firstCopyErr != nil {
		return fmt.Errorf("upload node file failed: %w", firstCopyErr)
	}

	// 测试运行环境
	// 升级的节点暂时不列入测试
	if !nodeParams.IsUpgrading {
		_, stderr, err := this.client.Exec(env.HelperPath + " -cmd=test")
		if err != nil {
			return fmt.Errorf("test failed: %w", err)
		}
		if len(stderr) > 0 {
			return errors.New("test failed: " + stderr)
		}
	}

	// 如果是升级则优雅停止先前的进程
	exePath := dir + "/edge-dns/bin/edge-dns"
	if nodeParams.IsUpgrading {
		_, err = this.client.Stat(exePath)
		if err == nil {
			_, _, _ = this.client.Exec(exePath + " stop")

			// 删除可执行文件防止冲突
			err = this.client.Remove(exePath)
			if err != nil && err != os.ErrNotExist {
				return fmt.Errorf("remove old file failed: %w", err)
			}
		}
	}

	// 解压
	_, stderr, err := this.client.Exec(env.HelperPath + " -cmd=unzip -zip=\"" + targetZip + "\" -target=\"" + dir + "\"")
	if err != nil {
		return err
	}
	if len(stderr) > 0 {
		return errors.New("unzip installer failed: " + stderr)
	}

	// 修改配置文件
	{
		var configFile = dir + "/edge-dns/configs/api_dns.yaml"

		// sudo之后我们需要修改配置目录才能写入文件
		if this.client.sudo {
			_, _, _ = this.client.Exec("chown " + this.client.User() + " " + filepath.Dir(configFile))
		}

		var data = []byte(`rpc.endpoints: [ ${endpoints} ]
nodeId: "${nodeId}"
secret: "${nodeSecret}"`)

		data = bytes.ReplaceAll(data, []byte("${endpoints}"), []byte(nodeParams.QuoteEndpoints()))
		data = bytes.ReplaceAll(data, []byte("${nodeId}"), []byte(nodeParams.NodeId))
		data = bytes.ReplaceAll(data, []byte("${nodeSecret}"), []byte(nodeParams.Secret))

		_, err = this.client.WriteFile(configFile, data)
		if err != nil {
			return fmt.Errorf("write '%s': %w", configFile, err)
		}
	}

	// 在线安装/更新 Fluent Bit（与 DNS 节点安装流程联动）
	err = this.SetupFluentBit(nodeconfigs.NodeRoleDNS)
	if err != nil {
		installStatus.ErrorCode = "SETUP_FLUENT_BIT_FAILED"
		return fmt.Errorf("setup fluent-bit failed: %w", err)
	}

	// 测试
	_, stderr, err = this.client.Exec(dir + "/edge-dns/bin/edge-dns test")
	if err != nil {
		installStatus.ErrorCode = "TEST_FAILED"
		return fmt.Errorf("test edge node failed: %w, stderr: %s", err, stderr)
	}
	if len(stderr) > 0 {
		if regexp.MustCompile(`(?i)rpc`).MatchString(stderr) {
			installStatus.ErrorCode = "RPC_TEST_FAILED"
		}

		return errors.New("test edge dns node failed: " + stderr)
	}

	// 启动
	_, stderr, err = this.client.Exec(dir + "/edge-dns/bin/edge-dns start")
	if err != nil {
		return fmt.Errorf("start edge dns failed: %w", err)
	}

	if len(stderr) > 0 {
		return errors.New("start edge dns failed: " + stderr)
	}

	return nil
}