toss 0.0.4 → 0.0.5

data/bin/toss

@@ -1,3 +1,52 @@
 #!/usr/bin/env ruby
+lib = File.expand_path(File.dirname(__FILE__) + '/../lib')
+$LOAD_PATH.unshift(lib) if File.directory?(lib) && !$LOAD_PATH.include?(lib)
 
-puts "toss"
+require 'rubygems'
+require 'toss'
+
+puts "#{Toss::NAME} #{Toss::VERSION}"
+
+$options = Toss::Options.load
+
+# set environment
+set :env, $options[:environment]
+
+# load preset
+$options[:pre_set].each do |name, value|
+  set name, value
+end
+
+# load recipe
+if $options[:recipes].count > 0
+  $options[:recipes].each do |file|
+    require file  
+  end
+else
+  require "#{Dir.pwd}/Tossfile.rb"
+end
+
+# load set
+$options[:set].each do |name, value|
+  set name, value
+end
+
+if ARGV.count.to_i > 0 
+  @steps = []
+  ARGV.each do |func|
+    step func
+  end
+end
+
+# run steps
+steps && steps.each_with_index do |toss_step, index|
+  @step_index = index
+  puts "== " + cyan(cap_words(toss_step[0].to_s.gsub("_", " "))) + " =="
+  if method(toss_step[0]).call == false
+    rollback_steps
+    exit(1)
+  end
+end
+
+# close any open ssh sessions
+close_sessions

data/lib/toss.rb

@@ -1,5 +1,37 @@
+require "toss/options"
 require "toss/version"
+require "toss/meta"
+require "toss/ssh"
+require "toss/color"
+require "toss/server"
 
-module Toss
-  # Your code goes here...
+include Toss::Meta
+include Toss::Ssh
+
+@step_index ||= 0
+
+def steps
+  @steps ||= []
+end
+
+def step(step_name, options={})
+  steps << [step_name, options]
 end
+
+def rollback_steps
+  @step_index.downto(0) do |index|
+    m = method(steps[index][0])
+    if m.arity == -1
+      puts "== " + red("Rolling Back") + " " + bwhite(cap_words(steps[index][0].to_s.gsub("_", " "))) + " =="
+
+      if m.call(true) == false
+        puts "here"
+        exit(1)
+      end
+    end
+  end
+end
+
+def cap_words(str) 
+  str.split(" ").each{|word| word.capitalize!}.join(" ") 
+end

data/lib/toss/color.rb

@@ -0,0 +1,86 @@
+def colorize(str, beginColor, endColor = 0)
+  return str unless STDOUT.isatty
+
+  begin
+    require 'Win32/Console/ANSI' if RUBY_PLATFORM =~ /win32/
+  rescue LoadError
+    raise 'You must gem install win32console to use colorize on Windows'
+  end
+    
+  "\e[#{beginColor}m#{str}\e[#{endColor}m"
+end
+
+#30 Black
+def black(str)
+  colorize(str, "30")
+end
+
+#31 Red
+def red(str)
+  colorize(str, "31")
+end
+
+#31 Bright Red
+def bred(str)
+  colorize(str, "1;31")
+end
+
+#32 Green
+def green(str)
+  colorize(str, "32")
+end
+
+#32 Bright Green
+def bgreen(str)
+  colorize(str, "1;32")
+end
+
+#33 Yellow
+def yellow(str)
+  colorize(str, "33")
+end
+
+#33 Bright Yellow
+def byellow(str)
+  colorize(str, "1;33")
+end
+
+#34 Blue
+def blue(str)
+  colorize(str, "34")
+end
+
+#34 Bright Blue
+def bblue(str)
+  colorize(str, "1;34")
+end
+
+#35 Magenta
+def magenta(str)
+  colorize(str, "35")
+end
+
+#35 Bright Magenta
+def bmagenta(str)
+  colorize(str, "1;35")
+end
+
+#36 Cyan
+def cyan(str)
+  colorize(str, "36")
+end
+
+#36 Bright Cyan
+def bcyan(str)
+  colorize(str, "1;36")
+end
+
+#37 White
+def white(str)
+  colorize(str, "37")
+end
+
+#37 Bright White
+def bwhite(str)
+  colorize(str, "1;37")
+end

data/lib/toss/meta.rb

