hako 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b7b7ad2ffbeb17c94a9e690bf278bb4019852b3e
-  data.tar.gz: 27024dcecc995a1acc893e3072029cd693fd8e0a
+  metadata.gz: 41e144225804bd958a9c8be85ad1f2998cc3a70b
+  data.tar.gz: 746df8720d4697f12ed9b7fd66d73699af39e8c1
 SHA512:
-  metadata.gz: 874b5c28b65dd52d728c83a441f23d057807d1f445fe9edc5a5b44200eeb9b8fbcfd9770568a86f2ecee30686dcc389e713a2baed9dfef4e7ade3c05633905f1
-  data.tar.gz: 3218b993bcc4d924f0e6aaaf72a74dc84e2d97f07babcea01295c25625e5fcd13bcb9fad70ea254d30518e42778a14afc17d5df95cec657ddbd7d49f5909748c
+  metadata.gz: 47a8418855c3fe25132a0033375a00628df9cb07385e6ee1ccb50c6d18c1e14ae0deb1f9284cf32d354568631f7fbfe67d84a1a12b95e6ce1823d43dd48e7d92
+  data.tar.gz: d4274e025153b4cebaac5e6b4d81ef05443afa9a497cccb54b77cf54a8612decf9be2dc5d17980dae692e08cfa112e4f640122f6cf739dded359005657ebfdde

data/.rubocop.yml

@@ -0,0 +1,40 @@
+inherit_from: .rubocop_todo.yml
+
+AllCops:
+  DisplayCopNames: true
+
+Style/GuardClause:
+  Enabled: false
+
+Style/HashSyntax:
+  Exclude:
+    - 'Rakefile'
+
+Style/IfUnlessModifier:
+  Enabled: false
+
+Style/Next:
+  Enabled: false
+
+Style/NumericLiterals:
+  Enabled: false
+
+Style/PercentLiteralDelimiters:
+  PreferredDelimiters:
+    '%w': '[]'
+    '%i': '[]'
+
+Style/RaiseArgs:
+  EnforcedStyle: compact
+
+Style/SignalException:
+  Enabled: false
+
+Style/TrailingCommaInArguments:
+  Enabled: false
+
+Style/TrailingCommaInLiteral:
+  Enabled: false
+
+Performance/RedundantBlockCall:
+  Enabled: false

data/.rubocop_todo.yml

@@ -0,0 +1,23 @@
+Metrics/AbcSize:
+  Enabled: false
+
+Metrics/ClassLength:
+  Enabled: false
+
+Metrics/CyclomaticComplexity:
+  Enabled: false
+
+Metrics/LineLength:
+  Enabled: false
+
+Metrics/MethodLength:
+  Enabled: false
+
+Metrics/PerceivedComplexity:
+  Enabled: false
+
+Metrics/BlockNesting:
+  Enabled: false
+
+Style/Documentation:
+  Enabled: false

data/.travis.yml

@@ -1,4 +1,11 @@
 language: ruby
+sudo: false
 rvm:
-  - 2.2.3
-before_install: gem install bundler -v 1.10.6
+  - 2.1
+  - 2.2
+  - ruby-head
+before_install:
+  - gem install bundler
+matrix:
+  allow_failures:
+    - rvm: ruby-head

data/README.md

@@ -1,4 +1,7 @@
 # Hako
