diff options
Diffstat (limited to 'contrib/httpimg/httpimg.go')
| -rw-r--r-- | contrib/httpimg/httpimg.go | 15 |
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 { |