Подготовить HY2XS к production-сборке

apps/service/hysteria2.go

@@ -0,0 +1,149 @@
+package service
+
+import (
+	"errors"
+	"fmt"
+	"github.com/sirupsen/logrus"
+	"gopkg.in/yaml.v3"
+	"hy2xs-admin/dao"
+	"hy2xs-admin/model/constant"
+	"hy2xs-admin/model/vo"
+	"hy2xs-admin/util"
+	"os"
+)
+
+func InitHysteria2() error {
+	if !util.Exists(util.GetHysteria2BinPath()) {
+		return errors.New("systemd-managed hysteria binary not found")
+	}
+
+	config, err := dao.GetConfig("key = ?", constant.Hysteria2Enable)
+	if err != nil {
+		return err
+	}
+
+	if *config.Value == "1" {
+		logrus.Infof("hysteria2 lifecycle is managed by systemd in HY2XS production package")
+	}
+	return nil
+}
+
+func setHysteria2ConfigYAML() error {
+	serverConfig, err := GetHysteria2Config()
+	if err != nil {
+		return err
+	}
+	if serverConfig.Listen == nil || *serverConfig.Listen == "" {
+		return errors.New("hysteria2 config is empty")
+	}
+
+	authHttpUrl, err := GetAuthHttpUrl()
+	if err != nil {
+		return err
+	}
+
+	if serverConfig.Auth == nil || serverConfig.Auth.HTTP == nil || serverConfig.Auth.HTTP.URL == nil {
+		if err := UpdateHysteria2Config(serverConfig); err != nil {
+			return err
+		}
+		serverConfig, err = GetHysteria2Config()
+		if err != nil {
+			return err
+		}
+	}
+
+	// update auth http url
+	if *serverConfig.Auth.HTTP.URL != authHttpUrl {
+		serverConfig.Auth.HTTP.URL = &authHttpUrl
+		if err := UpdateHysteria2Config(serverConfig); err != nil {
+			return err
+		}
+	}
+
+	hysteria2Config, err := yaml.Marshal(&serverConfig)
+	if err != nil {
+		logrus.Errorf("marshal hysteria2 config err: %v", err)
+		return errors.New("marshal hysteria2 config err")
+	}
+	file, err := os.OpenFile(constant.Hysteria2ConfigPath, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
+	if err != nil {
+		logrus.Errorf("create hysteria2 server config file err: %v", err)
+		return errors.New("create hysteria2 server config file err")
+	}
+	_, err = file.WriteString(string(hysteria2Config))
+	if err != nil {
+		logrus.Errorf("write hysteria2 config.json file err: %v", err)
+		return errors.New("hysteria2 config.json file write err")
+	}
+	return nil
+}
+
+func Hysteria2IsRunning() bool {
+	_, err := util.Exec("systemctl is-active --quiet hysteria-server")
+	return err == nil
+}
+
+func StartHysteria2() error {
+	if err := setHysteria2ConfigYAML(); err != nil {
+		return err
+	}
+	return util.Systemctl("restart", "hysteria-server")
+}
+
+func StopHysteria2() error {
+	return util.Systemctl("stop", "hysteria-server")
+}
+
+func RestartHysteria2() error {
+	if err := StopHysteria2(); err != nil {
+		return err
+	}
+	if err := StartHysteria2(); err != nil {
+		return err
+	}
+	return nil
+}
+
+func ReleaseHysteria2() error {
+	return nil
+}
+
+func Hysteria2AcmePath() (vo.Hysteria2AcmePathVo, error) {
+	hysteria2AcmePathVo := vo.Hysteria2AcmePathVo{}
+	hysteria2Config, err := GetHysteria2Config()
+	if err != nil {
+		return hysteria2AcmePathVo, err
+	}
+	if hysteria2Config.TLS != nil &&
+		hysteria2Config.TLS.Cert != nil && *hysteria2Config.TLS.Cert != "" &&
+		hysteria2Config.TLS.Key != nil && *hysteria2Config.TLS.Key != "" {
+		if util.Exists(*hysteria2Config.TLS.Cert) && util.Exists(*hysteria2Config.TLS.Key) {
+			hysteria2AcmePathVo.CrtPath = *hysteria2Config.TLS.Cert
+			hysteria2AcmePathVo.KeyPath = *hysteria2Config.TLS.Key
+			return hysteria2AcmePathVo, nil
+		}
+		return hysteria2AcmePathVo, errors.New("cert not found")
+	} else if hysteria2Config.ACME != nil &&
+		hysteria2Config.ACME.Domains != nil &&
+		len(hysteria2Config.ACME.Domains) > 0 &&
+		hysteria2Config.ACME.CA != nil &&
+		*hysteria2Config.ACME.CA != "" &&
+		hysteria2Config.ACME.Dir != nil &&
+		*hysteria2Config.ACME.Dir != "" {
+		acmeDir := *hysteria2Config.ACME.Dir
+		for _, domain := range hysteria2Config.ACME.Domains {
+			crtPath, err := util.FindFile(acmeDir, fmt.Sprintf("%s.crt", domain))
+			if err != nil {
+				continue
+			}
+			keyPath, err := util.FindFile(acmeDir, fmt.Sprintf("%s.key", domain))
+			if err != nil {
+				continue
+			}
+			hysteria2AcmePathVo.CrtPath = crtPath
+			hysteria2AcmePathVo.KeyPath = keyPath
+			return hysteria2AcmePathVo, nil
+		}
+	}
+	return vo.Hysteria2AcmePathVo{}, errors.New("cert not found")
+}