From 61fec06f5194c61bf4b4f346c55fb4722cd72efe Mon Sep 17 00:00:00 2001
From: Mike Tougeron <github@e.tougeron.com>
Date: Mon, 19 Dec 2016 18:39:38 -0800
Subject: [PATCH] Add the archive file's MD5 to the output

---
 .../archive/data_source_archive_file.go       | 20 +++++++++++++++----
 .../docs/providers/archive/d/archive_file.md  |  2 ++
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/builtin/providers/archive/data_source_archive_file.go b/builtin/providers/archive/data_source_archive_file.go
index 77bd49d0e..d99b52f56 100644
--- a/builtin/providers/archive/data_source_archive_file.go
+++ b/builtin/providers/archive/data_source_archive_file.go
@@ -1,6 +1,7 @@
 package archive
 
 import (
+	"crypto/md5"
 	"crypto/sha1"
 	"crypto/sha256"
 	"encoding/base64"
@@ -68,6 +69,12 @@ func dataSourceFile() *schema.Resource {
 				ForceNew:    true,
 				Description: "Base64 Encoded SHA256 checksum of output file",
 			},
+			"output_md5": &schema.Schema{
+				Type:        schema.TypeString,
+				Computed:    true,
+				ForceNew:    true,
+				Description: "MD5 of output file",
+			},
 		},
 	}
 }
@@ -94,13 +101,14 @@ func dataSourceFileRead(d *schema.ResourceData, meta interface{}) error {
 		return err
 	}
 
-	sha1, base64sha256, err := genFileShas(outputPath)
+	sha1, base64sha256, md5, err := genFileShas(outputPath)
 	if err != nil {
 
 		return fmt.Errorf("could not generate file checksum sha256: %s", err)
 	}
 	d.Set("output_sha", sha1)
 	d.Set("output_base64sha256", base64sha256)
+	d.Set("output_md5", md5)
 
 	d.Set("output_size", fi.Size())
 	d.SetId(d.Get("output_sha").(string))
@@ -136,10 +144,10 @@ func archive(d *schema.ResourceData) error {
 	return nil
 }
 
-func genFileShas(filename string) (string, string, error) {
+func genFileShas(filename string) (string, string, string, error) {
 	data, err := ioutil.ReadFile(filename)
 	if err != nil {
-		return "", "", fmt.Errorf("could not compute file '%s' checksum: %s", filename, err)
+		return "", "", "", fmt.Errorf("could not compute file '%s' checksum: %s", filename, err)
 	}
 	h := sha1.New()
 	h.Write([]byte(data))
@@ -150,5 +158,9 @@ func genFileShas(filename string) (string, string, error) {
 	shaSum := h256.Sum(nil)
 	sha256base64 := base64.StdEncoding.EncodeToString(shaSum[:])
 
-	return sha1, sha256base64, nil
+	md5 := md5.New()
+	md5.Write([]byte(data))
+	md5Sum := hex.EncodeToString(md5.Sum(nil))
+
+	return sha1, sha256base64, md5Sum, nil
 }
diff --git a/website/source/docs/providers/archive/d/archive_file.md b/website/source/docs/providers/archive/d/archive_file.md
index 73d101ab2..2f8c8292c 100644
--- a/website/source/docs/providers/archive/d/archive_file.md
+++ b/website/source/docs/providers/archive/d/archive_file.md
@@ -48,3 +48,5 @@ The following attributes are exported:
 * `output_sha` - The SHA1 checksum of output archive file.
 
 * `output_base64sha256` - The base64-encoded SHA256 checksum of output archive file.
+
+* `output_md5` - The MD5 checksum of output archive file.