capricorn 0.2.25 → 2.0.0

data/ext/Makefile

@@ -0,0 +1,2 @@
+install:
+	@(cd ../erlang && ./rebar compile && ./rebar force=1 generate)

data/ext/extconf.rb

@@ -0,0 +1 @@
+# see Makefile

data/lib/capricorn-client.rb

@@ -0,0 +1,86 @@
+module Capricorn
+  require 'thor'
+  require 'yaml'
+  require 'bertrpc'
+  require 'highline'
+  require 'fileutils'
+  
+  require 'capricorn-client/helpers'
+  
+  module CLI
+    require 'capricorn-client/cli/applications'
+    require 'capricorn-client/cli/gems'
+    require 'capricorn-client/cli/machines'
+  end
+  
+  module Runner
+    
+    HELP = <<-EOH
+Topics:
+  CLUSTER applications
+  # manage applications
+   
+  CLUSTER gems
+  # manage gems
+   
+  CLUSTER machines
+  # manage machines
+EOH
+    
+    DEFAULT_CONFIG = <<-EOH
+cluster_name:
+  host: localhost
+  port: 3457
+  username: dummy
+  password: dummy
+EOH
+    
+    def start
+      cluster_name = ARGV.shift
+      unless cluster_name
+        help
+        exit 1
+      end
+      
+      @cluster = config[cluster_name]
+      unless @cluster
+        puts "I don't know this cluster"
+        exit 1
+      end
+      
+      case ARGV.shift
+      when 'applications'
+        Capricorn::CLI::Applications.start
+      when 'gems'
+        Capricorn::CLI::Gems.start
+      when 'machines'
+        Capricorn::CLI::Machines.start
+      else
+        help
+      end
+    end
+    
+    def help
+      puts HELP
+    end
+    
+    def cluster
+      @cluster
+    end
+    
+    def config
+      @config ||= begin
+        config_path = File.expand_path('~/.capricorn/config.yml')
+        unless File.file?(config_path)
+          FileUtils.mkdir_p(File.dirname(config_path))
+          File.open(config_path, 'w+', 0600) { |file| file.write DEFAULT_CONFIG }
+          puts "Please edit your config file in ~/.capricorn/config.yml"
+          exit 2
+        end
+        YAML.load_file(config_path)
+      end
+    end
+    
+    extend self
+  end
+end

data/lib/capricorn-client/cli/applications.rb