@@ -0,0 +1,34 @@
+module Toss
+  module Meta
+    module ClassMethods
+    end
+    
+    module InstanceMethods
+      def set(name, value)
+        create_accessor(name)
+        Object.send("#{name}=", value)
+      end
+
+      def create_method(name, &block)
+        Object.send(:define_method, name, &block)
+      end
+
+      def create_accessor(name)
+        create_method("#{name}=") do |value| 
+          $globals =  $globals || {}
+          $globals[name] = value
+        end
+        create_method(name.to_sym) do 
+          $globals =  $globals || {}
+          $globals[name]
+        end
+      end
+    end
+    
+    def self.included(receiver)
+      receiver.extend ClassMethods
+      receiver.send :include, InstanceMethods
+    end
+  end
+end
+

data/lib/toss/options.rb

@@ -0,0 +1,50 @@
+require 'optparse'
+
+module Toss
+  module Options
+
+    def self.load
+      options = { :recipes => [], :pre_set => {}, :set => {}, :verbose => false }
+
+      optparse = OptionParser.new do |opts|
+        opts.on('-e', '--environment ENVIRONMENT', 'Environment to deploy as') do |environment|
+          options[:environment] = environment
+        end
+
+        opts.on('-f', '--file FILE', 'A recipe file to run. Option may be used more than once.') do |file|
+           options[:recipes] << value 
+        end
+
+        opts.on("-S", "--set-before NAME=VALUE", "Set a variable before the recipes are loaded.") do |name_value|
+          name, value = name_value.split("=", 2)
+          options[:pre_set][name.to_sym] = value
+        end
+
+        opts.on("-s", "--set NAME=VALUE", "Set a variable after the recipes are loaded.") do |name_value|
+          name, value = name_value.split("=", 2)
+          options[:set][name.to_sym] = value
+        end
+
+        opts.on('-v', '--verbose', 'Show more information about the process') do
+          options[:verbose] = true
+        end
+
+        opts.on('-h', '--help', 'Display this screen') do
+          puts opts
+          exit
+        end
+      end
+
+      begin
+        optparse.parse!
+      rescue OptionParser::InvalidOption, OptionParser::MissingArgument
+        puts $!.to_s
+        puts optparse
+        exit
+      end
+
+      options
+    end
+
+  end
+end

data/lib/toss/server.rb

@@ -0,0 +1,78 @@
+def server(properties)
+  return Server.new(properties)
+end
+
+def with(obj, &block)
+  obj.instance_eval &block
+end
+
+class Server
+
+  attr_accessor :host, :user, :password, :type
+  attr_accessor :working_directory
+  attr_accessor :ssh
+
+  def initialize(properties)
+    @host = properties[:host]
+    @user = properties[:user]
+    @password = properties[:password]
+    @type = properties[:type]
+  end
+
+  def run(command='', &block)
+    if block_given?
+      self.instance_eval &block
+    else
+      result = exec(command, false)
+      return result.first
+    end
+  end
+
+  def exec(command, quiet=false)
+    ssh = open_sessions[host] || start_session(self) 
+    # capture all stderr and stdout output from a remote process
+
+    if $options[:verbose] 
+      puts "#{bwhite host} executing: #{command}"
+    end
+
+    if !working_directory.nil? && working_directory != ""
+      command = "cd #{working_directory} && " + command
+    end
+
+    stdout_data, stderr_data, exit_code, exit_signal = ssh_exec! ssh, command
+
+    if !quiet
+      puts "#{bwhite host} said: #{red stderr_data.strip!}" if !stderr_data.nil? && stderr_data != ""
+      puts "#{bwhite host} said: #{stdout_data.strip!}" if !stdout_data.nil? && stdout_data != ""
+    end
+
+    return exit_code, stdout_data, stderr_data, exit_signal
+  end
+
+  def exists?(path)
+    exit_code = run("[ -e #{path} ] && exit 0 || exit 1")
+    return exit_code == 0
+  end
+
+  def cd(path)
+    if $options[:verbose] 
+      puts "#{bwhite host} cd to: #{path}"
+    end
+    if block_given?
+      previous_path = @working_directory
+      @working_directory = path
+      yield
+      @working_directory =  previous_path
+    else
+      @working_directory = path
+    end
+  end
+
+  def delete(path)
+    if exists? "#{path}"
+      return run "unlink #{path}"
+    end
+  end
+
+end

