Initial commit (code only without large binaries)

EdgeAdmin/internal/web/actions/default/settings/server/admin-server-utils/utils.go

@@ -0,0 +1,86 @@
+package adminserverutils
+
+import (
+	"errors"
+	"github.com/iwind/TeaGo"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/types"
+	"gopkg.in/yaml.v3"
+	"net"
+	"os"
+	"time"
+)
+
+var ServerConfigIsChanged = false
+
+const configFilename = "server.yaml"
+
+// LoadServerConfig 读取当前服务配置
+func LoadServerConfig() (*TeaGo.ServerConfig, error) {
+	configFile := Tea.ConfigFile(configFilename)
+	data, err := os.ReadFile(configFile)
+	if err != nil {
+		return nil, err
+	}
+	var serverConfig = &TeaGo.ServerConfig{}
+	err = yaml.Unmarshal(data, serverConfig)
+	if err != nil {
+		return nil, err
+	}
+	return serverConfig, nil
+}
+
+// WriteServerConfig 保存当前服务配置
+func WriteServerConfig(serverConfig *TeaGo.ServerConfig) error {
+	data, err := yaml.Marshal(serverConfig)
+	if err != nil {
+		return err
+	}
+	err = os.WriteFile(Tea.ConfigFile(configFilename), data, 0666)
+	if err != nil {
+		return err
+	}
+
+	ServerConfigIsChanged = true
+
+	return nil
+}
+
+// ReadServerHTTPS 检查HTTPS地址
+func ReadServerHTTPS() (port int, err error) {
+	config, err := LoadServerConfig()
+	if err != nil {
+		return 0, err
+	}
+	if config == nil {
+		return 0, errors.New("could not load server config")
+	}
+
+	if config.Https.On && len(config.Https.Listen) > 0 {
+		for _, listen := range config.Https.Listen {
+			_, portString, splitErr := net.SplitHostPort(listen)
+			if splitErr == nil {
+				var portInt = types.Int(portString)
+				if portInt > 0 {
+					// 是否已经启动
+					checkErr := func() error {
+						conn, connErr := net.DialTimeout("tcp", ":"+portString, 1*time.Second)
+						if connErr != nil {
+							return connErr
+						}
+						_ = conn.Close()
+						return nil
+					}()
+					if checkErr != nil {
+						continue
+					}
+
+					port = portInt
+					err = nil
+					break
+				}
+			}
+		}
+	}
+	return
+}

EdgeAdmin/internal/web/actions/default/settings/server/index.go

@@ -0,0 +1,28 @@
+package server
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	adminserverutils "github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/settings/server/admin-server-utils"
+)
+
+type IndexAction struct {
+	actionutils.ParentAction
+}
+
+func (this *IndexAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *IndexAction) RunGet(params struct{}) {
+	this.Data["serverIsChanged"] = adminserverutils.ServerConfigIsChanged
+
+	serverConfig, err := adminserverutils.LoadServerConfig()
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Data["serverConfig"] = serverConfig
+
+	this.Show()
+}

EdgeAdmin/internal/web/actions/default/settings/server/init.go

@@ -0,0 +1,21 @@
+package server
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/configloaders"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/settings/settingutils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/helpers"
+	"github.com/iwind/TeaGo"
+)
+
+func init() {
+	TeaGo.BeforeStart(func(server *TeaGo.Server) {
+		server.
+			Helper(helpers.NewUserMustAuth(configloaders.AdminModuleCodeServer)).
+			Helper(settingutils.NewHelper("server")).
+			Prefix("/settings/server").
+			Get("", new(IndexAction)).
+			GetPost("/updateHTTPPopup", new(UpdateHTTPPopupAction)).
+			GetPost("/updateHTTPSPopup", new(UpdateHTTPSPopupAction)).
+			EndAll()
+	})
+}

EdgeAdmin/internal/web/actions/default/settings/server/updateHTTPPopup.go

@@ -0,0 +1,69 @@
+package server
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/utils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	adminserverutils "github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/settings/server/admin-server-utils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/langs/codes"
+	"github.com/iwind/TeaGo/actions"
+	"net"
+)
+
+type UpdateHTTPPopupAction struct {
+	actionutils.ParentAction
+}
+
+func (this *UpdateHTTPPopupAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *UpdateHTTPPopupAction) RunGet(params struct{}) {
+	serverConfig, err := adminserverutils.LoadServerConfig()
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	this.Data["serverConfig"] = serverConfig
+
+	this.Show()
+}
+
+func (this *UpdateHTTPPopupAction) RunPost(params struct {
+	IsOn    bool
+	Listens []string
+
+	Must *actions.Must
+}) {
+	defer this.CreateLogInfo(codes.AdminServer_LogUpdateServerHTTPSettings)
+
+	if len(params.Listens) == 0 {
+		this.Fail("请输入绑定地址")
+	}
+
+	serverConfig, err := adminserverutils.LoadServerConfig()
+	if err != nil {
+		this.Fail("保存失败：" + err.Error())
+	}
+
+	serverConfig.Http.On = params.IsOn
+
+	listen := []string{}
+	for _, addr := range params.Listens {
+		addr = utils.FormatAddress(addr)
+		if len(addr) == 0 {
+			continue
+		}
+		if _, _, err := net.SplitHostPort(addr); err != nil {
+			addr += ":80"
+		}
+		listen = append(listen, addr)
+	}
+	serverConfig.Http.Listen = listen
+
+	err = adminserverutils.WriteServerConfig(serverConfig)
+	if err != nil {
+		this.Fail("保存失败：" + err.Error())
+	}
+
+	this.Success()
+}

