MeBox/internal/service/runtime_settings.go

package service

import (
	"context"
	"strconv"
	"strings"

	"go.uber.org/zap"

	"github.com/truewhile/MeBox/internal/config"
	"github.com/truewhile/MeBox/internal/repository"
)

func ApplyRuntimeSettings(ctx context.Context, cfg *config.Config, repos *repository.Container, log *zap.Logger) {
	if cfg == nil || repos == nil || repos.Setting == nil {
		return
	}
	rows, err := repos.Setting.All(ctx)
	if err != nil {
		if log != nil {
			log.Warn("load runtime settings failed", zap.Error(err))
		}
		return
	}
	for _, row := range rows {
		ApplyRuntimeSetting(cfg, row.Key, row.Value)
	}
}

func ApplyRuntimeSetting(cfg *config.Config, key, value string) {
	config.RuntimeMu.Lock()
	defer config.RuntimeMu.Unlock()
	if cfg == nil {
		return
	}
	value = strings.TrimSpace(value)
	switch key {
	case "ffmpeg.path", "app.ffmpeg_path":
		cfg.App.FFmpegPath = value
	case "app.server_url", "server.url", "public.server_url", "strm.base_url":
		cfg.App.ServerURL = strings.TrimRight(value, "/")
	case "ffprobe.path", "app.ffprobe_path":
		cfg.App.FFprobePath = value
	case "ffprobe.max_concurrent", "app.ffprobe_max_concurrent":
		if n, err := strconv.Atoi(value); err == nil {
			if n < 1 {
				n = 1
			}
			if n > 8 {
				n = 8
			}
			cfg.App.FFprobeMaxConcurrent = n
		}
	case "cloud.scan_max_concurrent", "cloud.scan_list_concurrency", "app.cloud_scan_max_concurrent":
		if n, err := strconv.Atoi(value); err == nil {
			if n < 1 {
				n = 1
			}
			if n > 16 {
				n = 16
			}
			cfg.App.CloudScanMaxConcurrent = n
		}
	case "app.max_cpu_threads", "runtime.max_cpu_threads":
		if n, err := strconv.Atoi(value); err == nil {
			if n < 1 {
				n = 1
			}
			if n > 8 {
				n = 8
			}
			cfg.App.MaxCPUThreads = n
		}
	case "transcode.enabled", "transcoder.enabled":
		cfg.Transcoder.Enabled = parseBoolSetting(value, true)
	case "transcode.hw_enabled", "transcoder.hardware_accel":
		cfg.Transcoder.HardwareAccel = parseBoolSetting(value, false)
	case "transcode.hw_accel", "transcoder.encoder":
		switch value {
		case "", "auto", "none", "software":
			cfg.Transcoder.Encoder = ""
		case "nvenc", "qsv", "vaapi":
			cfg.Transcoder.Encoder = value
		}
	case "transcode.max_height", "transcoder.max_height":
		if n, err := strconv.Atoi(value); err == nil {
			cfg.Transcoder.MaxHeight = n
		}
	case "transcode.max_jobs", "transcoder.max_concurrent":
		if n, err := strconv.Atoi(value); err == nil {
			cfg.Transcoder.MaxConcurrent = n
		}
	case "transcode.realtime", "transcoder.realtime":
		cfg.Transcoder.Realtime = parseBoolSetting(value, true)
	case "transcode.threads", "transcoder.threads":
		if n, err := strconv.Atoi(value); err == nil {
			cfg.Transcoder.Threads = n
		}
	case "transcode.idle_timeout_seconds", "transcoder.idle_timeout_seconds":
		if n, err := strconv.Atoi(value); err == nil {
			cfg.Transcoder.IdleTimeoutSeconds = n
		}
	case "transcode.video_bitrate", "transcoder.video_bitrate":
		cfg.Transcoder.VideoBitrate = value
	case "https.enabled":
		cfg.App.HTTPSEnabled = parseBoolSetting(value, false)
	case "https.cert":
		cfg.App.SSLCert = value
	case "https.key":
		cfg.App.SSLKey = value
	case "https.cert_path":
		cfg.App.SSLCertPath = strings.TrimSpace(value)
	case "https.key_path":
		cfg.App.SSLKeyPath = strings.TrimSpace(value)
	case "cache.images_max_size_mb":
		if n, err := strconv.Atoi(value); err == nil {
			if n < 0 {
				n = 0
			}
			cfg.Cache.ImagesMaxSizeMB = n
		}
	case "cache.memory_max_size_mb":
		n := config.DefaultCacheMemoryMaxSizeMB
		if value != "" {
			parsed, err := strconv.Atoi(value)
			if err != nil {
				return
			}
			n = parsed
		}
		if n <= 0 {
			n = config.DefaultCacheMemoryMaxSizeMB
		}
		if n > 4096 {
			n = 4096
		}
		cfg.Cache.MemoryMaxSizeMB = n
	}
}

// ParseBoolSetting is the exported variant of parseBoolSetting for handlers
// that need to interpret persisted on/off settings (accepts zh + en tokens).
func ParseBoolSetting(value string, fallback bool) bool {
	return parseBoolSetting(value, fallback)
}

func parseBoolSetting(value string, fallback bool) bool {
	switch strings.ToLower(strings.TrimSpace(value)) {
	case "1", "true", "yes", "on", "enabled", "启用", "开启":
		return true
	case "0", "false", "no", "off", "disabled", "禁用", "关闭":
		return false
	default:
		return fallback
	}
}