ec2-mini 0.0.1 → 0.0.2

data/README.md

@@ -0,0 +1,31 @@
+Install
+---
+
+`gem install ec2-mini`
+
+Setup
+---
+
+vi ~/.ec2-mini
+
+```
+access_key_id: 'XXX'
+secret_access_key: 'XXX'
+region: 'XXX'
+```
+
+Command
+---
+
+`ec2-mini [role] [command]`
+
+Example
+---
+
+`ec2-mini ap-server backup`
+
+`ec2-mini ap-server +3`
+
+`ec2-mini ap-server -1`
+
+`ec2-mini ap-server count`

data/bin/ec2-mini

@@ -4,4 +4,9 @@ $:.unshift File.expand_path("../../lib", __FILE__)
 
 require "ec2-mini/cli"
 
-EC2Mini::CLI.start
+home_config = "#{ENV['HOME']}/.ec2-mini"
+current_config = "#{ENV['PWD']}/.ec2-mini"
+config_file = File.exist?(home_config) ? home_config : current_config
+
+client = EC2Mini::CLI.new(nil, config_file)
+client.start

data/lib/ec2-mini/cli.rb

@@ -3,115 +3,145 @@ require 'yaml'
 
 module EC2Mini
   class CLI
-    def self.start(options = {})
 
-      unless File.exist?(ENV['HOME'] + '/.ec2-mini')
-        puts ".ec2-mini is not found."
-        exit 0
+    def initialize(options = nil, config_file = '')
+      @options = options || load_config_file(config_file)
+
+      ['access_key_id', 'secret_access_key', 'region'].each do |attribute|
+        error "Not set #{attribute}." unless @options[attribute]
       end
+    end
 
-      options = YAML.load_file(ENV['HOME'] + '/.ec2-mini')
-
-      exit 0 unless ARGV[0] =~ /^[\w\-]+$/
-      exit 0 unless ARGV[1] =~ /^([+-]?[0-9]+)|backup|count$/
-
-      role = ARGV[0]
-      operation = ARGV[1]
-
-      access_key_id = options['access_key_id']
-      secret_access_key =  options['secret_access_key']
-      region = options['region']
-      ec2 = AWS::EC2.new(access_key_id: access_key_id, secret_access_key: secret_access_key, region: region)
-
-      #operation
-      case operation
-      when /^\+/
-
-        command = operation.scan(/([0-9]+$)/)[0][0].to_i
-
-        # search instance
-        running_instance = ''
-        ec2.instances.tagged('Mini', role).to_a.each do |instance|
-          next if instance.status != :running
-          running_instance = instance
-          break
-        end
-
-        # search ami
-        ami = ec2.images.with_owner("self").filter("name", role).first
-
-        # for create instance
-        image_id = ami.id
-        security_groups = running_instance.groups
-        zone = running_instance.availability_zone
-        key_name = running_instance.key_name
-        instance_type = running_instance.instance_type
-
-        command.times do
-          running_instance = ec2.instances.create(
-            instance_type: instance_type,
-            key_name: key_name,
-            image_id: image_id,
-            availability_zone: zone,
-            security_groups: security_groups
-          )
-          running_instance.add_tag('Name', value: role)
-          running_instance.add_tag('Mini', value: role)
-        end
-        puts "successfully created #{role} #{operation}"
-
-      when /^\-/
-
-        command = operation.scan(/([0-9]+$)/)[0][0].to_i
-
-        # search instance
-        # TODO Warning launch_time sort
-        instances = ec2.instances.tagged('Mini', role)
-        instances = instances.to_a.reverse
-
-        instances.each do |instance|
-          next if instance.status != :running || command <= 0
-          instance.terminate
-          command -= 1
-        end
-
-        puts "successfully removed #{role} #{operation}"
+    def start(role = nil, command = nil)
+      role_regex = /^[\w\-]+$/
+      command_regex = /^([+-][0-9]+)|backup|count$/
 
-      when 'backup'
+      error "Not set role." if !(role =~ role_regex) && !(ARGV[0] =~ role_regex)
+      error "Not set command." if !(command =~ command_regex) && !(ARGV[1] =~ command_regex)
 
-        # TODO
-        # deregister old ami
-        amis = ec2.images.with_owner("self").filter("name", role)
-        amis.first.deregister unless amis.count.zero?
-
-        # search instance
-        running_instance = ''
-        ec2.instances.tagged('Mini', role).to_a.each do |instance|
-          next if instance.status != :running
-          running_instance = instance
-          break
-        end
-
-        # create ami
-        image = running_instance.create_image(role, {:description => role, :no_reboot => true})
-        begin
-          sleep 1
-          print '.'
-          image = ec2.images[image.id]
-        end until image.state != :pending
-        if image.state == :failure
-          puts "create image failed: #{image.state_reason}"
-          exit 1
-        end
-        puts 'successfully created backup'
+      role = role || ARGV[0]
+      command = command || ARGV[1]
 
