rsh2 0.0.2 → 0.0.3

data/Rakefile

@@ -2,8 +2,8 @@ require 'rake_ext'
 
 project(
   name: "rsh2",
-  version: "0.0.2",
-  summary: "Tiny wrapper over Net::SSH and Net::SFTP",
+  version: "0.0.3",
+  summary: "Tiny wrapper over Net::SSH/SFTP + small rake addon for cluster configuration management",
 
   author: "Alexey Petrushin",
   homepage: "http://github.com/alexeypetrushin/rssh"

data/lib/ros/rake.rb

@@ -0,0 +1,9 @@
+def package *a, &b
+  task *a do |task, *args|
+    Ros.each_box_cached do |box|
+      package = Ros::Package.new task.name
+      package.configure_with &b
+      package.apply_to box
+    end    
+  end  
+end

data/lib/ros/ros.rb

@@ -0,0 +1,57 @@
+module Ros
+  class Dsl
+    attr_reader :package
+    
+    def initialize package
+      @package = package
+    end
+    
+    def applied? &b        
+      package.applied = b
+    end
+  
+    def apply &b
+      package.apply = b
+    end
+  end
+  
+  class Package
+    attr_accessor :applied, :apply, :name
+    
+    def initialize name
+      @name = name
+    end
+    
+    def configure_with &b
+      dsl = Dsl.new self
+      dsl.instance_eval &b
+    end
+    
+    def apply_to box
+      unless applied and applied.call(box)
+        print "applying '#{name}' to '#{box.options[:host]}'\n"
+        apply.call box
+        print "done\n"
+      end
+    end
+  end
+  
+  class << self
+    def each_box_cached &b
+      unless @each_box_cached
+        unless Object.private_instance_methods(false).include?(:each_box)
+          raise "you must define 'each_box' method!"
+        end
+        
+        @each_box_cached = []
+        each_box do |box| 
+          @each_box_cached << box
+          
+          # cache ssh connection
+          box.open_connection
+        end
+      end
+      @each_box_cached.each &b
+    end
+  end
+end

data/lib/ros.rb

@@ -0,0 +1,4 @@
+%w(
+  ros
+  rake
+).each{|f| require "ros/#{f}"}

data/lib/rsh/box.rb

@@ -142,7 +142,33 @@ module Rsh
       driver.exec cmd
     end
     
+    def home path = nil
+      @home ||= bash('cd ~; pwd').gsub("\n", '')    
+      "#{@home}#{path}"
+    end
+    
+    def mark key
+      ensure_mark_requrements!
+      bash "touch #{marks_dir}/#{key}"
+    end
+
+    def has_mark? key
+      ensure_mark_requrements!
+      file_exist? "#{marks_dir}/#{key}"
+    end
+
     protected
+      def marks_dir
+        home "/.marks"
+      end
+
+      def ensure_mark_requrements!
+        unless @ensure_mark_requrements
+          create_directory marks_dir unless directory_exist? marks_dir
+          @ensure_mark_requrements = true
+        end
+      end
+    
       def method_missing m, *a, &b
         driver.send m, *a, &b
       end

data/lib/rsh/drivers/ssh.rb

@@ -9,14 +9,14 @@ module Rsh
     
       def upload_file from_local_path, to_remote_path
         remote do |ssh, sftp|
-          sftp.upload! from_local_path, to_remote_path
+          sftp.upload! from_local_path, fix_path(to_remote_path)
         end
       end
 
       def download_file from_remote_path, to_local_path
         File.open to_local_path, "w" do |out|
           remote do |ssh, sftp|
-            sftp.download! from_remote_path, out #, :recursive => true          
+            sftp.download! fix_path(from_remote_path), out #, :recursive => true          
           end
         end
       end
@@ -24,7 +24,7 @@ module Rsh
       def exist? remote_file_path
         remote do |ssh, sftp|
           begin
-            fattrs = sftp.stat! remote_file_path
+            fattrs = sftp.stat! fix_path(remote_file_path)
             fattrs.directory? or fattrs.file? or fattrs.symlink?
           rescue Net::SFTP::StatusException
             false
@@ -36,7 +36,7 @@ module Rsh
 
       def remove_file remote_file_path
         remote do |ssh, sftp|
-          sftp.remove! remote_file_path
+          sftp.remove! fix_path(remote_file_path)
         end
       end
 
@@ -82,17 +82,31 @@ module Rsh
       
       def upload_directory from_local_path, to_remote_path
         remote do |ssh, sftp|
-          sftp.upload! from_local_path, to_remote_path
+          sftp.upload! from_local_path, fix_path(to_remote_path)
         end
       end
       
       def download_directory from_remote_path, to_local_path
         remote do |ssh, sftp|
-          sftp.download! from_remote_path, to_local_path, :recursive => true
+          sftp.download! fix_path(from_remote_path), to_local_path, :recursive => true
         end
       end
     
       protected
+        def fix_path path
+          path.sub(/^\~/, home)
+        end
+        
+        def home
+          unless @home
+            command = 'cd ~; pwd'
+            code, stdout, stderr = exec command
+            raise "can't execute '#{command}'!" unless code == 0
+            @home = stdout.gsub("\n", '')    
+          end
+          @home
+        end
+      
         # taken from here http://stackoverflow.com/questions/3386233/how-to-get-exit-status-with-rubys-netssh-library/3386375#3386375
         def hacked_exec!(ssh, command, &block)
           stdout_data = ""

data/readme.md

@@ -1,12 +1,9 @@
-# Tiny wrapper over Net::SSH and Net::SFTP
+# Rsh - Tiny wrapper over Net::SSH/SFTP
 
 Because they are too hard to use and have terrible API design.
 
     box = Rsh::Box.new host: 'webapp.com', ssh: {user: 'root', password: 'secret'}
 
-    stdout = box.bash 'ls /'
-    code, stdout, stderr = box.exec 'ls /'
-
     box.upload_directory '/my_project', '/apps/my_project'
     box.bash 'nohup /apps/my_project/server_start'
   
@@ -14,6 +11,49 @@ Honestly my wrapper also not very good. I would like to make API looks like the
 but it requires a lots of time, maybe I'll do it later.
 So, for now it's just a small wrapper to do ssh/io operations not so painfull.
 
+# Ros - Small rake addon for configuration/depoyment management and automation
+
+It may be **usefull if Your claster has about 1-10 boxes**, and tools like Chef, Puppet, Capistrano are too complex and proprietary for your needs.
+**It's extremely easy**, there are only 3 methods.
+
+Define your packages, they are just rake tasks, so you probably know how to work with them:
+
+    namespace :os do
+      package :ruby do
+        applied?{|box| box.has_mark? :ruby}
+        apply do |box| 
+          box.bash 'apt-get install ruby'
+          box.mark :ruby
+        end
+      end
+
+      package :rails => :ruby do
+        applied?{|box| box.has_mark? :rails}
+        apply do |box| 
+          box.bash 'gem install rails'
+          box.mark :rails
+        end
+      end
+    end
+    
+Define to what it should be applied:
+
+    def each_box &b
+      host = ENV['host'] || raise(":host not defined!")
+      box = Rsh::Box.new host: host, ssh: {user: 'root', password: 'secret'}
+      b.call box
+    end
+    
+Run it:
+
+    $ rake os:rails host=webapp.com
+    
+**You can use it also for deployment**, exactly the same way, configure it the way you like, it's just rake 
+tasks. And by the way, the *box.mark ...* is just an example check, you can use anything there.
+
+It checks if the package already has been applied to box, so you can evolve your configuration and apply 
+it multiple times, it will apply only missing packages (or drop the *applied?* clause and it will be applied every run).
+
 ## TODO
 
 - introduce Entity/Dir/File (the same as in Rush)

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 0
-  - 2
-  version: 0.0.2
+  - 3
+  version: 0.0.3
 platform: ruby
 authors: 
 - Alexey Petrushin
@@ -30,6 +30,9 @@ files:
 - Rakefile
 - readme.md
 - lib/old/ssh.rb
+- lib/ros/rake.rb
+- lib/ros/ros.rb
+- lib/ros.rb
 - lib/rsh/box.rb
 - lib/rsh/drivers/abstract.rb
 - lib/rsh/drivers/local.rb
@@ -79,6 +82,6 @@ rubyforge_project:
 rubygems_version: 1.3.7
 signing_key: 
 specification_version: 3
-summary: Tiny wrapper over Net::SSH and Net::SFTP
+summary: Tiny wrapper over Net::SSH/SFTP + small rake addon for cluster configuration management
 test_files: []