parse url to png

weaming · Nov 1, 2018 · 9d88a1a · 9d88a1a
1 parent c8a2f6f
commit 9d88a1a
Show file tree

Hide file tree

Showing 4 changed files with 68 additions and 28 deletions.
diff --git a/fs.go b/fs.go
@@ -2,9 +2,9 @@ package main
 
 import (
 	"io/ioutil"
+	"log"
 	"os"
 	"path"
-	"log"
 )
 
 func ExistFile(filename string) bool {

diff --git a/listFlag.go b/listFlag.go
@@ -1,8 +1,8 @@
 package main
 
 import (
-	"strings"
 	"fmt"
+	"strings"
 )
 
 type ArrayFlags []string

diff --git a/main.go b/main.go
@@ -3,6 +3,10 @@ package main
 import (
 	"flag"
 	"fmt"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"os"
 	"path"
 
 	"github.com/golang-commonmark/markdown"
@@ -27,39 +31,44 @@ func main() {
 
 	print := flag.Bool("print", false, "print generated html")
 	includeCSS := flag.Bool("include-css", false, "include css file as html header when read content from *.html file")
+	url := flag.String("url", "", "capture screenshot of the url")
 	flag.Parse()
 
-	if *cssName != "" {
-		cssUrlList = append(cssUrlList, getCssUrl(*cssName))
-	}
-
-	if *outputPath == "" {
-		*outputPath = ReplaceExt(path.Base(*markdownPath), "png")
-	}
-
-	//prepare static files
-	//go staticServer(*staticPath)
+	html := ""
+	if *url != "" {
+		html = GetHTMLFromURL(*url)
+	} else {
+		if *cssName != "" {
+			cssUrlList = append(cssUrlList, getCssUrl(*cssName))
+		}
 
-	header := ""
-	if *htmlFile == "" || *includeCSS {
-		header = `<meta http-equiv="content-Type" content="text/html; charset=UTF-8" />`
-		for _, f := range cssFileList {
-			header += renderCssPath(f)
+		if *outputPath == "" {
+			*outputPath = ReplaceExt(path.Base(*markdownPath), "png")
 		}
-		for _, u := range cssUrlList {
-			header += renderCssUrl(u)
+
+		//prepare static files
+		//go staticServer(*staticPath)
+
+		header := ""
+		if *htmlFile == "" || *includeCSS {
+			header = `<meta http-equiv="content-Type" content="text/html; charset=UTF-8" />`
+			for _, f := range cssFileList {
+				header += renderCssPath(f)
+			}
+			for _, u := range cssUrlList {
+				header += renderCssUrl(u)
+			}
 		}
-	}
 
-	html := ""
-	if *htmlFile == "" {
-		md := ReadFile(*markdownPath)
-		html = fmt.Sprintf("%v\n\n<div class='main container content article'>\n%v\n</div>", header, markdown2html(md))
-	} else {
-		if *includeCSS {
-			html = header + ReadFile(*htmlFile)
+		if *htmlFile == "" {
+			md := ReadFile(*markdownPath)
+			html = fmt.Sprintf("%v\n\n<div class='main container content article'>\n%v\n</div>", header, markdown2html(md))
 		} else {
-			html = ReadFile(*htmlFile)
+			if *includeCSS {
+				html = header + ReadFile(*htmlFile)
+			} else {
+				html = ReadFile(*htmlFile)
+			}
 		}
 	}
 
@@ -82,6 +91,35 @@ func markdown2html(text string) string {
 	return md.RenderToString([]byte(text))
 }
 
+func GetHTMLFromURL(URL string) string {
+	u, err := url.Parse(URL)
+	if err != nil {
+		panic(err)
+	}
+	baseUrl := path.Dir(u.Path)
+	if baseUrl != "." && baseUrl != "" {
+		u.Path = baseUrl
+	}
+	base := u.String()
+
+	rv := ""
+	response, err := http.Get(URL)
+	if err != nil {
+		fmt.Printf("%s", err)
+		os.Exit(1)
+	} else {
+		defer response.Body.Close()
+		contents, err := ioutil.ReadAll(response.Body)
+		if err != nil {
+			fmt.Printf("%s", err)
+			os.Exit(1)
+		}
+		rv += fmt.Sprintf("<head><base href=\"%v\"></head>", base)
+		rv += string(contents)
+	}
+	return rv
+}
+
 func (r *ImageRender) generateImage(html, format, output string, width, quality int) []byte {
 	c := ImageOptions{
 		BinaryPath: *r.BinaryPath,

diff --git a/wkhtmltoimage.go b/wkhtmltoimage.go
@@ -101,6 +101,8 @@ func buildParams(options *ImageOptions) ([]string, error) {
 	// might want to add --javascript-delay too?
 	a = append(a, "-q")
 	a = append(a, "--disable-plugins")
+	a = append(a, "--javascript-delay")
+	a = append(a, "2000")
 
 	// support unicode
 	a = append(a, "--encoding")
-Original file line number
+Diff line change
@@ Expand Up / @@ -2,9 +2,9 @@ package main @@
     import (
     	"io/ioutil"
+    	"log"
     	"os"
     	"path"
-    	"log"
     )
     func ExistFile(filename string) bool {
@@ Expand Down @@