mirror of
https://github.com/wgh136/nysoure.git
synced 2025-09-27 20:27:23 +00:00
81 lines
2.2 KiB
Go
81 lines
2.2 KiB
Go
package config
|
|
|
|
import (
|
|
"encoding/json"
|
|
"nysoure/server/utils"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
var config *ServerConfig
|
|
|
|
type ServerConfig struct {
|
|
// MaxUploadingSizeInMB is the maximum size of files that are being uploaded at the same time.
|
|
MaxUploadingSizeInMB int `json:"max_uploading_size_in_mb"`
|
|
// MaxFileSizeInMB is the maximum size of a single file that can be uploaded.
|
|
MaxFileSizeInMB int `json:"max_file_size_in_mb"`
|
|
// MaxDownloadsForSingleIP is the maximum number of downloads allowed from a single IP address.
|
|
MaxDownloadsPerDayForSingleIP int `json:"max_downloads_per_day_for_single_ip"`
|
|
// AllowRegister indicates whether user registration is allowed.
|
|
AllowRegister bool `json:"allow_register"`
|
|
// CloudflareTurnstileSiteKey is the site key for Cloudflare Turnstile.
|
|
CloudflareTurnstileSiteKey string `json:"cloudflare_turnstile_site_key"`
|
|
// CloudflareTurnstileSecretKey is the secret key for Cloudflare Turnstile.
|
|
CloudflareTurnstileSecretKey string `json:"cloudflare_turnstile_secret_key"`
|
|
}
|
|
|
|
func init() {
|
|
filepath := filepath.Join(utils.GetStoragePath(), "config.json")
|
|
if _, err := os.Stat(filepath); os.IsNotExist(err) {
|
|
config = &ServerConfig{
|
|
MaxUploadingSizeInMB: 20 * 1024, // 20GB
|
|
MaxFileSizeInMB: 8 * 1024, // 8GB
|
|
MaxDownloadsPerDayForSingleIP: 20,
|
|
AllowRegister: true,
|
|
CloudflareTurnstileSiteKey: "",
|
|
CloudflareTurnstileSecretKey: "",
|
|
}
|
|
} else {
|
|
data, err := os.ReadFile(filepath)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
config = &ServerConfig{}
|
|
if err := json.Unmarshal(data, config); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func GetConfig() ServerConfig {
|
|
return *config
|
|
}
|
|
|
|
func SetConfig(newConfig ServerConfig) {
|
|
config = &newConfig
|
|
data, err := json.MarshalIndent(config, "", " ")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
filepath := filepath.Join(utils.GetStoragePath(), "config.json")
|
|
if err := os.WriteFile(filepath, data, 0644); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func MaxUploadingSize() int64 {
|
|
return int64(config.MaxUploadingSizeInMB) * 1024 * 1024
|
|
}
|
|
|
|
func MaxFileSize() int64 {
|
|
return int64(config.MaxFileSizeInMB) * 1024 * 1024
|
|
}
|
|
|
|
func AllowRegister() bool {
|
|
return config.AllowRegister
|
|
}
|
|
|
|
func MaxDownloadsPerDayForSingleIP() int {
|
|
return config.MaxDownloadsPerDayForSingleIP
|
|
}
|