add missing model

converter/converter.go

@@ -127,12 +127,12 @@ func (c *Converter) Convert(files ...models.File) error {
 }
 
 // Convert converts all given input files
-func (c *Converter) ConvertHtml(html []byte, outputPath string) error {
-
-	if len(html) == 0 {
+func (c *Converter) ConvertHtmls(htmls ...models.Html) error {
+	for _, h := range htmls {
+		if len(h.Html) == 0 {
 			return fmt.Errorf("no .html input provided")
-	} else if outputPath == "" || filepath.Ext(outputPath) != ".pdf" {
-		return fmt.Errorf("no .pdf output file path provided: %s", outputPath)
+		} else if h.Output == "" || filepath.Ext(h.Output) != ".pdf" {
+			return fmt.Errorf("no .pdf output file path provided: %s", h.Output)
 		}
 
 		taskCtx, taskCancel := chromedp.NewContext(c.browserCtx)
@@ -149,7 +149,7 @@ func (c *Converter) ConvertHtml(html []byte, outputPath string) error {
 					return err
 				}
 
-			return page.SetDocumentContent(frameTree.Frame.ID, string(html)).Do(ctx)
+				return page.SetDocumentContent(frameTree.Frame.ID, string(h.Html)).Do(ctx)
 			}),
 			chromedp.WaitReady("body", chromedp.ByQuery),
 			chromedp.ActionFunc(func(ctx context.Context) error {
@@ -173,9 +173,10 @@ func (c *Converter) ConvertHtml(html []byte, outputPath string) error {
 		}
 
 		// Save PDF to file
-	if err := os.WriteFile(outputPath, pdfData, 0644); err != nil {
+		if err := os.WriteFile(h.Output, pdfData, 0644); err != nil {
 			return err
 		}
+	}
 	return nil
 }
 

html2pdf_test.go

@@ -94,7 +94,7 @@ func TestConvertHtml(t *testing.T) {
 	}
 	defer c.Close()
 
-	err = c.ConvertHtml(html, outputPath)
+	err = c.ConvertHtmls(models.Html{Html: html, Output: outputPath})
 	if err != nil {
 		t.Fatalf("ConvertHtml failed: %v", err)
 	}

models/html.go

@@ -0,0 +1,9 @@
+package models
+
+type Htmls []Html
+
+// input model for converter
+type Html struct {
+	Html   []byte
+	Output string
+}