data/lib/toss/ssh.rb

@@ -0,0 +1,65 @@
+require "net/ssh"
+require "net/scp"
+
+module Toss
+  module Ssh
+    module ClassMethods
+    end
+    
+    module InstanceMethods
+      def open_sessions
+        @open_sessions ||= {}
+      end
+
+      def start_session(server)
+        puts "Starting ssh session to #{server.host}"
+        open_sessions[server.host] = Net::SSH.start(server.host, server.user, :password => server.password)
+        open_sessions[server.host]
+      end
+
+      def close_sessions
+        open_sessions.each.each do |target_host,session|
+          puts "Closing ssh session for #{target_host}"
+          session.close
+          open_sessions.delete(target_host)
+        end
+      end
+
+      def ssh_exec!(ssh, command)
+        stdout_data = ""
+        stderr_data = ""
+        exit_code = nil
+        exit_signal = nil
+        ssh.open_channel do |channel|
+          channel.exec(command) do |ch, success|
+            unless success
+              abort "FAILED: couldn't execute command (ssh.channel.exec)"
+            end
+            channel.on_data do |ch,data|
+              stdout_data+=data
+            end
+
+            channel.on_extended_data do |ch,type,data|
+              stderr_data+=data
+            end
+
+            channel.on_request("exit-status") do |ch,data|
+              exit_code = data.read_long
+            end
+
+            channel.on_request("exit-signal") do |ch, data|
+              exit_signal = data.read_long
+            end
+          end
+        end
+        ssh.loop
+        [stdout_data, stderr_data, exit_code, exit_signal]
+      end      
+    end
+    
+    def self.included(receiver)
+      receiver.extend         ClassMethods
+      receiver.send :include, InstanceMethods
+    end
+  end
+end

data/lib/toss/version.rb

@@ -1,3 +1,4 @@
 module Toss
-  VERSION = "0.0.4"
+  NAME = "Toss"
+  VERSION = "0.0.5"
 end

data/toss.gemspec

@@ -8,7 +8,7 @@ Gem::Specification.new do |s|
   s.platform    = Gem::Platform::RUBY
   s.authors     = ["Tim Santeford"]
   s.email       = ["tsantef@gmail.com"]
-  s.homepage    = "https://github.com/tsantef/toss"
+  s.homepage    = %q{https://github.com/tsantef/toss}
   s.default_executable = %q{toss}
   s.summary     = %q{toss - Simple deployment}
   s.description = %q{toss - Simple deployment}
@@ -20,7 +20,8 @@ Gem::Specification.new do |s|
   s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
   s.require_paths = ["lib"]
 
-  # specify any dependencies here; for example:
-  # s.add_development_dependency "rspec"
-  # s.add_runtime_dependency "rest-client"
+  s.add_dependency(%q<net-ssh>, [">= 2.0.14"])
+  s.add_dependency(%q<net-scp>, [">= 1.0.0"])
+  s.add_dependency(%q<highline>, [">= 0"])
+  
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: toss
 version: !ruby/object:Gem::Version 
-  hash: 23
+  hash: 21
   prerelease: 
   segments: 
   - 0
   - 0
-  - 4
-  version: 0.0.4
+  - 5
+  version: 0.0.5
 platform: ruby
 authors: 
 - Tim Santeford
@@ -15,9 +15,54 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-02-04 00:00:00 Z
-dependencies: []
-
+date: 2012-03-06 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: net-ssh
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 19
+        segments: 
+        - 2
+        - 0
+        - 14
+        version: 2.0.14
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: net-scp
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 23
+        segments: 
+        - 1
+        - 0
+        - 0
+        version: 1.0.0
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: highline
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id003
 description: toss - Simple deployment
 email: 
 - tsantef@gmail.com
@@ -34,6 +79,11 @@ files:
 - Rakefile
 - bin/toss
 - lib/toss.rb
+- lib/toss/color.rb
+- lib/toss/meta.rb
+- lib/toss/options.rb
+- lib/toss/server.rb
+- lib/toss/ssh.rb
 - lib/toss/version.rb
 - toss.gemspec
 homepage: https://github.com/tsantef/toss