Andreas/siyuan
1
0
Fork 0
mirror of https://github.com/siyuan-note/siyuan.git synced 2025-12-30 05:18:49 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

✨ Support adding the watermark on export PDF https://github.com/siyuan-note/siyuan/issues/9961

Browse source
This commit is contained in:
Daniel 2023-12-27 11:23:37 +08:00
parent 44060c6e0d
commit ec0989c32f
No known key found for this signature in database
GPG key ID: 86211BA83DF03017
9 changed files with 157 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

54
kernel/model/export.go
View file

@ -20,6 +20,7 @@ import (
"bytes"
"errors"
"fmt"
"github.com/88250/pdfcpu/pkg/font"
"net/http"
"net/url"
"os"
@ -781,7 +782,7 @@ func processIFrame(tree *parse.Tree) {
}
}
func ProcessPDF(id, p string, merge, removeAssets bool) (err error) {
func ProcessPDF(id, p string, merge, removeAssets, watermark bool) (err error) {
tree, _ := loadTreeByBlockID(id)
if nil == tree {
return
@ -810,6 +811,12 @@ func ProcessPDF(id, p string, merge, removeAssets bool) (err error) {
return ast.WalkContinue
})
pdfcpu.ConfigPath = "disable"
font.UserFontDir = filepath.Join(util.HomeDir, ".config", "siyuan", "fonts")
if mkdirErr := os.MkdirAll(font.UserFontDir, 0755); nil != mkdirErr {
logging.LogErrorf("mkdir [%s] failed: %s", font.UserFontDir, mkdirErr)
return
}
pdfCtx, ctxErr := api.ReadContextFile(p)
if nil != ctxErr {
logging.LogErrorf("read pdf context failed: %s", ctxErr)
@ -818,7 +825,7 @@ func ProcessPDF(id, p string, merge, removeAssets bool) (err error) {
processPDFBookmarks(pdfCtx, headings)
processPDFLinkEmbedAssets(pdfCtx, assetDests, removeAssets)
processPDFWatermark(pdfCtx)
processPDFWatermark(pdfCtx, watermark)
pdfcpu.VersionStr = "SiYuan v" + util.Ver
if writeErr := api.WriteContextFile(pdfCtx, p); nil != writeErr {
@ -828,10 +835,14 @@ func ProcessPDF(id, p string, merge, removeAssets bool) (err error) {
return
}
func processPDFWatermark(pdfCtx *pdfcpu.Context) {
func processPDFWatermark(pdfCtx *pdfcpu.Context, watermark bool) {
// Support adding the watermark on export PDF https://github.com/siyuan-note/siyuan/issues/9961
// https://pdfcpu.io/core/watermark
if !watermark {
return
}
str := Conf.Export.PDFWatermarkStr
if "" == str {
return
@ -842,9 +853,39 @@ func processPDFWatermark(pdfCtx *pdfcpu.Context) {
}
desc := Conf.Export.PDFWatermarkDesc
f, e := pdfCtx.ExtractFont(1)
if nil == e {
desc = "fontname:" + f.Name
descParts := strings.Split(desc, ",")
m := map[string]string{}
for _, descPart := range descParts {
kv := strings.Split(descPart, ":")
if 2 != len(kv) {
continue
}
m[kv[0]] = kv[1]
}
if "" == m["fontname"] {
m["fontname"] = "LXGW WenKai Lite"
}
descBuilder := bytes.Buffer{}
for k, v := range m {
descBuilder.WriteString(k)
descBuilder.WriteString(":")
descBuilder.WriteString(v)
descBuilder.WriteString(",")
}
desc = descBuilder.String()
desc = desc[:len(desc)-1]
fontPath := filepath.Join(util.AppearancePath, "fonts", "LxgwWenKai-Lite-1.311", "LXGWWenKaiLite-Regular.ttf")
err := api.InstallFonts([]string{fontPath})
if nil != err {
logging.LogErrorf("install font [%s] failed: %s", fontPath, err)
}
fonts, err := api.ListFonts()
if nil != err {
logging.LogErrorf("list fonts failed: %s", err)
} else {
for _, f := range fonts {
logging.LogInfof("installed font: %s", f)
}
}
mode := "text"
@ -859,7 +900,6 @@ func processPDFWatermark(pdfCtx *pdfcpu.Context) {
logging.LogInfof("add PDF watermark [mode=%s, str=%s, desc=%s]", mode, str, desc)
var wm *pdfcpu.Watermark
var err error
switch mode {
case "text":
wm, err = pdfcpu.ParseTextWatermarkDetails(str, desc, false, pdfcpu.POINTS)