rest_connection 0.0.1

data/lib/rest_connection/rightscale/rightscale_api_resources.rb

@@ -0,0 +1,22 @@
+#    This file is part of RestConnection 
+#
+#    RestConnection is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License as published by
+#    the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    RestConnection is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with RestConnection.  If not, see <http://www.gnu.org/licenses/>.
+
+require 'rest_connection/rightscale/executable'
+require 'rest_connection/rightscale/server'
+require 'rest_connection/rightscale/deployment'
+require 'rest_connection/rightscale/status'
+require 'rest_connection/rightscale/server_template'
+require 'rest_connection/rightscale/right_script'
+require 'rest_connection/rightscale/instance'

data/lib/rest_connection/rightscale/server.rb

@@ -0,0 +1,121 @@
+#    This file is part of RestConnection 
+#
+#    RestConnection is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License as published by
+#    the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    RestConnection is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with RestConnection.  If not, see <http://www.gnu.org/licenses/>.
+
+require 'rest_connection/ssh_hax'
+
+class Server < RightScale::Api::Base
+  include SshHax
+  def wait_for_state(st)
+    reload
+    connection.logger("#{nickname} is #{self.state}")
+    while(1)
+      return true if state == st
+      raise "FATAL error, this server is stranded and needs to be #{st}: #{nickname}, see audit: #{self.audit_link}" if state.include?('stranded')
+      sleep 5
+      connection.logger("waiting for server #{nickname} to go #{st}, state is #{state}")
+      reload
+    end
+  end
+
+  def wait_for_operational_with_dns
+    wait_for_state("operational")
+    while(1)
+      connection.logger "waiting for dns-name for #{self.nickname}"
+      break if self['dns-name'] && !self['dns-name'].empty?
+      self.settings
+      sleep 2
+    end
+    connection.logger "got DNS: #{self['dns-name']}"
+  end
+
+  def audit_link
+    # proof of concept for now
+    server_id = self.href.split(/\//).last
+    "https://my.rightscale.com/servers/#{server_id}#audit_entries"
+  end
+
+  def start
+    if self.state == "stopped"
+      t = URI.parse(self.href)
+      return connection.post(t.path + '/start')
+    else
+      connection.logger("WARNING: was in #{self.state} so skiping start call")
+    end
+  end
+
+  def stop
+    if self.state != "stopped"
+      t = URI.parse(self.href)
+      connection.post(t.path + '/stop')
+    else
+      connection.logger("WARNING: was in #{self.state} so skiping stop call")
+    end
+  end
+  
+  def run_script(script)
+    serv_href = URI.parse(self.href)
+    location = connection.post(serv_href.path + '/run_script', :right_script => script.href)
+    Status.new('href' => location)
+  end
+
+  def set_input(name, value)
+    serv_href = URI.parse(self.href)
+    connection.put(serv_href.path, :server => {:parameters => {name.to_sym => value} })
+  end
+
+  def self.create(opts)
+    location = connection.post("servers", :server => opts)
+    Server.new('href' => location)
+  end
+
+  def set_template(href)
+    serv_href = URI.parse(self.href)
+    connection.put(serv_href.path, :server => {:server_template_href => href})
+  end
+
+  def settings
+    serv_href = URI.parse(self.href)
+    @params.merge! connection.get(serv_href.path + "/settings")
+  end
+
+  def monitors
+    serv_href = URI.parse(self.href)
+    @params.merge! connection.get(serv_href.path + "/monitors")
+  end
+
+  def reboot
+    serv_href = URI.parse(self.href)
+    connection.post(serv_href.path + "/reboot") 
+  end
+
+  def relaunch
+    unless state == "stopped"
+      wind_monkey
+      server_id = self.href.split(/\//).last
+      base_url = URI.parse(self.href)
+      base_url.path = "/servers/#{server_id}"
+
+      s = agent.get(base_url.to_s)
+      relaunch = s.links.detect {|d| d.to_s == "Relaunch"}
+      prelaunch_page = agent.get(relaunch.href)
+      launch_form = prelaunch_page.forms[2]
+      launch_form.radiobuttons_with(:name => 'launch_immediately').first.check
+      agent.submit(launch_form, launch_form.buttons.first)
+    else
+      connection.logger("WARNING: detected server is #{self.state}, skipping relaunch")
+    end
+  end
+end
+

data/lib/rest_connection/rightscale/server_template.rb

@@ -0,0 +1,31 @@
+#    This file is part of RestConnection 
+#
+#    RestConnection is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License as published by
+#    the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    RestConnection is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with RestConnection.  If not, see <http://www.gnu.org/licenses/>.
+
+class ServerTemplate < RightScale::Api::Base
+  def initialize(params)
+    @params = params
+    fetch_executables
+  end
+  
+  def fetch_executables
+    my_href = URI.parse(self.href)
+    #@params.merge! :executables =? connection.get(my_href.path + "/executables")
+    ex = []
+    connection.get(my_href.path + "/executables").each do |e|
+      ex << Executable.new(e)
+    end
+    @params["executables"] = ex
+  end
+end

data/lib/rest_connection/rightscale/status.rb

@@ -0,0 +1,26 @@
+#    This file is part of RestConnection 
+#
+#    RestConnection is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License as published by
+#    the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    RestConnection is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with RestConnection.  If not, see <http://www.gnu.org/licenses/>.
+
+class Status < RightScale::Api::Base
+  def wait_for_completed(audit_link = "no audit link available")
+    while(1)
+      reload
+      return true if self.state == "completed"
+      raise "FATAL error, script failed\nSee Audit: #{audit_link}" if self.state == 'failed'
+      sleep 5
+      connection.logger("querying status of right_script.. got: #{self.state}")
+    end
+  end
+end

data/lib/rest_connection/ssh_hax.rb

@@ -0,0 +1,141 @@
+#    This file is part of RestConnection 
+#
+#    RestConnection is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License as published by
+#    the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    RestConnection is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#    You should have received a copy of the GNU General Public License
+#    along with RestConnection.  If not, see <http://www.gnu.org/licenses/>.
+
+
+require 'net/ssh'
+
+# This is a mixin module run_recipes until api support is ready for checking result of recipe run.
+# The mixin typically used from Server#run_recipe
+module SshHax
+  # recipe can be either a String, or an Executable
+  # host_dns is optional and will default to objects self.dns_name
+  def run_recipe(recipe, host_dns=self.dns_name)
+    if recipe.is_a?(Executable)
+      recipe = recipe.recipe
+    end
+    # legacy used this key by default
+    if connection.settings[:ssh_key]
+      ssh_key = connection.settings[:ssh_key] 
+    else
+      ssh_key='~/.ssh/publish-test'
+    end
+    status = nil
+    result = nil
+    output = ""
+    tail_command ="tail -f -n1 /var/log/messages"
+    expect = /RightLink.*RS> ([completed|failed]+: < #{recipe} >)/
+    run_this = "rs_run_recipe -n '#{recipe}'"
+    Net::SSH.start(host_dns, 'root', :keys => [ssh_key]) do |ssh|
+      cmd_channel = ssh.open_channel do |ch1|
+        ch1.on_request('exit-status') do |ch, data|
+          status = data.read_long
+        end
+        ch1.exec(run_this) do |ch2, success|
+          unless success
+            output = "ERROR: SSH cmd failed to exec"
+            status = 1
+          end
+          ch2.on_data do |ch, data|
+            output += data
+          end
+          ch2.on_extended_data do |ch, type, data|
+            output += data
+          end
+
+        end
+      end
+      log_channel = ssh.open_channel do |ch2|
+        ch2.exec tail_command do |ch, success|
+          raise "could not execute command" unless success
+          # "on_data" is called when the process writes something to stdout
+          ch.on_data do |c, data|
+            output += data
+            if data =~ expect
+              result = $1
+            end
+          end
+          # "on_extended_data" is called when the process writes something to stderr
+          ch.on_extended_data do |c, type, data|
+            #STDERR.print data
+          end
+          ch.on_close do 
+          end
+          ch.on_process do |c|
+            if result
+              ch.close
+              ssh.exec("killall tail")
+            end
+          end
+        end
+      end 
+      cmd_channel.wait
+      log_channel.wait
+    end
+    success = result.include?('completed')
+    connection.logger output
+    connection.logger "Converge failed. See server audit: #{self.audit_link}" unless success
+    return {:status => success, :output => output}
+  end
+
+  def spot_check(command, ssh_key="~/.ssh/publish-test", host_dns=self.dns_name, &block)
+    ssh_key = connection.settings[:ssh_key] if connection.settings[:ssh_key]
+    connection.logger "SSHing to #{host_dns} using key #{ssh_key}"
+    Net::SSH.start(host_dns, 'root', :keys => [ssh_key]) do |ssh|
+      result = ssh.exec!(command)
+      yield result
+    end
+  end 
+
+  # returns true or false based on command success
+  def spot_check_command?(command, ssh_key="~/.ssh/publish-test", host_dns=self.dns_name)
+    ssh_key = connection.settings[:ssh_key] if connection.settings[:ssh_key]
+    results = spot_check_command(command, ssh_key, host_dns)
+    return results[:status] == 0
+  end
+
+
+  # returns hash of exit_status and output from command
+  def spot_check_command(command, ssh_key="~/.ssh/publish-test", host_dns=self.dns_name)
+    ssh_key = connection.settings[:ssh_key] if connection.settings[:ssh_key]
+    connection.logger "SSHing to #{host_dns} using key #{ssh_key}"
+    status = nil
+    output = ""
+    Net::SSH.start(host_dns, 'root', :keys => [ssh_key]) do |ssh|
+      cmd_channel = ssh.open_channel do |ch1|
+        ch1.on_request('exit-status') do |ch, data|
+          status = data.read_long
+        end
+        output += "Running: #{command}\n"
+        ch1.exec(command) do |ch2, success|
+          unless success
+            output = "ERROR: SSH cmd failed to exec"
+            status = 1
+          end
+          ch2.on_data do |ch, data|
+            output += data
+          end
+          ch2.on_extended_data do |ch, type, data|
+            output += data
+          end
+        end
+      end
+    end
+    connection.logger output
+    return {:status => status, :output => output}
+  end 
+
+end
+
+

data/rest_connection.gemspec

@@ -0,0 +1,81 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{rest_connection}
+  s.version = "0.0.1"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Jeremy Deininger"]
+  s.date = %q{2010-02-12}
+  s.description = %q{provides rest_connection}
+  s.email = %q{jeremy@rubyonlinux.org}
+  s.extra_rdoc_files = [
+    "README"
+  ]
+  s.files = [
+    "README",
+     "Rakefile",
+     "VERSION",
+     "config/rest_api_config.yaml.sample",
+     "examples/dev_setup.rb",
+     "examples/relaunch_deployment.rb",
+     "examples/restart_instance_agent.rb",
+     "examples/right_scale_ec2_instances_api_test.rb",
+     "examples/run_ebs_sequence.rb",
+     "examples/run_ebs_terminate.rb",
+     "examples/run_mysql_chef_sequence.rb",
+     "examples/run_php_chef_sequence.rb",
+     "examples/set_deployment_template_href.rb",
+     "lib/rest_connection.rb",
+     "lib/rest_connection/mechanize_connection.rb",
+     "lib/rest_connection/rightscale/deployment.rb",
+     "lib/rest_connection/rightscale/executable.rb",
+     "lib/rest_connection/rightscale/instance.rb",
+     "lib/rest_connection/rightscale/right_script.rb",
+     "lib/rest_connection/rightscale/rightscale_api_base.rb",
+     "lib/rest_connection/rightscale/rightscale_api_resources.rb",
+     "lib/rest_connection/rightscale/server.rb",
+     "lib/rest_connection/rightscale/server_template.rb",
+     "lib/rest_connection/rightscale/status.rb",
+     "lib/rest_connection/ssh_hax.rb",
+     "rest_connection.gemspec"
+  ]
+  s.homepage = %q{http://github.com/jeremyd/rest_connection}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{lib for restful connections to the rightscale api}
+  s.test_files = [
+    "examples/console.rb",
+     "examples/dev_setup.rb",
+     "examples/mysql_dev_rollback.rb",
+     "examples/relaunch_deployment.rb",
+     "examples/restart_instance_agent.rb",
+     "examples/right_scale_ec2_instances_api_test.rb",
+     "examples/run_ebs_sequence.rb",
+     "examples/run_ebs_terminate.rb",
+     "examples/run_mysql_chef_sequence.rb",
+     "examples/run_php_chef_sequence.rb",
+     "examples/set_deployment_template_href.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<mechanize>, [">= 0.9.3"])
+      s.add_runtime_dependency(%q<activesupport>, [">= 0"])
+    else
+      s.add_dependency(%q<mechanize>, [">= 0.9.3"])
+      s.add_dependency(%q<activesupport>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<mechanize>, [">= 0.9.3"])
+    s.add_dependency(%q<activesupport>, [">= 0"])
+  end
+end
+

metadata

@@ -0,0 +1,109 @@
+--- !ruby/object:Gem::Specification 
+name: rest_connection
+version: !ruby/object:Gem::Version 
+  version: 0.0.1
+platform: ruby
+authors: 
+- Jeremy Deininger
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-02-12 00:00:00 -08:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: mechanize
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.9.3
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: activesupport
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+description: provides rest_connection
+email: jeremy@rubyonlinux.org
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README
+files: 
+- README
+- Rakefile
+- VERSION
+- config/rest_api_config.yaml.sample
+- examples/dev_setup.rb
+- examples/relaunch_deployment.rb
+- examples/restart_instance_agent.rb
+- examples/right_scale_ec2_instances_api_test.rb
+- examples/run_ebs_sequence.rb
+- examples/run_ebs_terminate.rb
+- examples/run_mysql_chef_sequence.rb
+- examples/run_php_chef_sequence.rb
+- examples/set_deployment_template_href.rb
+- lib/rest_connection.rb
+- lib/rest_connection/mechanize_connection.rb
+- lib/rest_connection/rightscale/deployment.rb
+- lib/rest_connection/rightscale/executable.rb
+- lib/rest_connection/rightscale/instance.rb
+- lib/rest_connection/rightscale/right_script.rb
+- lib/rest_connection/rightscale/rightscale_api_base.rb
+- lib/rest_connection/rightscale/rightscale_api_resources.rb
+- lib/rest_connection/rightscale/server.rb
+- lib/rest_connection/rightscale/server_template.rb
+- lib/rest_connection/rightscale/status.rb
+- lib/rest_connection/ssh_hax.rb
+- rest_connection.gemspec
+has_rdoc: true
+homepage: http://github.com/jeremyd/rest_connection
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: lib for restful connections to the rightscale api
+test_files: 
+- examples/console.rb
+- examples/dev_setup.rb
+- examples/mysql_dev_rollback.rb
+- examples/relaunch_deployment.rb
+- examples/restart_instance_agent.rb
+- examples/right_scale_ec2_instances_api_test.rb
+- examples/run_ebs_sequence.rb
+- examples/run_ebs_terminate.rb
+- examples/run_mysql_chef_sequence.rb
+- examples/run_php_chef_sequence.rb
+- examples/set_deployment_template_href.rb