feat: 备注折叠与导入自动测速

Author: cheluen

backend/api/handlers.go

@@ -466,6 +466,40 @@ func (h *Handler) UpdateNode(c *gin.Context) {
 	c.JSON(http.StatusOK, req)
 }
 
+func (h *Handler) UpdateNodeRemark(c *gin.Context) {
+	id, err := strconv.Atoi(c.Param("id"))
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid id"})
+		return
+	}
+
+	var req struct {
+		Remark string `json:"remark"`
+	}
+	if err := c.BindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid request"})
+		return
+	}
+
+	res, err := h.db.Exec(`
+		UPDATE proxy_nodes
+		SET remark = ?, updated_at = CURRENT_TIMESTAMP
+		WHERE id = ?
+	`, req.Remark, id)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "failed to update remark"})
+		return
+	}
+
+	affected, _ := res.RowsAffected()
+	if affected == 0 {
+		c.JSON(http.StatusNotFound, gin.H{"error": "node not found"})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"id": id, "remark": req.Remark})
+}
+
 // DeleteNode deletes a proxy node
 func (h *Handler) DeleteNode(c *gin.Context) {
 	id, err := strconv.Atoi(c.Param("id"))

backend/api/handlers_test.go

@@ -7,6 +7,7 @@ import (
 	"net/http"
 	"net/http/httptest"
 	"strconv"
+	"strings"
 	"testing"
 
 	"sb-proxy/backend/models"
@@ -171,3 +172,39 @@ func TestCheckNodeIPRejectsSocksFallback(t *testing.T) {
 		t.Fatalf("expected cleared status after fallback, got ip=%s location=%s country=%s latency=%d", ip, location, countryCode, latency)
 	}
 }
+
+func TestUpdateNodeRemarkUpdatesRemark(t *testing.T) {
+	gin.SetMode(gin.TestMode)
+
+	handler := newTestHandler(t, func(proxyAddr, username, password string) (*services.IPInfo, error) {
+		return nil, fmt.Errorf("not used")
+	})
+
+	nodeID := insertTestNode(t, handler.db)
+
+	rec := httptest.NewRecorder()
+	ctx, _ := gin.CreateTestContext(rec)
+	ctx.Params = gin.Params{gin.Param{Key: "id", Value: strconv.Itoa(nodeID)}}
+
+	req, _ := http.NewRequest(
+		http.MethodPut,
+		"/api/nodes/"+strconv.Itoa(nodeID)+"/remark",
+		strings.NewReader(`{"remark":"hello"}`),
+	)
+	req.Header.Set("Content-Type", "application/json")
+	ctx.Request = req
+
+	handler.UpdateNodeRemark(ctx)
+
+	if rec.Code != http.StatusOK {
+		t.Fatalf("unexpected status %d", rec.Code)
+	}
+
+	var remark string
+	if err := handler.db.QueryRow("SELECT remark FROM proxy_nodes WHERE id = ?", nodeID).Scan(&remark); err != nil {
+		t.Fatalf("query node: %v", err)
+	}
+	if remark != "hello" {
+		t.Fatalf("expected remark to be updated, got %q", remark)
+	}
+}

backend/main.go

@@ -174,6 +174,7 @@ func main() {
 		authorized.POST("/nodes/batch-delete", handler.BatchDeleteNodes)
 		authorized.POST("/nodes/batch-export", handler.BatchExportNodes)
 		authorized.PUT("/nodes/:id", handler.UpdateNode)
+		authorized.PUT("/nodes/:id/remark", handler.UpdateNodeRemark)
 		authorized.PUT("/nodes/:id/replace", handler.ReplaceNode)
 		authorized.DELETE("/nodes/:id", handler.DeleteNode)
 		authorized.GET("/nodes/:id/export", handler.ExportNode)