mybot 0.0.2 → 0.0.3

data/vendor/lib/core/ubuntu/mysql.rb

@@ -1,6 +1,7 @@
 module Core
 	module Ubuntu
 		class Mysql < Mybot::Lib
+			PROMPT = "mysql -u %s --password=%s -e \"%s\""
 
 			def initialize(node)
 				@node = node
@@ -78,6 +79,136 @@ module Core
 				options[:dir] = "/var/lib/mysql"
 				@node.fs.remove(options)
 			end
+
+			def database_list(options = {})
+				error "root_password is required" unless options[:root_password]
+
+				query = "SHOW DATABASES;"
+				options[:cmd] = sprintf(PROMPT, "root", options[:root_password], query)
+
+				databases = @node.run(options).stdout.split("\r\n")
+				databases[3..-2].map do |db|
+					db.gsub("|", "").strip
+				end
+			end
+
+			def database_exists?(options = {})
+				error "database is required" unless options[:database]
+
+				database_list(options).include?(options[:database])
+			end
+
+			def database_create(options = {})
+				error "root_password is required" unless options[:root_password]
+				error "database is required" unless options[:database]
+
+				options[:character_set] ||= "utf8"
+				options[:collate] ||= "utf8_general_ci"
+
+				query = sprintf("CREATE DATABASE %s CHARACTER SET %s COLLATE %s;",
+					options[:database], options[:character_set], options[:collate])
+
+				options[:cmd] = sprintf(PROMPT, "root", options[:root_password], query)
+				@node.run(options)
+			end
+
+			def database_remove(options = {})
+				error "root_password is required" unless options[:root_password]
+				error "database is required" unless options[:database]
+
+				query = sprintf("DROP DATABASE %s;", options[:database])
+
+				options[:cmd] = sprintf(PROMPT, "root", options[:root_password], query)
+				@node.run(options)
+			end
+
+			def user_list(options = {})
+				error "root_password is required" unless options[:root_password]
+
+				query = "USE mysql; SELECT User from user;"
+				options[:cmd] = sprintf(PROMPT, "root", options[:root_password], query)
+				users = @node.run(options).stdout.split("\r\n")
+				users[3..-2].map do |u|
+					u.gsub("|", "").strip
+				end
+			end
+
+			def user_exists?(options = {})
+				user_list(options).include?(options[:username])
+			end
+
+			def user_create(options = {})
+				error "root_password is required" unless options[:root_password]
+				error "username is required" unless options[:username]
+				error "password is required" unless options[:password]
+
+				options[:host] ||= "localhost"
+
+				query = "CREATE USER '#{options[:username]}'@"
+				query += "'#{options[:host]}' IDENTIFIED BY '#{options[:password]}';"
+
+				options[:cmd] = sprintf(PROMPT, "root", options[:root_password], query)
+				@node.run(options)
+			end
+
+			def user_remove(options = {})
+				error "root_password is required" unless options[:root_password]
+				error "username is required" unless options[:username]
+
+				options[:host] ||= "localhost"
+
+				query = "DROP USER '#{options[:username]}'@'#{options[:host]}';"
+
+				options[:cmd] = sprintf(PROMPT, "root", options[:root_password], query)
+				@node.run(options)
+			end
+
+			def grant_privileges(options = {})
+				error "root_password is required" unless options[:root_password]
+				error "username is required" unless options[:username]
+
+				options[:host] ||= "localhost"
+				options[:database] ||= "*"
+				options[:table] ||= "*"
+
+				query = sprintf("GRANT ALL PRIVILEGES ON %s.%s TO '%s'@'%s';",
+					options[:database], options[:table], options[:username],
+					options[:host])
+
+				options[:cmd] = sprintf(PROMPT, "root", options[:root_password], query)
+				@node.run(options)
+			end
+
+			def flush_privileges(options = {})
+				error "root_password is required" unless options[:root_password]
+
+				query = "FLUSH PRIVILEGES;"
+
+				options[:cmd] = sprintf(PROMPT, "root", options[:root_password], query)
+				@node.run(options)
+			end
+
+			def dump(options = {})
+				error "root_password is required" unless options[:root_password]
+				error "database is required" unless options[:database]
+
+				tmp = @node.fs.mktemp({ :pattern => "/tmp/dump.XXXX" })
+
+				options[:cmd] = sprintf("mysqldump -u %s --password=%s %s > %s",
+					"root", options[:root_password], options[:database], tmp)
+				@node.run(options)
+				tmp
+			end
+
+			def load(options = {})
+				error "root_password is required" unless options[:root_password]
+				error "database is required" unless options[:database]
+				error "file is required" unless options[:file]
+
+				options[:cmd] = sprintf("mysql -u %s --password=%s %s < %s",
+					"root", options[:root_password], options[:database], options[:file])
+				@node.run(options)
+			end
 		end
 	end
 end

