change chrome path priority close #1
This commit is contained in:
@@ -2,6 +2,7 @@ package converter
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
@@ -22,9 +23,13 @@ type Converter struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// NewConverter starts a new converter instance with a chrome headless shell executable
|
// NewConverter starts a new converter instance with a chrome headless shell executable
|
||||||
func NewConverter(chromePath string) *Converter {
|
func NewConverter(chromePath string) (*Converter, error) {
|
||||||
|
var err error
|
||||||
c := &Converter{chromePath: chromePath}
|
c := &Converter{chromePath: chromePath}
|
||||||
chromePath = c.getChromePath()
|
chromePath, err = c.getChromePath()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
opts := append(chromedp.DefaultExecAllocatorOptions[:],
|
opts := append(chromedp.DefaultExecAllocatorOptions[:],
|
||||||
chromedp.ExecPath(chromePath),
|
chromedp.ExecPath(chromePath),
|
||||||
@@ -36,7 +41,7 @@ func NewConverter(chromePath string) *Converter {
|
|||||||
var allocCtx context.Context
|
var allocCtx context.Context
|
||||||
allocCtx, c.cancel = chromedp.NewExecAllocator(context.Background(), opts...)
|
allocCtx, c.cancel = chromedp.NewExecAllocator(context.Background(), opts...)
|
||||||
c.ctx, c.cancel = chromedp.NewContext(allocCtx)
|
c.ctx, c.cancel = chromedp.NewContext(allocCtx)
|
||||||
return c
|
return c, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Convert converts all given input files
|
// Convert converts all given input files
|
||||||
@@ -97,10 +102,22 @@ func (c *Converter) Convert(files ...models.File) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 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{
|
||||||
@@ -124,14 +141,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)
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -11,11 +11,14 @@ func Convert(chromePath, inputFile, outputFile string) error {
|
|||||||
Input: inputFile,
|
Input: inputFile,
|
||||||
Output: outputFile,
|
Output: outputFile,
|
||||||
}
|
}
|
||||||
c := converter.NewConverter(chromePath)
|
c, err := converter.NewConverter(chromePath)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
return c.Convert(input)
|
return c.Convert(input)
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewConverterInstance start new chrome headless shell instance
|
// NewConverterInstance start new chrome headless shell instance
|
||||||
func NewConverterInstance(chromePath string) *converter.Converter {
|
func NewConverterInstance(chromePath string) (*converter.Converter, error) {
|
||||||
return converter.NewConverter(chromePath)
|
return converter.NewConverter(chromePath)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -41,7 +41,10 @@ func TestConvertFiles(t *testing.T) {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
c := converter.NewConverter("./assets")
|
c, err := converter.NewConverter("./assets")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
if err := c.Convert(input...); err != nil {
|
if err := c.Convert(input...); err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
|
|||||||
Reference in New Issue
Block a user