+[![Gem Version](https://badge.fury.io/rb/hako.svg)](http://badge.fury.io/rb/hako)
+[![Build Status](https://travis-ci.org/eagletmt/hako.svg)](https://travis-ci.org/eagletmt/hako)
+
 Deploy Docker container.
 
 ## Status
@@ -36,7 +39,7 @@ I, [2015-10-02T12:56:12.262760 #8141]  INFO -- : Deployment isn't needed
 Load balancer:
   hako-hello-XXXXXXXXXX.ap-northeast-1.elb.amazonaws.com:80 -> front:80
 Deployments:
-  [PRIMARY] desired_count=2, pending_count=0, running_count=2
+  [PRIMARY] hello:30 desired_count=2, pending_count=0, running_count=2
 Tasks:
   [RUNNING]: i-XXXXXXXX (ecs-001)
   [RUNNING]: i-YYYYYYYY (ecs-002)

data/Rakefile

@@ -1,6 +1,8 @@
-require "bundler/gem_tasks"
-require "rspec/core/rake_task"
+require 'bundler/gem_tasks'
+require 'rspec/core/rake_task'
+require 'rubocop/rake_task'
 
 RSpec::Core::RakeTask.new(:spec)
+RuboCop::RakeTask.new(:rubocop)
 
-task :default => :spec
+task :default => [:spec, :rubocop]

data/bin/console

@@ -1,7 +1,7 @@
 #!/usr/bin/env ruby
 
-require "bundler/setup"
-require "hako"
+require 'bundler/setup'
+require 'hako'
 
 # You can add fixtures and/or initialization code here to make experimenting
 # with your gem easier. You can also use a different console, if you like.
@@ -10,5 +10,5 @@ require "hako"
 # require "pry"
 # Pry.start
 
-require "irb"
+require 'irb'
 IRB.start

data/examples/hello-lb.yml

@@ -17,6 +17,7 @@ scheduler:
   elb:
     listeners:
       - load_balancer_port: 80
+        protocol: HTTP
     subnets:
       - subnet-XXXXXXXX
       - subnet-YYYYYYYY
@@ -29,3 +30,8 @@ front:
     region: ap-northeast-1
     bucket: nanika
     prefix: hako/front_config
+  extra:
+    locations:
+      /:
+        allow_only_from:
+          - 10.0.0.0/24

data/examples/hello.yml

@@ -2,7 +2,7 @@ image: ryotarai/hello-sinatra
 env:
   $providers:
     - type: file
-      path: examples/hello.env
+      path: hello.env
   PORT: 3000
   MESSAGE: '#{username}-san'
 port: 3000

data/hako.gemspec

@@ -4,24 +4,25 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 require 'hako/version'
 
 Gem::Specification.new do |spec|
-  spec.name          = "hako"
+  spec.name          = 'hako'
   spec.version       = Hako::VERSION
-  spec.authors       = ["Kohei Suzuki"]
-  spec.email         = ["eagletmt@gmail.com"]
+  spec.authors       = ['Kohei Suzuki']
+  spec.email         = ['eagletmt@gmail.com']
 
-  spec.summary       = %q{Deploy Docker container}
-  spec.description   = %q{Deploy Docker container}
-  spec.homepage      = "https://github.com/eagletmt/hako"
+  spec.summary       = 'Deploy Docker container'
+  spec.description   = 'Deploy Docker container'
+  spec.homepage      = 'https://github.com/eagletmt/hako'
 
   spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
-  spec.bindir        = "exe"
+  spec.bindir        = 'exe'
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
-  spec.require_paths = ["lib"]
+  spec.require_paths = ['lib']
 
-  spec.add_dependency "aws-sdk", "~> 2.1.0"
-  spec.add_dependency "thor"
+  spec.add_dependency 'aws-sdk', '>= 2.1.0'
+  spec.add_dependency 'thor'
 
-  spec.add_development_dependency "bundler"
-  spec.add_development_dependency "rake"
-  spec.add_development_dependency "rspec"
+  spec.add_development_dependency 'bundler'
+  spec.add_development_dependency 'rake'
+  spec.add_development_dependency 'rspec'
+  spec.add_development_dependency 'rubocop', '>= 0.36.0'
 end

data/lib/hako.rb

@@ -1,8 +1,12 @@
 require 'logger'
-require "hako/version"
+require 'hako/version'
 
 module Hako
   def self.logger
-    @logger ||= Logger.new($stdout)
+    @logger ||=
+      begin
+        $stdout.sync = true
+        Logger.new($stdout)
+      end
   end
 end

data/lib/hako/after_scripts.rb

@@ -0,0 +1,4 @@
+module Hako
+  module AfterScripts
+  end
+end

data/lib/hako/application.rb

@@ -0,0 +1,22 @@
+require 'yaml'
+
+module Hako
+  class Application
+    attr_reader :id, :root_path, :yaml
+
+    def initialize(yaml_path)
+      path = Pathname.new(yaml_path)
+      @id = path.basename.sub_ext('').to_s
+      @root_path = path.parent
+      @yaml = YAML.load(load_default_yaml(@root_path) + path.read)
+    end
+
+    private
+
+    def load_default_yaml(root_path)
+      root_path.join('default.yml').read
+    rescue Errno::ENOENT
+      ''
+    end
+  end
+end

data/lib/hako/cli.rb

@@ -4,15 +4,33 @@ module Hako
   class CLI < Thor
     desc 'deploy FILE', 'Run deployment'
     option :force, aliases: %w[-f], type: :boolean, default: false, desc: 'Run deployment even if nothing is changed'
+    option :tag, aliases: %w[-t], type: :string, default: 'latest', desc: 'Specify tag (default: latest)'
     def deploy(yaml_path)
+      require 'hako/application'
       require 'hako/commander'
-      Commander.new(yaml_path).deploy(force: options[:force])
+      Commander.new(Application.new(yaml_path)).deploy(force: options[:force], tag: options[:tag])
+    end
+
+    desc 'oneshot FILE COMMAND ARG...', 'Run oneshot task'
+    option :tag, aliases: %w[-t], type: :string, default: 'latest', desc: 'Specify tag (default: latest)'
+    def oneshot(yaml_path, command, *args)
+      require 'hako/application'
+      require 'hako/commander'
+      Commander.new(Application.new(yaml_path)).oneshot([command, *args], tag: options[:tag])
     end
 
     desc 'status FILE', 'Show deployment status'
     def status(yaml_path)
+      require 'hako/application'
+      require 'hako/commander'
+      Commander.new(Application.new(yaml_path)).status
+    end
+
+    desc 'remove FILE', 'Destroy the application'
+    def remove(yaml_path)
+      require 'hako/application'
       require 'hako/commander'
-      Commander.new(yaml_path).status
+      Commander.new(Application.new(yaml_path)).remove
     end
   end
 end

data/lib/hako/commander.rb

@@ -1,4 +1,4 @@
-require 'yaml'
+require 'hako/after_scripts'
 require 'hako/env_expander'
 require 'hako/error'
 require 'hako/front_config'
@@ -7,32 +7,53 @@ require 'hako/schedulers'
 
 module Hako
   class Commander
-    PROVIDERS_KEY = '$providers'
+    PROVIDERS_KEY = '$providers'.freeze
 
-    def initialize(yaml_path)
-      @app_id = Pathname.new(yaml_path).basename.sub_ext('').to_s
-      @yaml = YAML.load_file(yaml_path)
+    def initialize(app)
+      @app = app
+      $LOAD_PATH << @app.root_path.join('lib')
     end
 
-    def deploy(force: false)
-      env = @yaml['env'].dup
-      providers = load_providers(env.delete(PROVIDERS_KEY) || [])
-      env = EnvExpander.new(providers).expand(env)
-
-      front = load_front(@yaml['front'])
+    def deploy(force: false, tag: 'latest')
+      env = load_environment(@app.yaml['env'])
+      front = load_front(@app.yaml['front'])
+      scheduler = load_scheduler(@app.yaml['scheduler'])
+      app_port = @app.yaml.fetch('port', nil)
+      image = @app.yaml.fetch('image') { raise Error.new('image must be set') }
+      image_tag = "#{image}:#{tag}"
+      after_scripts = @app.yaml.fetch('after_scripts', []).map { |config| load_after_script(config) }
 
-      scheduler = load_scheduler(@yaml['scheduler'])
-      app_port = @yaml.fetch('port', nil)
-      image_tag = @yaml['image']  # TODO: Append revision
       scheduler.deploy(image_tag, env, app_port, front, force: force)
+
+      after_scripts.each(&:after_deploy)
+    end
+
+    def oneshot(commands, tag: 'latest')
+      env = load_environment(@app.yaml['env'])
+      scheduler = load_scheduler(@app.yaml['scheduler'])
+      image = @app.yaml.fetch('image') { raise Error.new('image must be set') }
+      image_tag = "#{image}:#{tag}"
+      exit scheduler.oneshot(image_tag, env, commands)
     end
 
     def status
-      load_scheduler(@yaml['scheduler']).status
+      load_scheduler(@app.yaml['scheduler']).status
+    end
+
+    def remove
+      after_scripts = @app.yaml.fetch('after_scripts', []).map { |config| load_after_script(config) }
+      load_scheduler(@app.yaml['scheduler']).remove
+      after_scripts.each(&:after_remove)
     end
 
     private
 
+    def load_environment(env)
+      env = env.dup
+      providers = load_providers(env.delete(PROVIDERS_KEY) || [])
+      EnvExpander.new(providers).expand(env)
+    end
+
     def load_providers(provider_configs)
       provider_configs.map do |config|
         type = config['type']
@@ -40,7 +61,7 @@ module Hako
           raise Error.new("type must be set in each #{PROVIDERS_KEY} element")
         end
         require "hako/env_providers/#{type}"
-        Hako::EnvProviders.const_get(camelize(type)).new(config)
+        Hako::EnvProviders.const_get(camelize(type)).new(@app.root_path, config)
       end
     end
 
@@ -50,7 +71,7 @@ module Hako
         raise Error.new('type must be set in scheduler')
       end
       require "hako/schedulers/#{type}"
-      Hako::Schedulers.const_get(camelize(type)).new(@app_id, scheduler_config)
+      Hako::Schedulers.const_get(camelize(type)).new(@app.id, scheduler_config)
     end
 
     def load_front(yaml)
@@ -59,6 +80,12 @@ module Hako
       Hako::Fronts.const_get(camelize(front_config.type)).new(front_config)
     end
 
+    def load_after_script(config)
+      type = config.fetch('type')
+      require "hako/after_scripts/#{type}"
+      Hako::AfterScripts.const_get(camelize(type)).new(@app, config)
+    end
+
     def camelize(name)
       name.split('_').map(&:capitalize).join('')
     end

data/lib/hako/env_expander.rb

@@ -8,11 +8,8 @@ module Hako
     class ExpansionError < Error
     end
 
-    class Literal < Struct.new(:literal)
-    end
-
-    class Variable < Struct.new(:name)
-    end
+    Literal = Struct.new(:literal)
+    Variable = Struct.new(:name)
 
     def initialize(providers)
       @providers = providers
@@ -59,13 +56,13 @@ module Hako
       tokens = []
       pos = 0
       while s.scan_until(/#\{(.*?)\}/)
-        pre = s.string.byteslice(pos ... (s.pos - s.matched.size))
+        pre = s.string.byteslice(pos...(s.pos - s.matched.size))
         var = s[1]
         unless pre.empty?
           tokens << Literal.new(pre)
         end
         if var.empty?
-          raise ExpansionError.new("Empty interpolation is not allowed")
+          raise ExpansionError.new('Empty interpolation is not allowed')
         else
           tokens << Variable.new(var)
         end

data/lib/hako/env_provider.rb

@@ -5,7 +5,7 @@ module Hako
     class ValidationError < Error
     end
 
-    def initialize(_options)
+    def initialize(_root_path, _options)
       raise NotImplementedError
     end
 

data/lib/hako/env_providers/file.rb

@@ -3,11 +3,11 @@ require 'hako/env_provider'
 module Hako
   module EnvProviders
     class File < EnvProvider
-      def initialize(options)
+      def initialize(root_path, options)
         unless options['path']
-          validation_error!("path must be set")
+          validation_error!('path must be set')
         end
-        @path = options['path']
+        @path = root_path.join(options['path'])
       end
 
       def ask(variables)

data/lib/hako/front_config.rb

@@ -1,8 +1,9 @@
 require 'erb'
 
 module Hako
-  class FrontConfig < Struct.new(:type, :image_tag, :s3)
-    class S3Config < Struct.new(:region, :bucket, :prefix)
+  FrontConfig = Struct.new(:type, :image_tag, :s3, :extra)
+  class FrontConfig
+    S3Config = Struct.new(:region, :bucket, :prefix) do
       def initialize(options)
         self.region = options.fetch('region')
         self.bucket = options.fetch('bucket')
@@ -22,6 +23,7 @@ module Hako
       self.type = options.fetch('type')
       self.image_tag = options.fetch('image_tag')
       self.s3 = S3Config.new(options.fetch('s3'))
+      self.extra = options.fetch('extra', {})
     end
   end
 end

data/lib/hako/fronts/nginx.rb

@@ -11,8 +11,48 @@ module Hako
 
       private
 
+      def templates_directory
+        File.expand_path('../../templates', __FILE__)
+      end
+
       def nginx_conf_erb
-        File.expand_path('../../templates/nginx.conf.erb', __FILE__)
+        File.join(templates_directory, 'nginx.conf.erb')
+      end
+
+      def nginx_location_conf_erb
+        File.join(templates_directory, 'nginx.location.conf.erb')
+      end
+
+      def locations
+        locs = @config.extra.fetch('locations', {}).dup
+        locs['/'] ||= {}
+        locs.keys.each do |k|
+          locs[k] = Location.new(locs[k])
+        end
+        locs
+      end
+
+      def client_max_body_size
+        @config.extra.fetch('client_max_body_size', nil)
+      end
+
+      def render_location(listen_spec, location)
+        ERB.new(File.read(nginx_location_conf_erb), nil, '-').result(binding).each_line.map do |line|
+          "    #{line}"
+        end.join('')
+      end
+
+      class Location
+        def initialize(config)
+          @config = config
+        end
+
+        def allow_only_from
+          allow = @config.fetch('allow_only_from', nil)
+          if allow
+            allow.flatten
+          end
+        end
       end
     end
   end

data/lib/hako/scheduler.rb

@@ -16,6 +16,10 @@ module Hako
       raise NotImplementedError
     end
 
+    def remove
+      raise NotImplementedError
+    end
+
     def upload_front_config(app_id, front, app_port)
       front_conf = front.generate_config(app_port)
       s3_config = front.config.s3

data/lib/hako/schedulers/echo.rb

@@ -0,0 +1,28 @@
+require 'hako/scheduler'
+
+module Hako
+  module Schedulers
+    class Echo < Scheduler
+      def initialize(app_id, _options)
+        @app_id = app_id
+      end
+
+      def deploy(image_tag, env, app_port, _front, force: false)
+        puts "Deploy #{image_tag} with app_port=#{app_port}, force=#{force}"
+        puts 'Environment variables:'
+        env.each do |key, val|
+          puts "  #{key}=#{val.inspect}"
+        end
+      end
+
+      def oneshot(image_tag, env, commands)
+        puts "Run #{image_tag} with oneshot commands=#{commands.inspect}"
+        puts 'Environment variables:'
+        env.each do |key, val|
+          puts "  #{key}=#{val.inspect}"
+        end
+        0
+      end
+    end
+  end
+end

data/lib/hako/schedulers/ecs.rb

@@ -1,13 +1,13 @@
 require 'aws-sdk'
 require 'hako'
-require 'hako/error'
 require 'hako/scheduler'
 require 'hako/schedulers/ecs_definition_comparator'
+require 'hako/schedulers/ecs_elb'
 
 module Hako
   module Schedulers
     class Ecs < Scheduler
-      DEFAULT_CLUSTER = 'default'
+      DEFAULT_CLUSTER = 'default'.freeze
       DEFAULT_FRONT_PORT = 10000
 
       def initialize(app_id, options)
@@ -19,9 +19,8 @@ module Hako
         region = options.fetch('region') { validation_error!('region must be set') }
         @role = options.fetch('role', nil)
         @ecs = Aws::ECS::Client.new(region: region)
-        @elb = Aws::ElasticLoadBalancing::Client.new(region: region)
+        @elb = EcsElb.new(app_id, Aws::ElasticLoadBalancing::Client.new(region: region), options.fetch('elb', nil))
         @ec2 = Aws::EC2::Client.new(region: region)
-        @elb_config = options.fetch('elb', nil)
       end
 
       def deploy(image_tag, env, app_port, front, force: false)
@@ -31,7 +30,7 @@ module Hako
           'S3_CONFIG_BUCKET' => front.config.s3.bucket,
           'S3_CONFIG_KEY' => front.config.s3.key(@app_id),
         }
-        front_port = determine_front_port(front)
+        front_port = determine_front_port
         task_definition = register_task_definition(image_tag, env, front.config, front_env, front_port)
         if task_definition == :noop
           Hako.logger.info "Task definition isn't changed"
@@ -48,11 +47,21 @@ module Hako
           Hako.logger.info "Updated service: #{service.service_arn}"
           wait_for_ready(service)
         end
-        Hako.logger.info "Deployment completed"
+        Hako.logger.info 'Deployment completed'
+      end
+
+      def oneshot(image_tag, env, commands)
+        task_definition = register_task_definition_for_oneshot(image_tag)
+        Hako.logger.info "Registered task definition: #{task_definition.task_definition_arn}"
+        task = run_task(task_definition, env, commands)
+        Hako.logger.info "Started task: #{task.task_arn}"
+        exit_code = wait_for_task(task)
+        Hako.logger.info 'Oneshot task finished'
+        exit_code
       end
 
       def status
-        service = @ecs.describe_services(cluster: @cluster, services: [@app_id]).services[0]
+        service = describe_service
         unless service
           puts 'Unavailable'
           exit 1
@@ -60,7 +69,7 @@ module Hako
 
         unless service.load_balancers.empty?
           lb = service.load_balancers[0]
-          lb_detail = @elb.describe_load_balancers(load_balancer_names: [lb.load_balancer_name]).load_balancer_descriptions[0]
+          lb_detail = @elb.describe_load_balancer
           puts 'Load balancer:'
           lb_detail.listener_descriptions.each do |ld|
             l = ld.listener
@@ -70,11 +79,12 @@ module Hako
 
         puts 'Deployments:'
         service.deployments.each do |d|
-          puts "  [#{d.status}] desired_count=#{d.desired_count}, pending_count=#{d.pending_count}, running_count=#{d.running_count}"
+          abbrev_task_definition = d.task_definition.slice(%r{task-definition/(.+)\z}, 1)
+          puts "  [#{d.status}] #{abbrev_task_definition} desired_count=#{d.desired_count}, pending_count=#{d.pending_count}, running_count=#{d.running_count}"
         end
 
         puts 'Tasks:'
-        @ecs.list_tasks(cluster: @cluster, service_name: @app_id).each do |page|
+        @ecs.list_tasks(cluster: @cluster, service_name: service.service_arn).each do |page|
           unless page.task_arns.empty?
             tasks = @ecs.describe_tasks(cluster: @cluster, tasks: page.task_arns).tasks
             container_instances = {}
@@ -108,25 +118,49 @@ module Hako
         end
       end
 
+      def remove
+        service = describe_service
+        if service
+          @ecs.delete_service(cluster: @cluster, service: @app_id)
+          Hako.logger.info "#{service.service_arn} is deleted"
+        else
+          puts "Service #{@app_id} doesn't exist"
+        end
+
+        @elb.destroy
+      end
+
       private
 
-      def determine_front_port(front)
+      def describe_service
         service = @ecs.describe_services(cluster: @cluster, services: [@app_id]).services[0]
+        if service && service.status != 'INACTIVE'
+          service
+        end
+      end
+
+      def determine_front_port
+        service = describe_service
         if service
           find_front_port(service)
         else
           max_port = -1
           @ecs.list_services(cluster: @cluster).each do |page|
             unless page.service_arns.empty?
-              @ecs.describe_services(cluster: @cluster, services: page.service_arns).services.each do |service|
-                max_port = [max_port, find_front_port(service)].max
+              @ecs.describe_services(cluster: @cluster, services: page.service_arns).services.each do |s|
+                if s.status != 'INACTIVE'
+                  port = find_front_port(s)
+                  if port
+                    max_port = [max_port, port].max
+                  end
+                end
               end
             end
           end
           if max_port == -1
             DEFAULT_FRONT_PORT
           else
-            max_port+1
+            max_port + 1
           end
         end
       end
@@ -137,7 +171,9 @@ module Hako
         task_definition.container_definitions.each do |c|
           container_definitions[c.name] = c
         end
-        container_definitions['front'].port_mappings[0].host_port
+        if container_definitions.size == 2 && container_definitions['front'] && container_definitions['app']
+          container_definitions['front'].port_mappings[0].host_port
+        end
       end
 
       def task_definition_changed?(front, app)
@@ -172,6 +208,23 @@ module Hako
         end
       end
 
+      def register_task_definition_for_oneshot(image_tag)
+        @ecs.register_task_definition(
+          family: "#{@app_id}-oneshot",
+          container_definitions: [
+            {
+              name: 'oneshot',
+              image: image_tag,
+              cpu: @cpu,
+              memory: @memory,
+              links: [],
+              port_mappings: [],
+              environment: [],
+            },
+          ],
+        ).task_definition
+      end
+
       def front_container(front_config, env, front_port)
         environment = env.map { |k, v| { name: k, value: v } }
         {
@@ -180,7 +233,7 @@ module Hako
           cpu: 100,
           memory: 100,
           links: ['app:app'],
-          port_mappings: [{container_port: 80, host_port: front_port, protocol: 'tcp'}],
+          port_mappings: [{ container_port: 80, host_port: front_port, protocol: 'tcp' }],
           essential: true,
           environment: environment,
         }
@@ -200,9 +253,69 @@ module Hako
         }
       end
 
+      def run_task(task_definition, env, commands)
+        environment = env.map { |k, v| { name: k, value: v } }
+        @ecs.run_task(
+          cluster: @cluster,
+          task_definition: task_definition.task_definition_arn,
+          overrides: {
+            container_overrides: [
+              {
+                name: 'oneshot',
+                command: commands,
+                environment: environment,
+              },
+            ],
+          },
+          count: 1,
+          started_by: "hako oneshot #{@app_id}",
+        ).tasks[0]
+      end
+
+      def wait_for_task(task)
+        task_arn = task.task_arn
+        container_instance_arn = nil
+        started_at = nil
+        loop do
+          task = @ecs.describe_tasks(cluster: @cluster, tasks: [task_arn]).tasks[0]
+          if container_instance_arn != task.container_instance_arn
+            container_instance_arn = task.container_instance_arn
+            report_container_instance(container_instance_arn)
+          end
+          unless started_at
+            started_at = task.started_at
+            if started_at
+              Hako.logger.info "Started at #{started_at}"
+            end
+          end
+
+          Hako.logger.info "  status #{task.last_status}"
+
+          if task.last_status == 'STOPPED'
+            Hako.logger.info "Stopped at #{task.stopped_at}"
+            container = task.containers[0]
+            Hako.logger.info "Exit code is #{container.exit_code}"
+            return container.exit_code
+          end
+          sleep 1
+        end
+      end
+
+      def report_container_instance(container_instance_arn)
+        container_instance = @ecs.describe_container_instances(cluster: @cluster, container_instances: [container_instance_arn]).container_instances[0]
+        @ec2.describe_tags(filters: [{ name: 'resource-id', values: [container_instance.ec2_instance_id] }]).each do |page|
+          tag = page.tags.find { |t| t.key == 'Name' }
+          if tag
+            Hako.logger.info "Container instance is #{container_instance_arn} (#{tag.value} #{container_instance.ec2_instance_id})"
+          else
+            Hako.logger.info "Container instance is #{container_instance_arn} (#{container_instance.ec2_instance_id})"
+          end
+        end
+      end
+
       def create_or_update_service(task_definition_arn, front_port)
-        services = @ecs.describe_services(cluster: @cluster, services: [@app_id]).services
-        if services.empty?
+        service = describe_service
+        if service.nil?
           params = {
             cluster: @cluster,
             service_name: @app_id,
@@ -210,24 +323,18 @@ module Hako
             desired_count: @desired_count,
             role: @role,
           }
-          if @elb_config
-            name = find_or_create_load_balancer(front_port)
-            params.merge!(
-              load_balancers: [
-                {
-                  load_balancer_name: name,
-                  container_name: 'front',
-                  container_port: 80,
-                },
-              ],
-            )
+          name = @elb.find_or_create_load_balancer(front_port)
+          if name
+            params[:load_balancers] = [
+              {
+                load_balancer_name: name,
+                container_name: 'front',
+                container_port: 80,
+              },
+            ]
           end
           @ecs.create_service(params).service
         else
-          service = services[0]
-          if service.status != 'ACTIVE'
-            raise Error.new("Service #{service.service_arn} is already exist but the status is #{service.status}")
-          end
           params = {
             cluster: @cluster,
             service: @app_id,
@@ -242,7 +349,7 @@ module Hako
         end
       end
 
-      SERVICE_KEYS = %i[desired_count task_definition]
+      SERVICE_KEYS = %i[desired_count task_definition].freeze
 
       def service_changed?(service, params)
         SERVICE_KEYS.each do |key|
@@ -254,7 +361,7 @@ module Hako
       end
 
       def wait_for_ready(service)
-        latest_event_id = service.events[0].id
+        latest_event_id = find_latest_event_id(service.events)
         loop do
           s = @ecs.describe_services(cluster: service.cluster_arn, services: [service.service_arn]).services[0]
           s.events.each do |e|
@@ -263,7 +370,7 @@ module Hako
             end
             Hako.logger.info "#{e.created_at}: #{e.message}"
           end
-          latest_event_id = s.events[0].id
+          latest_event_id = find_latest_event_id(s.events)
           finished = s.deployments.all? { |d| d.status != 'ACTIVE' }
           if finished
             return
@@ -273,37 +380,12 @@ module Hako
         end
       end
 
-      def find_or_create_load_balancer(front_port)
-        unless load_balancer_exist?(elb_name)
-          listeners = @elb_config.fetch('listeners').map do |l|
-            {
-              protocol: 'tcp',
-              load_balancer_port: l.fetch('load_balancer_port'),
-              instance_port: front_port,
-              ssl_certificate_id: l.fetch('ssl_certificate_id', nil),
-            }
-          end
-          lb = @elb.create_load_balancer(
-            load_balancer_name: elb_name,
-            listeners: listeners,
-            subnets: @elb_config.fetch('subnets'),
-            security_groups: @elb_config.fetch('security_groups'),
-            tags: @elb_config.fetch('tags', {}).map { |k, v| { key: k, value: v.to_s } },
-          )
-          Hako.logger.info "Created ELB #{lb.dns_name} with instance_port=#{front_port}"
+      def find_latest_event_id(events)
+        if events.empty?
+          nil
+        else
+          events[0].id
         end
-        elb_name
-      end
-
-      def load_balancer_exist?(name)
-        @elb.describe_load_balancers(load_balancer_names: [elb_name])
-        true
-      rescue Aws::ElasticLoadBalancing::Errors::LoadBalancerNotFound
-        false
-      end
-
-      def elb_name
-        "hako-#{@app_id}"
       end
     end
   end

data/lib/hako/schedulers/ecs_definition_comparator.rb

@@ -5,9 +5,9 @@ module Hako
         @expected_container = expected_container
       end
 
-      CONTAINER_KEYS = %i[image cpu memory links]
-      PORT_MAPPING_KEYS = %i[container_port host_port protocol]
-      ENVIRONMENT_KEYS = %i[name value]
+      CONTAINER_KEYS = %i[image cpu memory links].freeze
+      PORT_MAPPING_KEYS = %i[container_port host_port protocol].freeze
+      ENVIRONMENT_KEYS = %i[name value].freeze
 
       def different?(actual_container)
         unless actual_container

data/lib/hako/schedulers/ecs_elb.rb

@@ -0,0 +1,64 @@
+require 'aws-sdk'
+require 'hako'
+
+module Hako
+  module Schedulers
+    class EcsElb
+      def initialize(app_id, elb, elb_config)
+        @app_id = app_id
+        @elb = elb
+        @elb_config = elb_config
+      end
+
+      def describe_load_balancer
+        @elb.describe_load_balancers(load_balancer_names: [name]).load_balancer_descriptions[0]
+      end
+
+      def find_or_create_load_balancer(front_port)
+        if @elb_config
+          unless exist?
+            listeners = @elb_config.fetch('listeners').map do |l|
+              {
+                protocol: l.fetch('protocol'),
+                load_balancer_port: l.fetch('load_balancer_port'),
+                instance_port: front_port,
+                ssl_certificate_id: l.fetch('ssl_certificate_id', nil),
+              }
+            end
+            lb = @elb.create_load_balancer(
+              load_balancer_name: name,
+              listeners: listeners,
+              subnets: @elb_config.fetch('subnets'),
+              security_groups: @elb_config.fetch('security_groups'),
+              tags: @elb_config.fetch('tags', {}).map { |k, v| { key: k, value: v.to_s } },
+            )
+            Hako.logger.info "Created ELB #{lb.dns_name} with instance_port=#{front_port}"
+          end
+          name
+        end
+      end
+
+      def destroy
+        if exist?
+          @elb.delete_load_balancer(load_balancer_name: name)
+          Hako.logger.info "Deleted ELB #{name}"
+        else
+          Hako.logger.info "ELB #{name} doesn't exist"
+        end
+      end
+
+      def exist?
+        describe_load_balancer
+        true
+      rescue Aws::ElasticLoadBalancing::Errors::LoadBalancerNotFound
+        false
+      end
+
+      private
+
+      def name
+        "hako-#{@app_id}"
+      end
+    end
+  end
+end

data/lib/hako/templates/nginx.conf.erb

@@ -1,13 +1,13 @@
 server {
   listen 80;
 
-  location / {
-    proxy_pass http://<%= listen_spec %>;
-    proxy_set_header Host $host;
-    proxy_set_header Connection ""; # for upstream keepalive
-    proxy_http_version 1.1;         # for upstream keepalive
-    proxy_connect_timeout 5s;
-    proxy_send_timeout 20s;
-    proxy_read_timeout 20s;
+  <%- if client_max_body_size -%>
+  client_max_body_size <%= client_max_body_size %>;
+  <%- end -%>
+
+  <%- locations.each do |path, location| -%>
+  location <%= path %> {
+<%= render_location(listen_spec, location) %>
   }
+<%- end -%>
 }

data/lib/hako/templates/nginx.location.conf.erb

@@ -0,0 +1,14 @@
+<%- if location.allow_only_from -%>
+  <%- location.allow_only_from.each do |ip| -%>
+allow <%= ip %>;
+  <%- end -%>
+deny all;
+<%- end -%>
+
+proxy_pass http://<%= listen_spec %>;
+proxy_set_header Host $host;
+proxy_set_header Connection ""; # for upstream keepalive
+proxy_http_version 1.1;         # for upstream keepalive
+proxy_connect_timeout 5s;
+proxy_send_timeout 20s;
+proxy_read_timeout 20s;

data/lib/hako/version.rb

@@ -1,3 +1,3 @@
 module Hako
-  VERSION = "0.1.0"
+  VERSION = '0.2.0'.freeze
 end

metadata

@@ -1,27 +1,27 @@
 --- !ruby/object:Gem::Specification
 name: hako
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Kohei Suzuki
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2015-10-05 00:00:00.000000000 Z
+date: 2016-01-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
         version: 2.1.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
         version: 2.1.0
 - !ruby/object:Gem::Dependency
@@ -80,6 +80,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.36.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.36.0
 description: Deploy Docker container
 email:
 - eagletmt@gmail.com
@@ -90,6 +104,8 @@ extra_rdoc_files: []
 files:
 - ".gitignore"
 - ".rspec"
+- ".rubocop.yml"
+- ".rubocop_todo.yml"
 - ".travis.yml"
 - Gemfile
 - README.md
@@ -102,6 +118,8 @@ files:
 - exe/hako
 - hako.gemspec
 - lib/hako.rb
+- lib/hako/after_scripts.rb
+- lib/hako/application.rb
 - lib/hako/cli.rb
 - lib/hako/commander.rb
 - lib/hako/env_expander.rb
@@ -115,9 +133,12 @@ files:
 - lib/hako/fronts/nginx.rb
 - lib/hako/scheduler.rb
 - lib/hako/schedulers.rb
+- lib/hako/schedulers/echo.rb
 - lib/hako/schedulers/ecs.rb
 - lib/hako/schedulers/ecs_definition_comparator.rb
+- lib/hako/schedulers/ecs_elb.rb
 - lib/hako/templates/nginx.conf.erb
+- lib/hako/templates/nginx.location.conf.erb
 - lib/hako/version.rb
 homepage: https://github.com/eagletmt/hako
 licenses: []