refactor(ec2): EC2 API cleanups

NOTICE

@@ -1,2 +1,3 @@
 AWS Cloud Development Kit (AWS CDK)
 Copyright 2018-2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+

packages/@aws-cdk/aws-autoscaling/lib/auto-scaling-group.ts

@@ -175,7 +175,17 @@ export interface AutoScalingGroupProps extends CommonAutoScalingGroupProps {
   /**
    * AMI to launch
    */
-  readonly machineImage: ec2.IMachineImageSource;
+  readonly machineImage: ec2.IMachineImage;
+
+  /**
+   * Specific UserData to use
+   *
+   * The UserData may still be mutated after creation.
+   *
+   * @default - A UserData object appropriate for the MachineImage's
+   * Operating System is created.
+   */
+  readonly userData?: ec2.UserData;
 
   /**
    * An IAM role to associate with the instance profile assigned to this Auto Scaling Group.
@@ -354,7 +364,11 @@ export class AutoScalingGroup extends AutoScalingGroupBase implements
    */
   public readonly autoScalingGroupArn: string;
 
-  private readonly userDataLines = new Array<string>();
+  /**
+   * UserData for the instances
+   */
+  public readonly userData: ec2.UserData;
+
   private readonly autoScalingGroup: CfnAutoScalingGroup;
   private readonly securityGroup: ec2.ISecurityGroup;
   private readonly securityGroups: ec2.ISecurityGroup[] = [];
@@ -381,12 +395,13 @@ export class AutoScalingGroup extends AutoScalingGroupBase implements
     });
 
     // use delayed evaluation
