reploy 0.1

data/lib/reploy.rb

@@ -0,0 +1,176 @@
+require 'fileutils'
+require 'net/ssh'
+require 'popen4'
+
+=begin
+	Reploy is simple deploying tool, similar to Vlad or Capistrano, but even simplier and faster.
+
+	How to use:
+
+	* Install a gem
+	* Write a config
+	* rake reploy:deploy
+=end
+
+class Reploy
+	def config
+		@@config
+	end
+
+	def config=(v)
+		@@config=(v)
+	end
+
+	def self.[](k)
+		@@config[k]
+	end
+
+	def self.[]=(k,v)
+		@@config[k] = v
+	end
+
+	# load Reploy with given config
+	def Reploy.load(opts)
+		@@config = opts
+		@@config.default = {}
+		if opts[:recipes].is_a? Array
+			opts[:recipes].each do |r|
+				if r =~ /\.rb$/ # It's local
+					require r
+				else
+					require "reploy/recipes/#{r}"
+				end
+			end
+		end
+		@@singleton = Reploy.new
+		self.add_tasks
+	end
+
+	def self.add_tasks
+		namespace :reploy do
+			# tasks for calling reploy from rake
+			# we can't (properly) override rake tasks, so it's methods. :(
+			%w|setup transfer deploy start stop restart nuke|.each do |t|
+				task t.to_sym do
+					self.send(t.to_sym)
+				end
+			end
+		end
+	end
+
+	def self.method_missing(s, *args, &block)
+		if args.empty?
+			@@singleton.send(s)
+		else
+			@@singleton.send(s, args)
+		end
+	end
+
+	# Run the command on remote host
+	def run(command)
+		@ssh ||= Net::SSH.start(config[:host],config[:user])
+		puts "#{config[:user]}@#{config[:host]}# #{command}"
+		@ssh.open_channel do |chan|
+			chan.on_request("exit-status") do |ch,data|
+				r = data.read_long
+				if r != 0
+					raise "Process failed with exit status #{r}"
+				end
+			end
+			chan.on_data do |ch, data|
+				data.each_line do |l|
+					puts "[r] #{l}"
+				end
+			end
+			chan.on_extended_data do |ch, type, data|
+				if type == 1
+					data.each_line do |l|
+						puts "[r] [e] #{l}"
+					end
+				end
+			end
+			chan.exec command do |ch, success|
+				raise "Running command failed." unless success
+			end
+			chan.wait
+		end
+		@ssh.loop
+	end
+
+	# Run the command on remote host, in "current" directory
+	def run_in_current(command)
+		run("cd #{@current} && #{command}")
+	end
+
+	# Create essential dir structure on remote host
+	def setup
+		run("mkdir -p "+(%w|current shared/log shared/tmp backups storage|.map {|v| "#{config[:path]}/#{v}"}.join(" ")))
+	end
+
+	# Check out all the files to be bulk-copied in directory, return its path
+	# This is naive implementation, this method is overridden in recipes like "git", which use version control to provide filelist
+	def checkout
+		yield '.' # dumb
+	end
+
+	# Run the command on local host
+	def run_local(command)
+		puts "localhost# #{command}"
+		status = POpen4::popen4 command do |stdout,stderr|
+			stdout.readlines.each do |l|
+				puts "[l] #{l}"
+			end
+			stderr.readlines.each do |l|
+				puts "[l] [e] #{l}"
+			end
+		end
+		raise "Process aborted in totally wrong way" if status.exitstatus != 0
+	end
+
+	# Prepare for transfer: create random named directory wih contents of current
+	def prepare
+		@current = "#{config[:path]}/tmp#{rand(66666)}"
+		run "cp -r #{config[:path]}/current #{@current}"
+	end
+
+	# Perform the file transfer: prepare, then sync, then copy files, then rename directory to current
+	# In case anything goes wrong, rollback and remove traces
+	def transfer
+		begin
+			prepare
+			checkout do |dir|
+				sync dir
+			end
+			copy
+			finalize
+		rescue Exception => e
+			puts "Exception: #{e}. Cleaning up..."
+			cleanup
+		end
+	end
+
+	# Finalize file transfer: rename temporary directory to current, remove or backup original "current"
+	def finalize
+		if config[:leave_backups] == true
+			run "mv #{config[:path]}/current #{config[:path]/backups}/#{Time.now.strftime("%d_%m_%Y_%H_%M")} && mv #{@current} #{config[:path]}/current"
+		else
+			run "rm -r #{config[:path]}/current && mv #{@current} #{config[:path]}/current"
+		end
+		@current = "#{config[:path]}/current"
+	end
+
+	# Remove temporary directory
+	def cleanup
+		run "rm -r #{@current}" if @current
+	end
+
+	# rm -rf in target directory, then setup
+	def nuke
+		run "rm -r #{config[:path]}/*"
+		setup
+	end
+
+	def deploy
+		transfer
+	end
+end

data/lib/reploy/recipes/git.rb

