Andreas/siyuan
1
0
Fork 0
mirror of https://github.com/siyuan-note/siyuan.git synced 2025-12-26 19:38:48 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

✨ 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 09:38:05 +08:00
parent 4fdaae2892
commit 94cc8b2e3b
No known key found for this signature in database
GPG key ID: 86211BA83DF03017
9 changed files with 57 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

33
kernel/model/export.go
View file

@ -818,7 +818,7 @@ func ProcessPDF(id, p string, merge, removeAssets bool) (err error) {
processPDFBookmarks(pdfCtx, headings)
processPDFLinkEmbedAssets(pdfCtx, assetDests, removeAssets)
// processPDFWatermark(pdfCtx, "text", "Test", "")
processPDFWatermark(pdfCtx)
pdfcpu.VersionStr = "SiYuan v" + util.Ver
if writeErr := api.WriteContextFile(pdfCtx, p); nil != writeErr {
@ -828,32 +828,45 @@ func ProcessPDF(id, p string, merge, removeAssets bool) (err error) {
return
}
func processPDFWatermark(pdfCtx *pdfcpu.Context, mode, watermark, desc string) {
func processPDFWatermark(pdfCtx *pdfcpu.Context) {
// Support adding the watermark on export PDF https://github.com/siyuan-note/siyuan/issues/9961
// https://pdfcpu.io/core/watermark
str := Conf.Export.PDFWatermarkStr
if "" == str {
return
}
if !IsPaidUser() {
return
}
if "" == watermark {
return
desc := Conf.Export.PDFWatermarkDesc
f, e := pdfCtx.ExtractFont(1)
if nil == e {
desc = "fontname:" + f.Name
}
if "text" != mode && "image" != mode && "pdf" != mode {
logging.LogErrorf("invalid watermark type: %s", mode)
return
mode := "text"
if gulu.File.IsExist(str) {
if ".pdf" == strings.ToLower(filepath.Ext(str)) {
mode = "pdf"
} else {
mode = "image"
}
}
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(watermark, desc, false, pdfcpu.POINTS)
wm, err = pdfcpu.ParseTextWatermarkDetails(str, desc, false, pdfcpu.POINTS)
case "image":
wm, err = pdfcpu.ParseImageWatermarkDetails(watermark, desc, false, pdfcpu.POINTS)
wm, err = pdfcpu.ParseImageWatermarkDetails(str, desc, false, pdfcpu.POINTS)
case "pdf":
wm, err = pdfcpu.ParsePDFWatermarkDetails(watermark, desc, false, pdfcpu.POINTS)
wm, err = pdfcpu.ParsePDFWatermarkDetails(str, desc, false, pdfcpu.POINTS)
}
if nil != err {