From e47bd659865bb82b02fa992476e487eabedbf768 Mon Sep 17 00:00:00 2001
From: Sean Malloy <spinelli85@gmail.com>
Date: Sat, 23 Dec 2017 00:13:17 -0600
Subject: [PATCH] Add vagrant post-processor support for Google

Add the ability to create Google vagrant boxes
using the vagrant post-processor. The Google
plugin for vagrant is linked below.

https://github.com/mitchellh/vagrant-google
---
 post-processor/vagrant/google.go              | 42 +++++++++++++++++++
 post-processor/vagrant/post-processor.go      |  3 ++
 .../docs/post-processors/vagrant.html.md      |  5 ++-
 3 files changed, 48 insertions(+), 2 deletions(-)
 create mode 100644 post-processor/vagrant/google.go

diff --git a/post-processor/vagrant/google.go b/post-processor/vagrant/google.go
new file mode 100644
index 000000000..a4d9dbf48
--- /dev/null
+++ b/post-processor/vagrant/google.go
@@ -0,0 +1,42 @@
+package vagrant
+
+import (
+	"bytes"
+	"text/template"
+
+	"github.com/hashicorp/packer/packer"
+)
+
+type googleVagrantfileTemplate struct {
+	Image string ""
+}
+
+type GoogleProvider struct{}
+
+func (p *GoogleProvider) KeepInputArtifact() bool {
+	return true
+}
+
+func (p *GoogleProvider) Process(ui packer.Ui, artifact packer.Artifact, dir string) (vagrantfile string, metadata map[string]interface{}, err error) {
+	// Create the metadata
+	metadata = map[string]interface{}{"provider": "google"}
+
+	// Build up the template data to build our Vagrantfile
+	tplData := &googleVagrantfileTemplate{}
+	tplData.Image = artifact.Id()
+
+	// Build up the Vagrantfile
+	var contents bytes.Buffer
+	t := template.Must(template.New("vf").Parse(defaultGoogleVagrantfile))
+	err = t.Execute(&contents, tplData)
+	vagrantfile = contents.String()
+	return
+}
+
+var defaultGoogleVagrantfile = `
+Vagrant.configure("2") do |config|
+  config.vm.provider :google do |google|
+    google.image = "{{ .Image }}"
+  end
+end
+`
diff --git a/post-processor/vagrant/post-processor.go b/post-processor/vagrant/post-processor.go
index aa65b2292..310270809 100644
--- a/post-processor/vagrant/post-processor.go
+++ b/post-processor/vagrant/post-processor.go
@@ -25,6 +25,7 @@ var builtins = map[string]string{
 	"mitchellh.vmware":          "vmware",
 	"mitchellh.vmware-esx":      "vmware",
 	"pearkes.digitalocean":      "digitalocean",
+	"packer.googlecompute":      "google",
 	"packer.parallels":          "parallels",
 	"MSOpenTech.hyperv":         "hyperv",
 	"transcend.qemu":            "libvirt",
@@ -232,6 +233,8 @@ func providerForName(name string) Provider {
 		return new(HypervProvider)
 	case "libvirt":
 		return new(LibVirtProvider)
+	case "google":
+		return new(GoogleProvider)
 	default:
 		return nil
 	}
diff --git a/website/source/docs/post-processors/vagrant.html.md b/website/source/docs/post-processors/vagrant.html.md
index 5f0a9bc50..8feabbbbf 100644
--- a/website/source/docs/post-processors/vagrant.html.md
+++ b/website/source/docs/post-processors/vagrant.html.md
@@ -32,6 +32,7 @@ providers.
 
 -   AWS
 -   DigitalOcean
+-   Google
 -   Hyper-V
 -   Parallels
 -   QEMU
@@ -100,8 +101,8 @@ Specify overrides within the `override` configuration by provider name:
 In the example above, the compression level will be set to 1 except for VMware,
 where it will be set to 0.
 
-The available provider names are: `aws`, `digitalocean`, `virtualbox`, `vmware`,
-and `parallels`.
+The available provider names are: `aws`, `digitalocean`, `google`, `virtualbox`,
+`vmware`, and `parallels`.
 
 ## Input Artifacts