|
|
|
|
@@ -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)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|