HY2XS_flamy/apps/service/config.go

package service

import (
	"errors"
	"fmt"
	"github.com/sirupsen/logrus"
	"gopkg.in/yaml.v3"
	"hy2xs-admin/dao"
	"hy2xs-admin/model/bo"
	"hy2xs-admin/model/constant"
	"hy2xs-admin/model/entity"
	"net"
	"os"
	"strconv"
	"strings"
)

// ConfigUpdate — одна пара «ключ/значение» из партии операторского API.
type ConfigUpdate struct {
	Key   string
	Value string
}

func UpdateConfig(key string, value string) error {
	return dao.UpdateConfig([]string{key}, map[string]interface{}{"value": value})
}

// UpdateConfigs применяет партию настроек целиком или не применяет вовсе.
//
// Проверка значений в неё не входит намеренно: партия обязана быть проверена
// ПОЛНОСТЬЮ до открытия транзакции. Транзакция, внутри которой ещё может
// возникнуть отказ по содержимому, — это откат вместо отказа, то есть лишняя
// работа с базой на каждом неверно заполненном поле формы.
func UpdateConfigs(updates []ConfigUpdate) error {
	if len(updates) == 0 {
		return nil
	}
	return dao.WithConfigTx(func(tx dao.ConfigTx) error {
		for _, item := range updates {
			if err := tx.UpdateConfig([]string{item.Key}, map[string]interface{}{"value": item.Value}); err != nil {
				return err
			}
		}
		return nil
	})
}

// ValidateConfigValue проверяет ЗНАЧЕНИЕ настройки до записи в базу.
//
// Раньше такой проверки не было вовсе: DTO ограничивал длину 128 символами, а
// поле расписания в панели — el-select с allow-create, то есть строка
// произвольная. Невалидное cron-выражение сохранялось, API отвечал успехом, а
// планировщик при следующем старте просто логировал ошибку и продолжал без
// джобы. Для оператора это выглядело как «настройка принята, сброс трафика
// молча перестал работать».
func ValidateConfigValue(key string, value string) error {
	switch key {
	case constant.ResetTrafficCron:
		return ValidateResetTrafficCron(value)
	default:
		return nil
	}
}

func GetConfig(key string) (entity.Config, error) {
	return dao.GetConfig("key = ?", key)
}

func ListConfig(keys []string) ([]entity.Config, error) {
	return dao.ListConfig("key in ?", keys)
}

// GetHysteria2Config читает ТОЛЬКО /etc/hysteria/config.yaml.
//
// Ветки «сначала посмотреть в SQLite, а если пусто — в файл» здесь больше нет.
// Ключ HYSTERIA2_CONFIG остался от предыдущего поколения, где панель сама
// владела серверным конфигом. В HY2XS владелец ровно один — install-оркестратор,
// маршруты записи конфига удалены, и писателя у этого ключа не осталось: clean
// install создавал его пустым, и пустым он оставался навсегда.
//
// Опасна была не бесполезность, а приоритет: БД читалась ПЕРВОЙ. Любое
// значение, попавшее в неё в обход продукта, молча становилось тем, что панель
// показывает оператору и из чего генерирует клиентские ссылки, пока Hysteria
// продолжала работать по файлу. То есть ключ был готовым вторым источником
// истины — ровно тем, что проект последовательно убирает.
func GetHysteria2Config() (bo.Hysteria2ServerConfig, error) {
	var serverConfig bo.Hysteria2ServerConfig
	content, err := os.ReadFile(constant.Hysteria2ConfigPath)
	if err != nil {
		return serverConfig, err
	}
	if err = yaml.Unmarshal(content, &serverConfig); err != nil {
		return serverConfig, err
	}
	return serverConfig, nil
}

func GetHysteria2ApiPort() (int64, error) {
	hysteria2Config, err := GetHysteria2Config()
	if err != nil {
		return 0, err
	}
	// Сообщение передаётся аргументом, а не форматной строкой: значение
	// приходит из /etc/hysteria/config.yaml, и знак процента в нём превращал
	// запись журнала в мусор вида `%!s(MISSING)`.
	if hysteria2Config.TrafficStats == nil || hysteria2Config.TrafficStats.Listen == nil {
		errMsg := "hysteria2 Traffic Stats API (HTTP) Listen is nil"
		logrus.Errorf("%s", errMsg)
		return 0, errors.New(errMsg)
	}
	apiPort, err := parseTrafficStatsPort(*hysteria2Config.TrafficStats.Listen)
	if err != nil {
		errMsg := fmt.Sprintf("apiPort: %s is invalid: %v", *hysteria2Config.TrafficStats.Listen, err)
		logrus.Errorf("%s", errMsg)
		return 0, errors.New(errMsg)
	}
	return apiPort, nil
}

func parseTrafficStatsPort(listen string) (int64, error) {
	trimmed := strings.TrimSpace(listen)
	if trimmed == "" {
		return 0, errors.New("empty listen")
	}

	hostPort := trimmed
	if strings.HasPrefix(trimmed, ":") {
		hostPort = "127.0.0.1" + trimmed
	}

	_, portStr, err := net.SplitHostPort(hostPort)
	if err != nil {
		return 0, err
	}
	port, err := strconv.ParseInt(portStr, 10, 64)
	if err != nil {
		return 0, err
	}
	if port <= 0 || port > 65535 {
		return 0, errors.New("port out of range")
	}
	return port, nil
}