@@ -0,0 +1,34 @@
+# Git recipe for Reploy
+
+class Reploy
+	def checkout
+		raise "That really shouldn't happen." if Reploy[:git].has_key? :from
+		temp_name = "#{Dir.tmpdir}/reploy#{rand(66666)}"
+		FileUtils.mkdir_p temp_name
+		`git archive HEAD | tar xf - -C #{temp_name}`
+		yield temp_name
+		FileUtils.rm_rf temp_name
+	end
+	if Reploy[:git].has_key? :from
+		def prepare
+			@current = "#{config[:path]}/tmp#{rand(66666)}"
+			run "mkdir -p #{@current}"
+		end
+		def transfer
+			begin
+				prepare
+				# A handy optimization, if repo is local, we can do checkout instead of cloning
+				if Reploy[:git][:from] =~ %r{^/}
+					run_in_current "GIT_DIR='#{Reploy[:git][:from]}' git archive HEAD | tar xf -"
+				else
+					run "export GIT_DIR='#{config[:path]}/storage/.git' && (test -d #{config[:path]}/storage/.git && git fetch master) || (mkdir -p #{config[:path]}/storage/.git && git clone --bare) && git archive HEAD | tar xf -"
+				end
+				copy
+				finalize
+			rescue Exception => e
+				puts "Exception: #{e}. Cleaning up..."
+				cleanup
+			end
+		end
+	end
+end

data/lib/reploy/recipes/merb.rb

@@ -0,0 +1,57 @@
+class Reploy
+	def merb_params_map
+		{ :host => "-h",
+			:port => "-p",
+			:adapter => "-a",
+			:environment => "-e"
+			}
+	end
+	def merb_get_args
+		(config[:merb] or {}).map do |k,v|
+			if merb_params_map[k]
+				"#{merb_params_map[k]} #{v}"
+			else
+				nil
+			end
+		end.join(" ")
+	end
+
+	def restart
+		run_in_current "merb -K #{config[:merb][:port]} && merb -d #{merb_get_args}"
+	end
+
+	def start
+		run_in_current "merb -d #{merb_get_args}"
+	end
+
+	def stop
+		run_in_current "merb -K #{config[:merb][:port]}"
+	end
+
+	def autoupgrade
+		run_in_current "MERB_ENV=#{config[:merb][:environment]} rake db:autoupgrade"
+	end
+
+	def symlink_shared
+		run("ln -s #{config[:path]}/shared/* #{config[:path]}/current/")
+	end
+
+
+	def deploy
+		transfer
+		symlink_shared
+		restart
+	end
+end
+
+# recipe-specific tasks
+
+desc "Perform Merb schema autoupgrade"
+task "reploy:merb:autoupgrade" do
+	Reploy.autoupgrade
+end
+
+desc "Deploy Merb"
+task "reploy:merb:deploy" do
+	Reploy.deploy
+end

data/lib/reploy/recipes/rsync.rb

@@ -0,0 +1,16 @@
+# Rsync recipe. Implements sync and copy
+
+class Reploy
+	def sync(dir)
+		run_local "rsync -avz --delete #{dir}/ #{config[:user]}@#{config[:host]}:#{@current}/"
+	end
+
+	def copy
+		if config[:copy].is_a? Array
+			config[:copy].each do |a|
+				k,v = a
+				run_local "rsync -tvaz #{k} #{config[:user]}@#{config[:host]}:#{@current}/#{v}"
+			end
+		end
+	end
+end

data/lib/reploy/recipes/scp.rb

@@ -0,0 +1,39 @@
+require 'net/scp'
+
+class Reploy
+
+	def prepare
+		@current = "#{config[:path]}/tmp#{rand(66666)}"
+		run "mkdir -p #{@current}"
+	end
+
+	def sync(dir)
+		if config[:scp].has_key? :tar
+			if %w|gz gzip|.include? config[:scp][:tar]
+				pipes = ["| gzip", "| gunzip" ]
+			elsif %w|bz2 bzip2|.include? config[:scp][:tar]
+				pipes = ["| bzip2", "| bunzip2" ]
+			elsif %w|lzma|.include? config[:scp][:tar]
+				pipes = ["| lzma", "| unlzma" ]
+			else
+				pipes = ["",""]
+			end
+			archive_name = "reploy#{rand(66666)}.tmp"
+			run_local "cd #{dir} && tar c . #{pipes[0]} > #{Dir.tmpdir}/#{archive_name}"
+			run_in_current "rm -r *"
+			Net::SCP.upload!(config[:host], config[:user],"#{Dir.tmpdir}/#{archive_name}", "#{config[:path]}/current")
+			run_in_current "cat #{archive_name} #{pipes[1]} | tar xf -"
+			run_in_current "rm #{archive_name}"
+		else
+			run_in_current "rm -r *"
+			Net::SCP.upload!(config[:host], config[:user],"#{Dir.tmpdir}/#{archive_name}", "#{config[:path]}/current", :recursive)
+		end
+	end
+
+	def copy(files)
+		files.each do |a|
+			k,v = a
+			Net::SCP.upload!(config[:host], config[:user],k, "#{config[:path]}/current/#{v}")
+		end
+	end
+end

metadata

@@ -0,0 +1,59 @@
+--- !ruby/object:Gem::Specification 
+name: reploy
+version: !ruby/object:Gem::Version 
+  version: "0.1"
+platform: ruby
+authors: 
+- Voker57
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-09-06 00:00:00 +04:00
+default_executable: 
+dependencies: []
+
+description: Simple tool for automation deployment
+email: voker57@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/reploy/recipes/git.rb
+- lib/reploy/recipes/merb.rb
+- lib/reploy/recipes/scp.rb
+- lib/reploy/recipes/rsync.rb
+- lib/reploy.rb
+has_rdoc: true
+homepage: http://bitcheese.net/wiki/reploy
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+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.4
+signing_key: 
+specification_version: 3
+summary: Simple deploying tool
+test_files: []
+