data/vendor/lib/core/ubuntu/nginx.rb

@@ -0,0 +1,43 @@
+module Core
+	module Ubuntu
+		class Nginx < Mybot::Lib
+			def initialize(node)
+				@node = node
+				@node.use [
+					"core/fs",
+					"core/ubuntu/apt",
+					"core/ubuntu/service"
+				]
+
+				@nginxconf_tpl = tpl_file "core/ubuntu/nginx/nginx.conf.erb"
+			end
+
+			def installed?(options = {})
+				@node.fs.exists?({ :file => "/usr/sbin/nginx"} )
+			end
+
+			def install(options = {})
+				options[:pkg] = "nginx"
+				@node.apt.install(options)
+			end
+
+			def remove(options = {})
+				options[:pkg] = "nginx"
+				@node.apt.remove(options)
+			end
+
+			def reconfigure(options = {})
+				error "config is required" unless options[:config]
+
+				options[:file] = "/etc/nginx/nginx.conf"
+				@node.fs.remove(options)
+
+				options[:content] = @nginxconf_tpl.render(options[:config])
+				@node.fs.create(options)
+
+				options[:service] = "nginx"
+				@node.service.restart(options)
+			end
+		end
+	end
+end

data/vendor/lib/core/ubuntu/php.rb

@@ -0,0 +1,76 @@
+module Core
+	module Ubuntu
+		class Php < Mybot::Lib
+			def initialize(node)
+				@node = node
+				@node.use [
+					"core/fs",
+					"core/ubuntu/apt",
+					"core/ubuntu/service"
+				]
+				@phpini_tpl = tpl_file "core/ubuntu/php/php.ini.erb"
+			end
+
+			def installed?(options = {})
+				options[:file] = "/usr/bin/php"
+				@node.fs.exists?(options)
+			end
+
+			def install(options = {})
+				options[:pkg] = "php5-common php5-cli php5-fpm"
+				@node.apt.install(options)
+
+				options[:service] = "php5-fpm"
+				@node.service.restart(options)
+			end
+
+			def remove(options = {})
+				options[:service] = "php5-fpm"
+				@node.service.stop(options)
+
+				options[:pkg] = "php5-common php5-cli php5-fpm"
+				@node.apt.remove(options)
+			end
+
+			def ext_installed?(options = {})
+				error "ext is required" unless options[:ext]
+
+				options[:file] = "/usr/lib/php5/20100525/#{options[:ext]}.so"
+				@node.fs.exists?(options)
+			end
+
+			def ext_install(options = {})
+				error "ext is required" unless options[:ext]
+
+				options[:pkg] = "php5-#{options[:ext]}"
+				@node.apt.install(options)
+
+				options[:service] = "php5-fpm"
+				@node.service.restart(options)
+			end
+
+			def ext_remove(options = {})
+				error "ext is required" unless options[:ext]
+
+				options[:pkg] = "php5-#{options[:ext]}"
+				@node.apt.remove(options)
+
+				options[:service] = "php5-fpm"
+				@node.service.restart(options)
+			end
+
+			def reconfigure(options = {})
+				error "config is required" unless options[:config]
+
+				options[:file] = "/etc/php5/fpm/php.ini"
+				@node.fs.remove(options)
+
+				options[:content] = @phpini_tpl.render(options[:config])
+				@node.fs.create(options)
+
+				options[:service] = "php5-fpm"
+				@node.service.restart(options)
+			end
+		end
+	end
+end

data/vendor/lib/core/ubuntu/rails.rb

