From ef4591c4fc720265d44e936407901f67b69aee74 Mon Sep 17 00:00:00 2001
From: Jerome Petazzoni <jerome.petazzoni@gmail.com>
Date: Wed, 28 Feb 2018 13:45:08 -0600
Subject: [PATCH] Allow to override instance type (closes #39)

---
 prepare-vms/docker-compose.yml | 1 +
 prepare-vms/lib/commands.sh    | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/prepare-vms/docker-compose.yml b/prepare-vms/docker-compose.yml
index aa607ad8..95220c3a 100644
--- a/prepare-vms/docker-compose.yml
+++ b/prepare-vms/docker-compose.yml
@@ -15,5 +15,6 @@ services:
             AWS_ACCESS_KEY_ID: ${AWS_ACCESS_KEY_ID}
             AWS_SECRET_ACCESS_KEY: ${AWS_SECRET_ACCESS_KEY}
             AWS_DEFAULT_REGION: ${AWS_DEFAULT_REGION}
+            AWS_INSTANCE_TYPE: ${AWS_INSTANCE_TYPE}
             USER: ${USER}
         entrypoint: /root/prepare-vms/workshopctl
diff --git a/prepare-vms/lib/commands.sh b/prepare-vms/lib/commands.sh
index 624d384a..c5dd969e 100644
--- a/prepare-vms/lib/commands.sh
+++ b/prepare-vms/lib/commands.sh
@@ -280,7 +280,7 @@ _cmd_start() {
     result=$(aws ec2 run-instances \
         --key-name $AWS_KEY_NAME \
         --count $COUNT \
-        --instance-type t2.medium \
+        --instance-type ${AWS_INSTANCE_TYPE-t2.medium} \
         --client-token $TOKEN \
         --image-id $AMI)
     reservation_id=$(echo "$result" | head -1 | awk '{print $2}')