+      case command
+      when /^\+[0-9]+/
+        number = command.scan(/([0-9]+$)/)[0][0].to_i
+        up(role, number)
+      when /^\-[0-9]+/
+        number = command.scan(/([0-9]+$)/)[0][0].to_i
+        down(role, number)
+      when 'backup'
+        backup(role)
       when 'count'
-        count = 0
-        ec2.instances.tagged('Mini', role).each do |instance|
-          count += 1 if instance.status == :running
-        end
-        puts "#{role}: #{count} instances running"
+        count(role)
+      else
+        error 'Not command'
       end
     end
+
+    private
+    def load_config_file(config_file)
+      error "Not found .ec2-mini file." unless File.exist?(config_file)
+      YAML.load_file(config_file)
+    end
+
+    def ec2_client
+      AWS::EC2.new(
+        access_key_id: @options['access_key_id'],
+        secret_access_key: @options['secret_access_key'],
+        region: @options['region']
+      )
+    end
+
+    def backup(role)
+
+      # TODO
+      # deregister old ami
+      amis = ec2_client.images.with_owner("self").filter("name", role)
+      amis.first.deregister unless amis.count
+
+      # search instance
+      running_instance = ''
+      ec2_client.instances.tagged('Mini', role).to_a.each do |instance|
+        next if instance.status != :running
+        running_instance = instance
+        break
+      end
+
+      # create ami
+      image = running_instance.create_image(role, { description: role, no_reboot: true })
+      begin
+        sleep 1
+        print '.'
+        image = ec2_client.images[image.id]
+      end until image.state != :pending
+      if image.state == :failure
+        error "create image failed: #{image.state_reason}"
+      end
+      puts 'successfully created backup'
+    end
+
+    def count(role)
+      count = 0
+      ec2_client.instances.tagged('Mini', role).each do |instance|
+        count += 1 if instance.status == :running
+      end
+      puts "#{role}: #{count} instances running"
+    end
+
+    def up(role, number)
+
+      # search instance
+      running_instance = ''
+      ec2_client.instances.tagged('Mini', role).to_a.each do |instance|
+        next if instance.status != :running
+        running_instance = instance
+        break
+      end
+
+      # search ami
+      ami = ec2_client.images.with_owner("self").filter("name", role).first
+
+      # for create instance
+      image_id = ami.id
+      security_groups = running_instance.groups
+      zone = running_instance.availability_zone
+      key_name = running_instance.key_name
+      instance_type = running_instance.instance_type
+
+      number.times do
+        running_instance = ec2_client.instances.create(
+          instance_type: instance_type,
+          key_name: key_name,
+          image_id: image_id,
+          availability_zone: zone,
+          security_groups: security_groups
+        )
+        running_instance.add_tag('Name', value: role)
+        running_instance.add_tag('Mini', value: role)
+      end
+      puts "successfully created #{role} #{operation}"
+    end
+
+    def down(role, number)
+
+      # search instance
+      # TODO Warning launch_time sort
+      instances = ec2_client.instances.tagged('Mini', role)
+      instances = instances.to_a.reverse
+
+      instances.each do |instance|
+        next if instance.status != :running || number <= 0
+        instance.terminate
+        number -= 1
+      end
+
+      puts "successfully removed #{role} #{operation}"
+    end
+
+    def error(message = '')
+      puts message
+      exit 1
+    end
+
   end
 end

data/lib/ec2-mini/version.rb

@@ -1,3 +1,3 @@
 module EC2Mini
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/spec/lib/cil_spec.rb

@@ -0,0 +1,17 @@
+require 'ec2-mini/cli'
+
+describe EC2Mini::CLI do
+  describe '#initialize' do
+    context 'not exist config file' do
+      it "is failure" do
+        expect{ EC2Mini::CLI.new(nil, '') }.to raise_error
+      end
+    end
+    context 'exist config file' do
+      it "is success" do
+        # TODO ENV['PWD']
+        expect(EC2Mini::CLI.new(nil, "#{ENV['PWD']}/spec/support/.ec2-mini").class).to eq EC2Mini::CLI
+      end
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ec2-mini
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-05-11 00:00:00.000000000 Z
+date: 2013-05-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk
@@ -37,6 +37,8 @@ files:
 - bin/ec2-mini
 - lib/ec2-mini/cli.rb
 - lib/ec2-mini/version.rb
+- README.md
+- spec/lib/cil_spec.rb
 homepage: http://github.com/camelmasa/ec2-mini
 licenses:
 - MIT