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"
	"net/http"
	"os"
)

func resolveUiBindHost() (string, error) {
	host := os.Getenv("HY2XS_UI_BIND_HOST")
	if host == "" {
		host = "127.0.0.1"
	}
	ip := net.ParseIP(host)
	if ip == nil || ip.To4() == nil {
		return "", errors.New("HY2XS_UI_BIND_HOST must be IPv4")
	}
	return host, nil
}

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
	}
	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
	}
	bindHost, err := resolveUiBindHost()
	if err != nil {
		return err
	}

	service.InitServer(fmt.Sprintf("%s:%d", bindHost, 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
}