spacialdb 0.0.1

data/lib/spacialdb/command.rb

@@ -0,0 +1,142 @@
+require 'spacialdb/helpers'
+require 'optparse'
+
+module Spacialdb
+  module Command
+    class InvalidCommand < RuntimeError; end
+    class CommandFailed  < RuntimeError; end
+
+    extend Spacialdb::Helpers
+
+    def self.load
+      Dir[File.join(File.dirname(__FILE__), "command", "*.rb")].each do |file|
+        require file
+      end
+    end
+
+    def self.commands
+      @@commands ||= {}
+    end
+
+    def self.command_aliases
+      @@command_aliases ||= {}
+    end
+
+    def self.namespaces
+      @namespaces ||= {}
+    end
+
+    def self.register_command(command)
+      commands[command[:command]] = command
+    end
+
+    def self.register_namespace(namespace)
+      namespaces[namespace[:name]] = namespace
+    end
+
+    def self.current_command
+      @current_command
+    end
+
+    def self.global_options
+      @global_options ||= []
+    end
+
+    def self.global_option(name, *args)
+      global_options << { :name => name, :args => args }
+    end
+
+    global_option :db,      "--db DB", "-d"
+    global_option :confirm, "--confirm DB"
+
+    def self.run(cmd, args=[])
+
+      command = parse(cmd)
+
+      unless command
+        error " !   #{cmd} is not a spacialdb command. See 'spacialdb help'."
+        return
+      end
+
+      @current_command = cmd
+
+      opts = {}
+      invalid_options = []
+
+      parser = OptionParser.new do |parser|
+        global_options.each do |global_option|
+          parser.on(*global_option[:args]) do |value|
+            opts[global_option[:name]] = value
+          end
+        end
+        command[:options].each do |name, option|
+          parser.on("-#{option[:short]}", "--#{option[:long]}", option[:desc]) do |value|
+            opts[name.gsub("-", "_").to_sym] = value
+          end
+        end
+      end
+
+      begin
+        parser.order!(args) do |nonopt|
+          invalid_options << nonopt
+        end
+      rescue OptionParser::InvalidOption => ex
+        invalid_options << ex.args.first
+        retry
+      end
+
+      if opts[:help]
+        run "help", [cmd]
+        return
+      end
+
+      args.concat(invalid_options)
+
+      begin
+        object = command[:klass].new(args.dup, opts.dup)
+        object.send(command[:method])
+      rescue InvalidCommand
+        error "Unknown command. Run 'spacialdb help' for usage information."
+      rescue RestClient::Unauthorized
+        puts "Authentication failure"
+        run "login"
+        retry
+      rescue RestClient::ResourceNotFound => e
+        error extract_not_found(e.http_body)
+      rescue RestClient::RequestFailed => e
+        error extract_error(e.http_body)
+      rescue RestClient::RequestTimeout
+        error "API request timed out. Please try again, or contact support@spacialdb.com if this issue persists."
+      rescue CommandFailed => e
+        error e.message
+      end
+    rescue OptionParser::ParseError => ex
+      commands[cmd] ? run("help", [cmd]) : run("help")
+    rescue Interrupt => e
+      error "\n[canceled]"
+    end
+
+    def self.parse(cmd)
+      commands[cmd] || commands[command_aliases[cmd]]
+    end
+
+    def self.extract_not_found(body)
+      body =~ /^[\w\s]+ not found$/ ? body : "Resource not found"
+    end
+
+    def self.extract_error(body)
+      msg = parse_error_json(body) || parse_error_plain(body) || 'Internal server error'
+      msg.split("\n").map { |line| ' !   ' + line }.join("\n")
+    end
+    
+    def self.parse_error_json(body)
+      json = json_decode(body.to_s)
+      json ? json['error'] : nil
+    end
+
+    def self.parse_error_plain(body)
+      return unless body.respond_to?(:headers) && body.headers[:content_type].to_s.include?("text/plain")
+      body.to_s
+    end
+  end
+end

data/lib/spacialdb/helpers.rb

@@ -0,0 +1,81 @@
+require "vendor/okjson"
+
+module Spacialdb
+  module Helpers
+
+    def display(msg="", newline=true)
+      if newline
+        puts(msg)
+      else
+        print(msg)
+        STDOUT.flush
+      end
+    end
+
+    def error(msg)
+      STDERR.puts(msg)
+      exit 1
+    end
+
+    def ask
+      gets.strip
+    end
+
+    def echo_off
+      system "stty -echo"
+    end
+
+    def echo_on
+      system "stty echo"
+    end
+
+    def home_directory
+      ENV['HOME']
+    end
+
+    def redisplay(line, line_break = false)
+      display("\r\e[0K#{line}", line_break)
+    end
+
+    def confirm_command(db = db)
+      raise(Spacialdb::Command::CommandFailed, "No db specified.\nSet it by adding --db <db name>") unless db
+
+      confirmed_db = extract_option('--confirm', false)
+      if confirmed_db
+        unless confirmed_db == db
+          raise(Spacialdb::Command::CommandFailed, "Confirmed db #{confirmed_db} did not match the selected db #{db}.")
+        end
+        return true
+      else
+        display
+        display " !    WARNING: Potentially Destructive Action"
+        display " !    This command will affect the db: #{db}"
+        display " !    To proceed, type \"#{db}\" or re-run this command with --confirm #{db}"
+        display
+        display "> ", false
+        if ask.downcase != db
+          display " !    Input did not match #{db}. Aborted."
+          false
+        else
+          true
+        end
+      end
+    end
+
+    def longest(items)
+      items.map(&:to_s).map(&:length).sort.last
+    end
+
+    def json_encode(object)
+      OkJson.encode(object)
+    rescue OkJson::ParserError
+      nil
+    end
+
+    def json_decode(json)
+      OkJson.decode(json)
+    rescue OkJson::ParserError
+      nil
+    end
+  end
+end

data/lib/spacialdb/version.rb

@@ -0,0 +1,3 @@
+module Spacialdb
+  VERSION = "0.0.1"
+end

data/lib/spacialdb.rb

@@ -0,0 +1,3 @@
+module Spacialdb end;
+
+require 'spacialdb/client'