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

kernel/model/upload.go

@@ -0,0 +1,197 @@
+// 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 model
+
+import (
+	"crypto/sha256"
+	"errors"
+	"fmt"
+	"io"
+	"os"
+	"path"
+	"path/filepath"
+	"strings"
+
+	"github.com/88250/gulu"
+	"github.com/88250/lute/ast"
+	"github.com/gin-gonic/gin"
+	"github.com/siyuan-note/siyuan/kernel/sql"
+	"github.com/siyuan-note/siyuan/kernel/treenode"
+	"github.com/siyuan-note/siyuan/kernel/util"
+)
+
+func InsertLocalAssets(id string, assetPaths []string) (succMap map[string]interface{}, err error) {
+	succMap = map[string]interface{}{}
+
+	bt := treenode.GetBlockTree(id)
+	if nil == bt {
+		err = errors.New(Conf.Language(71))
+		return
+	}
+
+	docDirLocalPath := filepath.Join(util.DataDir, bt.BoxID, path.Dir(bt.Path))
+	assets := getAssetsDir(filepath.Join(util.DataDir, bt.BoxID), docDirLocalPath)
+	for _, p := range assetPaths {
+		fName := filepath.Base(p)
+		fName = util.FilterUploadFileName(fName)
+		ext := filepath.Ext(fName)
+		fName = strings.TrimSuffix(fName, ext)
+		ext = strings.ToLower(ext)
+		fName += ext
+		baseName := fName
+		if gulu.File.IsDir(p) {
+			succMap[baseName] = "file://" + p
+			continue
+		}
+
+		var f *os.File
+		f, err = os.Open(p)
+		if nil != err {
+			return
+		}
+
+		var data []byte
+		data, err = io.ReadAll(f)
+		f.Close()
+		if nil != err {
+			return
+		}
+
+		hash := fmt.Sprintf("%x", sha256.Sum256(data))
+		if existAsset := sql.QueryAssetByHash(hash); nil != existAsset {
+			// 已经存在同样数据的资源文件的话不重复保存
+			succMap[baseName] = existAsset.Path
+		} else {
+			ext := path.Ext(fName)
+			fName = fName[0 : len(fName)-len(ext)]
+			fName = fName + "-" + ast.NewNodeID() + ext
+			writePath := filepath.Join(assets, fName)
+			if err = gulu.File.WriteFileSafer(writePath, data, 0644); nil != err {
+				return
+			}
+			succMap[baseName] = "assets/" + fName
+		}
+	}
+	IncWorkspaceDataVer()
+	return
+}
+
+func Upload(c *gin.Context) {
+	ret := gulu.Ret.NewResult()
+	defer c.JSON(200, ret)
+
+	form, err := c.MultipartForm()
+	if nil != err {
+		util.LogErrorf("insert asset failed: %s", err)
+		ret.Code = -1
+		ret.Msg = err.Error()
+		return
+	}
+	assetsDirPath := filepath.Join(util.DataDir, "assets")
+	if nil != form.Value["id"] {
+		id := form.Value["id"][0]
+		bt := treenode.GetBlockTree(id)
+		if nil == bt {
+			ret.Code = -1
+			ret.Msg = Conf.Language(71)
+			return
+		}
+		docDirLocalPath := filepath.Join(util.DataDir, bt.BoxID, path.Dir(bt.Path))
+		assetsDirPath = getAssetsDir(filepath.Join(util.DataDir, bt.BoxID), docDirLocalPath)
+	}
+	if nil != form.Value["assetsDirPath"] {
+		assetsDirPath = form.Value["assetsDirPath"][0]
+		assetsDirPath = filepath.Join(util.DataDir, assetsDirPath)
+		if err := os.MkdirAll(assetsDirPath, 0755); nil != err {
+			ret.Code = -1
+			ret.Msg = err.Error()
+			return
+		}
+	}
+
+	var errFiles []string
+	succMap := map[string]interface{}{}
+	files := form.File["file[]"]
+	for _, file := range files {
+		fName := file.Filename
+		fName = util.FilterUploadFileName(fName)
+		ext := filepath.Ext(fName)
+		fName = strings.TrimSuffix(fName, ext)
+		ext = strings.ToLower(ext)
+		fName += ext
+		baseName := fName
+		f, err := file.Open()
+		if nil != err {
+			errFiles = append(errFiles, fName)
+			ret.Msg = err.Error()
+			break
+		}
+
+		data, err := io.ReadAll(f)
+		if nil != err {
+			errFiles = append(errFiles, fName)
+			ret.Msg = err.Error()
+			break
+		}
+		f.Close()
+
+		hash := fmt.Sprintf("%x", sha256.Sum256(data))
+		if existAsset := sql.QueryAssetByHash(hash); nil != existAsset {
+			// 已经存在同样数据的资源文件的话不重复保存
+			succMap[baseName] = existAsset.Path
+		} else {
+			_, id := util.LastID(fName)
+			ext := path.Ext(fName)
+			fName = fName[0 : len(fName)-len(ext)]
+			if !util.IsIDPattern(id) {
+				id = ast.NewNodeID()
+				fName = fName + "-" + id + ext
+			} else {
+				if !util.IsIDPattern(fName) {
+					fName = fName[:len(fName)-len(id)-1] + "-" + id + ext
+				} else {
+					fName = fName + ext
+				}
+			}
+			writePath := filepath.Join(assetsDirPath, fName)
+			if err = gulu.File.WriteFileSafer(writePath, data, 0644); nil != err {
+				errFiles = append(errFiles, fName)
+				ret.Msg = err.Error()
+				break
+			}
+			succMap[baseName] = "assets/" + fName
+		}
+	}
+
+	ret.Data = map[string]interface{}{
+		"errFiles": errFiles,
+		"succMap":  succMap,
+	}
+
+	IncWorkspaceDataVer()
+}
+
+func getAssetsDir(boxLocalPath, docDirLocalPath string) (assets string) {
+	assets = filepath.Join(docDirLocalPath, "assets")
+	if !gulu.File.IsExist(assets) {
+		assets = filepath.Join(boxLocalPath, "assets")
+		if !gulu.File.IsExist(assets) {
+			assets = filepath.Join(util.DataDir, "assets")
+		}
+	}
+	return
+}