webbynode 0.1.2 → 0.2.0

data/lib/webbynode/io.rb

@@ -0,0 +1,175 @@
+require 'yaml'
+
+module Webbynode
+  class DirectoryNotFound < StandardError; end
+  
+  class Io
+    class KeyAlreadyExists < StandardError; end
+    
+    TemplatesPath = File.join(File.dirname(__FILE__), '..', 'templates')
+    
+    def app_name
+      Dir.pwd.split("/").last.gsub(/[\.| ]/, "_")
+    end
+    
+    def exec(s, redirect_stderr=true)
+      `#{s}#{redirect_stderr ? " 2>&1" : ""}`
+    end
+    
+    def directory?(s)
+      File.directory?(s)
+    end
+    
+    def file_exists?(f)
+      File.exists?(f)
+    end
+    
+    def read_file(f)
+      File.read(f)
+    end
+    
+    def open_file(f, a, &blk)
+      File.open(f, a, &blk)
+    end
+    
+    def log(text, notify=false)
+      notify = :action unless notify
+
+      case notify
+      when :notify
+        notify = true
+        puts "#{text}" 
+      
+      when :start
+        notify = true
+        puts "[Webbynode] #{text}" 
+      
+      when :quiet_start
+        notify = false
+        puts "[Webbynode] #{text}" 
+      
+      when :action
+        notify = false
+        puts "            #{text}"
+      
+      when :warning
+        notify = false
+        puts "            WARNING: #{text}"
+      
+      when :action
+        notify = false
+        puts "            #{text}"
+      
+      when :finish
+        notify = true
+        puts
+        puts "[Webbynode] #{text}"
+        
+      when :error
+        notify = true
+        puts
+        puts "[Webbynode] ERROR: #{text}"
+        
+      else
+        notify = false
+        puts "            #{text}"
+        
+      end
+          
+      Webbynode::Notify.message(text) if notify
+    end
+
+    def log_and_exit(text, notify=false)
+      log(text, notify)
+      exit
+    end
+    
+    def create_local_key(passphrase="")
+      unless File.exists?(Webbynode::Commands::AddKey::LocalSshKey)
+        exec "ssh-keygen -t rsa -N \"#{passphrase}\" -f #{Webbynode::Commands::AddKey::LocalSshKey}"
+      end
+    end
+    
+    def create_file(file_name, contents, executable=nil)
+      File.open(file_name, "w") do |file|
+        file.write(contents)
+      end
+      FileUtils.chmod 0755, file_name if executable
+    end
+    
+    def delete_file(file_name)
+      File.delete(file_name)
+    end
+    
+    def templates_path
+      TemplatesPath
+    end
+    
+    def read_from_template(template)
+      read_file File.join(templates_path, template)
+    end
+    
+    def create_from_template(file, template)
+      contents = read_from_template(template)
+      create_file(file, contents)
+    end
+    
+    def properties(s)
+      (@properties||={})[s] = Properties.new(s)
+    end
+    
+    def general_settings
+      @general_settings ||= properties("#{ENV['HOME']}/.webbynode")
+    end
+    
+    def with_settings_for(file, &blk)
+      settings = properties(file)
+      yield settings
+      settings.save
+    end
+    
+    def with_general_settings(&blk)
+      yield general_settings
+      general_settings.save
+    end
+    
+    def with_setting(&blk)
+      with_settings_for ".webbynode/settings", &blk
+    end
+    
+    def add_general_setting(key, value)
+      with_general_settings { |s| s[key] = value }
+    end
+    
+    def remove_setting(key)
+      with_setting { |s| s.remove key }
+    end
+    
+    def add_setting(key, value)
+      with_setting { |s| s[key] = value }
+    end
+    
+    def config_multi_add(key, new_value)
+      raise "Missing Webbynode config file" unless file_exists?(".webbynode/config")
+      config = read_yaml(".webbynode/config")
+      (config[key] ||= []) << new_value
+      write_yaml(config)
+    end
+    
+    def read_yaml(file)
+      YAML.load_file(file)
+    end
+    
+    def write_yaml(obj)
+      create_file(file, obj.to_yaml)
+    end
+    
+    def read_config(config_file)
+      read_file(config_file).split("\n").inject({}) do |hash, line|
+        line_parts = line.split("=")
+        hash[line_parts.first.strip.to_sym] = line_parts.last.strip
+        hash
+      end
+    end
+  end
+end

