paadi/html2pdf
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add function convert html

Browse Source
This commit is contained in:
Adrian Zürcher
2026-01-06 09:54:47 +01:00
parent 7fec8e1101
commit bd07add64e
2 changed files with 113 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
converter/converter.go
View File

@@ -126,6 +126,59 @@ func (c *Converter) Convert(files ...models.File) error {
return nil return nil
} }
// Convert converts all given input files
func (c *Converter) ConvertHtml(html []byte, outputPath string) error {
if len(html) == 0 {
return fmt.Errorf("no .html input provided")
} else if outputPath == "" || filepath.Ext(outputPath) != ".pdf" {
return fmt.Errorf("no .pdf output file path provided: %s", outputPath)
}
taskCtx, taskCancel := chromedp.NewContext(c.browserCtx)
timeoutCtx, timeoutCancel := context.WithTimeout(taskCtx, 60*time.Second)
var pdfData []byte
err := chromedp.Run(timeoutCtx,
// Start with a blank page
chromedp.Navigate("about:blank"),
// Inject HTML directly
chromedp.ActionFunc(func(ctx context.Context) error {
frameTree, err := page.GetFrameTree().Do(ctx)
if err != nil {
return err
}
return page.SetDocumentContent(frameTree.Frame.ID, string(html)).Do(ctx)
}),
chromedp.WaitReady("body", chromedp.ByQuery),
chromedp.ActionFunc(func(ctx context.Context) error {
buf, _, err := page.PrintToPDF().
WithPrintBackground(true).
WithPaperWidth(8.27).
WithPaperHeight(11.69).
Do(ctx)
if err != nil {
return err
}
pdfData = buf
return nil
}),
)
timeoutCancel()
taskCancel()
if err != nil {
return err
}
// Save PDF to file
if err := os.WriteFile(outputPath, pdfData, 0644); err != nil {
return err
}
return nil
}
func (c *Converter) Close() { func (c *Converter) Close() {
// Close browser first, then allocator // Close browser first, then allocator
if c.browserCancel != nil { if c.browserCancel != nil {

60
html2pdf_test.go
View File

@@ -1,6 +1,7 @@
package html2pdf package html2pdf
import ( import (
"bytes"
"fmt" "fmt"
"os" "os"
"path/filepath" "path/filepath"
@@ -57,3 +58,62 @@ func TestConvertFiles(t *testing.T) {
} }
t.Log("test successfull") t.Log("test successfull")
} }
func TestConvertHtml(t *testing.T) {
t.Log("start test ConvertHtml")
// Skip in short mode (useful for CI)
if testing.Short() {
t.Skip("skipping ConvertHtml integration test in short mode")
}
html := []byte(`
<!DOCTYPE html>
<html>
<head>
<meta charset="utf-8">
<title>Test PDF</title>
<style>
body { font-family: Arial, sans-serif; }
h1 { color: #333; }
</style>
</head>
<body>
<h1>Hello PDF</h1>
<p>This is a test.</p>
</body>
</html>
`)
tmpDir := t.TempDir()
outputPath := filepath.Join(tmpDir, "test.pdf")
c, err := converter.NewConverter("assets/chrome-headless-shell/win64")
if err != nil {
t.Fatal(err)
}
defer c.Close()
err = c.ConvertHtml(html, outputPath)
if err != nil {
t.Fatalf("ConvertHtml failed: %v", err)
}
// Assert PDF exists
data, err := os.ReadFile(outputPath)
if err != nil {
t.Fatalf("PDF not created: %v", err)
}
// Assert non-empty
if len(data) == 0 {
t.Fatal("PDF file is empty")
}
// Assert valid PDF header
if !bytes.HasPrefix(data, []byte("%PDF-")) {
t.Fatalf("output is not a valid PDF (missing %%PDF- header)")
}
t.Log("ConvertHtml test successful")
}