add flag to export only inner HTML of output file

gardenerik · gardenerik · commit 35e2a71253a4 · 2023-02-14T16:44:34.000+01:00
diff --git a/webtex_render/config.go b/webtex_render/config.go
@@ -12,4 +12,5 @@ type Config struct {
 	InputURL          string
 	OutputURL         string
 	Template          *template.Template
+	OnlyInnerHTML     bool
 }
diff --git a/webtex_render/html.go b/webtex_render/html.go
@@ -31,7 +31,13 @@ func parseHtml(config Config) error {
 		s.SetAttr("src", sha)
 	})
 
-	html, err := doc.Html()
+	var toExport *goquery.Selection
+	if config.OnlyInnerHTML {
+		toExport = doc.Find("body")
+	} else {
+		toExport = doc.Selection
+	}
+	html, err := toExport.Html()
 	if err != nil {
 		log.Println("failed generating output html:", err)
 		return err
diff --git a/webtex_render/main.go b/webtex_render/main.go
@@ -15,6 +15,7 @@ func main() {
 	var inputUrl = flag.String("inurl", "eqn://", "input webtex URL prefix")
 	var outputUrl = flag.String("outurl", "equations", "image src root")
 	var templateFilename = flag.String("template", "", "TeX template file")
+	var onlyInner = flag.Bool("innerhtml", false, "export only inner HTML of the result (without <html><body> tags)")
 	flag.Parse()
 
 	var err error
@@ -41,6 +42,7 @@ func main() {
 	config.EquationDirectory = *equationDirectory
 	config.InputURL = *inputUrl
 	config.OutputURL = *outputUrl
+	config.OnlyInnerHTML = *onlyInner
 
 	if *templateFilename == "" {
 		log.Println("please provide a template file.")
