feat: add platform configuration tables and admin settings

Add four DB-backed configuration tables for site-wide settings: site_settings (logo, top-left title, header/footer text with zh/en variants), upload_configs (master switch, default size, storage dir), upload_file_types (per-extension whitelist with per-type size limits), and download_baseurls (multiple download sources with priority/default). - Models, AutoMigrate, and first-run seed (18 common file types) - Process-level config cache warmed at startup, refreshed on admin save - SetUserContext injects cached site settings into templates - base.html renders logo (local upload or external URL), title, header banner, and footer with i18n fallback - Upload validator drives avatar uploads (form + AJAX) through the configured switch/type/size policy - Admin pages for site/upload/download settings with i18n keys Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
2026-06-21 20:32:28 +08:00
parent b474ee009a
commit e3367e6e12
17 changed files with 1277 additions and 20 deletions
@@ -0,0 +1,329 @@
+package handlers
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"path/filepath"
+	"strconv"
+	"strings"
+
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+
+	"go_blog/models"
+)
+
+// mbToBytes converts a megabyte count (string) to bytes. Returns 0 on parse
+// failure. Values <= 0 are treated as 0 (meaning "use default" for per-type
+// limits).
+func mbToBytes(s string) int64 {
+	s = strings.TrimSpace(s)
+	if s == "" {
+		return 0
+	}
+	n, err := strconv.ParseFloat(s, 64)
+	if err != nil {
+		return 0
+	}
+	return int64(n * 1024 * 1024)
+}
+
+// bytesToMB renders a byte count as megabytes (one decimal) for form display.
+func bytesToMB(b int64) string {
+	return fmt.Sprintf("%.1f", float64(b)/float64(1024*1024))
+}
+
+// userIDFromSession extracts the logged-in user's ID, or 0 if absent.
+func userIDFromSession(c *gin.Context) uint {
+	session := sessions.Default(c)
+	if id, ok := session.Get("user_id").(uint); ok {
+		return id
+	}
+	return 0
+}
+
+// ---------------- Site settings ----------------
+
+// SiteSettingsPage renders the site display settings form.
+func SiteSettingsPage(db *gorm.DB) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		tr := getTr(c)
+		var s models.SiteSetting
+		if err := db.First(&s, 1).Error; err != nil {
+			s = models.SiteSetting{ID: 1}
+		}
+
+		data := DefaultData(c)
+		data["Title"] = tr["settings_site_title"]
+		data["Site"] = s
+		if msg := c.Query("saved"); msg == "1" {
+			data["Success"] = tr["settings_saved"]
+		}
+		c.HTML(http.StatusOK, "settings_site", data)
+	}
+}
+
+// SiteSettingsSave handles logo upload and text fields for site settings.
+func SiteSettingsSave(db *gorm.DB, storagePath string) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		var s models.SiteSetting
+		if err := db.First(&s, 1).Error; err != nil {
+			s = models.SiteSetting{ID: 1}
+		}
+
+		s.LogoTextZh = strings.TrimSpace(c.PostForm("logo_text_zh"))
+		s.LogoTextEn = strings.TrimSpace(c.PostForm("logo_text_en"))
+		s.HeaderTextZh = strings.TrimSpace(c.PostForm("header_text_zh"))
+		s.HeaderTextEn = strings.TrimSpace(c.PostForm("header_text_en"))
+		s.FooterTextZh = strings.TrimSpace(c.PostForm("footer_text_zh"))
+		s.FooterTextEn = strings.TrimSpace(c.PostForm("footer_text_en"))
+		s.UpdatedBy = userIDFromSession(c)
+
+		// Logo upload (optional). A logo_url form field takes precedence over an
+		// uploaded file, so admins can set either a local file or an external link.
+		if logoURL := strings.TrimSpace(c.PostForm("logo_url")); logoURL != "" {
+			s.Logo = logoURL
+		} else if file, header, err := c.Request.FormFile("logo"); err == nil {
+			defer file.Close()
+			check := ValidateUpload(header)
+			if !check.OK || check.Type.Category != models.CategoryImage {
+				c.Redirect(http.StatusFound, "/admin/settings/site")
+				return
+			}
+			logoDir := filepath.Join(storagePath, "logos")
+			os.MkdirAll(logoDir, 0755)
+			// Remove the previous local logo (skip external URLs).
+			if s.Logo != "" && !s.LogoIsURL() {
+				os.Remove(filepath.Join(logoDir, s.Logo))
+			}
+			ext := strings.ToLower(filepath.Ext(header.Filename))
+			savedName := fmt.Sprintf("logo%s", ext)
+			dst, err := os.Create(filepath.Join(logoDir, savedName))
+			if err != nil {
+				c.Redirect(http.StatusFound, "/admin/settings/site")
+				return
+			}
+			defer dst.Close()
+			if _, err := io.Copy(dst, file); err != nil {
+				c.Redirect(http.StatusFound, "/admin/settings/site")
+				return
+			}
+			s.Logo = savedName
+		}
+
+		// Remove logo entirely if requested.
+		if c.PostForm("logo_clear") == "1" {
+			if s.Logo != "" && !s.LogoIsURL() {
+				os.Remove(filepath.Join(storagePath, "logos", s.Logo))
+			}
+			s.Logo = ""
+		}
+
+		if err := db.Save(&s).Error; err != nil {
+			c.Redirect(http.StatusFound, "/admin/settings/site")
+			return
+		}
+		models.RefreshConfigCache(db)
+		c.Redirect(http.StatusFound, "/admin/settings/site?saved=1")
+	}
+}
+
+// ---------------- Upload settings ----------------
+
+// fileTypeView augments an UploadFileType with a pre-rendered max-size MB
+// string for the template (avoids needing a template FuncMap for division).
+type fileTypeView struct {
+	models.UploadFileType
+	MaxSizeMB string
+}
+
+// UploadSettingsPage renders the upload policy + file-type management page.
+func UploadSettingsPage(db *gorm.DB) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		tr := getTr(c)
+		var u models.UploadConfig
+		if err := db.First(&u, 1).Error; err != nil {
+			u = models.UploadConfig{ID: 1, Enabled: true, DefaultMaxSize: models.DefaultUploadMaxSize, StorageDir: "attachments"}
+		}
+		var types []models.UploadFileType
+		db.Order("category asc, sort asc, id asc").Find(&types)
+
+		views := make([]fileTypeView, 0, len(types))
+		for _, t := range types {
+			views = append(views, fileTypeView{UploadFileType: t, MaxSizeMB: bytesToMB(t.MaxSize)})
+		}
+
+		data := DefaultData(c)
+		data["Title"] = tr["settings_upload_title"]
+		data["Upload"] = u
+		data["FileTypes"] = views
+		data["DefaultMaxSizeMB"] = bytesToMB(u.DefaultMaxSize)
+		if msg := c.Query("saved"); msg == "1" {
+			data["Success"] = tr["settings_saved"]
+		}
+		c.HTML(http.StatusOK, "settings_upload", data)
+	}
+}
+
+// UploadSettingsSave dispatches upload-config and file-type actions.
+func UploadSettingsSave(db *gorm.DB) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		switch c.PostForm("action") {
+		case "save_config":
+			saveUploadConfig(db, c)
+		case "add_type":
+			addUploadFileType(db, c)
+		case "toggle_type":
+			toggleUploadFileType(db, c)
+		case "size_type":
+			sizeUploadFileType(db, c)
+		case "delete_type":
+			deleteUploadFileType(db, c)
+		}
+		models.RefreshConfigCache(db)
+		c.Redirect(http.StatusFound, "/admin/settings/upload?saved=1")
+	}
+}
+
+func saveUploadConfig(db *gorm.DB, c *gin.Context) {
+	var u models.UploadConfig
+	if err := db.First(&u, 1).Error; err != nil {
+		u = models.UploadConfig{ID: 1}
+	}
+	u.Enabled = c.PostForm("enabled") == "1"
+	u.DefaultMaxSize = mbToBytes(c.PostForm("default_max_size"))
+	if u.DefaultMaxSize <= 0 {
+		u.DefaultMaxSize = models.DefaultUploadMaxSize
+	}
+	if dir := strings.TrimSpace(c.PostForm("storage_dir")); dir != "" {
+		u.StorageDir = dir
+	}
+	u.UpdatedBy = userIDFromSession(c)
+	db.Save(&u)
+}
+
+func addUploadFileType(db *gorm.DB, c *gin.Context) {
+	ext := strings.ToLower(strings.TrimSpace(c.PostForm("extension")))
+	if ext == "" {
+		return
+	}
+	if !strings.HasPrefix(ext, ".") {
+		ext = "." + ext
+	}
+	t := models.UploadFileType{
+		Extension: ext,
+		MimeType:  strings.TrimSpace(c.PostForm("mime_type")),
+		Category:  strings.TrimSpace(c.PostForm("category")),
+		MaxSize:   mbToBytes(c.PostForm("max_size")),
+		Enabled:   c.PostForm("enabled") != "0",
+		Sort:      50,
+	}
+	if t.Category == "" {
+		t.Category = models.CategoryOther
+	}
+	// Ignore duplicate-extension errors silently.
+	db.Where("extension = ?", t.Extension).FirstOrCreate(&t)
+}
+
+func toggleUploadFileType(db *gorm.DB, c *gin.Context) {
+	id, _ := strconv.Atoi(c.PostForm("id"))
+	var t models.UploadFileType
+	if db.First(&t, id).Error != nil {
+		return
+	}
+	t.Enabled = !t.Enabled
+	db.Save(&t)
+}
+
+func sizeUploadFileType(db *gorm.DB, c *gin.Context) {
+	id, _ := strconv.Atoi(c.PostForm("id"))
+	var t models.UploadFileType
+	if db.First(&t, id).Error != nil {
+		return
+	}
+	t.MaxSize = mbToBytes(c.PostForm("max_size"))
+	db.Save(&t)
+}
+
+func deleteUploadFileType(db *gorm.DB, c *gin.Context) {
+	id, _ := strconv.Atoi(c.PostForm("id"))
+	db.Delete(&models.UploadFileType{}, id)
+}
+
+// ---------------- Download settings ----------------
+
+// DownloadSettingsPage renders the download base-URL management page.
+func DownloadSettingsPage(db *gorm.DB) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		tr := getTr(c)
+		var urls []models.DownloadBaseURL
+		db.Order("is_default desc, priority asc, id asc").Find(&urls)
+
+		data := DefaultData(c)
+		data["Title"] = tr["settings_download_title"]
+		data["BaseURLs"] = urls
+		if msg := c.Query("saved"); msg == "1" {
+			data["Success"] = tr["settings_saved"]
+		}
+		c.HTML(http.StatusOK, "settings_download", data)
+	}
+}
+
+// DownloadSettingsSave dispatches download base-URL actions.
+func DownloadSettingsSave(db *gorm.DB) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		switch c.PostForm("action") {
+		case "add":
+			addDownloadBaseURL(db, c)
+		case "toggle":
+			toggleDownloadBaseURL(db, c)
+		case "default":
+			defaultDownloadBaseURL(db, c)
+		case "delete":
+			deleteDownloadBaseURL(db, c)
+		}
+		models.RefreshConfigCache(db)
+		c.Redirect(http.StatusFound, "/admin/settings/download?saved=1")
+	}
+}
+
+func addDownloadBaseURL(db *gorm.DB, c *gin.Context) {
+	name := strings.TrimSpace(c.PostForm("name"))
+	base := strings.TrimSpace(c.PostForm("base_url"))
+	if base == "" {
+		return
+	}
+	prio, _ := strconv.Atoi(c.PostForm("priority"))
+	b := models.DownloadBaseURL{
+		Name:     name,
+		BaseURL:  base,
+		Priority: prio,
+		Enabled:  c.PostForm("enabled") != "0",
+	}
+	db.Create(&b)
+}
+
+func toggleDownloadBaseURL(db *gorm.DB, c *gin.Context) {
+	id, _ := strconv.Atoi(c.PostForm("id"))
+	var b models.DownloadBaseURL
+	if db.First(&b, id).Error != nil {
+		return
+	}
+	b.Enabled = !b.Enabled
+	db.Save(&b)
+}
+
+func defaultDownloadBaseURL(db *gorm.DB, c *gin.Context) {
+	id, _ := strconv.Atoi(c.PostForm("id"))
+	// Only one default at a time.
+	db.Model(&models.DownloadBaseURL{}).Where("1=1").Update("is_default", false)
+	db.Model(&models.DownloadBaseURL{}).Where("id = ?", id).Update("is_default", true)
+}
+
+func deleteDownloadBaseURL(db *gorm.DB, c *gin.Context) {
+	id, _ := strconv.Atoi(c.PostForm("id"))
+	db.Delete(&models.DownloadBaseURL{}, id)
+}