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

apps/cmd/cmd.go

@@ -0,0 +1,50 @@
+package cmd
+
+import (
+	"fmt"
+	"github.com/spf13/cobra"
+	"hy2xs-admin/model/constant"
+	"hy2xs-admin/util"
+	"os"
+)
+
+var rootCmd = &cobra.Command{
+	Use:   "hy2xs-admin",
+	Short: "HY2XS admin panel for Hysteria2",
+	Long:  "HY2XS admin panel for Hysteria2.",
+	Run:   run,
+}
+
+var (
+	version bool
+	port    string
+)
+
+func init() {
+	rootCmd.Flags().BoolVarP(&version, "version", "v", false, "Show version")
+	rootCmd.Flags().StringVarP(&port, "port", "p", "", "The port of the web server")
+}
+
+func run(cmd *cobra.Command, args []string) {
+	if version {
+		fmt.Println("HY2XS admin version", constant.Version)
+		return
+	}
+
+	if err := util.VerifyPort(port); err != nil {
+		fmt.Println(err.Error())
+		os.Exit(1)
+	}
+	for {
+		if err := runServer(port); err != nil {
+			fmt.Println(err.Error())
+			os.Exit(1)
+		}
+	}
+}
+
+func Execute() {
+	if err := rootCmd.Execute(); err != nil {
+		os.Exit(1)
+	}
+}

apps/cmd/reset.go

@@ -0,0 +1,51 @@
+package cmd
+
+import (
+	"fmt"
+	"github.com/spf13/cobra"
+	"hy2xs-admin/dao"
+	"hy2xs-admin/util"
+	"os"
+)
+
+var resetCmd = &cobra.Command{
+	Use:   "reset",
+	Short: "Reset username and password",
+	Long:  "Reset username and password.",
+	Run:   runReset,
+}
+
+func init() {
+	rootCmd.AddCommand(resetCmd)
+}
+
+func runReset(cmd *cobra.Command, args []string) {
+	username, err := util.RandomString(6)
+	if err != nil {
+		fmt.Println(err.Error())
+		os.Exit(1)
+	}
+	password, err := util.RandomString(6)
+	if err != nil {
+		fmt.Println(err.Error())
+		os.Exit(1)
+	}
+	if err = dao.InitSqliteDB(); err != nil {
+		fmt.Println(err.Error())
+		os.Exit(1)
+	}
+	if err = dao.UpdateAccount([]int64{1}, map[string]interface{}{
+		"username": username,
+		"pass":     util.SHA224String(password),
+		"con_pass": fmt.Sprintf("%s.%s", username, password)}); err != nil {
+		fmt.Println(err.Error())
+		os.Exit(1)
+	}
+	if err = dao.CloseSqliteDB(); err != nil {
+		fmt.Println(err.Error())
+		os.Exit(1)
+	}
+	fmt.Println(fmt.Sprintf("HY2XS admin Login Username: %s", username))
+	fmt.Println(fmt.Sprintf("HY2XS admin Login Password: %s", password))
+	fmt.Println(fmt.Sprintf("HY2XS admin Connection Password: %s", fmt.Sprintf("%s.%s", username, password)))
+}

apps/cmd/server.go

@@ -0,0 +1,85 @@
+package cmd
+
+import (
+	"errors"
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"github.com/sirupsen/logrus"
+	"hy2xs-admin/dao"
+	"hy2xs-admin/middleware"
+	"hy2xs-admin/model/constant"
+	"hy2xs-admin/router"
+	"hy2xs-admin/service"
+	"hy2xs-admin/util"
+	"net/http"
+	"os"
+)
+
+func runServer(port string) error {
+	defer releaseResource()
+
+	middleware.InitLog()
+	service.InitForward()
+	if err := initFile(); err != nil {
+		return err
+	}
+	if err := dao.InitSql(port); err != nil {
+		return err
+	}
+	if err := middleware.InitCron(); err != nil {
+		return err
+	}
+	if err := service.InitHysteria2(); err != nil {
+		return err
+	}
+	if err := service.InitTableAndChain(); err != nil {
+		logrus.Errorf(err.Error())
+	}
+	if err := service.InitPortHopping(); err != nil {
+		logrus.Errorf(err.Error())
+	}
+	config, err := dao.GetConfig("key = ?", constant.HUIWebContext)
+	if err != nil {
+		return err
+	}
+
+	r := gin.Default()
+	router.Router(r, config.Value)
+
+	serverPort, crtPath, keyPath, err := service.GetServerPortAndCert()
+	if err != nil {
+		return err
+	}
+
+	service.InitServer(fmt.Sprintf(":%d", serverPort), r)
+	if err := service.StartServer(crtPath, keyPath); err != nil && err != http.ErrServerClosed {
+		logrus.Errorf("start server err: %v", err)
+		return errors.New("start server err")
+	}
+	return nil
+}
+
+func releaseResource() {
+	if err := dao.CloseSqliteDB(); err != nil {
+		logrus.Errorf(err.Error())
+	}
+	if err := service.ReleaseHysteria2(); err != nil {
+		logrus.Errorf(err.Error())
+	}
+	if err := service.RemoveByComment(); err != nil {
+		logrus.Errorf(err.Error())
+	}
+}
+
+func initFile() error {
+	var dirs = []string{constant.LogDir, constant.SqliteDBDir, constant.BinDir, constant.ExportPathDir}
+	for _, item := range dirs {
+		if !util.Exists(item) {
+			if err := os.Mkdir(item, os.ModePerm); err != nil {
+				logrus.Errorf("%s create err: %v", item, err)
+				return errors.New(fmt.Sprintf("%s create err", item))
+			}
+		}
+	}
+	return nil
+}

apps/cmd/version.go

@@ -0,0 +1,22 @@
+package cmd
+
+import (
+	"fmt"
+	"github.com/spf13/cobra"
+	"hy2xs-admin/model/constant"
+)
+
+var versionCmd = &cobra.Command{
+	Use:   "version",
+	Short: "Show version",
+	Long:  "Show version.",
+	Run:   runVersion,
+}
+
+func init() {
+	rootCmd.AddCommand(versionCmd)
+}
+
+func runVersion(cmd *cobra.Command, args []string) {
+	fmt.Println("HY2XS admin version", constant.Version)
+}