❤️ 完整开源界面和内核 https://github.com/siyuan-note/siyuan/issues/5013

kernel/api/graph.go

@@ -0,0 +1,131 @@
+// SiYuan - Build Your Eternal Digital Garden
+// Copyright (c) 2020-present, b3log.org
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+package api
+
+import (
+	"net/http"
+
+	"github.com/88250/gulu"
+	"github.com/gin-gonic/gin"
+	"github.com/siyuan-note/siyuan/kernel/conf"
+	"github.com/siyuan-note/siyuan/kernel/model"
+	"github.com/siyuan-note/siyuan/kernel/util"
+)
+
+func resetGraph(c *gin.Context) {
+	ret := gulu.Ret.NewResult()
+	defer c.JSON(http.StatusOK, ret)
+
+	graph := conf.NewGlobalGraph()
+	model.Conf.Graph.Global = graph
+	model.Conf.Save()
+	ret.Data = map[string]interface{}{
+		"conf": graph,
+	}
+}
+
+func resetLocalGraph(c *gin.Context) {
+	ret := gulu.Ret.NewResult()
+	defer c.JSON(http.StatusOK, ret)
+
+	graph := conf.NewLocalGraph()
+	model.Conf.Graph.Local = graph
+	model.Conf.Save()
+	ret.Data = map[string]interface{}{
+		"conf": graph,
+	}
+}
+
+func getGraph(c *gin.Context) {
+	ret := gulu.Ret.NewResult()
+	defer c.JSON(http.StatusOK, ret)
+
+	arg, ok := util.JsonArg(c, ret)
+	if !ok {
+		return
+	}
+
+	query := arg["k"].(string)
+
+	graphConf, err := gulu.JSON.MarshalJSON(arg["conf"])
+	if nil != err {
+		ret.Code = -1
+		ret.Msg = err.Error()
+		return
+	}
+
+	global := conf.NewGlobalGraph()
+	if err = gulu.JSON.UnmarshalJSON(graphConf, global); nil != err {
+		ret.Code = -1
+		ret.Msg = err.Error()
+		return
+	}
+
+	model.Conf.Graph.Global = global
+	model.Conf.Save()
+
+	boxID, nodes, links := model.BuildGraph(query)
+	ret.Data = map[string]interface{}{
+		"nodes": nodes,
+		"links": links,
+		"conf":  global,
+		"box":   boxID,
+		"reqId": arg["reqId"],
+	}
+	util.RandomSleep(200, 500)
+}
+
+func getLocalGraph(c *gin.Context) {
+	ret := gulu.Ret.NewResult()
+	defer c.JSON(http.StatusOK, ret)
+
+	arg, ok := util.JsonArg(c, ret)
+	if !ok {
+		return
+	}
+
+	keyword := arg["k"].(string)
+	id := arg["id"].(string)
+
+	graphConf, err := gulu.JSON.MarshalJSON(arg["conf"])
+	if nil != err {
+		ret.Code = -1
+		ret.Msg = err.Error()
+		return
+	}
+
+	local := conf.NewLocalGraph()
+	if err = gulu.JSON.UnmarshalJSON(graphConf, local); nil != err {
+		ret.Code = -1
+		ret.Msg = err.Error()
+		return
+	}
+
+	model.Conf.Graph.Local = local
+	model.Conf.Save()
+
+	boxID, nodes, links := model.BuildTreeGraph(id, keyword)
+	ret.Data = map[string]interface{}{
+		"id":    id,
+		"box":   boxID,
+		"nodes": nodes,
+		"links": links,
+		"conf":  local,
+		"reqId": arg["reqId"],
+	}
+	util.RandomSleep(200, 500)
+}