data/lib/webbynode/notify.rb

@@ -0,0 +1,19 @@
+module Webbynode
+  class Notify
+    
+    TITLE       = "Webbynode"
+    IMAGE_PATH  = File.join(File.dirname(__FILE__), '..', '..', 'assets', 'webbynode.png')
+    
+    def self.message(message)
+      if self.installed? and !$testing
+        %x(growlnotify -t "#{TITLE}" -m "#{message}" --image "#{IMAGE_PATH}")
+      end
+    end
+    
+    def self.installed?
+      return false if %x(which growlnotify).chomp.empty?
+      true
+    end
+    
+  end
+end

data/lib/webbynode/option.rb

@@ -0,0 +1,84 @@
+module Webbynode
+  class Option
+    attr_reader :name, :kind, :desc, :options, :errors
+    attr_accessor :value
+    
+    def Option.name_for(s)
+      return $1 if s =~ /^--(\w+)(=("[^"]+"|[\w]+))*/
+    end
+    
+    def initialize(*args)
+      raise "Cannot initialize Parameter without a name" unless args.first
+      
+      @errors = []
+      @value = nil
+      @options = args.pop if args.last.is_a?(Hash)
+      @options ||= {}
+      @options[:required] = true if @options[:required].nil?
+      
+      @name = args[0]
+      if args[1].is_a?(String)
+        @desc = args[1]
+      else
+        @kind = args[1]
+        @desc = args[2]
+      end
+      
+      @kind ||= String
+      @value = [] if @kind == Array
+    end
+    
+    def validate!
+      raise Webbynode::Command::InvalidCommand, errors.join("\n") unless valid?
+    end
+    
+    def valid?
+      @errors = []
+      if (validations = @options[:validate])
+        validations.each_pair do |key, value|
+          @errors << send("#{key}_error", value) unless send(key, value)
+        end
+      end
+      @errors.empty?
+    end
+    
+    def in(allowed_values)
+      allowed_values.include?(self.value)
+    end
+    
+    def in_error(allowed_values)
+      opts = allowed_values.map { |v| "'#{v}'"}
+      "Invalid value '#{value}' for #{self.class.name.split("::").last.downcase} '#{self.name}'. It should be one of #{opts.to_phrase("or")}."
+    end
+    
+    def parse(s)
+      if s =~ /^--(\w+)(=("[^"]+"|[\w]+))*/
+        self.value = $3 ? $3.gsub(/"/, "") : true
+      end
+    end
+    
+    def array?
+      kind == Array
+    end
+    
+    def reset!
+      self.value = default_value
+    end
+    
+    def default_value
+      array? ? [] : nil
+    end
+    
+    def required?
+      @options[:required]
+    end
+    
+    def take
+      @options[:take]
+    end
+    
+    def to_s
+      "--#{name.to_s}#{take ? "=#{take.to_s}" : ""}"
+    end
+  end
+end

data/lib/webbynode/parameter.rb

@@ -0,0 +1,27 @@
+module Webbynode
+  class Parameter < Webbynode::Option
+    def initialize(*args)
+      super
+      @options[:required] = true if @options[:required].nil?
+    end
+    
+    def validate!
+      if required? and self.value === self.default_value
+        raise Webbynode::Command::InvalidCommand, "Missing '#{name}' parameter."
+      end
+      super
+    end
+    
+    def required?
+      @options[:required]
+    end
+    
+    def to_s
+      if required?
+        "#{name}"
+      else
+        "[#{name}]"
+      end
+    end
+  end
+end

data/lib/webbynode/properties.rb

@@ -0,0 +1,43 @@
+module Webbynode
+  class Properties < Hash
+    attr_accessor :file, :properties
+    
+    def io
+      @io ||= Webbynode::Io.new
+    end
+
+    def initialize(file, create=true)
+      raise "Tried to instantiate a real property file" if $testing
+      @file = file
+
+      begin
+        IO.foreach(file) do |line|
+          self[$1.strip.to_s] = $2.strip if line = ~ /([^=]*)=(.*)\/\/(.*)/ || line =~ /([^=]*)=(.*)/
+        end
+      rescue
+      end
+    end
+
+    def to_s
+      output = "File name #{@file}\n"
+      self.each { |key, value| output += " #{key} = #{value}\n" }
+      output
+    end
+
+    def add(key, value = nil)
+      return unless key.length > 0
+      self[key] = value
+    end
+
+    def remove(key)
+      return unless key.length > 0
+      self.delete(key)
+    end
+
+    def save
+      file = File.new(@file, "w+")
+      self.each { |key, value| file.puts "#{key}=#{value}\n" }
+      file.close
+    end
+  end
+end

data/lib/webbynode/push_and.rb

@@ -0,0 +1,16 @@
+module Webbynode
+  class PushAndError < StandardError; end
+  class PushAndFileNotFound < StandardError; end
+
+  class PushAnd
+    def io; @io ||= Webbynode::Io.new; end
+    
+    def present?
+      io.file_exists?(".pushand")
+    end
+    
+    def parse_remote_app_name
+      io.read_file(".pushand")[/^phd \$0 ([^ ]+)/, 1]
+    end
+  end
+end

data/lib/webbynode/remote_executor.rb

@@ -0,0 +1,21 @@
+module Webbynode  
+  class RemoteExecutor
+    attr_accessor :ip
+        
+    def initialize(ip)
+      @ip = ip
+    end
+    
+    def ssh
+      @ssh ||= Ssh.new(ip)
+    end
+    
+    def create_folder(folder)
+      ssh.execute "mkdir -p #{folder}"
+    end
+    
+    def exec(cmd, echo=false)
+      ssh.execute(cmd, echo)
+    end
+  end
+end

data/lib/webbynode/server.rb

@@ -0,0 +1,39 @@
+module Webbynode
+  class InvalidAuthentication < StandardError; end
+  class PermissionError < StandardError; end
+  class ApplicationNotDeployed < StandardError; end
+  
+  class Server
+    attr_accessor :ip
+    
+    def initialize(ip)
+      @ssh  = Ssh.new(ip)
+      @ip   = ip
+    end
+    
+    def io
+      @io ||= Webbynode::Io.new
+    end
+   
+    def remote_executor
+      @remote_executor ||= Webbynode::RemoteExecutor.new(ip)
+    end
+    
+    def pushand
+      @pushand ||= Webbynode::PushAnd.new
+    end
+
+    def add_ssh_key(key_file, passphrase="")
+      io.create_local_key(key_file, passphrase) unless io.file_exists?(key_file)
+      remote_executor.create_folder("~/.ssh")
+      
+      key_contents = io.read_file(key_file)
+      remote_executor.exec "echo \"#{key_contents}\" >> ~/.ssh/authorized_keys; chmod 644 ~/.ssh/authorized_keys"
+    end
+    
+    def application_pushed?
+      return false if remote_executor.exec("cd #{pushand.parse_remote_app_name}") =~ /No such file or directory/
+      true
+    end
+  end
+end

data/lib/webbynode/ssh.rb

@@ -0,0 +1,65 @@
+module Webbynode
+  class Ssh
+    def initialize(remote_ip)
+      @remote_ip = remote_ip
+    end
+    
+    def io
+      @io ||= Webbynode::Io.new
+    end
+    
+    def connect
+      raise "No IP given" unless @remote_ip
+      @conn = nil if @conn and @conn.closed?
+      @conn ||= Net::SSH.start(@remote_ip, 'git', :auth_methods => %w(publickey hostbased))
+    rescue Net::SSH::AuthenticationFailed
+      HighLine.track_eof = false
+      
+      begin
+        @password ||= ask("Enter your deployment password for #{@remote_ip}: ") { |q| q.echo = '' }
+        @conn     ||= Net::SSH.start(@remote_ip, 'git', :password => @password)  
+      rescue Net::SSH::AuthenticationFailed
+        io.log "Could not connect to server: invalid authentication.", true
+        exit
+      end
+      
+    rescue Net::SSH::Disconnect
+      io.log "Could not connect to the server: Wrong IP or Server Offline.", true
+      exit
+    
+    end
+    
+    def execute(script, echo=false)
+      connect
+      output = ""
+      error_output = ""
+      
+      exit_code = nil
+      channel = @conn.open_channel do |chan|
+        chan.on_request('exit-status') do |ch, data|
+          exit_code = data.read_long
+        end
+        
+        chan.on_data do |ch, data|
+          puts data if echo
+          output << data
+        end
+        
+        chan.on_extended_data do |ch, type, data|
+          next unless type == 1  # only handle stderr
+          puts data if echo
+          output << data
+          error_output << data
+        end
+        
+        chan.exec("#{script} < /dev/null") do |ch, s|
+          raise Exceptions::SshInstallationError, "Error executing script \"#{script[:name]}\"" unless s
+        end
+      end
+      
+      channel.wait
+      
+      output
+    end
+  end
+end

data/lib/webbynode/ssh_keys.rb

@@ -0,0 +1,36 @@
+module Webbynode
+  module SshKeys
+    # adds a given key string to a remote server,
+    # creating the key folder structure if needed
+    def add_key_to_remote(key)
+      run_remote_command "mkdir ~/.ssh 2>/dev/null; chmod 700 ~/.ssh; echo \"#{key}\" >> ~/.ssh/authorized_keys; chmod 644 ~/.ssh/authorized_keys"
+    end
+    
+    # lists all authorized keys on the remote server
+    def remote_authorized_keys
+      run_remote_command "cat \$HOME/.ssh/authorized_keys"
+    end
+    
+    # checks if a remote server has a given key
+    def remote_has_key?(key)
+      if keys = remote_authorized_keys
+        keys.index(key) 
+      end
+    end
+    
+    # name of the local publish ssh key file
+    def local_key_file
+      "#{ENV['HOME']}/.ssh/id_rsa.pub"
+    end
+    
+    # creates the local key with an optional passphrase
+    def create_local_key(passphrase="")
+      run "ssh-keygen -t rsa -N \"#{passphrase}\" -f #{local_key_file}"
+    end
+    
+    def local_key(create_when_missing=true)
+      create_local_key(named_options["passphrase"]) if create_when_missing and !File.exists?(local_key_file)
+      @local_key ||= File.read(local_key_file) 
+    end
+  end
+end

data/lib/webbynode.rb

@@ -0,0 +1,56 @@
+$:.unshift(File.dirname(__FILE__)) unless
+  $:.include?(File.dirname(__FILE__)) || $:.include?(File.expand_path(File.dirname(__FILE__)))
+
+require 'rubygems'
+require 'domainatrix'
+require 'net/ssh'
+require 'highline/import'
+require 'pp'
+
+require File.join(File.dirname(__FILE__), 'webbynode', 'io')
+require File.join(File.dirname(__FILE__), 'webbynode', 'git')
+require File.join(File.dirname(__FILE__), 'webbynode', 'ssh')
+require File.join(File.dirname(__FILE__), 'webbynode', 'server')
+require File.join(File.dirname(__FILE__), 'webbynode', 'push_and')
+require File.join(File.dirname(__FILE__), 'webbynode', 'command')
+require File.join(File.dirname(__FILE__), 'webbynode', 'option')
+require File.join(File.dirname(__FILE__), 'webbynode', 'parameter')
+require File.join(File.dirname(__FILE__), 'webbynode', 'api_client')
+require File.join(File.dirname(__FILE__), 'webbynode', 'remote_executor')
+require File.join(File.dirname(__FILE__), 'webbynode', 'notify')
+require File.join(File.dirname(__FILE__), 'webbynode', 'properties')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'init')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'push')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'config')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'add_key')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'add_backup')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'change_dns')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'delete')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'remote')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'tasks')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'start')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'stop')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'restart')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'alias')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'help')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'webbies')
+require File.join(File.dirname(__FILE__), 'webbynode', 'commands', 'version')
+
+require File.join(File.dirname(__FILE__), 'webbynode', 'application')
+
+module Webbynode
+  VERSION = '0.1.2'
+end
+
+class Array
+  def to_phrase(last_join="and")
+    return "" if empty?
+
+    array = self.clone
+    last = array.pop
+
+    return last if array.empty?
+
+    "#{array.join(", ")} #{last_join} #{last}"
+  end
+end

