2 Commits

Author SHA1 Message Date
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
2 changed files with 37 additions and 1 deletions

View File

@@ -8,6 +8,7 @@ import (
"path/filepath"
"runtime"
"strings"
"syscall"
"time"
"gitea.tecamino.com/paadi/html2pdf/models"
@@ -20,11 +21,17 @@ type Converter struct {
chromePath string
ctx context.Context
cancel context.CancelFunc
progress func(progress int)
}
// NewConverter starts a new converter instance with a chrome headless shell executable
func NewConverter(chromePath string) (*Converter, error) {
var err error
if runtime.GOOS == "windows" {
hideConsole()
}
c := &Converter{chromePath: chromePath}
chromePath, err = c.getChromePath()
if err != nil {
@@ -44,9 +51,17 @@ func NewConverter(chromePath string) (*Converter, error) {
return c, nil
}
func (c *Converter) SetProgressCallback(cb func(progress int)) {
c.progress = cb
}
// Convert converts all given input files
func (c *Converter) Convert(files ...models.File) error {
for _, f := range files {
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" {
@@ -147,3 +162,19 @@ func (c *Converter) getChromePath() (string, error) {
return "", errors.New("chrome path not found")
}
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,6 +1,7 @@
package html2pdf
import (
"fmt"
"os"
"path/filepath"
"strings"
@@ -46,6 +47,10 @@ func TestConvertFiles(t *testing.T) {
t.Fatal(err)
}
c.SetProgressCallback(func(progress int) {
fmt.Println(progress)
})
if err := c.Convert(input...); err != nil {
t.Fatal(err)
}