capricorn 0.2.25 → 2.0.0

data/lib/capricorn.rb

@@ -1,101 +1,7 @@
-require 'rubygems'
+module Capricorn
 
-autoload :URI,         'uri'
-autoload :Etc,         'etc'
-autoload :DRb,         'drb'
-autoload :TSort,       'tsort'
-autoload :Logger,      'logger'
-autoload :OpenSSL,     'drb/ssl'
-autoload :FileUtils,   'fileutils'
-autoload :DRbUndumped, 'drb'
+  require 'capricorn/version'
+  require 'capricorn/driver'
+  require 'capricorn/system_context'
 
-module Capricorn
-  base = File.expand_path(File.dirname(__FILE__))
-  
-  autoload :Actor,            (base+'/capricorn/actor')
-  autoload :Server,           (base+'/capricorn/server')
-  autoload :Client,           (base+'/capricorn/client')
-  autoload :System,           (base+'/capricorn/system')
-  autoload :Daemon,           (base+'/capricorn/daemon')
-  autoload :JobQueue,         (base+'/capricorn/job_queue')
-  autoload :Satellite,        (base+'/capricorn/satellite')
-  autoload :AppRunner,        (base+'/capricorn/app_runner')
-  autoload :ExceptionHandler, (base+'/capricorn/exception_handler')
-  
-  module Actors
-    base = File.expand_path(File.dirname(__FILE__))
-    
-    autoload :BaseActor,      (base+'/capricorn/actors/base_actor')
-    autoload :MysqlActor,     (base+'/capricorn/actors/mysql_actor')
-    autoload :PleskActor,     (base+'/capricorn/actors/plesk_actor')
-    autoload :ApacheActor,    (base+'/capricorn/actors/apache_actor')
-    autoload :Sqlite3Actor,   (base+'/capricorn/actors/sqlite3_actor')
-    autoload :HostFileActor,  (base+'/capricorn/actors/host_file_actor')
-    autoload :PassengerActor, (base+'/capricorn/actors/passenger_actor')
-  end
-  
-  module Apps
-    base = File.expand_path(File.dirname(__FILE__))
-    
-    autoload :Dev,       (base+'/capricorn/apps/dev')
-    autoload :Jobs,      (base+'/capricorn/apps/jobs')
-    autoload :Server,    (base+'/capricorn/apps/server')
-    autoload :Engines,   (base+'/capricorn/apps/engines')
-    autoload :Satellite, (base+'/capricorn/apps/satellite')
-  end
-  
-  DEFAULT_ROOT_SYSTEM_DIR = '/var/capricorn'
-  DEFAULT_USER_SYSTEM_DIR = '~/.capricorn'
-  STOP_STATUS    = 101
-  RESTART_STATUS = 102
-  RELOAD_STATUS  = 103
-  QUICK_CERT     = "http://segment7.net/projects/ruby/QuickCert/QuickCert-1.0.2.tar.gz"
-  
-  THOR_VERSION          = '= 0.9.9'
-  RUBIGEN_VERSION       = '>= 1.5.2'
-  
-  Capricorn::ExceptionHandler.setup
-  extend ExceptionHandler
-  
-  def self.client(token=nil)
-    unless @client
-      @client = Capricorn::Client.current(token)
-      unless @client
-        puts "Failed to connect to the capricorn!"
-        exit(1)
-      end
-    end
-    @client
-  end
-  
-  def self.system
-    @system ||= Capricorn::System.shared
-  end
-  
-  def self.version
-    unless @version
-      if __FILE__ =~ /\/capricorn-([^\/]+)\//
-        @version = $1
-      else
-        @version = 'edge'
-      end
-    end
-    @version
-  end
-  
-  def self.server?(value=nil)
-    @is_server = value unless value.nil?
-    @is_server
-  end
-  
-  def self.runtime_gem(gem, version='>= 0.0.0', lib=nil)
-    begin
-      gem(gem, version)
-      require(lib || gem)
-    rescue LoadError
-      puts "You must install #{gem} (#{version}) to use this.\nPlease run: [sudo] gem install #{gem}"
-      exit(1)
-    end
-  end
-  
-end
+end

data/lib/capricorn/driver.rb