@@ -0,0 +1,256 @@
+class Capricorn::CLI::Applications < Thor
+  include Capricorn::Helpers
+  
+  desc "list", "list all applications on MACHINE"
+  def list
+    machine = select_machine
+    
+    apps = client.call.applications.all(machine.to_sym).last || []
+    apps.collect! do |app|
+      [app[1], app[3]]
+    end
+    apps.flatten!
+    
+    ui.table %w( Id Name ), apps, :columns_across, 2
+  end
+  
+  desc "info", "get info of an application on MACHINE"
+  def info
+    machine, id = *select_application
+    app = find_app(machine, id)
+    show_app(app)
+  end
+  
+  desc "create", "create an application on MACHINE"
+  def create
+    machine     = select_machine
+    name        = ui.ask("name: ") do |q|
+      q.validate = /^.+$/
+    end
+    environment = ui.choose do |q|
+      q.choices(*%w( production development staging testing ))
+      q.prompt = "environment: "
+      q.default = 'production'
+    end
+    domains = []
+    domains << ui.ask("domain: ") do |q|
+      q.validate  = /^[\w\d]+(\.[\w\d]+)*$/
+    end
+    loop do
+      domain = ui.ask("more:   ") do |q|
+        q.validate = /^([\w\d]+(\.[\w\d]+)*)?$/
+      end
+      break if domain.strip.empty?
+      domains << domain
+    end
+    
+    domains = domains.flatten.compact
+    domains = domains.collect { |d| d.downcase.sub(/^www\./, '') }
+    domains = domains.uniq
+    
+    environment = environment.to_sym
+    
+    p client.call.applications.create(machine.to_sym, name, domains, environment)
+  end
+  
+  desc "import", "import an application on MACHINE (only use this from an application dir)"
+  def import
+    root = File.expand_path(Dir.pwd)
+    
+    unless File.file?(File.join(root, "host/config/milkshake.yml"))
+      puts ui.color('This doesn\'t appear to be a milkshake app!', :red)
+      puts ui.color('no milkshake config was found', :red)
+      exit 1
+    end
+    
+    unless File.directory?(File.join(root, "shared"))
+      puts ui.color('This doesn\'t appear to be a milkshake app!', :red)
+      puts ui.color('no shared dir was found', :red)
+      exit 1
+    end
+    
+    gems = YAML.load_file(File.join(root, "host/config/milkshake.yml"))
+    gems = (gems['gems'].keys rescue [])
+    if gems.empty?
+      puts ui.color('This doesn\'t appear to be a milkshake app!', :red)
+      puts ui.color('no gems were found', :red)
+      exit 1
+    end
+    
+    stats = (File.stat(File.join(root, "host/config/environment.rb")) rescue nil)
+    unless stats
+      puts ui.color('This doesn\'t appear to be a milkshake app!', :red)
+      puts ui.color('no environment.rb was found', :red)
+      exit 1
+    end
+    
+    uid = (Etc.getpwuid(stats.uid).name rescue nil)
+    unless uid
+      puts ui.color('This doesn\'t appear to be a milkshake app!', :red)
+      puts ui.color('no uid was found', :red)
+      exit 1
+    end
+    
+    gid = (Etc.getgrgid(stats.gid).name rescue nil)
+    unless gid
+      puts ui.color('This doesn\'t appear to be a milkshake app!', :red)
+      puts ui.color('no gid was found', :red)
+      exit 1
+    end
+    
+    machine     = select_machine
+    name        = ui.ask("name: ") do |q|
+      q.validate = /^.+$/
+    end
+    environment = ui.choose do |q|
+      q.choices(*%w( production development staging testing ))
+      q.prompt = "environment: "
+      q.default = 'production'
+    end
+    domains = []
+    domains << ui.ask("domain: ") do |q|
+      q.validate  = /^[\w\d]+(\.[\w\d]+)*$/
+    end
+    loop do
+      domain = ui.ask("more:   ") do |q|
+        q.validate = /^([\w\d]+(\.[\w\d]+)*)?$/
+      end
+      break if domain.strip.empty?
+      domains << domain
+    end
+    
+    domains = domains.flatten.compact
+    domains = domains.collect { |d| d.downcase.sub(/^www\./, '') }
+    domains = domains.uniq
+    
+    environment = environment.to_sym
+    
+    p client.call.applications.import(machine.to_sym, name, domains, environment, root, gems, uid, gid)
+  end
+  
+  desc "update", "update an application on MACHINE"
+  def update
+    machine, id = *select_application
+    app = find_app(machine, id)
+    
+    app[4]  ||= []
+    app[9]  ||= []
+    app[10] ||= []
+    
+    loop do
+      print "\e[H\e[2J" ; $stdout.flush
+      
+      show_app(app)
+      
+      puts
+      
+      ui.choose do |q|
+        q.header = "Actions"
+        q.prompt = 'What do you wan to do? '
+        
+        q.choice('gem-add') do
+          name = ui.ask("enter a gem name: ") do |q|
+            q.validate = /^([a-zA-Z0-9_-]+)?$/
+          end
+          app[10] << name unless name.strip.empty?
+        end
+        
+        q.choice('gem-remove') do
+          g = ui.choose do |q|
+            q.header = "Gems"
+            q.prompt = 'choose a gem to remove: '
+            q.choices(*app[10])
+          end
+          app[10].delete(g)
+        end
+        
+        q.choice('domain-add') do
+          domain = ui.ask("enter a domain: ") do |q|
+            q.validate = /^([\w\d]+(\.[\w\d]+)*)?$/
+          end
+          app[4] << domain unless domain.strip.empty?
+        end
+        
+        q.choice('domain-remove') do
+          domain = ui.choose do |q|
+            q.header = "Domains"
+            q.prompt = 'choose a domain to remove: '
+            q.choices(*app[4])
+          end
+          app[4].delete(domain)
+        end
+        
+        q.choice('abort') do
+          exit(0)
+        end
+        
+        q.choice('commit') do
+          p client.call.applications.update(machine.to_sym, id, app[4], app[10])
+          exit(0)
+        end
+        
+      end
+    end
+  end
+  
+  desc "fupdate", "force update an application on MACHINE"
+  def fupdate
+    machine, id = *select_application
+    p client.call.applications.fupdate(machine.to_sym, id)
+  end
+  
+  desc "relink", "relink an application on MACHINE"
+  def relink
+    machine, id = *select_application
+    p client.call.applications.relink(machine.to_sym, id)
+  end
+  
+  desc "restart", "restart an application on MACHINE"
+  def restart
+    machine, id = *select_application
+    p client.call.applications.restart(machine.to_sym, id)
+  end
+  
+  desc "start", "start an application on MACHINE"
+  def start
+    machine, id = *select_application
+    p client.call.applications.start(machine.to_sym, id)
+  end
+  
+  desc "stop", "stop an application on MACHINE"
+  def stop
+    machine, id = *select_application
+    p client.call.applications.stop(machine.to_sym, id)
+  end
+  
+private
+  
+  def find_app(machine ,id)
+    apps = client.call.applications.all(machine.to_sym).last || []
+    app = apps.detect do |app|
+      app[1] == id
+    end
+    
+    unless app
+      puts ui.color('Application not found!', :red)
+      exit(1)
+    end
+    
+    app
+  end
+  
+  def show_app(app)
+    app = app.to_a.dup
+    app.shift
+    app.pop
+    app.collect! do |v|
+      case v
+      when Array then v.join(', ')
+      else v.to_s
+      end
+    end
+    
+    ui.table %w( Id Machine Name Domains Environment User Group Root Installed\ Gems Requested\ Gems  ), app, :columns_down, 2
+  end
+  
+end

