Merge pull request #3 from circa10a/convert_images

#minor: support converting images to pdfs

Author: circa10a

Makefile

@@ -3,7 +3,7 @@ HOSTNAME=github.com
 NAMESPACE=circa10a
 NAME=mailform
 BINARY=terraform-provider-${NAME}
-VERSION=0.2.0
+VERSION=0.4.0
 OS_ARCH=darwin_amd64
 
 default: install

README.md

@@ -103,11 +103,3 @@ If you wish to work on the provider, you'll first need [Go](http://www.golang.or
 To compile the provider, run `go install`. This will build the provider and put the provider binary in the `$GOPATH/bin` directory.
 
 To generate or update documentation, run `go generate`.
-
-In order to run the full suite of Acceptance tests, run `make testacc`.
-
-*Note:* Acceptance tests create real resources, and often cost money to run.
-
-```sh
-make testacc
-```

docs/resources/pdf.md

@@ -17,9 +17,13 @@ Render a PDF and write to a local file.
 
 ### Required
 
-- `content` (String) Content of PDF
 - `filename` (String) The path to the PDF file that will be created
+
+### Optional
+
+- `content` (String) Content of PDF
 - `header` (String) Header/title of PDF
+- `image_filename` (String) The image file to be converted to a PDF. Typically used for postcards
 
 ### Read-Only
 

examples/resources/mailform_pdf/pdf.tf

@@ -11,3 +11,8 @@ resource "mailform_pdf" "example" {
   content  = "Some resume contents"
   filename = "./test.pdf"
 }
+
+resource "mailform_pdf" "converted_image" {
+  image_filename = "./myimage.jpg"
+  filename       = "./myimage.pdf"
+}

internal/provider/resource_pdf.go

@@ -23,23 +23,39 @@ func resourcePDF() *schema.Resource {
 		DeleteContext: resourcePDFDelete,
 
 		Schema: map[string]*schema.Schema{
+			"filename": {
+				Description: "The path to the PDF file that will be created",
+				Type:        schema.TypeString,
+				Required:    true,
+				ForceNew:    true,
+			},
 			"header": {
 				Description: "Header/title of PDF",
 				Type:        schema.TypeString,
-				Required:    true,
+				Optional:    true,
 				ForceNew:    true,
+				ConflictsWith: []string{
+					"image_filename",
+				},
 			},
 			"content": {
 				Description: "Content of PDF",
 				Type:        schema.TypeString,
-				Required:    true,
+				Optional:    true,
 				ForceNew:    true,
+				ConflictsWith: []string{
+					"image_filename",
+				},
 			},
-			"filename": {
-				Description: "The path to the PDF file that will be created",
+			"image_filename": {
+				Description: "The image file to be converted to a PDF. Typically used for postcards",
 				Type:        schema.TypeString,
-				Required:    true,
+				Optional:    true,
 				ForceNew:    true,
+				ConflictsWith: []string{
+					"header",
+					"content",
+				},
 			},
 		},
 	}
@@ -48,30 +64,25 @@ func resourcePDF() *schema.Resource {
 func resourcePDFCreate(ctx context.Context, d *schema.ResourceData, meta any) diag.Diagnostics {
 	var diags diag.Diagnostics
 
-	pdf := gofpdf.New(gofpdf.OrientationPortrait, "mm", gofpdf.PageSizeLetter, "")
-
-	header := d.Get("header").(string)
-	content := d.Get("content").(string)
+	// Used for generating pdfs and also converting pdf's to images
 	filename := d.Get("filename").(string)
 
-	pdf.AddPage()
-	pdf.SetTitle(d.Get("header").(string), false)
-	pdf.SetFont("Arial", "B", 16)
-	// Calculate width of title and position
-	wd := pdf.GetStringWidth(header) + 6
-	pdf.SetX((210 - wd) / 2)
-	// Title
-	pdf.CellFormat(wd, 9, header, "", 1, "C", false, 0, "")
-	// Line break
-	pdf.Ln(10)
-	pdf.SetFont("Arial", "", 11)
-	pdf.SetAutoPageBreak(true, 2.00)
-	// Write ze content
-	pdf.Write(8, content)
-
-	err := pdf.OutputFileAndClose(filename)
-	if err != nil {
-		return diag.FromErr(err)
+	// If an image, convert to pdf
+	if imageFilename, ok := d.GetOk("image_filename"); ok {
+		err := convertImage(imageFilename.(string), filename)
+		if err != nil {
+			defer resourcePDFDelete(ctx, d, filename)
+			return diag.FromErr(err)
+		}
+	} else {
+		// Generate content if not image
+		header := d.Get("header").(string)
+		content := d.Get("content").(string)
+
+		err := renderPDF(header, content, filename)
+		if err != nil {
+			return diag.FromErr(err)
+		}
 	}
 
 	outputContent, err := ioutil.ReadFile(filename)
@@ -116,3 +127,38 @@ func resourcePDFDelete(ctx context.Context, d *schema.ResourceData, meta any) di
 	os.Remove(d.Get("filename").(string))
 	return nil
 }
+
+// renderPDF converts header + content to a pdf and writes to an output file
+func renderPDF(header string, content string, outputFilePath string) error {
+	pdf := gofpdf.New(gofpdf.OrientationPortrait, "mm", gofpdf.PageSizeLetter, "")
+	pdf.AddPage()
+	pdf.SetTitle(header, false)
+	pdf.SetFont("Arial", "B", 16)
+	// Calculate width of title and position
+	wd := pdf.GetStringWidth(header) + 6
+	pdf.SetX((210 - wd) / 2)
+	// Title
+	pdf.CellFormat(wd, 9, header, "", 1, "C", false, 0, "")
+	// Line break
+	pdf.Ln(10)
+	pdf.SetFont("Arial", "", 11)
+	pdf.SetAutoPageBreak(true, 2.00)
+	// Write ze content
+	pdf.Write(8, content)
+
+	return pdf.OutputFileAndClose(outputFilePath)
+}
+
+// convertImage converts input image path to pdf file
+func convertImage(inputFilePath, outputFilePath string) error {
+	pdf := gofpdf.New(gofpdf.OrientationPortrait, "mm", gofpdf.PageSizeLetter, "")
+	pdf.AddPage()
+	pdf.Image(inputFilePath, 0, 0, 240, 480, false, "", 0, "")
+
+	err := pdf.OutputFileAndClose(outputFilePath)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}