flatten results

bjesus · Sep 12, 2024 · 30fddcc · 30fddcc
1 parent ac6642f
commit 30fddcc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/cmd/pipet/main.go b/cmd/pipet/main.go
@@ -125,6 +125,7 @@ func runPipet(c *cli.Context, specFile string) error {
 		}
 
 		log.Println("Generating output")
+
 		if jsonOutput {
 			newValue = outputs.OutputJSON(pipet)
 		} else if templateFile != "" {

diff --git a/parsers/html.go b/parsers/html.go
@@ -6,7 +6,7 @@ import (
 	"github.com/PuerkitoBio/goquery"
 )
 
-func ParseHTMLQueries(htmlData []byte, queries []string, nextPage string) (interface{}, string, error) {
+func ParseHTMLQueries(htmlData []byte, queries []string, nextPage string) ([]interface{}, string, error) {
 
 	result := []interface{}{}
 
@@ -43,7 +43,13 @@ func ParseHTMLQueries(htmlData []byte, queries []string, nextPage string) (inter
 					html = "<table>" + html + "</table>"
 				}
 				value2, _, _ := ParseHTMLQueries([]byte(html), lines, "")
-				subresult = append(subresult, value2)
+
+				if len(value2) == 1 {
+					subresult = append(subresult, value2[0])
+				} else {
+					subresult = append(subresult, value2)
+				}
+
 			})
 
 			result = append(result, subresult)