add function convert html

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

View File

@@ -126,6 +126,59 @@ func (c *Converter) Convert(files ...models.File) error {
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() {
// Close browser first, then allocator
if c.browserCancel != nil {