data/spec/fixtures/api/credentials

@@ -0,0 +1,3 @@
+--- 
+:token: apitoken
+:email: fcoury@me.com

data/spec/fixtures/api/dns

@@ -0,0 +1,30 @@
+HTTP/1.1 200 OK
+Server: nginx/0.6.35
+Date: Sat, 30 Jan 2010 20:23:40 GMT
+Content-Type: text/yaml; charset=utf-8
+Connection: keep-alive
+Set-Cookie: _webbymanager_session=BAh7BjoPc2Vzc2lvbl9pZCIlODc4ZGUwY2I2YzI2N2E3NjJiMTU4MzUyODZhNjBmM2Q%3D--6fd4a124a19e997aec94772e04070acdde120887; path=/; HttpOnly
+Status: 200
+ETag: "ba0c974d33641165d7dcd0b6c0c0363e"
+X-Runtime: 97
+Cache-Control: private, max-age=0, must-revalidate
+Content-Length: 300
+
+--- 
+zones: 
+- :ttl: 86400
+  :domain: rubyista.info.
+  :status: Active
+  :id: 14
+- :ttl: 86400
+  :domain: controledefrota.com.
+  :status: Active
+  :id: 137
+- :ttl: 86400
+  :domain: androidreference.com.
+  :status: Active
+  :id: 307
+- :ttl: 86400
+  :domain: webbyapp.com.
+  :status: Active
+  :id: 979

