joekhoobyar-capistrano-extensions 0.0.0

data/Rakefile

@@ -0,0 +1,40 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'rake/rdoctask'
+require 'rake/testtask'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |s|
+    s.name = "capistrano-extensions"
+    s.homepage = "http://github.com/joekhoobyar/capistrano-extensions"
+    s.summary = "Joe Khoobyar's Capistrano Extensions"
+    s.description = %Q{Joe Khoobyar's Capistrano Extensions}
+
+    s.authors = ["Joe Khoobyar"]
+    s.email = "joe@ankhcraft.com"
+  end
+rescue LoadError
+  puts "Jeweler not available. Install it with: sudo gem install technicalpickles-jeweler -s http://gems.github.com"
+end
+
+# ---------  RDoc Documentation ---------
+desc "Generate RDoc documentation"
+Rake::RDocTask.new(:rdoc) do |rdoc|
+  rdoc.options << '--line-numbers' << '--inline-source' <<
+    '--main' << 'README' <<
+    '--title' << "Joe Khoobyar's Capistrano Extensions" <<
+    '--charset' << 'utf-8'
+  rdoc.rdoc_dir = "doc"
+  rdoc.rdoc_files.include 'README'
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+
+Rake::TestTask.new do |t|
+  t.libs << "test"
+  t.libs << "lib"
+end
+
+task :default => :package
+

data/VERSION.yml

@@ -0,0 +1,4 @@
+--- 
+:minor: 0
+:patch: 0
+:major: 0

data/lib/capistrano_extensions/files.rb

@@ -0,0 +1,18 @@
+require File.join(File.dirname(__FILE__), %w(files local.rb))
+require File.join(File.dirname(__FILE__), %w(files remote.rb))
+
+module CapistranoExtensions::Files
+  class_eval(Local.public_instance_methods(false).map do |m|
+    "def #{m}(*f) send(_via.to_s + '_files').#{m}(*f) end"
+  end.join("\n"))
+
+  def _via
+    if ! @config.exists?(:files_via) or @config.files_via != :local then
+      :remote
+    else
+      :local
+    end
+  end
+end
+
+Capistrano.plugin :files, CapistranoExtensions::Files

data/lib/capistrano_extensions/files/local.rb

@@ -0,0 +1,40 @@
+require 'fileutils'
+
+module CapistranoExtensions
+  module Files
+    module Local
+
+      def tail_f(file, n=10)
+        unless defined? File::Tail::Logfile
+          gem 'file-tail'
+          require 'file/tail'
+        end
+        File::Tail::Logfile.tail(file, :backward=>n) do |line| puts line end
+      end
+
+      def upload(from, to)
+        cp(from, to)
+      end
+
+      def download(from, to)
+        cp(from, to)
+      end
+
+
+      include FileUtils::Verbose
+
+      public *FileUtils::Verbose.methods(false)
+      private *%w(copy_entry copy_file copy_stream
+                  remove_entry remove_entry_secure remove_file
+                  compare_file compare_stream
+                  uptodate?)
+
+      class_eval(%w(exists? directory? executable?).map do |m|
+        "def #{m}(f) File.#{m}(f) end"
+      end.join("\n"))
+
+    end
+  end
+end
+
+Capistrano.plugin :local_files, CapistranoExtensions::Files::Local

data/lib/capistrano_extensions/files/remote.rb

@@ -0,0 +1,189 @@
+require 'fileutils'
+
+module CapistranoExtensions
+  module Files
+    module Remote
+
+      def tail_f(file, n=10)
+        cmd = "tail -n #{n} -f #{_q file}"
+        _via == :system ? system(cmd) : stream(cmd)
+      end
+
+      def upload(*args)
+        _via == :system ? cp(*args) : upload(*args)
+      end
+
+      def download(*args)
+        _via == :system ? cp(*args) : download(*args)
+      end
+
+      def cd(dir, options={})
+        if block_given?
+          dir, dir2 = pwd, dir
+          _r 'cd', dir2
+          yield
+        end  
+        _r 'cd', dir
+      end
+
+      def pwd
+        capture("pwd", :via => _via)
+      end
+
+      def mkdir(*args)
+        options = args.pop if Hash === args.last
+        _r 'mkdir', args
+      end
+
+      def mkdir_p(*args)
+        options = args.pop if Hash === args.last
+        _r 'mkdir -p', args
+      end
+
+      def rmdir(*args)
+        options = args.pop if Hash === args.last
+        _r 'rmdir', args
+      end
+
+      def ln(*args)
+        options = args.pop if Hash === args.last
+        _r 'ln', args, 2
+      end
+
+      def ln_s(*args)
+        options = args.pop if Hash === args.last
+        _r 'ln -s', args, 2
+      end
+
+      def ln_sf(*args)
+        options = args.pop if Hash === args.last
+        _r 'ln -sf', args, 2
+      end
+
+      def cp(*args)
+        options = args.pop if Hash === args.last
+        _r 'cp', args
+      end
+
+      def cp_r(*args)
+        options = args.pop if Hash === args.last
+        _r 'cp -r', args
+      end
+
+      def mv(*args)
+        options = args.pop if Hash === args.last
+        _r 'mv', args, 2
+      end
+
+      def rm(*args)
+        options = args.pop if Hash === args.last
+        _r 'rm', args
+      end
+
+      def rm_r(*args)
+        options = args.pop if Hash === args.last
+        _r 'rm -r', args
+      end
+
+      def rm_rf(*args)
+        options = args.pop if Hash === args.last
+        _r 'rm -rf', args
+      end
+
+      def install(*args)
+        options = args.pop if Hash === args.last
+        _r 'install', args, 2
+      end
+
+      def chmod(*args)
+        options = args.pop if Hash === args.last
+        _r 'chmod', args
+      end
+
+      def chmod_R(*args)
+        options = args.pop if Hash === args.last
+        _r 'chmod -R', args
+      end
+
+      def chown(*args)
+        options = args.pop if Hash === args.last
+        _r 'chown', args
+      end
+
+      def chown_R(*args)
+        options = args.pop if Hash === args.last
+        _r 'chown -R', args
+      end
+
+      def touch(*args)
+        options = args.pop if Hash === args.last
+        _r 'touch', args
+      end
+
+      def exists?(a, options={:verbose=>false})
+        silence!(options) { _r "[ -f #{_q a} ]" }
+      end
+
+      def directory?(a, options={:verbose=>false})
+        silence!(options) { _r "[ -d #{_q a} ]" }
+      end
+
+      def executable?(a, options={:verbose=>false})
+        silence!(options) { _r "[ -x #{_q a} ]" }
+      end
+
+
+    private
+
+      def silence!(quiet=false)
+        quiet = (FalseClass===quiet[:verbose]) if Hash === quiet
+        orig_quiet, @quiet = @quiet, quiet
+        yield
+      ensure
+        @quiet = orig_quiet
+      end
+
+      def _r(cmd, args=nil, min=nil)
+        case args
+        when NilClass
+          raise ArgumentError unless min.nil? or min.zero?
+        when Array
+          args = args.flatten
+          raise ArgumentError if (min || 1) > args.length
+          cmd = "#{cmd} #{_q(*args)}" if args.any?
+        else
+          raise ArgumentError if min.nil? or min < 1
+          cmd = "#{cmd} #{_q args}"
+        end
+
+        case (v = _via)
+        when :system
+          @quiet or $stderr.puts cmd
+          system cmd
+        when :sudo
+          sudo cmd, :as => @config.fetch(:runner, 'app')
+        else
+          invoke_command cmd, :via => v
+        end          
+      end
+
+      def _q(*list)
+        list.map { |l| "'#{l.gsub("'", "\\'")}'" }.join ' '
+      end
+
+      def _via
+        case (v = files_via)
+        when :local
+          :system
+        when :remote, NilClass
+          :runner
+        else 
+          v
+        end
+      end
+
+    end
+  end
+end
+
+Capistrano.plugin :remote_files, CapistranoExtensions::Files::Remote

data/lib/capistrano_extensions/service.rb

@@ -0,0 +1,109 @@
+module CapistranoExtensions
+  module Service
+
+    LSB_DEFAULT_ACTIONS = %w(status start stop restart)
+
+    CRM_DEFAULT_ACTIONS = [	[ :status, '-W' ],
+                            [ :summary, "-x | awk '/^raw xml:/ { exit }; { print }'" ],
+                            [ :start, "--meta -d 'target_role'" ],
+                            [ :stop, "--meta -p 'target_role' -v 'stopped'" ] ]
+
+    CRM_OCF_DEFAULT_ACTIONS = [ [ :validate, '-c -C' ],
+                                [ :monitor, '-c -m' ] ]
+
+    SVC_ACTION_CAPTIONS = Hash.new do |h,k|
+      h[k] = "#{k.to_s.capitalize} Service"
+    end
+    SVC_ACTION_CAPTIONS.update :status => 'Check Status', :check => 'Check Config', :summary => 'Status Summary'
+
+    def crm(id,*args)
+      svc_desc = next_description(:reset)
+      svc_cmd = "/usr/sbin/crm_resource -r #{id.to_s.split(':').last}"
+      svc_actions = CRM_DEFAULT_ACTIONS
+
+      if Hash === args.last
+        options = args.pop
+        svc_desc = id.to_s.capitalize unless svc_desc or options.delete(:hide)
+        svc_actions += args.shift if Array === args.first
+      else
+        options = {}
+      end
+
+      namespace id do
+        desc "#{svc_desc}: #{SVC_ACTION_CAPTIONS[:status]}" if svc_desc
+        task :default, options do
+            sudo "#{svc_cmd} -W"
+        end
+
+        svc_actions.each do |svc_action,svc_args|
+          svc_action = svc_action.intern unless Symbol===svc_action
+          desc "#{svc_desc}: #{SVC_ACTION_CAPTIONS[svc_action]}" if svc_desc
+          task svc_action, options do
+            sudo "#{svc_cmd} #{svc_args}"
+          end
+        end
+
+        instance_eval { yield } if block_given?
+      end
+    end
+
+    def crm_ocf(id,*args)
+      svc_desc = next_description(:reset)
+      svc_cmd = "ocf_resource -g #{id.to_s.split(':').last}"
+      svc_actions = CRM_OCF_DEFAULT_ACTIONS
+
+      if Hash === args.last
+        options = args.pop
+        svc_desc = id.to_s.capitalize unless svc_desc or options.delete(:hide)
+        svc_actions += args.shift if Array === args.first
+      else
+        options = {}
+      end
+
+      namespace id do
+        svc_actions.each do |svc_action,svc_args|
+          svc_action = svc_action.intern unless Symbol===svc_action
+          desc "#{svc_desc}: #{SVC_ACTION_CAPTIONS[svc_action]}" if svc_desc
+          task svc_action, options do
+            sudo "#{svc_cmd} #{svc_args}"
+          end
+        end
+
+        instance_eval { yield } if block_given?
+      end
+    end
+
+    def lsb(id,*args)
+      svc_desc = next_description(:reset)
+      svc_cmd = "/etc/init.d/#{id.to_s.split(':').last}"
+      svc_actions = LSB_DEFAULT_ACTIONS 
+
+      if Hash === args.last
+        options = args.pop
+        svc_desc = id.to_s.capitalize unless svc_desc or options.delete(:hide)
+        svc_actions += args.shift if Array === args.first
+      else
+        options = {}
+      end
+
+      namespace id do
+        desc "#{svc_desc}: #{SVC_ACTION_CAPTIONS[:status]}" if svc_desc
+        task :default, options do
+            sudo "#{svc_cmd} status"
+        end
+
+        svc_actions.each do |svc_action|
+          svc_action = svc_action.intern
+          desc "#{svc_desc}: #{SVC_ACTION_CAPTIONS[svc_action]}" if svc_desc
+          task svc_action, options do
+            sudo "#{svc_cmd} #{svc_action}"
+          end
+        end
+
+        instance_eval { yield } if block_given?
+      end
+    end
+  end
+end
+
+Capistrano.plugin :service, CapistranoExtensions::Service

data/lib/jk_capistrano_extensions.rb

@@ -0,0 +1,8 @@
+unless Capistrano::Configuration.respond_to?(:instance)
+  abort "jk-capistrano-extensions requires Capistrano 2"
+end
+require 'capistrano'
+
+require "#{File.dirname(__FILE__)}/capistrano_extensions/file_transfer.rb"
+require "#{File.dirname(__FILE__)}/capistrano_extensions/files.rb"
+require "#{File.dirname(__FILE__)}/capistrano_extensions/service.rb"

metadata

@@ -0,0 +1,60 @@
+--- !ruby/object:Gem::Specification 
+name: joekhoobyar-capistrano-extensions
+version: !ruby/object:Gem::Version 
+  version: 0.0.0
+platform: ruby
+authors: 
+- Joe Khoobyar
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-04-23 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: Joe Khoobyar's Capistrano Extensions
+email: joe@ankhcraft.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README
+files: 
+- Rakefile
+- VERSION.yml
+- lib/capistrano_extensions/files.rb
+- lib/capistrano_extensions/files/local.rb
+- lib/capistrano_extensions/files/remote.rb
+- lib/capistrano_extensions/service.rb
+- lib/jk_capistrano_extensions.rb
+- README
+has_rdoc: true
+homepage: http://github.com/joekhoobyar/capistrano-extensions
+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.2.0
+signing_key: 
+specification_version: 3
+summary: Joe Khoobyar's Capistrano Extensions
+test_files: []
+