123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231 |
- // 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"
- "strings"
- "github.com/88250/gulu"
- "github.com/88250/lute/html"
- "github.com/gin-gonic/gin"
- "github.com/siyuan-note/siyuan/kernel/model"
- "github.com/siyuan-note/siyuan/kernel/util"
- )
- func findReplace(c *gin.Context) {
- ret := gulu.Ret.NewResult()
- defer c.JSON(http.StatusOK, ret)
- arg, ok := util.JsonArg(c, ret)
- if !ok {
- return
- }
- k := arg["k"].(string)
- r := arg["r"].(string)
- methodArg := arg["method"]
- var method int // 0:文本,1:查询语法,2:SQL,3:正则表达式
- if nil != methodArg {
- method = int(methodArg.(float64))
- }
- idsArg := arg["ids"].([]interface{})
- var ids []string
- for _, id := range idsArg {
- ids = append(ids, id.(string))
- }
- err := model.FindReplace(k, r, ids, method)
- if nil != err {
- ret.Code = -1
- ret.Msg = err.Error()
- ret.Data = map[string]interface{}{"closeTimeout": 5000}
- return
- }
- return
- }
- func searchAsset(c *gin.Context) {
- ret := gulu.Ret.NewResult()
- defer c.JSON(http.StatusOK, ret)
- arg, ok := util.JsonArg(c, ret)
- if !ok {
- return
- }
- k := arg["k"].(string)
- ret.Data = model.SearchAssetsByName(k)
- return
- }
- func searchTag(c *gin.Context) {
- ret := gulu.Ret.NewResult()
- defer c.JSON(http.StatusOK, ret)
- arg, ok := util.JsonArg(c, ret)
- if !ok {
- return
- }
- k := arg["k"].(string)
- tags := model.SearchTags(k)
- ret.Data = map[string]interface{}{
- "tags": tags,
- "k": k,
- }
- }
- func searchWidget(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)
- blocks := model.SearchWidget(keyword)
- ret.Data = map[string]interface{}{
- "blocks": blocks,
- "k": keyword,
- }
- }
- func searchTemplate(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)
- blocks := model.SearchTemplate(keyword)
- ret.Data = map[string]interface{}{
- "blocks": blocks,
- "k": keyword,
- }
- }
- func searchEmbedBlock(c *gin.Context) {
- ret := gulu.Ret.NewResult()
- defer c.JSON(http.StatusOK, ret)
- arg, ok := util.JsonArg(c, ret)
- if !ok {
- return
- }
- embedBlockID := arg["embedBlockID"].(string)
- stmt := arg["stmt"].(string)
- excludeIDsArg := arg["excludeIDs"].([]interface{})
- var excludeIDs []string
- for _, excludeID := range excludeIDsArg {
- excludeIDs = append(excludeIDs, excludeID.(string))
- }
- headingMode := 0 // 0:带标题下方块
- headingModeArg := arg["headingMode"]
- if nil != headingModeArg {
- headingMode = int(headingModeArg.(float64))
- }
- breadcrumb := false
- breadcrumbArg := arg["breadcrumb"]
- if nil != breadcrumbArg {
- breadcrumb = breadcrumbArg.(bool)
- }
- blocks := model.SearchEmbedBlock(embedBlockID, stmt, excludeIDs, headingMode, breadcrumb)
- ret.Data = map[string]interface{}{
- "blocks": blocks,
- }
- }
- func searchRefBlock(c *gin.Context) {
- ret := gulu.Ret.NewResult()
- defer c.JSON(http.StatusOK, ret)
- arg, ok := util.JsonArg(c, ret)
- if !ok {
- return
- }
- reqId := arg["reqId"]
- ret.Data = map[string]interface{}{"reqId": reqId}
- if nil == arg["id"] {
- return
- }
- rootID := arg["rootID"].(string)
- id := arg["id"].(string)
- keyword := arg["k"].(string)
- beforeLen := int(arg["beforeLen"].(float64))
- blocks, newDoc := model.SearchRefBlock(id, rootID, keyword, beforeLen)
- ret.Data = map[string]interface{}{
- "blocks": blocks,
- "newDoc": newDoc,
- "k": html.EscapeHTMLStr(keyword),
- "reqId": arg["reqId"],
- }
- }
- func fullTextSearchBlock(c *gin.Context) {
- ret := gulu.Ret.NewResult()
- defer c.JSON(http.StatusOK, ret)
- arg, ok := util.JsonArg(c, ret)
- if !ok {
- return
- }
- query := arg["query"].(string)
- pathArg := arg["path"]
- var path string
- if nil != pathArg {
- path = pathArg.(string)
- }
- var box string
- if "" != path {
- box = strings.Split(path, "/")[0]
- path = strings.TrimPrefix(path, box)
- }
- var types map[string]bool
- if nil != arg["types"] {
- typesArg := arg["types"].(map[string]interface{})
- types = map[string]bool{}
- for t, b := range typesArg {
- types[t] = b.(bool)
- }
- }
- methodArg := arg["method"]
- var method int // 0:文本,1:查询语法,2:SQL,3:正则表达式
- if nil != methodArg {
- method = int(methodArg.(float64))
- }
- groupByArg := arg["groupBy"]
- var groupBy int // 0:不分组,1:按文档分组
- if nil != groupByArg {
- groupBy = int(groupByArg.(float64))
- }
- blocks, matchedBlockCount, matchedRootCount := model.FullTextSearchBlock(query, box, path, types, method, groupBy)
- ret.Data = map[string]interface{}{
- "blocks": blocks,
- "matchedBlockCount": matchedBlockCount,
- "matchedRootCount": matchedRootCount,
- }
- }
|