EdgeAdmin/internal/web/actions/default/settings/server/updateHTTPSPopup.go

@@ -0,0 +1,145 @@
+package server
+
+import (
+	"context"
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeAdmin/internal/utils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	adminserverutils "github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/settings/server/admin-server-utils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/langs/codes"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/sslconfigs"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/actions"
+	"net"
+	"os"
+)
+
+type UpdateHTTPSPopupAction struct {
+	actionutils.ParentAction
+}
+
+func (this *UpdateHTTPSPopupAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *UpdateHTTPSPopupAction) RunGet(params struct{}) {
+	serverConfig, err := adminserverutils.LoadServerConfig()
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	this.Data["serverConfig"] = serverConfig
+
+	// 证书
+	certConfigs := []*sslconfigs.SSLCertConfig{}
+	if len(serverConfig.Https.Cert) > 0 && len(serverConfig.Https.Key) > 0 {
+		certData, err := os.ReadFile(Tea.Root + "/" + serverConfig.Https.Cert)
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		keyData, err := os.ReadFile(Tea.Root + "/" + serverConfig.Https.Key)
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		certConfig := &sslconfigs.SSLCertConfig{
+			Id:       0,
+			Name:     "-",
+			CertData: certData,
+			KeyData:  keyData,
+		}
+		_ = certConfig.Init(context.TODO())
+		certConfig.CertData = nil
+		certConfig.KeyData = nil
+		certConfigs = append(certConfigs, certConfig)
+	}
+	this.Data["certConfigs"] = certConfigs
+
+	this.Show()
+}
+
+func (this *UpdateHTTPSPopupAction) RunPost(params struct {
+	IsOn        bool
+	Listens     []string
+	CertIdsJSON []byte
+
+	Must *actions.Must
+}) {
+	defer this.CreateLogInfo(codes.AdminServer_LogUpdateServerHTTPSSettings)
+
+	if len(params.Listens) == 0 {
+		this.Fail("请输入绑定地址")
+	}
+
+	serverConfig, err := adminserverutils.LoadServerConfig()
+	if err != nil {
+		this.Fail("保存失败：" + err.Error())
+	}
+
+	serverConfig.Https.On = params.IsOn
+
+	listen := []string{}
+	for _, addr := range params.Listens {
+		addr = utils.FormatAddress(addr)
+		if len(addr) == 0 {
+			continue
+		}
+		if _, _, err := net.SplitHostPort(addr); err != nil {
+			addr += ":80"
+		}
+		listen = append(listen, addr)
+	}
+	serverConfig.Https.Listen = listen
+
+	// 证书
+	certIds := []int64{}
+	err = json.Unmarshal(params.CertIdsJSON, &certIds)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	if params.IsOn && len(certIds) == 0 {
+		this.Fail("要启用HTTPS，需要先选择或上传一个可用的证书")
+	}
+
+	// 保存证书到本地
+	if len(certIds) > 0 && certIds[0] != 0 {
+		certResp, err := this.RPC().SSLCertRPC().FindEnabledSSLCertConfig(this.AdminContext(), &pb.FindEnabledSSLCertConfigRequest{
+			SslCertId: certIds[0],
+		})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		if len(certResp.SslCertJSON) == 0 {
+			this.Fail("选择的证书已失效，请换一个")
+		}
+
+		certConfig := &sslconfigs.SSLCertConfig{}
+		err = json.Unmarshal(certResp.SslCertJSON, certConfig)
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		err = os.WriteFile(Tea.ConfigFile("https.key.pem"), certConfig.KeyData, 0666)
+		if err != nil {
+			this.Fail("保存密钥失败：" + err.Error())
+		}
+		err = os.WriteFile(Tea.ConfigFile("https.cert.pem"), certConfig.CertData, 0666)
+		if err != nil {
+			this.Fail("保存证书失败：" + err.Error())
+		}
+
+		serverConfig.Https.Key = "configs/https.key.pem"
+		serverConfig.Https.Cert = "configs/https.cert.pem"
+	}
+
+	err = adminserverutils.WriteServerConfig(serverConfig)
+	if err != nil {
+		this.Fail("保存配置失败：" + err.Error())
+	}
+
+	this.Success()
+}