data/spec/fixtures/api/dns_a_record

@@ -0,0 +1,20 @@
+HTTP/1.1 200 OK
+Server: nginx/0.6.35
+Date: Sat, 30 Jan 2010 21:08:17 GMT
+Content-Type: text/yaml; charset=utf-8
+Connection: keep-alive
+Set-Cookie: _webbymanager_session=BAh7BjoPc2Vzc2lvbl9pZCIlYWY1NmJlODIzMjVhMjQ3OThhMTJlODg2YWUzN2NlNDY%3D--68c4db1775911eaf87ad612ab27ef05ae5366943; path=/; HttpOnly
+Status: 200
+ETag: "7c4ab3a7f92166784ae648ae2df75520"
+X-Runtime: 78
+Cache-Control: private, max-age=0, must-revalidate
+Content-Length: 110
+
+--- 
+record: 
+  :ttl: 86400
+  :type: A
+  :aux: 0
+  :name: xyz
+  :data: 200.100.200.100
+  :id: 7360

data/spec/fixtures/api/dns_a_record_already_exists

@@ -0,0 +1,15 @@
+HTTP/1.1 100 Continue
+
+HTTP/1.1 500 Internal Server Error
+Server: nginx/0.6.35
+Date: Sat, 30 Jan 2010 22:40:01 GMT
+Content-Type: text/yaml; charset=utf-8
+Connection: keep-alive
+Set-Cookie: _webbymanager_session=BAh7BjoPc2Vzc2lvbl9pZCIlNTcxODBjZjkyNThiZTUxZWY4MTBkYjQ4NGEwNzNmNTk%3D--5945514460235bcbdd1a97871c0d451be71c2433; path=/; HttpOnly
+Status: 500
+X-Runtime: 131
+Cache-Control: no-cache
+Content-Length: 41
+
+--- 
+errors: Data has already been taken