@@ -0,0 +1,105 @@
+module Core
+	module Ubuntu
+		class Rails < Mybot::Lib
+			def initialize(node)
+				@node = node
+				@node.use [
+					"core/fs",
+					"core/ubuntu/ruby"
+				]
+			end
+
+			def setup_deployment(options = {})
+				error "app_name is required" unless options[:app_name]
+
+				dirs = [
+					"/home/#{options[:app_name]}/releases",
+					"/home/#{options[:app_name]}/shared",
+					"/home/#{options[:app_name]}/shared/bundle",
+					"/home/#{options[:app_name]}/shared/log",
+					"/home/#{options[:app_name]}/shared/pids",
+					"/home/#{options[:app_name]}/shared/system",
+				]
+
+				dirs.each do |d|
+					options[:dir] = d
+					@node.fs.require_created(options)
+				end
+			end
+
+			def deploy(options = {})
+				error "app_name is required" unless options[:app_name]
+				error "release is required" unless options[:release]
+
+				releases_path = "/home/#{options[:app_name]}/releases"
+				release_path = "#{releases_path}/#{options[:release]}"
+				shared_path = "/home/#{options[:app_name]}/shared"
+				current_path = "/home/#{options[:app_name]}/current"
+
+				# upload release archive
+				options[:from] = options[:to] = "/tmp/#{options[:release]}.tar.gz"
+				@node.fs.upload(options)
+
+				# extract archive
+				options[:to] = releases_path
+				@node.fs.extract(options)
+
+				# install bundle
+				options[:ruby_version] = "1.9.3-p194"
+				options[:cmd] = "bundle install " +
+					"--gemfile #{release_path}/Gemfile " +
+					"--path /home/#{options[:app_name]}/shared/bundle " +
+					"--deployment --without development test"
+				options[:cwd] = release_path
+				@node.ruby.rbenv_exec(options)
+
+				# link public/system
+				options[:dir] = "#{release_path}/public/system"
+				@node.fs.remove(options)
+
+				options[:dir] = "#{release_path}/public"
+				@node.fs.require_created(options)
+
+				options[:from] = "#{shared_path}/public/system"
+				options[:to] = "#{release_path}/public/system"
+				@node.fs.link(options)
+
+				# link log
+				options[:dir] = "#{release_path}/log"
+				@node.fs.remove(options)
+
+				options[:from] = "#{shared_path}/log"
+				options[:to] = "#{release_path}/log"
+				@node.fs.link(options)
+
+				# link tmp/pids
+				options[:dir] = "#{release_path}/tmp/pids"
+				@node.fs.remove(options)
+
+				options[:dir] = "#{release_path}/tmp"
+				@node.fs.require_created(options)
+
+				options[:from] = "#{shared_path}/pids"
+				options[:to] = "#{release_path}/tmp/pids"
+				@node.fs.link(options)
+
+				# link current
+				options[:dir] = current_path
+				@node.fs.remove(options)
+
+				options[:from] = release_path
+				options[:to] = current_path
+				@node.fs.link(options)
+
+				# restart passenger
+				options[:file] = "#{release_path}/tmp/restart.txt"
+
+				# repair permissions
+				options[:dir] = "/home/#{options[:app_name]}"
+				options[:user] = options[:app_name]
+				options[:group] =  options[:app_name]
+				@node.fs.chown(options)
+			end
+		end
+	end
+end

data/vendor/lib/core/ubuntu/ruby.rb

@@ -38,6 +38,7 @@ module Core
 				options[:pkg] = "ruby-build"
 				@node.apt.install(options)
 
+				options.delete(:file)
 				options[:dir] = RBENV_ROOT
 				@node.fs.create(options)
 			end
@@ -58,7 +59,7 @@ module Core
 					"RBENV_ROOT" => RBENV_ROOT
 				}
 				options[:cmd] = "rbenv versions"
-				versions = @node.run(options)
+				versions = @node.run(options).stdout
 				versions.split("\r\n").map { |v| v.strip }
 			end
 
@@ -102,14 +103,14 @@ module Core
 				}
 
 				options[:cmd] = "rbenv exec #{options[:cmd]}"
-				@node.run(options)
+				@node.run(options).stdout
 			end
 
 			def rbenv_gem_list(options = {})
 				options[:cmd] = "gem list"
 				gems = rbenv_exec(options)
 				gems = gems.split("\r\n").reject do |g|
-					g == "" || g == "*** LOCAL GEMS ***"
+					g.strip == "" || g == "*** LOCAL GEMS ***"
 				end.map do |g|
 					parts = g.split("(")
 					{

data/vendor/lib/core/ubuntu/static.rb

@@ -0,0 +1,50 @@
+module Core
+	module Ubuntu
+		class Static < Mybot::Lib
+			def initialize(node)
+				@node = node
+				@node.use [
+					"core/fs"
+				]
+			end
+
+			def setup_deployment(options = {})
+				error "app_name is required" unless options[:app_name]
+
+				options[:dir] = "/home/#{options[:app_name]}/releases"
+				@node.fs.require_created(options)
+			end
+
+			def deploy(options = {})
+				error "app_name is required" unless options[:app_name]
+				error "release is required" unless options[:release]
+
+				releases_path = "/home/#{options[:app_name]}/releases"
+				release_path = "#{releases_path}/#{options[:release]}"
+				current_path = "/home/#{options[:app_name]}/current"
+
+				# upload release archive
+				options[:from] = options[:to] = "/tmp/#{options[:release]}.tar.gz"
+				@node.fs.upload(options)
+
+				# extract archive
+				options[:to] = releases_path
+				@node.fs.extract(options)
+
+				# link current
+				options[:dir] = current_path
+				@node.fs.remove(options)
+
+				options[:from] = release_path
+				options[:to] = current_path
+				@node.fs.link(options)
+
+				# repair permissions
+				options[:dir] = "/home/#{options[:app_name]}"
+				options[:user] = options[:app_name]
+				options[:group] =  options[:app_name]
+				@node.fs.chown(options)
+			end
+		end
+	end
+end