res-downloader/core/config.go

package core

import (
	"encoding/json"
	"runtime"
	"strconv"
	"strings"
	"sync"
)

type MimeInfo struct {
	Type   string `json:"Type"`
	Suffix string `json:"Suffix"`
}

// Config struct
type Config struct {
	storage       *Storage
	Theme         string              `json:"Theme"`
	Locale        string              `json:"Locale"`
	Host          string              `json:"Host"`
	Port          string              `json:"Port"`
	Quality       int                 `json:"Quality"`
	SaveDirectory string              `json:"SaveDirectory"`
	FilenameLen   int                 `json:"FilenameLen"`
	FilenameTime  bool                `json:"FilenameTime"`
	UpstreamProxy string              `json:"UpstreamProxy"`
	OpenProxy     bool                `json:"OpenProxy"`
	DownloadProxy bool                `json:"DownloadProxy"`
	AutoProxy     bool                `json:"AutoProxy"`
	WxAction      bool                `json:"WxAction"`
	TaskNumber    int                 `json:"TaskNumber"`
	UserAgent     string              `json:"UserAgent"`
	UseHeaders    string              `json:"UseHeaders"`
	MimeMap       map[string]MimeInfo `json:"MimeMap"`
}

var (
	mimeMux sync.RWMutex
)

