zfben_rails_rake 0.0.10 → 0.0.11

data/lib/zfben_rails_rake/tasks/remote.rb

@@ -0,0 +1,94 @@
+desc 'do remote job'
+task :remote, [:job] do |task, args|
+  args = args.to_hash
+  err 'No job found' unless args.has_key? :job
+  job = File.join Rails.root, args[:key] + '.rb'
+  err args[:job] + '.rb is not exists!' unless File.exists? release
+  
+  require 'net/ssh'
+  @config = {}
+  
+  def ssh user_host, *options
+    user_host = user_host.split '@'
+    if user_host.length != 2
+      err 'ssh error, you should set ssh like `username@host`'
+    else
+      @config[:ssh] = {
+        :user => user_host[0],
+        :host => user_host[1],
+        :options => options[0]
+      }
+    end
+  end
+  
+  def remote_path path
+    @config[:remote_path] = path
+  end
+  
+  def before_connect &block
+    if block
+      @cmd_list = []
+      block.call
+      if @cmd_list.length > 0
+        @config[:before_connect] = @cmd_list
+      end
+    end
+  end
+  
+  def after_connect &block
+    if block
+      @cmd_list = []
+      block.call
+      if @cmd_list.length > 0
+        @config[:after_connect] = @cmd_list
+      end
+    end
+  end
+  
+  def rake task
+    @cmd_list.push 'rake ' + task
+  end
+  
+  def run cmd
+    @cmd_list.push 'run ' + cmd
+  end
+  
+  load job
+  
+  p @config
+  
+  if @config.has_key?(:before_connect)
+    @config[:before_connect].each do |cmd|
+      sys cmd
+    end
+  end
+  
+  Net::SSH.start(@config[:ssh][:host], @config[:ssh][:user]) do |net|
+    @net = net
+    def exec cmd
+      p "#{@config[:ssh][:user]}@#{@config[:ssh][:host]}: " + cmd
+      cmd = "cd #{@config[:remote_path]};#{cmd}"
+      channel = @net.open_channel do |ch|
+        ch.exec cmd do |ch, success|
+          err "could not execute command: #{cmd}" unless success
+          
+          ch.on_data do |c, data|
+            data.split("\n").each{ |s| p s }
+          end
+          
+          ch.on_extended_data do |c, type, data|
+            err data
+          end
+        end
+      end
+      channel.wait
+    end
+    
+    p 'ssh connect success'
+    if @config.has_key?(:after_connect)
+      @config[:after_connect].each do |cmd|
+        exec cmd
+      end
+    end
+  end
+end

data/lib/zfben_rails_rake/tasks/unicorn.rb

@@ -17,13 +17,13 @@ namespace :unicorn do
   task :restart, [:env, :config] => [:stop, :start]
   
   desc 'copy unicorn.rb to root path'
-  task :copy, [:processes, :port] do |task, args|
+  task :copy, [:processes] do |task, args|
     path = File.join(Rails.root, 'unicorn.rb')
     if File.exists? path
       err 'unicorn.rb is exists! Please remove it and run again.'
     else
-      args = { :processes => 1, :port => 8080 }.merge args.to_hash
-      p file = "# Added by zfben_rails_rake\nworker_processes #{args[:processes]}\nlisten #{args[:port]}, :tcp_nopush => true, :tcp_nodelay => true\npid 'tmp/unicorn.pid'\n# End zfben_rails_rake"
+      args = { :processes => 1 }.merge args.to_hash
+      p file = "# Added by zfben_rails_rake\nworker_processes #{args[:processes]}\nlisten File.realpath(File.dirname(__FILE__)) << '/tmp/unicorn.sock', :tcp_nopush => true, :tcp_nodelay => true\npid 'tmp/unicorn.pid'\npreload_app true\n# End zfben_rails_rake"
       File.open(path, 'w'){ |f| f.write file }
     end
   end

data/lib/zfben_rails_rake/version.rb

@@ -1,3 +1,3 @@
 module ZfbenRailsRake
-  VERSION = "0.0.10"
+  VERSION = "0.0.11"
 end

data/zfben_rails_rake.gemspec

@@ -22,4 +22,5 @@ Gem::Specification.new do |s|
   s.add_dependency 'rainbow'
   s.add_dependency 'unicorn'
   s.add_dependency 'request-log-analyzer'
+  s.add_dependency 'net-ssh'
 end

metadata

@@ -2,7 +2,7 @@
 name: zfben_rails_rake
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.0.10
+  version: 0.0.11
 platform: ruby
 authors: 
 - Ben
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-06-09 00:00:00 Z
+date: 2011-06-11 00:00:00 Z
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: rainbow
@@ -45,6 +45,17 @@ dependencies:
         version: "0"
   type: :runtime
   version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: net-ssh
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+  type: :runtime
+  version_requirements: *id004
 description: ""
 email: 
 - ben@zfben.com
@@ -64,6 +75,7 @@ files:
 - lib/zfben_rails_rake/tasks/git.rb
 - lib/zfben_rails_rake/tasks/helper.rb
 - lib/zfben_rails_rake/tasks/log.rb
+- lib/zfben_rails_rake/tasks/remote.rb
 - lib/zfben_rails_rake/tasks/unicorn.rb
 - lib/zfben_rails_rake/version.rb
 - zfben_rails_rake.gemspec