CloudyScripts 0.0.11 → 0.0.12

data/lib/scripts/ec2/ec2_script.rb

@@ -1,52 +1,101 @@
-# Base class for any script on EC2.
-class Ec2Script
-  # Initialization. Common Input parameters:
-  # * aws_access_key => the Amazon AWS Access Key (see Your Account -> Security Credentials)
-  # * aws_secret_key => the Amazon AWS Secret Key
-  # * ec2_api_server => the API Server to connect to (optional, default is us-east-1 (=> <ec2_api_server>.ec2.amazonaws.com)
-  # * logger => allows to pass a ruby logger object used for logging (optional, default is a stdout logger with level WARN)
-  # Scripts may add specific key/value pairs.
-  def initialize(input_params)
-    @input_params = input_params
-    @state_change_listeners = []
-    @progress_message_listeners = []
-    if input_params[:logger] == nil
-      @logger = Logger.new(STDOUT)
-      @logger .level = Logger::WARN
-      input_params[:logger] = @logger
-    end
-    @result = {:done => false, :failed => false}
-    @input_params[:result] = @result
-  end
-
-  def register_state_change_listener(listener)
-    @state_change_listeners << listener
-  end
-
-  def register_progress_message_listener(listener)
-    @progress_message_listeners << listener
-  end
-
-  def start_script
-    raise Exception.new("must be implemented")
-  end
-
-  # Return a hash of results. Common values are:
-  # * :done => is true when the script has terminated, otherwise false
-  # * :failed => is false when the script succeeded
-  # * :failure_reason => returns a failure reason (string)
-  # * :end_state => returns the state, in which the script terminated (#Help::ScriptExecutionState)
-  # Scripts may add specific key/value pairs.
-  # * 
-  def get_execution_result
-    raise Exception.new("must be implemented")
-  end
-
-  def post_message(message, level = Logger::DEBUG)
-    @progress_message_listeners.each() {|listener|
-      listener.new_message(message, level)
-    }
-  end
-  
-end
-
+# Base class for any script on EC2.
+class Ec2Script
+  # Initialization. Common Input parameters:
+  # * aws_access_key => the Amazon AWS Access Key (see Your Account -> Security Credentials)
+  # * aws_secret_key => the Amazon AWS Secret Key
+  # * ec2_api_server => the API Server to connect to (optional, default is us-east-1 (=> <ec2_api_server>.ec2.amazonaws.com)
+  # * logger => allows to pass a ruby logger object used for logging (optional, default is a stdout logger with level WARN)
+  # Scripts may add specific key/value pairs.
+  def initialize(input_params)
+    @input_params = input_params
+    @state_change_listeners = []
+    @progress_message_listeners = []
+    if input_params[:logger] == nil
+      @logger = Logger.new(STDOUT)
+      @logger .level = Logger::WARN
+      input_params[:logger] = @logger
+    end
+    @result = {:done => false, :failed => false}
+    @input_params[:result] = @result
+  end
+
+  def register_state_change_listener(listener)
+    @state_change_listeners << listener
+  end
+
+  def register_progress_message_listener(listener)
+    @progress_message_listeners << listener
+  end
+
+  # Check input parameters (in @input_parameters object variable)
+  # and set default values.
+  # Abstract method to be implemented by extending classes.
+  def check_input_parameters()
+    raise Exception.new("check_input_parameters must be implemented")
+  end
+
+  # Load the initial state for the script.
+  # Abstract method to be implemented by extending classes.
+  def load_initial_state()
+    raise Exception.new("load_initial_state must be implemented")
+  end
+
+  # Executes the script.
+  def start_script()
+    # optional parameters and initialization
+    check_input_parameters()
+    @input_params[:script] = self
+    begin
+      current_state = load_initial_state()
+      @state_change_listeners.each() {|listener|
+        current_state.register_state_change_listener(listener)
+      }
+      end_state = current_state.start_state_machine()
+      if end_state.failed?
+        @result[:failed] = true
+        @result[:failure_reason] = current_state.end_state.failure_reason
+        @result[:end_state] = current_state.end_state
+      else
+        @result[:failed] = false
+      end
+    rescue Exception => e
+      @logger.warn "exception during encryption: #{e}"
+      @logger.warn e.backtrace.join("\n")
+      err = e.to_s
+      err += " (in #{current_state.end_state.to_s})" unless current_state == nil
+      @result[:failed] = true
+      @result[:failure_reason] = err
+      @result[:end_state] = current_state.end_state unless current_state == nil
+    ensure
+      begin
+      @input_params[:remote_command_handler].disconnect
+      rescue Exception => e2
+      end
+    end
+    #
+    @result[:done] = true
+  end
+
+
+  # Return a hash of results. Common values are:
+  # * :done => is true when the script has terminated, otherwise false
+  # * :failed => is false when the script succeeded
+  # * :failure_reason => returns a failure reason (string)
+  # * :end_state => returns the state, in which the script terminated (#Help::ScriptExecutionState)
+  # Scripts may add specific key/value pairs.
+  # * 
+  # Returns a hash with the following information:
+  # :done => if execution is done
+  #
+  def get_execution_result
+    @result
+  end
+
+  def post_message(message, level = Logger::DEBUG)
+    @progress_message_listeners.each() {|listener|
+      listener.new_message(message, level)
+    }
+  end
+  
+end
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: CloudyScripts
 version: !ruby/object:Gem::Version 
-  version: 0.0.11
+  version: 0.0.12
 platform: ruby
 authors: 
 - Matthias Jung
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-02-08 00:00:00 +01:00
+date: 2010-03-17 00:00:00 +01:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -52,8 +52,10 @@ files:
 - lib/help/remote_command_handler.rb
 - lib/help/script_execution_state.rb
 - lib/help/state_change_listener.rb
+- lib/help/state_transition_helper.rb
 - lib/scripts/ec2/ami2_ebs_conversion.rb
 - lib/scripts/ec2/dm_encrypt.rb
+- lib/scripts/ec2/download_snapshot.rb
 - lib/scripts/ec2/ec2_script.rb
 has_rdoc: true
 homepage: http://elastic-security.com