@@ -0,0 +1,86 @@
+require 'bert'
+
+unless Object.respond_to?(:instance_exec)
+  class Object
+    module InstanceExecHelper; end
+    include InstanceExecHelper
+    def instance_exec(*args, &block) # !> method redefined; discarding old instance_exec
+      mname = "__instance_exec_#{Thread.current.object_id.abs}_#{object_id.abs}"
+      InstanceExecHelper.module_eval{ define_method(mname, &block) }
+      begin
+        ret = __send__(mname, *args)
+      ensure
+        InstanceExecHelper.module_eval{ undef_method(mname) } rescue nil
+      end
+      ret
+    end
+  end
+end
+
+module ErlangHelpers
+
+  def setup
+    return true if @setup
+    $erlin  = IO.new(3, 'r')
+    $erlout = IO.new(4, 'w')
+    $erlin.sync = true
+    @setup = true
+  end
+
+  def send(obj)
+    berp = BERT.encode(obj)
+    $erlout.write [berp.size].pack('N')+berp
+    $erlout.flush
+  end
+
+  def receive
+    exit(0) if $erlin.eof?
+    h = $erlin.read(4)
+    while h == nil
+      sleep(0.1)
+      exit(0) if $erlin.eof?
+      h = $erlin.read(4)
+    end
+    l = h.unpack('N')[0]
+    exit(0) if $erlin.eof?
+    BERT.decode($erlin.read(l))
+  end
+
+  def error(*args)
+    raise InternalError, args
+  end
+
+  extend self
+end
+
+class InternalError < RuntimeError
+  attr_accessor :args
+  def initialize(args)
+    @args = args
+    super
+  end
+end
+
+def Erlang(&block)
+  ErlangHelpers.setup
+
+  loop do
+    cmd = ErlangHelpers.receive
+
+    if cmd == :stop
+      exit(0)
+    end
+
+    begin
+      ctx = Object.new
+      ctx.extend ErlangHelpers
+      ctx.instance_exec(cmd, &block)
+    rescue InternalError => e
+      ErlangHelpers.send BERT::Tuple[:error, BERT::Tuple[*e.args]]
+    rescue RuntimeError => e
+      ErlangHelpers.send BERT::Tuple[:error, BERT::Tuple[:runtime, e.class.to_s, e.message, e.backtrace]]
+    rescue Exception => e
+      ErlangHelpers.send BERT::Tuple[:error, BERT::Tuple[:system, e.class.to_s, e.message, e.backtrace]]
+    end
+  end
+end

data/lib/capricorn/recipes/apache-debian.rb

