From 9eda3fa327351482c4e27aa0a45371a01d1c8739 Mon Sep 17 00:00:00 2001
From: Jeremy Carroll <jeremy@pinterest.com>
Date: Wed, 7 Aug 2013 14:33:20 -0700
Subject: [PATCH] Testing adding user-data support to Packer

---
 builder/amazon/common/run_config.go               | 1 +
 builder/amazon/common/step_run_source_instance.go | 2 ++
 builder/amazon/ebs/builder.go                     | 1 +
 builder/amazon/instance/builder.go                | 1 +
 4 files changed, 5 insertions(+)

diff --git a/builder/amazon/common/run_config.go b/builder/amazon/common/run_config.go
index 805a34c4b..80093fefe 100644
--- a/builder/amazon/common/run_config.go
+++ b/builder/amazon/common/run_config.go
@@ -12,6 +12,7 @@ type RunConfig struct {
 	SourceAmi          string `mapstructure:"source_ami"`
 	IamInstanceProfile string `mapstructure:"iam_instance_profile"`
 	InstanceType       string `mapstructure:"instance_type"`
+	UserData           string `mapstructure:"user_data"`
 	RawSSHTimeout      string `mapstructure:"ssh_timeout"`
 	SSHUsername        string `mapstructure:"ssh_username"`
 	SSHPort            int    `mapstructure:"ssh_port"`
diff --git a/builder/amazon/common/step_run_source_instance.go b/builder/amazon/common/step_run_source_instance.go
index 58f6f0f07..4767f88c4 100644
--- a/builder/amazon/common/step_run_source_instance.go
+++ b/builder/amazon/common/step_run_source_instance.go
@@ -11,6 +11,7 @@ import (
 type StepRunSourceInstance struct {
 	ExpectedRootDevice string
 	InstanceType       string
+	UserData           string
 	SourceAMI          string
 	IamInstanceProfile string
 	SubnetId           string
@@ -28,6 +29,7 @@ func (s *StepRunSourceInstance) Run(state map[string]interface{}) multistep.Step
 		KeyName:            keyName,
 		ImageId:            s.SourceAMI,
 		InstanceType:       s.InstanceType,
+		UserData:           s.UserData,
 		MinCount:           0,
 		MaxCount:           0,
 		SecurityGroups:     []ec2.SecurityGroup{ec2.SecurityGroup{Id: securityGroupId}},
diff --git a/builder/amazon/ebs/builder.go b/builder/amazon/ebs/builder.go
index cd5ba1452..04478ba3c 100644
--- a/builder/amazon/ebs/builder.go
+++ b/builder/amazon/ebs/builder.go
@@ -99,6 +99,7 @@ func (b *Builder) Run(ui packer.Ui, hook packer.Hook, cache packer.Cache) (packe
 		&awscommon.StepRunSourceInstance{
 			ExpectedRootDevice: "ebs",
 			InstanceType:       b.config.InstanceType,
+			UserData:           b.config.UserData,
 			SourceAMI:          b.config.SourceAmi,
 			IamInstanceProfile: b.config.IamInstanceProfile,
 			SubnetId:           b.config.SubnetId,
diff --git a/builder/amazon/instance/builder.go b/builder/amazon/instance/builder.go
index 3517553c3..5257eb1ff 100644
--- a/builder/amazon/instance/builder.go
+++ b/builder/amazon/instance/builder.go
@@ -164,6 +164,7 @@ func (b *Builder) Run(ui packer.Ui, hook packer.Hook, cache packer.Cache) (packe
 		&awscommon.StepRunSourceInstance{
 			ExpectedRootDevice: "instance-store",
 			InstanceType:       b.config.InstanceType,
+			UserData:           b.config.UserData,
 			SourceAMI:          b.config.SourceAmi,
 			SubnetId:           b.config.SubnetId,
 		},