位置变更，Preview正在重构

internal/api/handler.go

@@ -1,6 +1,9 @@
 package api
 
 import (
+	"bytes"
+	"fmt"
+	"html/template"
 	"net"
 	"net/http"
 	"strconv"
@@ -415,3 +418,125 @@ func readTailLog(filePath string, n int) ([]string, error) {
 	// 为了编译通过，先简单返回空
 	return []string{}, nil
 }
+
+// ================================================================
+// 导入/导出 TOML
+// ================================================================
+
+func ImportTomlHandler(c *gin.Context) {
+	file, err := c.FormFile("file")
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"code": 1, "msg": "请选择文件"})
+		return
+	}
+
+	f, err := file.Open()
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "读取文件失败"})
+		return
+	}
+	defer f.Close()
+
+	buf := new(bytes.Buffer)
+	if _, err := buf.ReadFrom(f); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "读取文件失败"})
+		return
+	}
+
+	parsed, err := frp.ParseToml(buf.String())
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"code": 1, "msg": "解析 TOML 失败: " + err.Error()})
+		return
+	}
+
+	cfg := parsed.ToGlobalConfig()
+	cfg.TcpMux = true
+	if err := db.UpdateGlobalConfig(cfg); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "更新配置失败: " + err.Error()})
+		return
+	}
+
+	if _, err := db.DB.Exec("DELETE FROM proxies"); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "清空隧道失败"})
+		return
+	}
+
+	proxies := parsed.ToProxies()
+	for _, p := range proxies {
+		if err := db.CreateProxy(&p); err != nil {
+			c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "导入隧道失败: " + err.Error()})
+			return
+		}
+	}
+
+	if err := frp.GenerateConfig(); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "生成配置失败: " + err.Error()})
+		return
+	}
+
+	if err := frp.Reload(); err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"code": 0,
+			"msg":  fmt.Sprintf("导入成功！共 %d 条隧道，但热加载失败: %s", len(proxies), err.Error()),
+		})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"code": 0,
+		"msg":  fmt.Sprintf("导入成功！共 %d 条隧道", len(proxies)),
+	})
+}
+
+func ExportTomlHandler(c *gin.Context) {
+	cfg, err := db.GetGlobalConfig()
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "读取配置失败: " + err.Error()})
+		return
+	}
+
+	proxies, err := db.GetProxies()
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "读取隧道失败: " + err.Error()})
+		return
+	}
+
+	var activeProxies []db.Proxy
+	for _, p := range proxies {
+		if p.Enabled {
+			activeProxies = append(activeProxies, p)
+		}
+	}
+
+	data := struct {
+		*db.GlobalConfig
+		Proxies          []db.Proxy
+		WireProtocolLine string
+	}{
+		GlobalConfig: cfg,
+		Proxies:      activeProxies,
+	}
+
+	if cfg.WireProtocolV2 {
+		data.WireProtocolLine = `wireProtocol = "v2"`
+	} else {
+		data.WireProtocolLine = ""
+	}
+
+	// 这里需要 frp.FrpcTemplateContent，需要从 frp 包导出
+	tmpl, err := template.New("frpc").Parse(frp.FrpcTemplateContent)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "解析模板失败: " + err.Error()})
+		return
+	}
+
+	var buf bytes.Buffer
+	if err := tmpl.Execute(&buf, data); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"code": 2, "msg": "渲染模板失败: " + err.Error()})
+		return
+	}
+
+	c.Header("Content-Type", "text/plain; charset=utf-8")
+	c.Header("Content-Disposition", "attachment; filename=frpc.toml")
+	c.String(http.StatusOK, buf.String())
+}