Merge branch 'jelmersnoeck-contribution-package'

1 files changed, 32 insertions, 0 deletions

diff --git a/contrib/httpimg/httpimg.go b/contrib/httpimg/httpimg.go
new file mode 100644
index 0000000..10afcc8
--- /dev/null
+++ b/contrib/httpimg/httpimg.go
@@ -0,0 +1,32 @@
+package httpimg
+
+import (
+	"github.com/jung-kurt/gofpdf"
+	"net/http"
+)
+
+// 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) {
+	info = f.GetImageInfo(urlStr)
+
+	if info != nil {
+		return
+	}
+
+	resp, err := http.Get(urlStr)
+
+	if err != nil {
+		f.SetError(err)
+		return
+	}
+
+	defer resp.Body.Close()
+
+	if tp == "" {
+		tp = f.ImageTypeFromMime(resp.Header["Content-Type"][0])
+	}
+
+	return f.RegisterImageReader(urlStr, tp, resp.Body)
+}