func initConfig() *Config {
	if globalConfig == nil {
		def := `
{
  "Host": "127.0.0.1",
  "Port": "8899",
  "Theme": "lightTheme",
  "Quality": 0,
  "SaveDirectory": "",
  "FilenameLen": 0,
  "FilenameTime": true,
  "UpstreamProxy": "",
  "OpenProxy": false,
  "DownloadProxy": false,
  "AutoProxy": true,
  "WxAction": true,
  "TaskNumber": __TaskNumber__,
  "UserAgent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36",
  "UseHeaders": "User-Agent,Referer,Authorization,Cookie",
  "MimeMap": {
	  "image/png": { "Type": "image", "Suffix": ".png" },
	  "image/webp": { "Type": "image", "Suffix": ".webp" },
	  "image/jpeg": { "Type": "image", "Suffix": ".jpeg" },
	  "image/jpg": { "Type": "image", "Suffix": ".jpg" },
	  "image/gif": { "Type": "image", "Suffix": ".gif" },
	  "image/avif": { "Type": "image", "Suffix": ".avif" },
	  "image/bmp": { "Type": "image", "Suffix": ".bmp" },
	  "image/tiff": { "Type": "image", "Suffix": ".tiff" },
	  "image/heic": { "Type": "image", "Suffix": ".heic" },
	  "image/x-icon": { "Type": "image", "Suffix": ".ico" },
	  "image/svg+xml": { "Type": "image", "Suffix": ".svg" },
	  "image/vnd.adobe.photoshop": { "Type": "image", "Suffix": ".psd" },
	  "image/jp2": { "Type": "image", "Suffix": ".jp2" },
	  "image/jpeg2000": { "Type": "image", "Suffix": ".jp2" },
	  "image/apng": { "Type": "image", "Suffix": ".apng" },
	  "audio/mpeg": { "Type": "audio", "Suffix": ".mp3" },
	  "audio/mp3": { "Type": "audio", "Suffix": ".mp3" },
	  "audio/wav": { "Type": "audio", "Suffix": ".wav" },
	  "audio/aiff": { "Type": "audio", "Suffix": ".aiff" },
	  "audio/x-aiff": { "Type": "audio", "Suffix": ".aiff" },
	  "audio/aac": { "Type": "audio", "Suffix": ".aac" },
	  "audio/ogg": { "Type": "audio", "Suffix": ".ogg" },
	  "audio/flac": { "Type": "audio", "Suffix": ".flac" },
	  "audio/midi": { "Type": "audio", "Suffix": ".mid" },
	  "audio/x-midi": { "Type": "audio", "Suffix": ".mid" },
	  "audio/x-ms-wma": { "Type": "audio", "Suffix": ".wma" },
	  "audio/opus": { "Type": "audio", "Suffix": ".opus" },
	  "audio/webm": { "Type": "audio", "Suffix": ".webm" },
	  "audio/mp4": { "Type": "audio", "Suffix": ".m4a" },
	  "audio/amr": { "Type": "audio", "Suffix": ".amr" },
	  "video/mp4": { "Type": "video", "Suffix": ".mp4" },
	  "video/webm": { "Type": "video", "Suffix": ".webm" },
	  "video/ogg": { "Type": "video", "Suffix": ".ogv" },
	  "video/x-msvideo": { "Type": "video", "Suffix": ".avi" },
	  "video/mpeg": { "Type": "video", "Suffix": ".mpeg" },
	  "video/quicktime": { "Type": "video", "Suffix": ".mov" },
	  "video/x-ms-wmv": { "Type": "video", "Suffix": ".wmv" },
	  "video/3gpp": { "Type": "video", "Suffix": ".3gp" },
	  "video/x-matroska": { "Type": "video", "Suffix": ".mkv" },
	  "audio/video": { "Type": "live", "Suffix": ".flv" },
	  "video/x-flv": { "Type": "live", "Suffix": ".flv" },
	  "application/dash+xml": { "Type": "live", "Suffix": ".mpd" },
	  "application/vnd.apple.mpegurl": { "Type": "m3u8", "Suffix": ".m3u8" },
	  "application/x-mpegurl": { "Type": "m3u8", "Suffix": ".m3u8" },
	  "application/x-mpeg": { "Type": "m3u8", "Suffix": ".m3u8" },
	  "application/pdf": { "Type": "pdf", "Suffix": ".pdf" },
	  "application/vnd.ms-powerpoint": { "Type": "ppt", "Suffix": ".ppt" },
	  "application/vnd.openxmlformats-officedocument.presentationml.presentation": { "Type": "ppt", "Suffix": ".pptx" },
	  "application/vnd.ms-excel": { "Type": "xls", "Suffix": ".xls" },
	  "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet": { "Type": "xls", "Suffix": ".xlsx" },
	  "text/csv": { "Type": "xls", "Suffix": ".csv" },
	  "application/msword": { "Type": "doc", "Suffix": ".doc" },
	  "application/rtf": { "Type": "doc", "Suffix": ".rtf" },
	  "text/rtf": { "Type": "doc", "Suffix": ".rtf" },
	  "application/vnd.oasis.opendocument.text": { "Type": "doc", "Suffix": ".odt" },
	  "application/vnd.openxmlformats-officedocument.wordprocessingml.document": { "Type": "doc", "Suffix": ".docx" },
	  "font/woff": { "Type": "font", "Suffix": ".woff" }
	}
}
`
		def = strings.ReplaceAll(def, "__TaskNumber__", strconv.Itoa(runtime.NumCPU()*2))
		globalConfig = &Config{
			storage: NewStorage("config.json", []byte(def)),
		}

		defaultMap := make(map[string]interface{})
		_ = json.Unmarshal([]byte(def), &defaultMap)

		data, err := globalConfig.storage.Load()
		if err == nil {
			var loadedMap map[string]interface{}
			_ = json.Unmarshal(data, &loadedMap)

			for key, val := range defaultMap {
				if _, ok := loadedMap[key]; !ok {
					loadedMap[key] = val
				}
			}

			finalBytes, _ := json.Marshal(loadedMap)
			_ = json.Unmarshal(finalBytes, &globalConfig)

		} else {
			globalLogger.Esg(err, "load config err")
		}
	}
	return globalConfig
}

func (c *Config) setConfig(config Config) {
	oldProxy := c.UpstreamProxy
	c.Host = config.Host
	c.Port = config.Port
	c.Theme = config.Theme
	c.Locale = config.Locale
	c.Quality = config.Quality
	c.SaveDirectory = config.SaveDirectory
	c.FilenameLen = config.FilenameLen
	c.FilenameTime = config.FilenameTime
	c.UpstreamProxy = config.UpstreamProxy
	c.UserAgent = config.UserAgent
	c.OpenProxy = config.OpenProxy
	c.DownloadProxy = config.DownloadProxy
	c.AutoProxy = config.AutoProxy
	c.TaskNumber = config.TaskNumber
	c.WxAction = config.WxAction
	c.UseHeaders = config.UseHeaders
	if oldProxy != c.UpstreamProxy {
		proxyOnce.setTransport()
	}

	mimeMux.Lock()
	c.MimeMap = config.MimeMap
	mimeMux.Unlock()

	jsonData, err := json.Marshal(c)
	if err == nil {
		_ = globalConfig.storage.Store(jsonData)
	}
}