data/spec/fixtures/api/dns_a_record_error

@@ -0,0 +1,13 @@
+HTTP/1.1 500 Internal Server Error
+Server: nginx/0.6.35
+Date: Sat, 30 Jan 2010 21:40:30 GMT
+Content-Type: text/yaml; charset=utf-8
+Connection: keep-alive
+Set-Cookie: _webbymanager_session=BAh7BjoPc2Vzc2lvbl9pZCIlZjA1ZDFkNTVjZDlkNjQ0OGQ3YzRmZTVjODYwZGQyZjc%3D--3f95ecf6d9eae211aee4f2340dce691424f27655; path=/; HttpOnly
+Status: 500
+X-Runtime: 5
+Cache-Control: no-cache
+Content-Length: 38
+
+--- 
+error: No DNS entry for id 99999

data/spec/fixtures/api/dns_new_zone

@@ -0,0 +1,17 @@
+HTTP/1.1 200 OK
+Server: nginx/0.6.35
+Date: Sat, 30 Jan 2010 22:59:34 GMT
+Content-Type: text/yaml; charset=utf-8
+Connection: keep-alive
+Set-Cookie: _webbymanager_session=BAh7BjoPc2Vzc2lvbl9pZCIlZjhjNmEwZDE5ZjcyMGFmZTA4MmE2NTM1MDMwNDEyNGU%3D--18e7af8560b1ee3180a9aa1100fdead75243169d; path=/; HttpOnly
+Status: 200
+ETag: "08e60ad87837d5d3b26199f5d7b7df92"
+X-Runtime: 96
+Cache-Control: private, max-age=0, must-revalidate
+Content-Length: 70
+
+--- 
+:ttl: 86400
+:domain: newzone.com.
+:status: Active
+:id: 1045