Merge pull request #44 from jelmersnoeck/contribution-specific-interfaces

Contribution packages specific interfaces.

1 files changed, 13 insertions, 2 deletions

diff --git a/contrib/httpimg/httpimg.go b/contrib/httpimg/httpimg.go
index 10afcc8..7cbf281 100644
--- a/contrib/httpimg/httpimg.go
+++ b/contrib/httpimg/httpimg.go
@@ -1,14 +1,25 @@
 package httpimg
 
 import (
-	"github.com/jung-kurt/gofpdf"
+	"io"
 	"net/http"
+
+	"github.com/jung-kurt/gofpdf"
 )
 
+// httpimgPdf is a partial interface that only implements the functions we need
+// from the PDF generator to put the HTTP images on the PDF.
+type httpimgPdf interface {
+	GetImageInfo(imageStr string) *gofpdf.ImageInfoType
+	ImageTypeFromMime(mimeStr string) string
+	RegisterImageReader(imgName, tp string, r io.Reader) *gofpdf.ImageInfoType
+	SetError(err error)
+}
+
 // Register registers a HTTP image. Downloading the image from the provided URL
 // and adding it to the PDF but not adding it to the page. Use Image() with the
 // same URL to add the image to the page.
-func Register(f *gofpdf.Fpdf, urlStr, tp string) (info *gofpdf.ImageInfoType) {
+func Register(f httpimgPdf, urlStr, tp string) (info *gofpdf.ImageInfoType) {
 	info = f.GetImageInfo(urlStr)
 
 	if info != nil {