-    const machineImage = props.machineImage.getImage(this);
-    const userDataToken = Lazy.stringValue({ produce: () => Fn.base64((machineImage.os.createUserData(this.userDataLines))) });
+    const imageConfig = props.machineImage.getImage(this);
+    this.userData = props.userData || imageConfig.userData || ec2.UserData.forOperatingSystem(imageConfig.osType);
+    const userDataToken = Lazy.stringValue({ produce: () => Fn.base64(this.userData.render()) });
     const securityGroupsToken = Lazy.listValue({ produce: () => this.securityGroups.map(sg => sg.securityGroupId) });
 
     const launchConfig = new CfnLaunchConfiguration(this, 'LaunchConfig', {
-      imageId: machineImage.imageId,
+      imageId: imageConfig.imageId,
       keyName: props.keyName,
       instanceType: props.instanceType.toString(),
       securityGroups: securityGroupsToken,
@@ -409,7 +424,7 @@ export class AutoScalingGroup extends AutoScalingGroupBase implements
       throw new Error(`Should have minCapacity (${minCapacity}) <= desiredCapacity (${desiredCapacity}) <= maxCapacity (${maxCapacity})`);
     }
 
-    const { subnetIds } = props.vpc.selectSubnets(props.vpcSubnets);
+    const { subnetIds, hasPublic } = props.vpc.selectSubnets(props.vpcSubnets);
     const asgProps: CfnAutoScalingGroupProps = {
       cooldown: props.cooldown !== undefined ? props.cooldown.toSeconds().toString() : undefined,
       minSize: minCapacity.toString(),
@@ -432,12 +447,12 @@ export class AutoScalingGroup extends AutoScalingGroupBase implements
       vpcZoneIdentifier: subnetIds
     };
 
-    if (!props.vpc.isPublicSubnets(subnetIds) && props.associatePublicIpAddress) {
+    if (!hasPublic && props.associatePublicIpAddress) {
       throw new Error("To set 'associatePublicIpAddress: true' you must select Public subnets (vpcSubnets: { subnetType: SubnetType.Public })");
     }
 
     this.autoScalingGroup = new CfnAutoScalingGroup(this, 'ASG', asgProps);
-    this.osType = machineImage.os.type;
+    this.osType = imageConfig.osType;
     this.autoScalingGroupName = this.autoScalingGroup.ref;
     this.autoScalingGroupArn = Stack.of(this).formatArn({
       service: 'autoscaling',
@@ -487,8 +502,8 @@ export class AutoScalingGroup extends AutoScalingGroupBase implements
    * Add command to the startup script of fleet instances.
    * The command must be in the scripting language supported by the fleet's OS (i.e. Linux/Windows).
    */
-  public addUserData(...scriptLines: string[]) {
-    scriptLines.forEach(scriptLine => this.userDataLines.push(scriptLine));
+  public addUserData(...commands: string[]) {
+    this.userData.addCommands(...commands);
   }
 
   /**

packages/@aws-cdk/aws-autoscaling/test/integ.amazonlinux2.expected.json

@@ -428,7 +428,7 @@
           }
         ],
         "UserData": {
-          "Fn::Base64": "#!/bin/bash\n"
+          "Fn::Base64": "#!/bin/bash"
         }
       },
       "DependsOn": [
@@ -467,4 +467,4 @@
       }
     }
   }
-}
+}

packages/@aws-cdk/aws-autoscaling/test/integ.amazonlinux2.ts

@@ -12,7 +12,7 @@ const vpc = new ec2.Vpc(stack, 'VPC', {
 
 new autoscaling.AutoScalingGroup(stack, 'Fleet', {
   vpc,
-  instanceType: new ec2.InstanceTypePair(ec2.InstanceClass.BURSTABLE2, ec2.InstanceSize.MICRO),
+  instanceType: ec2.InstanceType.of(ec2.InstanceClass.BURSTABLE2, ec2.InstanceSize.MICRO),
   machineImage: new ec2.AmazonLinuxImage({ generation: ec2.AmazonLinuxGeneration.AMAZON_LINUX_2 }),
 });
 

packages/@aws-cdk/aws-autoscaling/test/integ.asg-w-classic-loadbalancer.expected.json

@@ -602,7 +602,7 @@
           }
         ],
         "UserData": {
-          "Fn::Base64": "#!/bin/bash\n"
+          "Fn::Base64": "#!/bin/bash"
         }
       },
       "DependsOn": [
@@ -735,4 +735,4 @@
       ]
     }
   }
-}
+}

packages/@aws-cdk/aws-autoscaling/test/integ.asg-w-classic-loadbalancer.ts

@@ -13,7 +13,7 @@ const vpc = new ec2.Vpc(stack, 'VPC', {
 
 const asg = new autoscaling.AutoScalingGroup(stack, 'Fleet', {
   vpc,
-  instanceType: new ec2.InstanceTypePair(ec2.InstanceClass.BURSTABLE2, ec2.InstanceSize.MICRO),
+  instanceType: ec2.InstanceType.of(ec2.InstanceClass.BURSTABLE2, ec2.InstanceSize.MICRO),
   machineImage: new ec2.AmazonLinuxImage(),
 });
 

packages/@aws-cdk/aws-autoscaling/test/integ.asg-w-elbv2.expected.json

@@ -449,7 +449,7 @@
           }
         ],
         "UserData": {
-          "Fn::Base64": "#!/bin/bash\n"
+          "Fn::Base64": "#!/bin/bash"
         }
       },
       "DependsOn": [
@@ -667,4 +667,4 @@
       }
     }
   }
-}
+}

packages/@aws-cdk/aws-autoscaling/test/integ.asg-w-elbv2.ts

@@ -13,7 +13,7 @@ const vpc = new ec2.Vpc(stack, 'VPC', {
 
 const asg = new autoscaling.AutoScalingGroup(stack, 'Fleet', {
   vpc,
-  instanceType: new ec2.InstanceTypePair(ec2.InstanceClass.BURSTABLE2, ec2.InstanceSize.MICRO),
+  instanceType: ec2.InstanceType.of(ec2.InstanceClass.BURSTABLE2, ec2.InstanceSize.MICRO),
   machineImage: new ec2.AmazonLinuxImage(),
 });
 

packages/@aws-cdk/aws-autoscaling/test/integ.custom-scaling.expected.json

@@ -428,7 +428,7 @@
           }
         ],
         "UserData": {
-          "Fn::Base64": "#!/bin/bash\n"
+          "Fn::Base64": "#!/bin/bash"
         }
       },
       "DependsOn": [
@@ -502,4 +502,4 @@
       }
     }
   }
-}
+}

packages/@aws-cdk/aws-autoscaling/test/integ.custom-scaling.ts

@@ -12,7 +12,7 @@ const vpc = new ec2.Vpc(stack, 'VPC', {
 
 const asg = new autoscaling.AutoScalingGroup(stack, 'Fleet', {
   vpc,
-  instanceType: new ec2.InstanceTypePair(ec2.InstanceClass.BURSTABLE2, ec2.InstanceSize.MICRO),
+  instanceType: ec2.InstanceType.of(ec2.InstanceClass.BURSTABLE2, ec2.InstanceSize.MICRO),
   machineImage: new ec2.AmazonLinuxImage({ generation: ec2.AmazonLinuxGeneration.AMAZON_LINUX_2 }),
 });
 

packages/@aws-cdk/aws-autoscaling/test/integ.external-role.expected.json

@@ -581,7 +581,7 @@
           }
         ],
         "UserData": {
-          "Fn::Base64": "#!/bin/bash\n"
+          "Fn::Base64": "#!/bin/bash"
         }
       },
       "DependsOn": [
@@ -623,4 +623,4 @@
       }
     }
   }
-}
+}

packages/@aws-cdk/aws-autoscaling/test/integ.external-role.ts

@@ -13,7 +13,7 @@ class TestStack extends cdk.Stack {
     });
 
     new asg.AutoScalingGroup(this, 'ASG', {
-      instanceType: new ec2.InstanceTypePair(ec2.InstanceClass.T2, ec2.InstanceSize.MICRO),
+      instanceType: ec2.InstanceType.of(ec2.InstanceClass.T2, ec2.InstanceSize.MICRO),
       vpc,
       machineImage: new ec2.AmazonLinuxImage(),
       role

packages/@aws-cdk/aws-autoscaling/test/integ.spot-instances.expected.json

@@ -429,7 +429,7 @@
         ],
         "SpotPrice": "0.20",
         "UserData": {
-          "Fn::Base64": "#!/bin/bash\n"
+          "Fn::Base64": "#!/bin/bash"
         }
       },
       "DependsOn": [
@@ -468,4 +468,4 @@
       }
     }
   }
-}
+}

packages/@aws-cdk/aws-autoscaling/test/integ.spot-instances.ts

@@ -12,7 +12,7 @@ const vpc = new ec2.Vpc(stack, 'VPC', {
 
 new autoscaling.AutoScalingGroup(stack, 'Fleet', {
   vpc,
-  instanceType: new ec2.InstanceTypePair(ec2.InstanceClass.BURSTABLE2, ec2.InstanceSize.MICRO),
+  instanceType: ec2.InstanceType.of(ec2.InstanceClass.BURSTABLE2, ec2.InstanceSize.MICRO),
   machineImage: new ec2.AmazonLinuxImage({ generation: ec2.AmazonLinuxGeneration.AMAZON_LINUX_2 }),
   spotPrice: '0.20'
 });