🐛 gen pdf

yann0917 · Jan 13, 2024 · 852153e · 852153e
1 parent a9bd5b6
commit 852153e
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/utils/genPdf.go b/utils/genPdf.go
@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"fmt"
 	"os"
+	"path/filepath"
 	"runtime"
 
 	"github.com/SebastiaanKlippert/go-wkhtmltopdf"
@@ -28,11 +29,15 @@ func (p *PdfOption) GenPdf(buf *bytes.Buffer) (err error) {
 
 	if p.CoverPath != "" {
 		pdfg.Cover.EnableLocalFileAccess.Set(true)
-
+		dir, err := CurrentDir()
+		if err != nil {
+			pdfg.Cover.EnableLocalFileAccess.Set(false)
+		}
+		dir = filepath.Join(dir, p.CoverPath)
 		if runtime.GOOS == "windows" {
-			pdfg.Cover.Input = p.CoverPath
+			pdfg.Cover.Input = dir
 		} else {
-			pdfg.Cover.Input = "file://" + p.CoverPath
+			pdfg.Cover.Input = "file://" + dir
 		}
 	}