diff --git a/Grabazine.go b/Grabazine.go
index 9e2b8b1..c8d13b9 100644
--- a/Grabazine.go
+++ b/Grabazine.go
@@ -5,8 +5,10 @@ import (
"encoding/json"
"flag"
"fmt"
+ wkhtml "github.com/SebastiaanKlippert/go-wkhtmltopdf"
"github.com/pdfcpu/pdfcpu/pkg/api"
"io/ioutil"
+ "log"
"net/http"
"os"
"os/exec"
@@ -32,6 +34,7 @@ func main() {
passwordPtr := flag.String("p", "", "Zinio Password")
chromePtr := flag.String("c", "google-chrome", "Chrome executable")
zinioHostPtr := flag.String("e", "api-sec.ziniopro.com", "Zinio Host (Excluding port and URI Scheme). Known: `api-sec`, `api-sec-2`")
+ exportUsingWKHTML := flag.String("wkhtml", "false", "Use WKHTML instead of Chrome to generate PDF (false by default)")
flag.Parse()
@@ -45,7 +48,7 @@ func main() {
fmt.Println("Found " + strconv.Itoa(len(issues.Data)) + " issues in library.")
fmt.Println("Loading HTML template")
- defaultTemplate := ""
+ defaultTemplate := GetDefaultTemplate()
template, _ := ioutil.ReadFile("template.html")
if template == nil || len(template) == 0 {
@@ -71,6 +74,7 @@ func main() {
fmt.Println("Issue already found: " + issue.Publication.Name + " - " + issue.Name)
continue
}
+ fmt.Println("Downloading issue: " + issue.Publication.Name + " - " + issue.Name)
pages := GetPages(loginToken, issue, initialToken, *zinioHostPtr)
@@ -82,26 +86,72 @@ func main() {
pathString := issuePath + "_" + pages.Data[i].Index
- htmldata := strings.Replace(string(template), "SVG_PATH", pages.Data[i].Source, -1)
- //write html file, embedding svg
- ioutil.WriteFile(pathString+".html", []byte(htmldata), 0644)
+ resp, err := http.Get(pages.Data[i].Source)
+ // handle the error if there is one
+ if err != nil {
+ panic(err)
+ }
+ // do this now so it won't be forgotten
+ defer resp.Body.Close()
+ // reads html as a slice of bytes
+ html, err := ioutil.ReadAll(resp.Body)
+ if err != nil {
+ panic(err)
+ }
+ // show the HTML code as a string %s
+
+ htmldata := strings.Replace(string(template), "SVG_PATH", string(html), -1)
//convert to pdf
- cmd := exec.Command(*chromePtr, "--headless", "--disable-gpu", "--print-to-pdf="+pathString+".pdf", "--no-margins", pathString+".html")
+ if strings.ToLower(*exportUsingWKHTML) == "true" {
+
+ pdfg, err := wkhtml.NewPDFGenerator()
+ if err != nil {
+ return
+ }
+ pdfg.MarginBottom.Set(0)
+ pdfg.MarginTop.Set(0)
+ pdfg.MarginLeft.Set(0)
+ pdfg.MarginRight.Set(0)
+ pdfg.NoOutline.Set(true)
+ //pdfg.PageSize.Set(wkhtml.PageSizeCustom)
+ pdfg.AddPage(wkhtml.NewPageReader(strings.NewReader(htmldata)))
+
+ // Create PDF document in internal buffer
+ err = pdfg.Create()
+ if err != nil {
+ log.Fatal(err)
+ }
+
+ //Your Pdf Name
+ err = pdfg.WriteFile(pathString + ".pdf")
+ if err != nil {
+ log.Fatal(err)
+ }
+
+ } else {
+ //write html file, embedding svg
+ ioutil.WriteFile(pathString+".html", []byte(htmldata), 0644)
+ cmd := exec.Command(*chromePtr, "--headless", "--disable-gpu", "--print-to-pdf="+pathString+".pdf", "--no-margins", pathString+".html")
+ fmt.Println(cmd.Args)
+ err := cmd.Run()
+ if err != nil {
+ fmt.Printf("cmd.Run() failed with %s\n. You should retry this page.", err)
+ }
- fmt.Println(cmd.Args)
- err := cmd.Run()
- if err != nil {
- fmt.Printf("cmd.Run() failed with %s\n. You should retry this page.", err)
}
_ = os.Remove(pathString + ".html")
_ = os.Remove(pathString + ".svg")
+ filenames = append(filenames, pathString+".pdf")
+ }
+
+ for i := range filenames {
//remove last page
err = retry(5, 2*time.Second, func() (err error) {
- err = api.RemovePagesFile(pathString+".pdf", "", []string{"2-"}, nil)
+ err = api.RemovePagesFile(filenames[i], "", []string{"2-"}, nil)
if err != nil {
fmt.Printf("Removing extra pages failed with %s\n.", err)
@@ -111,8 +161,6 @@ func main() {
return
})
-
- filenames = append(filenames, pathString+".pdf")
}
_ = api.MergeCreateFile(filenames, completeName, nil)
@@ -266,3 +314,53 @@ func retry(attempts int, sleep time.Duration, f func() error) (err error) {
}
return fmt.Errorf("after %d attempts, last error: %s", attempts, err)
}
+
+func GetDefaultTemplate() string {
+ return `
+
+
+
+
+
+
+
+
+
+
+
+
+
+ `
+}
diff --git a/README.md b/README.md
index 40067c3..1d41e68 100644
--- a/README.md
+++ b/README.md
@@ -20,3 +20,6 @@ Each page is available as an SVG, which is then injected into an HTML page (base
google-chrome is then used to print the page to PDF, and all pages are combined into a single PDF.
+###Building
+Build for linux & windows on windows using the pwershell script in buildscripts
+
diff --git a/buildscripts/windows.ps1 b/buildscripts/windows.ps1
new file mode 100644
index 0000000..52d4d9f
--- /dev/null
+++ b/buildscripts/windows.ps1
@@ -0,0 +1,10 @@
+$env:GOOS = 'windows'
+$env:GOARCH = 'amd64'
+cd ..
+go build -o built/Zinigo_Windows_x64.exe Grabazine.go
+cd buildscripts
+$env:GOOS = 'linux'
+$env:GOARCH = 'amd64'
+cd ..
+go build -o built/Zinigo_Linux_AMD64 Grabazine.go
+cd buildscripts
\ No newline at end of file