7 Commits

Author SHA1 Message Date
Adrian Zürcher
39fb913f87 fix compile error for other os than windows 2025-12-29 11:20:50 +01:00
Adrian Zürcher
8df18a243f add console hide for windows 2025-12-29 11:07:36 +01:00
Adrian Zürcher
613b74ee55 add new feature to set progress callback 2025-12-25 01:14:03 +01:00
Adrian Zürcher
edb8d7e830 change chrome path priority close #1 2025-12-24 12:15:50 +01:00
Adrian Zürcher
537179af03 fix test model 2025-12-24 12:09:34 +01:00
Adrian Zürcher
48793c65bc add new model 2025-12-24 12:08:43 +01:00
Adrian Zürcher
5985854092 add new instance and provide multiple converting files close #2 2025-12-24 12:06:45 +01:00
6 changed files with 207 additions and 59 deletions

View File

@@ -2,41 +2,41 @@ package converter
import ( import (
"context" "context"
"errors"
"fmt"
"os" "os"
"path/filepath" "path/filepath"
"runtime" "runtime"
"strings"
"time" "time"
"gitea.tecamino.com/paadi/html2pdf/models"
"github.com/chromedp/cdproto/page" "github.com/chromedp/cdproto/page"
"github.com/chromedp/chromedp" "github.com/chromedp/chromedp"
) )
// html to pdf converter structure for
type Converter struct { type Converter struct {
chromePath string chromePath string
ctx context.Context
cancel context.CancelFunc
progress func(progress int)
} }
func NewConverter(chromePath string) *Converter { // NewConverter starts a new converter instance with a chrome headless shell executable
return &Converter{chromePath: chromePath} func NewConverter(chromePath string) (*Converter, error) {
} var err error
func (c *Converter) Convert(imputFile, outputFile string) error { if runtime.GOOS == "windows" {
hideConsole()
chromePath := c.getChromePath()
htmlURL := "file://"
switch runtime.GOOS {
case "windows":
htmlURL += "/"
} }
// Convert to absolute path c := &Converter{chromePath: chromePath}
absPath, err := filepath.Abs(imputFile) chromePath, err = c.getChromePath()
if err != nil { if err != nil {
return err return nil, err
} }
htmlURL += filepath.ToSlash(absPath)
opts := append(chromedp.DefaultExecAllocatorOptions[:], opts := append(chromedp.DefaultExecAllocatorOptions[:],
chromedp.ExecPath(chromePath), chromedp.ExecPath(chromePath),
chromedp.NoSandbox, chromedp.NoSandbox,
@@ -44,17 +44,50 @@ func (c *Converter) Convert(imputFile, outputFile string) error {
chromedp.DisableGPU, chromedp.DisableGPU,
) )
allocCtx, cancel := chromedp.NewExecAllocator(context.Background(), opts...) var allocCtx context.Context
defer cancel() allocCtx, c.cancel = chromedp.NewExecAllocator(context.Background(), opts...)
ctx, cancel := chromedp.NewContext(allocCtx) c.ctx, c.cancel = chromedp.NewContext(allocCtx)
defer cancel() return c, nil
}
ctx, cancel = context.WithTimeout(ctx, 60*time.Second) func (c *Converter) SetProgressCallback(cb func(progress int)) {
defer cancel() c.progress = cb
}
var pdf []byte // Convert converts all given input files
err = chromedp.Run(ctx, func (c *Converter) Convert(files ...models.File) error {
chromedp.Navigate(htmlURL), for i, f := range files {
if c.progress != nil {
c.progress(i + 1)
}
if f.Input == "" || filepath.Ext(f.Input) != ".html" {
return fmt.Errorf("no .html input file path provided: %s", f.Input)
} else if f.Output == "" || filepath.Ext(f.Output) != ".pdf" {
return fmt.Errorf("no .pdf output file path provided: %s", f.Output)
}
var htmlURL strings.Builder
htmlURL.WriteString("file://")
switch runtime.GOOS {
case "windows":
htmlURL.WriteString("/")
}
// Convert to absolute path
absPath, err := filepath.Abs(f.Input)
if err != nil {
return err
}
htmlURL.WriteString(filepath.ToSlash(absPath))
c.ctx, c.cancel = context.WithTimeout(c.ctx, 60*time.Second)
var pdfData []byte
err = chromedp.Run(c.ctx,
chromedp.Navigate(htmlURL.String()),
chromedp.WaitReady("body", chromedp.ByQuery), chromedp.WaitReady("body", chromedp.ByQuery),
chromedp.ActionFunc(func(ctx context.Context) error { chromedp.ActionFunc(func(ctx context.Context) error {
buf, _, err := page.PrintToPDF(). buf, _, err := page.PrintToPDF().
@@ -65,26 +98,43 @@ func (c *Converter) Convert(imputFile, outputFile string) error {
if err != nil { if err != nil {
return err return err
} }
pdf = buf pdfData = buf
return nil return nil
}), }),
) )
if err != nil { if err != nil {
c.cancel()
return err return err
} }
// Save PDF to file // Save PDF to file
if err := os.WriteFile(outputFile, pdf, 0644); err != nil { if err := os.WriteFile(f.Output, pdfData, 0644); err != nil {
c.cancel()
return err return err
} }
}
c.cancel()
return nil return nil
} }
// getChromePath checks for system Chrome, else falls back to bundled headless shell // getChromePath checks for system Chrome, else falls back to bundled headless shell
func (c *Converter) getChromePath() string { func (c *Converter) getChromePath() (string, error) {
chromeExec := "chrome-headless-shell"
if runtime.GOOS == "windows" {
chromeExec += ".exe"
}
path := filepath.Join(c.chromePath, chromeExec)
if _, err := os.Stat(path); err == nil {
return path, nil
}
// Candidate paths for system Chrome // Candidate paths for system Chrome
candidates := []string{} candidates := []string{}
// Fallback:
switch runtime.GOOS { switch runtime.GOOS {
case "windows": case "windows":
candidates = []string{ candidates = []string{
@@ -108,14 +158,9 @@ func (c *Converter) getChromePath() string {
// Check system paths first // Check system paths first
for _, path := range candidates { for _, path := range candidates {
if _, err := os.Stat(path); err == nil { if _, err := os.Stat(path); err == nil {
return path return path, nil
} }
} }
// Fallback: use bundled headless shell return "", errors.New("chrome path not found")
chromeExec := "chrome-headless-shell"
if runtime.GOOS == "windows" {
chromeExec += ".exe"
}
return filepath.Join(c.chromePath, chromeExec)
} }

9
converter/hideOthers.go Normal file
View File

@@ -0,0 +1,9 @@
//go:build !windows
package converter
// hideConsole does nothing on non-Windows systems
func hideConsole() {
// macOS and Linux don't have the same "console window" concept
// that needs manual hiding at runtime like Windows.
}

23
converter/hideWindows.go Normal file
View File

@@ -0,0 +1,23 @@
//go:build windows
package converter
import (
"syscall"
)
var (
kernel32 = syscall.NewLazyDLL("kernel32.dll")
user32 = syscall.NewLazyDLL("user32.dll")
getConsoleWindow = kernel32.NewProc("GetConsoleWindow")
showWindow = user32.NewProc("ShowWindow")
)
const SW_HIDE = 0
func hideConsole() {
hwnd, _, _ := getConsoleWindow.Call()
if hwnd != 0 {
showWindow.Call(hwnd, SW_HIDE)
}
}

View File

@@ -1,8 +1,24 @@
package html2pdf package html2pdf
import "gitea.tecamino.com/paadi/html2pdf/converter" import (
"gitea.tecamino.com/paadi/html2pdf/converter"
"gitea.tecamino.com/paadi/html2pdf/models"
)
// Convert converts one .html file to .pdf
func Convert(chromePath, inputFile, outputFile string) error { func Convert(chromePath, inputFile, outputFile string) error {
c := converter.NewConverter(chromePath) input := models.File{
return c.Convert(inputFile, outputFile) Input: inputFile,
Output: outputFile,
}
c, err := converter.NewConverter(chromePath)
if err != nil {
return err
}
return c.Convert(input)
}
// NewConverterInstance start new chrome headless shell instance
func NewConverterInstance(chromePath string) (*converter.Converter, error) {
return converter.NewConverter(chromePath)
} }

View File

@@ -1,12 +1,58 @@
package html2pdf package html2pdf
import "testing" import (
"fmt"
"os"
"path/filepath"
"strings"
"testing"
"gitea.tecamino.com/paadi/html2pdf/converter"
"gitea.tecamino.com/paadi/html2pdf/models"
)
func TestConvert(t *testing.T) { func TestConvert(t *testing.T) {
t.Log("start test") t.Log("start test convert one file")
err := Convert("./assets", "dst/test.html", "out.pdf") err := Convert("./assets", "dst/test.html", "out.pdf")
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
} }
func TestConvertFiles(t *testing.T) {
t.Log("start test convert files")
rootPath := "dst"
files, err := os.ReadDir(rootPath)
if err != nil {
t.Fatal(err)
}
var input []models.File
for _, f := range files {
ext := filepath.Ext(f.Name())
if ext != ".html" {
continue
}
input = append(input, models.File{
Input: filepath.Join(rootPath, f.Name()),
Output: strings.Replace(f.Name(), ext, ".pdf", 1),
})
}
c, err := converter.NewConverter("./assets")
if err != nil {
t.Fatal(err)
}
c.SetProgressCallback(func(progress int) {
fmt.Println(progress)
})
if err := c.Convert(input...); err != nil {
t.Fatal(err)
}
t.Log("test successfull")
}

9
models/file.go Normal file
View File

@@ -0,0 +1,9 @@
package models
type Files []File
// input model for converter
type File struct {
Input string
Output string
}