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

kernel/sql/aseet.go

@@ -0,0 +1,169 @@
+// 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 sql
+
+import (
+	"crypto/sha256"
+	"database/sql"
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/88250/gulu"
+	"github.com/88250/lute/ast"
+	"github.com/siyuan-note/siyuan/kernel/treenode"
+	"github.com/siyuan-note/siyuan/kernel/util"
+)
+
+type Asset struct {
+	ID      string
+	BlockID string
+	RootID  string
+	Box     string
+	DocPath string
+	Path    string
+	Name    string
+	Title   string
+	Hash    string
+}
+
+func docTagSpans(n *ast.Node) (ret []*Span) {
+	if tagsVal := n.IALAttr("tags"); "" != tagsVal {
+		tags := strings.Split(tagsVal, ",")
+		for _, tag := range tags {
+			markdown := "#" + tag + "#"
+			span := &Span{
+				ID:       ast.NewNodeID(),
+				BlockID:  n.ID,
+				RootID:   n.ID,
+				Box:      n.Box,
+				Path:     n.Path,
+				Content:  tag,
+				Markdown: markdown,
+				Type:     "tag",
+				IAL:      "",
+			}
+			ret = append(ret, span)
+		}
+	}
+	return
+}
+
+func docTitleImgAsset(root *ast.Node) *Asset {
+	if p := treenode.GetDocTitleImgPath(root); "" != p {
+		if !IsAssetLinkDest([]byte(p)) {
+			return nil
+		}
+
+		var hash string
+		absPath := filepath.Join(util.DataDir, p)
+		if data, err := os.ReadFile(absPath); nil != err {
+			util.LogErrorf("read asset [%s] data failed: %s", absPath, err)
+			hash = fmt.Sprintf("%x", sha256.Sum256([]byte(gulu.Rand.String(7))))
+		} else {
+			hash = fmt.Sprintf("%x", sha256.Sum256(data))
+		}
+		name, _ := util.LastID(p)
+		asset := &Asset{
+			ID:      ast.NewNodeID(),
+			BlockID: root.ID,
+			RootID:  root.ID,
+			Box:     root.Box,
+			DocPath: p,
+			Path:    p,
+			Name:    name,
+			Title:   "title-img",
+			Hash:    hash,
+		}
+		return asset
+	}
+	return nil
+}
+
+func QueryAssetsByName(name string) (ret []*Asset) {
+	ret = []*Asset{}
+	sqlStmt := "SELECT * FROM assets WHERE name LIKE ? GROUP BY id ORDER BY id DESC LIMIT 32"
+	rows, err := query(sqlStmt, "%"+name+"%")
+	if nil != err {
+		util.LogErrorf("sql query [%s] failed: %s", sqlStmt, err)
+		return
+	}
+	defer rows.Close()
+	for rows.Next() {
+		asset := scanAssetRows(rows)
+		ret = append(ret, asset)
+	}
+	return
+}
+
+func QueryAssetByHash(hash string) (ret *Asset) {
+	sqlStmt := "SELECT * FROM assets WHERE hash = ?"
+	row := queryRow(sqlStmt, hash)
+	var asset Asset
+	if err := row.Scan(&asset.ID, &asset.BlockID, &asset.RootID, &asset.Box, &asset.DocPath, &asset.Path, &asset.Name, &asset.Title, &asset.Hash); nil != err {
+		if sql.ErrNoRows != err {
+			util.LogErrorf("query scan field failed: %s", err)
+		}
+		return
+	}
+	ret = &asset
+	return
+}
+
+func QueryRootBlockAssets(rootID string) (ret []*Asset) {
+	sqlStmt := "SELECT * FROM assets WHERE root_id = ?"
+	rows, err := query(sqlStmt, rootID)
+	if nil != err {
+		util.LogErrorf("sql query [%s] failed: %s", sqlStmt, err)
+		return
+	}
+	defer rows.Close()
+	for rows.Next() {
+		asset := scanAssetRows(rows)
+		ret = append(ret, asset)
+	}
+	return
+}
+
+func scanAssetRows(rows *sql.Rows) (ret *Asset) {
+	var asset Asset
+	if err := rows.Scan(&asset.ID, &asset.BlockID, &asset.RootID, &asset.Box, &asset.DocPath, &asset.Path, &asset.Name, &asset.Title, &asset.Hash); nil != err {
+		util.LogErrorf("query scan field failed: %s", err)
+		return
+	}
+	ret = &asset
+	return
+}
+
+func assetLocalPath(linkDest, boxLocalPath, docDirLocalPath string) (ret string) {
+	ret = filepath.Join(docDirLocalPath, linkDest)
+	if gulu.File.IsExist(ret) {
+		return
+	}
+
+	ret = filepath.Join(boxLocalPath, linkDest)
+	if gulu.File.IsExist(ret) {
+		return
+	}
+
+	ret = filepath.Join(util.DataDir, linkDest)
+	if gulu.File.IsExist(ret) {
+		return
+	}
+	return ""
+}