data/lib/capricorn-client/cli/gems.rb

@@ -0,0 +1,57 @@
+class Capricorn::CLI::Gems < Thor
+  include Capricorn::Helpers
+  
+  desc "push PATHS...", "push gems to capricorn"
+  def push(*paths)
+    paths = paths.flatten.collect do |path|
+      Dir.glob(path)
+    end.flatten
+    paths.each do |path|
+      begin
+        client.call(:stream => path).gems.push
+        puts ui.color("Pushed: #{File.basename(path)}", :green)
+      rescue BERTRPC::UserError => e
+        puts ui.color(e.message, :red)
+      end
+    end
+  end
+  
+  desc "missing", "list missing gems"
+  def missing
+    ui.table %w( Gem Requirements ), clean_missing(client.call.gems.missing), :columns_across, 2
+  end
+  
+  desc "all", "list all gems"
+  def all
+    ui.table %w( Gem Versions ), clean_all(client.call.gems.all), :columns_across, 2
+  end
+  
+private
+  
+  def clean_all(gems)
+    gems = gems.last
+    gems.inject(Hash.new { |h,k| h[k] = [] }) do |m, g|
+      header, name, version = *(g.to_ary)
+      version = version.last.join('.')
+      m[name.to_s] << version
+      m
+    end.collect do |name, versions|
+      [name, versions.join(', ')]
+    end.flatten
+  end
+  
+  def clean_missing(deps)
+    deps = deps.last if deps and deps.first == :ok
+    return [] if deps.nil? or deps.empty?
+    deps.sort! { |a,b| a.first <=> b.first }
+    deps.collect do |dep|
+      reqs = dep.last.collect do |req|
+        op = req.shift
+        version = req.shift.to_a.last.join('.')
+        "#{op} #{version}"
+      end.join(', ')
+      [dep.first, reqs]
+    end.flatten
+  end
+  
+end

data/lib/capricorn-client/cli/machines.rb

@@ -0,0 +1,9 @@
+class Capricorn::CLI::Machines < Thor
+  include Capricorn::Helpers
+  
+  desc "list", "list all machines"
+  def list
+    ui.table("Machines", self.machines)
+  end
+  
+end

data/lib/capricorn-client/helpers.rb

@@ -0,0 +1,62 @@
+module Capricorn::Helpers
+  
+  def select_machine
+    if self.machines.size > 1
+      ui.choose(*self.machines)
+    elsif self.machines.size == 1 and ui.agree("Use machine #{self.machines.first}? ")
+      self.machines.first
+    else
+      puts ui.color('No machines found!!!', :red)
+      exit(1)
+    end
+  end
+  
+  def select_application
+    machine = select_machine
+    
+    apps = client.call.applications.all(machine.to_sym).last || []
+    apps.collect! do |app|
+      app[1]
+    end
+    apps.flatten!
+    
+    if apps.size > 1
+      [machine, ui.choose(*apps)]
+    elsif apps.size == 1 and ui.agree("Use application #{apps.first}? ")
+      [machine, apps.first]
+    else
+      puts ui.color('No application found!!!', :red)
+      exit(1)
+    end
+  end
+  
+  def machines
+    @machines ||= (client.call.machines.all || []).collect do |machine|
+      machine.to_s
+    end
+  end
+  
+  def client
+    cluster = Capricorn::Runner.cluster
+    host    = (cluster['host'] || 'localhost').to_s
+    port    = (cluster['port'] || 3457).to_i
+    @client ||= BERTRPC::Service.new(host, port)
+  end
+  
+  def ui
+    @ui ||= begin
+      @ui = HighLine.new
+      @ui.extend UI
+      @ui
+    end
+  end
+  
+  module UI
+    
+    def table(headers, items, mode=:rows, option=nil)
+      headers = [headers].flatten.collect { |h| color(h, :green) }
+      @output.puts list(headers+items, mode, option)
+    end
+    
+  end
+end