update elastic_whenever for FARGATE launch type

lib/elastic_whenever/option.rb

@@ -9,6 +9,10 @@ class Option
     attr_reader :identifier
     attr_reader :mode
     attr_reader :variables
+    attr_reader :assign_public_ip
+    attr_reader :launch_type
+    attr_reader :security_groups
+    attr_reader :subnets
     attr_reader :schedule_file
 
     class InvalidOptionException < StandardError; end
@@ -17,6 +21,10 @@ def initialize(args)
       @identifier = nil
       @mode = DRYRUN_MODE
       @variables = []
+      @assign_public_ip = 'DISABLED'
+      @launch_type = 'EC2'
+      @security_groups = nil
+      @subnets = nil
       @schedule_file = 'config/schedule.rb'
       @profile = nil
       @access_key = nil
@@ -47,6 +55,18 @@ def initialize(args)
             @variables << { key: key, value: value }
           end
         end
+        opts.on('--assign_public_ip', 'Assign a public IP.') do
+          @assign_public_ip = 'ENABLED'
+        end
+        opts.on('--launch_type launch_type', 'Launch type. Defualt: EC2') do |launch_type|
+          @launch_type = launch_type
+        end
+        opts.on('--security_groups groups', "Example: --security_groups 'sg-2c503655,sg-72f0cb0a'") do |groups|
+          @security_groups = groups
+        end
+        opts.on('--subnets subnets', "Example: --subnets 'subnet-4973d63f,subnet-45827d1d'") do |subnets|
+          @subnets = subnets
+        end
         opts.on('-f', '--file schedule_file', 'Default: config/schedule.rb') do |file|
           @schedule_file = file
         end

lib/elastic_whenever/task/target.rb

@@ -5,6 +5,10 @@ class Target
       attr_reader :definition
       attr_reader :container
       attr_reader :commands
+      attr_reader :assign_public_ip
+      attr_reader :launch_type
+      attr_reader :security_groups
+      attr_reader :subnets
 
       class InvalidContainerException < StandardError; end
 
@@ -37,25 +41,55 @@ def initialize(option, cluster:, definition:, container:, commands:, rule:, role
         @commands = commands
         @rule = rule
         @role = role
+        @assign_public_ip = option.assign_public_ip
+        @launch_type = option.launch_type
+        @security_groups = option.security_groups
+        @subnets = option.subnets
         @client = Aws::CloudWatchEvents::Client.new(option.aws_config)
       end
 
       def create
-        client.put_targets(
-          rule: rule.name,
-          targets: [
-            {
-              id: Digest::SHA1.hexdigest(commands.join("-")),
-              arn: cluster.arn,
-              input: input_json(container, commands),
-              role_arn: role.arn,
-              ecs_parameters: {
-                task_definition_arn: definition.arn,
-                task_count: 1,
+        if launch_type == 'FARGATE'
+          client.put_targets(
+            rule: rule.name,
+            targets: [
+              {
+                id: Digest::SHA1.hexdigest(commands.join("-")),
+                arn: cluster.arn,
+                input: input_json(container, commands),
+                role_arn: role.arn,
+                ecs_parameters: {
+                  launch_type: launch_type,
+                  task_definition_arn: definition.arn,
+                  task_count: 1,
+                  network_configuration: {
+                    awsvpc_configuration: {
+                      subnets: [ subnets ],
+                      security_groups: [ security_groups ],
+                      assign_public_ip: assign_public_ip,
+                    }
+                  }
+                }
               }
-            }
-          ]
-        )
+            ]
+          )
+        else
+          client.put_targets(
+            rule: rule.name,
+            targets: [
+              {
+                id: Digest::SHA1.hexdigest(commands.join("-")),
+                arn: cluster.arn,
+                input: input_json(container, commands),
+                role_arn: role.arn,
+                ecs_parameters: {
+                  task_definition_arn: definition.arn,
+                  task_count: 1,
+                }
+              }
+            ]
+          )
+        end
       end
 
       private

lib/elastic_whenever/version.rb

@@ -1,3 +1,3 @@
 module ElasticWhenever
-  VERSION = "0.3.2"
+  VERSION = "0.3.3"
 end