feat(scripts): 添加脚本和模板使用情况检查功能防止误删

Author: ZeroDeng01

api/script.go

@@ -2,6 +2,7 @@ package api
 
 import (
 	"strconv"
+	"sublink/database"
 	"sublink/models"
 	"sublink/utils"
 
@@ -49,6 +50,55 @@ func ScriptDel(c *gin.Context) {
 	utils.OkWithMsg(c, "删除成功")
 }
 
+func GetScriptUsage(c *gin.Context) {
+	scriptIDStr := c.Query("id")
+	if scriptIDStr == "" {
+		utils.FailWithMsg(c, "脚本ID不能为空")
+		return
+	}
+
+	scriptID, err := strconv.Atoi(scriptIDStr)
+	if err != nil || scriptID <= 0 {
+		utils.FailWithMsg(c, "脚本ID非法")
+		return
+	}
+
+	var subScripts []models.SubcriptionScript
+	if err := database.DB.Where("script_id = ?", scriptID).Find(&subScripts).Error; err != nil {
+		utils.FailWithMsg(c, "获取脚本使用情况失败")
+		return
+	}
+
+	if len(subScripts) == 0 {
+		utils.OkWithData(c, gin.H{
+			"subscriptions": []string{},
+			"count":         0,
+		})
+		return
+	}
+
+	subIDs := make([]int, 0, len(subScripts))
+	for _, subScript := range subScripts {
+		subIDs = append(subIDs, subScript.SubcriptionID)
+	}
+
+	var subs []models.Subcription
+	if err := database.DB.Where("id IN ?", subIDs).Find(&subs).Error; err != nil {
+		utils.FailWithMsg(c, "获取脚本使用情况失败")
+		return
+	}
+
+	usedBy := make([]string, 0, len(subs))
+	for _, sub := range subs {
+		usedBy = append(usedBy, sub.Name)
+	}
+
+	utils.OkWithData(c, gin.H{
+		"subscriptions": usedBy,
+		"count":         len(usedBy),
+	})
+}
+
 // ScriptUpdate 更新脚本
 func ScriptUpdate(c *gin.Context) {
 	var data models.Script

api/template.go

@@ -1,12 +1,14 @@
 package api
 
 import (
+	"encoding/json"
 	"errors"
 	"os"
 	"path/filepath"
 	"strconv"
 	"strings"
 	"sublink/cache"
+	"sublink/database"
 	"sublink/models"
 	"sublink/utils"
 
@@ -459,6 +461,79 @@ func DelTemp(c *gin.Context) {
 	utils.OkWithMsg(c, "删除成功")
 }
 
+func normalizeTemplateUsageValue(value string) string {
+	return strings.ReplaceAll(strings.TrimSpace(value), "\\", "/")
+}
+
+func buildTemplateMatchValues(filename string) map[string]struct{} {
+	normalizedFile := normalizeTemplateUsageValue(filename)
+	fileName := filepath.Base(normalizedFile)
+	values := []string{normalizedFile, fileName, "./template/" + fileName, "/template/" + fileName}
+	matchValues := make(map[string]struct{}, len(values))
+	for _, value := range values {
+		value = normalizeTemplateUsageValue(value)
+		if value == "" {
+			continue
+		}
+		matchValues[value] = struct{}{}
+	}
+	return matchValues
+}
+
+func getTemplateUsageSubscriptions(filename string) ([]string, error) {
+	var subs []models.Subcription
+	if err := database.DB.Find(&subs).Error; err != nil {
+		return nil, err
+	}
+
+	matchValues := buildTemplateMatchValues(filename)
+	usedBy := make([]string, 0)
+
+	for _, sub := range subs {
+		var config struct {
+			Clash string `json:"clash"`
+			Surge string `json:"surge"`
+		}
+
+		if sub.Config != "" {
+			if err := json.Unmarshal([]byte(sub.Config), &config); err != nil {
+				continue
+			}
+		}
+
+		clashValue := normalizeTemplateUsageValue(config.Clash)
+		surgeValue := normalizeTemplateUsageValue(config.Surge)
+		if _, ok := matchValues[clashValue]; ok {
+			usedBy = append(usedBy, sub.Name)
+			continue
+		}
+		if _, ok := matchValues[surgeValue]; ok {
+			usedBy = append(usedBy, sub.Name)
+		}
+	}
+
+	return usedBy, nil
+}
+
+func GetTemplateUsage(c *gin.Context) {
+	filename := c.Query("filename")
+	if filename == "" {
+		utils.FailWithMsg(c, "文件名不能为空")
+		return
+	}
+
+	usedBy, err := getTemplateUsageSubscriptions(filename)
+	if err != nil {
+		utils.FailWithMsg(c, "获取模板使用情况失败")
+		return
+	}
+
+	utils.OkWithData(c, gin.H{
+		"subscriptions": usedBy,
+		"count":         len(usedBy),
+	})
+}
+
 // ACL4SSRPreset ACL4SSR 规则预设
 type ACL4SSRPreset struct {
 	Name  string `json:"name"`

routers/script.go

@@ -15,6 +15,7 @@ func Script(r *gin.Engine) {
 		ScriptGroup.POST("/add", middlewares.DemoModeRestrict, api.ScriptAdd)
 		ScriptGroup.DELETE("/delete", middlewares.DemoModeRestrict, api.ScriptDel)
 		ScriptGroup.POST("/update", middlewares.DemoModeRestrict, api.ScriptUpdate)
+		ScriptGroup.GET("/usage", api.GetScriptUsage)
 		ScriptGroup.GET("/list", api.ScriptList)
 	}
 }

routers/template.go

@@ -13,6 +13,7 @@ func Templates(r *gin.Engine) {
 	{
 		TempsGroup.POST("/add", api.AddTemp)
 		TempsGroup.POST("/delete", api.DelTemp)
+		TempsGroup.GET("/usage", api.GetTemplateUsage)
 		TempsGroup.GET("/get", api.GetTempS)
 		TempsGroup.POST("/update", api.UpdateTemp)
 		TempsGroup.GET("/presets", api.GetACL4SSRPresets)