@@ -0,0 +1,112 @@
+require 'digest/sha1'
+
+domain = application[:domain]
+
+user       = 'www-data'
+
+db_name    = application[:id].downcase.gsub(/[^a-z]+/, '_')
+db_name    = "#{db_name[0,58]}_#{Digest::SHA1.hexdigest(application[:id])[0,5]}"
+db_user    = application[:id].downcase.gsub(/[^a-z]+/, '_')
+db_user    = "#{db_user[0,8]}_#{Digest::SHA1.hexdigest(application[:id])[0,5]}"
+db_pswd    = (rand(1_000_000_000) + 10_000).to_s
+
+httpdocs_path = "/var/www/vhosts/#{domain}/httpdocs"
+application_root = box["/var/www/vhosts/#{domain}/"]
+FileUtils.mkdir_p(application_root.full_path)
+has_db = application_root['shared/settings/database.yml'].exists?
+
+###############################################################################
+### Create host app                                                         ###
+###############################################################################
+if application_root['host/'].exists?
+  application_root.bash %{ mv host host-#{Time.now.strftime('%Y%m%d%H%M%S')} }
+end
+application_root.bash %{ milkshake create.host "host" --git-data --shared-data "shared" }
+
+unless has_db
+  ###############################################################################
+### Create Database                                                         ###
+###############################################################################
+  begin
+    box.bash(%{ mysql --user=root --password=Kendydig4488 --execute="CREATE DATABASE #{db_name} ; GRANT ALL ON #{db_name}.* TO '#{db_user}'@'%' IDENTIFIED BY  '#{db_pswd}' ;" })
+  rescue Rush::BashFailed
+    # ignore failure
+  end
+
+  ###############################################################################
+### Configure Database                                                      ###
+###############################################################################
+  database_conf = application_root['shared/settings/database.yml'].create
+  database_conf.write %{
+  default: &default
+    adapter: mysql
+    database: #{db_name}
+    username: #{db_user}
+    password: #{db_pswd}
+    host: localhost
+    encoding: utf8
+    socket: /var/run/mysqld/mysqld.sock
+
+  development:
+    <<: *default
+
+  test:
+    <<: *default
+
+  staging:
+    <<: *default
+
+  production:
+    <<: *default
+  }
+end
+
+###############################################################################
+### Configure VHost                                                         ###
+###############################################################################
+vhost_conf = box['/etc/apache2/sites-available/'+domain].create
+vhost_conf.write %[
+<VirtualHost *:80>
+  ServerName #{application[:domain]}
+  #{application[:aliases].collect {|d| 'ServerAlias '+d+"\n  "}}
+
+  RailsEnv #{application[:environment]}
+
+  DocumentRoot "#{application_root.full_path}/host/public"
+
+  ErrorDocument 503 /503.html
+  RewriteEngine on
+  RewriteCond %{DOCUMENT_ROOT}/../tmp/stop.txt -f
+  RewriteCond %{DOCUMENT_ROOT}/%{REQUEST_FILENAME} !-f
+  RewriteRule ^(.*)$ /$1 [R=503,L]
+
+  <Directory   "#{application_root.full_path}/host/public">
+    Options All
+    AllowOverride All
+    Order allow,deny
+    Allow from all
+  </Directory>
+</VirtualHost>
+]
+
+###############################################################################
+### Set Owner of Host                                                       ###
+###############################################################################
+box.bash %{ chown    #{user}:#{user} "#{application_root.full_path}" }
+box.bash %{ chown -R #{user}:#{user} "#{application_root.full_path}/host" }
+box.bash %{ chown -R #{user}:#{user} "#{application_root.full_path}/shared" }
+
+###############################################################################
+### Reconfigure vhost in Plesk                                              ###
+###############################################################################
+box.bash %{ /usr/sbin/a2ensite #{domain} }
+
+###############################################################################
+### Restart apache                                                          ###
+###############################################################################
+box.bash %{ /etc/init.d/apache2 restart }
+
+set :root_path, application_root.full_path.to_s
+set :www_user,  user.to_s
+set :www_group, user.to_s
+

data/lib/capricorn/recipes/centos-plesk.rb

@@ -0,0 +1,162 @@
+require 'digest/sha1'
+
+basedomain = application[:domain].split('.')[-2, 2].join('.')
+subdomain  = application[:domain].split('.')[0..-3].join('.')
+subdomain  = nil if subdomain.nil? or subdomain.empty?
+
+client     = 'mh1179'
+
+user       = application[:id].downcase.gsub(/[^a-z]+/, '_')
+short_user = "#{user[0,14]}_#{Digest::SHA1.hexdigest(application[:id])[0,5]}"
+passwd     = (rand(1_000_000_000) + 10_000).to_s
+
+db_name    = application[:id].downcase.gsub(/[^a-z]+/, '_')
+db_name    = "#{db_name[0,58]}_#{Digest::SHA1.hexdigest(application[:id])[0,5]}"
+db_user    = application[:id].downcase.gsub(/[^a-z]+/, '_')
+db_user    = "#{db_user[0,8]}_#{Digest::SHA1.hexdigest(application[:id])[0,5]}"
+db_pswd    = (rand(1_000_000_000) + 10_000).to_s
+
+###############################################################################
+### Find Unused Username                                                    ###
+###############################################################################
+begin
+  Etc.getpwnam(short_user)
+rescue ArgumentError
+  short_user = short_user.succ
+  # retry
+end
+
+###############################################################################
+### Create Plesk Domain                                                     ###
+###############################################################################
+begin
+  ### Check if the domain exists
+  box.bash(%{ /usr/local/psa/bin/domain -i #{basedomain} })
+  httpdocs_path = "/var/www/vhosts/#{basedomain}/httpdocs"
+  user = Etc.getpwuid(File.stat(httpdocs_path).uid).name
+  short_user = user
+  
+rescue Rush::BashFailed, Errno::ENOENT
+  ### try to create the base domain
+  box.bash(%{ /usr/local/psa/bin/domain -c #{basedomain} -clogin #{client} -status enabled -hosting true -hst_type phys -dns true -www true -login #{short_user} -passwd #{passwd} -shell /bin/bash })
+end
+application_root = box["/var/www/vhosts/#{basedomain}/"]
+
+###############################################################################
+### Create Plesk Subdomain                                                  ###
+###############################################################################
+if subdomain
+  begin
+    ### check is the subdomain exists
+    box.bash(%{ /usr/local/psa/bin/subdomain -i -s #{subdomain} -d #{basedomain} })
+    
+  rescue Rush::BashFailed
+    ### try to create the subdomain
+    box.bash(%{ /usr/local/psa/bin/subdomain -c #{subdomain} -d #{basedomain} })
+  end
+  application_root = application_root["subdomains/#{subdomain}/"]
+end
+
+###############################################################################
+### Add Domain Aliasses                                                     ###
+###############################################################################
+unless subdomain
+  application[:aliases].each do |aliasd|
+    begin
+      box.bash(%{ /usr/local/psa/bin/domalias --create #{aliasd} -domain #{basedomain} -status enabled -mail true -web true -dns true })
+    rescue Rush::BashFailed
+      # ignore failed aliases
+    end
+  end
+end
+
+has_db = application_root['shared/settings/database.yml'].exists?
+
+###############################################################################
+### Create host app                                                         ###
+###############################################################################
+if application_root['host/'].exists?
+  application_root.bash %{ mv host host-#{Time.now.strftime('%Y%m%d%H%M%S')} }
+end
+application_root.bash %{ milkshake create.host "host" --git-data --shared-data "shared" }
+
+unless has_db
+  ###############################################################################
+  ### Create Database                                                         ###
+  ###############################################################################
+  begin
+    box.bash(%{ /usr/local/psa/bin/database -c #{db_name} -domain #{basedomain} -server localhost:3306 -add_user #{db_user} -passwd #{db_pswd} })
+  rescue Rush::BashFailed
+    # ignore failure
+  end
+  
+  ###############################################################################
+  ### Configure Database                                                      ###
+  ###############################################################################
+  database_conf = application_root['shared/settings/database.yml'].create
+  database_conf.write %{
+  default: &default
+    adapter: mysql
+    database: #{db_name}
+    username: #{db_user}
+    password: #{db_pswd}
+    host: localhost
+    encoding: utf8
+    socket: /var/lib/mysql/mysql.sock
+  
+  development:
+    <<: *default
+  
+  test:
+    <<: *default
+  
+  staging:
+    <<: *default
+   
+  production:
+    <<: *default
+  }
+end
+
+###############################################################################
+### Configure VHost                                                         ###
+###############################################################################
+vhost_conf = application_root['conf/vhost.conf'].create
+vhost_conf.write %{
+DocumentRoot #{application_root.full_path}/host/public
+RailsEnv     #{application[:environment] || 'production'}
+
+ErrorDocument 503 /503.html
+RewriteEngine on
+RewriteCond %{DOCUMENT_ROOT}/../tmp/stop.txt -f
+RewriteCond %{DOCUMENT_ROOT}/%{REQUEST_FILENAME} !-f
+RewriteRule ^(.*)$ /$1 [R=503,L]
+
+<Directory  "#{application_root.full_path}/host/public">
+  Options All
+  AllowOverride All
+  Order allow,deny
+  Allow from all
+</Directory>
+}
+
+###############################################################################
+### Set Owner of Host                                                       ###
+###############################################################################
+box.bash %{ chown    #{short_user}:psaserv "#{application_root.full_path}" }
+box.bash %{ chown -R #{short_user}:psaserv "#{application_root.full_path}/host" }
+box.bash %{ chown -R #{short_user}:psaserv "#{application_root.full_path}/shared" }
+
+###############################################################################
+### Reconfigure vhost in Plesk                                              ###
+###############################################################################
+box.bash %{ /usr/local/psa/admin/sbin/websrvmng --reconfigure-vhost --vhost-name=#{basedomain} }
+
+###############################################################################
+### Restart apache                                                          ###
+###############################################################################
+box.bash %{ /etc/init.d/httpd restart }
+
+set :root_path, application_root.full_path.to_s
+set :www_user,  short_user.to_s
+set :www_group, 'psaserv'