instant_ec2 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 73d849bb6b5614cceddcdb42507e205d62db20c2
-  data.tar.gz: 310876d9dbe8088f169329d3e6dd9bcefa8212f4
+  metadata.gz: 416943632a687083890012e6991a51547c99a6be
+  data.tar.gz: 81ccf2521db6f8737edf9a6d037cf3e30a5e08cc
 SHA512:
-  metadata.gz: a932aee73fb96aee99650408738b6a5506292fab620738e5973fdc21516e443c2c740d6adbe9a9e7162cbb7afdb871b643eadcd89c6f158a8c4e316621956ee7
-  data.tar.gz: fd2d533ecf671566de3eef0428e244a94fa431d145fa46e68d11c715e95a15f36cc42696bb2cd2ebfa342f0665b0bd5e0cb8b3ef3a10d9bdfe510f9c987467b5
+  metadata.gz: 9d267345a6f3824e319ca91b894e8235aa5cbf0e41c9ee0df1eaff6c8c4dbfb5af22cf6996ea4816420fad7389bb9a340bd8293591642f101316390ef24ed304
+  data.tar.gz: 2cef48bad768c185593aa7b366c709b23270f0c1ba8de0a350bfe974cf116c82f5e0a288e8081eda39631d79fd3d6609d3f4e5af6e624057b006b93fb50633f3

data/lib/instant_ec2.rb

@@ -7,18 +7,25 @@ require 'aws-sdk'
 
 class EC2Instance < Hash
 
-  def initialize(h, ec2=nil)
+  def initialize(h, caller=nil)
 
-    @ec2 = ec2
+    @c = caller
     super().merge!(h)
 
   end
 
-  def start()
+  def start(duration: nil)
 
-    @ec2.start_instances instance_ids: [self[:instance_id]]
+    @c.start_instance self[:instance_id]       
+    Thread.new{ sleep duration * 60; self.stop} if duration
 
   end
+  
+  def stop()
+
+    @c.stop_instance self[:instance_id]
+
+  end  
 end
 
 class InstantEC2
@@ -39,9 +46,16 @@ class InstantEC2
     @images = image_names.zip(instance_ids).inject([]) do |r, x|
 
       name, id = x
-      r << EC2Instance.new({image_name: name, instance_id: id}, @ec2)
+      r << EC2Instance.new({image_name: name, instance_id: id}, self)
       
     end
+    
+    @hooks = {
+      running: ->(ip){ 
+        puts "%s: the instance is now accessible from %s" % [Time.now, ip]
+      },
+      stopping: ->(){ puts "%s: the instance is now stopping" % Time.now}
+    }
 
   end
 
@@ -62,52 +76,82 @@ class InstantEC2
     r.instances[0].public_ip_address if r
   end
   
-  def on_running()
-    
-    # timeout after 30 seconds
-    t1 = Time.now
-    sleep 1; ip = self.ip until ip or Time.now > t1 + 30
+  def on_running(&blk)
+    @hooks[:running] = blk
+  end    
+  
+  def running?
+    self.ip ? true : false
+  end
+  
+  # Launch an EC2 instance. Duration (optional) specified in minutes
+  #
+  def start(s, duration: nil)
     
-    if ip then 
-      yield(ip) 
-    else
-      puts 'on_running timedout'
-    end
+    self.find_image(s).start duration: duration
     
   end
-  
-  alias running find_running
 
-  def start(s)
-    self.find_image(s).start
+  def start_instance(id)
+    @ec2.start_instances instance_ids: [id]
+    Thread.new { trigger_on_start() }
   end
-
-  def stop
+  
+  def stopping
 
     r = self.find_running()
 
     if r then
       
       puts 'stopping ...'
-      instance_id = r.instances[0].instance_id      
-      @ec2.stop_instances instance_ids: [instance_id]
+      
+      self.stop_instance r.instances[0].instance_id      
       
     else
       puts 'no instances to stop'
     end
   end
   
-  def on_stopping()
+  def stop_instance(id)
+    @ec2.stop_instances instance_ids: [id]
+    trigger_on_stopping()    
+  end
+
+  def stopped?()
+    self.ip.nil?
+  end
+  
+  def on_stopped(&blk)
+    @hooks[:stopped] = blk
+  end  
+  
+  private
+  
+  def trigger_on_start()
+    
+    # timeout after 60 seconds
+    t1 = Time.now
+    sleep 1; ip = self.ip until ip or Time.now > t1 + 60
+    
+    if ip then 
+      @hooks[:running].call(ip)
+    else
+      puts 'on_running timedout'
+    end
+    
+  end
+  
+  def trigger_on_stopping()
     
     # timeout after 30 seconds
     t1 = Time.now
     sleep 1; ip = self.ip until ip.nil? or Time.now > t1 + 30
     
     if ip.nil? then 
-      yield
+      @hooks[:stopping].call()
     else
       puts 'on_stopping timedout'
     end
     
   end  
-end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: instant_ec2
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - James Robertson