jrubysql 0.1.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in jrubysql.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2012 Junegunn Choi
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,79 @@
+jrubysql
+========
+An SQL client for any JDBC-compliant database. Written in JRuby.
+
+Installation
+------------
+
+```
+gem install jrubysql
+```
+
+Usage
+-----
+
+```
+usage: jrubysql [options]
+       jrubysql -t DBMS_TYPE -h HOSTNAME [-u USERNAME -p [PASSWORD] [-d DATABASE]] [-f FILENAME]
+       jrubysql -c CLASSNAME -j JDBC_URL [-u USERNAME -p [PASSWORD] [-d DATABASE]] [-f FILENAME]
+
+    -t, --type DBMS_TYPE             Database type: mysql/oracle/postgres/sqlserver
+    -h, --host HOST                  DBMS host address
+
+    -c, --class-name CLASSNAME       Class name of the JDBC driver
+    -j, --jdbc-url JDBC_URL          JDBC URL for the connection
+
+    -u, --user USERNAME              Username
+    -p, --password [PASSWORD]        Password
+    -d, --database DATABASE          Name of the database (optional)
+
+    -f, --filename FILENAME          SQL script file
+    -o, --output OUTPUT_TYPE         Output type: cterm|term|csv (default: cterm)
+
+        --help                       Show this message
+        --version                    Show version
+```
+
+Connecting to the database
+--------------------------
+
+### Setting up CLASSPATH
+Add the appropriate JDBC drivers to the CLASSPATH.
+
+```
+export CLASSPATH=$CLASSPATH:~/lib/mysql-connector-java-5.1.17-bin.jar:~/lib/ojdbc6.jar
+```
+
+### With type (-t) and hostname (-h)
+
+```
+# Supports MySQL/Oracle/PostgreSQL/MSSQL
+
+jrubysql -t mysql -h localhost -d test -u user -p
+jrubysql -t oracle -h localhost:1521/orcl -u user -p password
+jrubysql -t postgres -h localhost -u root
+jrubysql -t sqlserver -h 192.168.62.26 -u user -p password
+```
+
+### Connect with class name of JDBC driver (-c) and JDBC URL (-j)
+
+```
+# You can connect to any database with its JDBC driver
+
+bin/jrubysql -corg.postgresql.Driver -jjdbc:postgresql://localhost/test
+bin/jrubysql -ccom.mysql.jdbc.Driver -jjdbc:mysql://localhost/test -uuser -p
+```
+
+Screenshot
+----------
+![](https://github.com/junegunn/jrubysql/raw/master/screenshots/simpsons.png)
+
+TODO
+----
+TESTS!!!
+
+Copyright
+---------
+Copyright (c) 2012 Junegunn Choi. See LICENSE.txt for
+further details.
+

data/Rakefile

@@ -0,0 +1,8 @@
+require "bundler/gem_tasks"
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end

data/bin/jrubysql

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+# encoding: UTF-8
+# Junegunn Choi (junegunn.c@gmail.com)
+# 2012/03/07-
+
+require 'rubygems'
+require 'jrubysql'
+
+JRubySQL.launch ARGV

data/jrubysql.gemspec

@@ -0,0 +1,31 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "jrubysql/version"
+
+Gem::Specification.new do |s|
+  s.name        = "jrubysql"
+  s.version     = JRubySQL::VERSION
+  s.authors     = ["Junegunn Choi"]
+  s.email       = ["junegunn.c@gmail.com"]
+  s.homepage    = "https://github.com/junegunn/jrubysql"
+  s.summary     = %q{An SQL client for any JDBC-compliant database.}
+  s.description = %q{An SQL client for any JDBC-compliant database. Written in JRuby.}
+
+  s.rubyforge_project = "jrubysql"
+
+  s.files         = `git ls-files`.split("\n").reject { |f| f =~ /^screenshots/ }
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  s.add_development_dependency "test-unit"
+
+  s.add_runtime_dependency "jdbc-helper", '~> 0.7.2'
+  s.add_runtime_dependency "insensitive_hash", '~> 0.2.3'
+  s.add_runtime_dependency "tabularize", '~> 0.1.1'
+  s.add_runtime_dependency "each_sql", '~> 0.3.1'
+  s.add_runtime_dependency "highline", '~> 1.6.11'
+  s.add_runtime_dependency "ansi", '~> 1.4.2'
+  s.add_runtime_dependency "erubis", '~> 2.7.0'
+end

data/lib/jrubysql/config.rb

@@ -0,0 +1,35 @@
+require 'yaml'
+require 'fileutils'
+
+module JRubySQL
+  # A simple key-value config in YAML
+  class Config
+    DEFAULT_PATH = File.join(ENV['HOME'], '.jrubysqlrc')
+
+    def initialize path = DEFAULT_PATH
+      @path = path
+      if @path && File.exists?(@path)
+        @yaml = YAML.load(File.read(@path))
+      end
+      @yaml = @yaml || {}
+    end
+
+    def [] key
+      @yaml[key]
+    end
+
+    def []= key, value
+      (@yaml[key] = value).tap { dump }
+    end
+
+  private
+    def dump
+      # Try to write atomically 
+      File.open(@path + '.tmp', 'w') do |f|
+        f << YAML.dump(@yaml)
+      end
+      FileUtils.mv(@path + '.tmp', @path)
+    end
+  end#Config
+end#JRubySQL
+

data/lib/jrubysql/constants.rb

@@ -0,0 +1,15 @@
+module JRubySQL
+module Constants
+  SUPPORTED_DBMS_TYPES = [ :mysql, :oracle, :postgres, :sqlserver ]
+
+  # .jrubysqlrc
+  MAX_COMMAND_HISTORY    = 100
+  MAX_CONNECTION_HISTORY = 10
+
+  # Terminal (TBD)
+  MAX_COLUMN_WIDTH = 80
+  MIN_SCREEN_ROWS  = 10
+  MAX_SCREEN_ROWS  = 50
+
+end#Constants
+end#JRubySQL

data/lib/jrubysql/controller.rb

@@ -0,0 +1,141 @@
+require 'quote_unquote'
+
+module JRubySQL
+class Controller
+  include JRubySQL::Messages
+
+  attr_reader :db_type
+
+  def initialize options, argv_str
+    @config = JRubySQL::Config.new
+    histories = @config['connections']
+
+    if options.nil?
+      if histories.nil? || histories.empty?
+        JRubySQL::OptionParser.parse []
+      else
+        # FIXME: Output
+        puts m(:choose_parameter)
+        histories.each_with_index do |history, idx|
+          puts "[#{idx + 1}] #{history.first}"
+        end
+        print '> '
+        select = $stdin.gets
+        select = select && select.chomp
+        if (1..(histories.length)).include?(select.to_i)
+          history = histories[select.to_i - 1]
+          @options = history.last
+          @argv_str = history.first
+        else
+          puts
+          JRubySQL::OptionParser.parse []
+        end
+      end
+    else
+      @options = options
+      @argv_str = argv_str
+    end
+
+    @db_type = JRubySQL::RDBMS.get_type(@options[:type] || @options[:driver])
+
+    # Setting up input: file or console (and more?)
+    if @options[:filename]
+      @input = JRubySQL::Input::File.new(self, @options[:filename])
+    else
+      @input = JRubySQL::Input::Console.new(self)
+    end
+
+    # Setting up output: Colored terminal 
+    case @options[:output]
+    when 'cterm'
+      @output = JRubySQL::Output::CTerm.new
+    when 'term'
+      @output = JRubySQL::Output::Term.new
+    when 'csv'
+      @output = JRubySQL::Output::CSV.new
+    end
+  end
+
+  def start
+    @output.welcome!
+    @output.info m(:connecting)
+    begin
+      @rdbms = JRubySQL::RDBMS.new @options
+    rescue Exception => e
+      @output.error e.to_s
+      exit 1
+    end
+    @output.info m(:connected)
+
+    history = @config['connections'] || []
+    history.unshift [@argv_str, @options]
+    history.uniq!
+    if history.length > JRubySQL::Constants::MAX_CONNECTION_HISTORY
+      history = history[0, JRubySQL::Constants::MAX_CONNECTION_HISTORY]
+    end
+    @config['connections'] = history
+
+    loop do
+      ret = @input.get
+
+      ret[:sqls].each do |sql|
+        begin
+          output @rdbms.execute(sql)
+        rescue Exception => e
+          @output.error e.to_s
+        end
+      end if ret.has_key?(:sqls)
+
+      ret[:commands].each do |command|
+        process_command command.keys.first, command.values.first
+      end if ret.has_key?(:commands)
+    end
+  end
+
+  def cursor empty = true
+    @output.cursor empty
+  end
+
+  def print_cursor empty = true
+    @output.print_cursor empty
+  end
+
+private
+  def output result
+    @output.print_result result
+  end
+
+  def process_command cmd, params
+    case cmd
+    when :help
+      @output.print_help
+    when :quit
+      @output.info m(:goodbye)
+      quit!
+    when :delimiter
+      @output.info m(:set_delimiter, params)
+      @input.delimiter = params
+    when :now
+      @output.info Time.now.strftime('%Y/%m/%d %H:%M:%S.%L')
+    when :autocommit
+      if params.nil?
+        @output.info m(:current_autocommit, @rdbms.autocommit ? 'on' : 'off')
+      elsif %[on off].include?(params.downcase)
+        @rdbms.autocommit = params.downcase == 'on'
+        @output.info m(:turn_autocommit, params.downcase)
+      else
+        @output.error m(:invalid_autocommit, params)
+      end
+    else
+      # TODO
+      @output.error m(:unknown_command)
+    end
+  end
+
+  def quit!
+    @rdbms.close rescue nil
+    exit 0
+  end
+
+end#Controller
+end#JRubySQL

data/lib/jrubysql/doc/help.txt.erb

@@ -0,0 +1,10 @@
+<% title = JRubySQL.name %>
+<%= title %>
+<%= '=' * title.length %>
+
+  help                   Display this message.
+  <SQL>                  Execute the SQL.
+  autocommit [<on|off>]  Show/enable/disable autocommit. (default: on)
+  delimiter <DELIMITER>  Change SQL delimiter.
+  now                    Display current timestamp.
+  exit                   Exit JRubySQL.

data/lib/jrubysql/input/console.rb

@@ -0,0 +1,84 @@
+require 'readline'
+
+module JRubySQL
+module Input
+class Console
+  def initialize controller
+    @controller = controller
+    @esql = JRubySQL::Input.get_parser @controller.db_type
+  end
+
+  def get
+    empty_response = { :sqls => [], :commands => [] }
+    line = Readline::readline(@controller.cursor(@esql.empty?), false)
+
+    return(
+      # CTRL-D
+      if line.nil?
+        puts
+        @esql.clear
+        empty_response
+      # Empty input (not inside a block)
+      elsif @esql.empty? && line.gsub(/\s/m, '').empty?
+        empty_response
+      # Console commands
+      elsif @esql.empty? && cmd = process_command(line)
+        { :commands => [cmd].compact }
+      # Line with delimiters
+      elsif line.include?(@esql.delimiter)
+        @esql << line + ' '
+        result = @esql.shift
+        result[:sqls].each do |sql|
+          Readline::HISTORY << sql + @esql.delimiter
+        end
+        { :sqls => result[:sqls] }
+      # SQL without delimiter
+      else
+        @esql << line + ' '
+        empty_response
+      end
+    )
+  end
+
+  def delimiter
+    @esql.delimiter
+  end
+
+  def delimiter= delim
+    @esql.delimiter = delim
+  end
+
+private
+
+  def process
+    result = @esql.shift
+    result[:sqls].each do |sql|
+      Readline::HISTORY << sql + ';'
+      @controller.execute sql
+    end
+  end
+
+  def process_command line
+    Readline::HISTORY << line
+    case line.chomp.downcase.strip
+    when /^help(#{Regexp.escape delimiter})?$/
+      { :help => nil }
+    when /^delimiter (\S+$)/
+      { :delimiter => $1 }
+    when 'autocommit'
+      { :autocommit => nil }
+    when /^autocommit (\S+?)(#{Regexp.escape delimiter})?$/
+      { :autocommit => $1 }
+    when 'now'
+      { :now => nil }
+    when /^(exit|quit)(#{Regexp.escape delimiter})?$/
+      { :quit => nil }
+    else
+      Readline::HISTORY.pop
+      nil
+    end
+  end
+end#Console
+end#Input
+end#JRubySQL
+

data/lib/jrubysql/input/file.rb

@@ -0,0 +1,19 @@
+require 'each_sql'
+
+module JRubySQL
+module Input
+class File
+  def initialize controller, file_path
+    @controller = controller
+    script = ::File.read(file_path)
+    sqls  = EachSQL(script, JRubySQL::Input.get_each_sql_type(@controller.db_type))
+    @ret = { :sqls => sqls }
+  end
+
+  def get
+    @ret.tap { @ret = { :commands => [{ :quit => nil }] } }
+  end
+end#Console
+end#Input
+end#JRubySQL
+

data/lib/jrubysql/input/input.rb

@@ -0,0 +1,17 @@
+require 'each_sql'
+
+module JRubySQL
+module Input
+  def self.get_each_sql_type db_type
+    {
+      :mysql => :mysql,
+      :oracle => :oracle,
+      :postgres => :postgres
+    }[db_type] || :default
+  end
+
+  def self.get_parser db_type, delimiter = ';'
+    EachSQL.new(get_each_sql_type(db_type), delimiter)
+  end
+end#Input
+end#JRubySQL

data/lib/jrubysql/messages.rb

@@ -0,0 +1,23 @@
+require 'yaml'
+require 'insensitive_hash/minimal'
+
+module JRubySQL
+module Messages
+  def m *args
+    args = args.dup
+    type = args.shift
+    msg = MESSAGES[type]
+    args.each do |arg|
+      msg = msg.sub('$$', arg.to_s)
+    end
+    msg
+  end
+
+private
+  MESSAGES = 
+    InsensitiveHash[
+      YAML.load(File.read File.join( File.dirname(__FILE__), 'messages.yml' ))
+    ].tap { |ih| ih.underscore = true }
+end#Messages
+end#JRubySQL
+

data/lib/jrubysql/messages.yml

@@ -0,0 +1,43 @@
+---
+choose parameter:
+  "Parameter not given. Choose one:"
+ask password:
+  "Password: "
+
+invalid connection:
+  Invalid connection specification
+unsupported database:
+  $$ is not supported yet. Try with -c and -j options instead.
+oracle service name required:
+  Oracle service name must be included in the hostname: e.g. localhost:1521/orcl
+invalid output:
+  Invalid output type.
+file not found:
+  "File not found: $$"
+
+connecting:
+  Connecting to the database ...
+connected:
+  Connected.
+
+set delimiter:
+  Setting delimiter to $$
+current autocommit:
+  "Current autocommit: $$"
+turn autocommit:
+  "Turning autocommit $$"
+invalid autocommit:
+  "Invalid option: '$$'. Required: [on|off]"
+unknown command:
+  Unknown command. Possibly a bug.
+goodbye:
+  Goodbye!
+
+interrupted:
+  Interrupted.
+
+rows returned:
+  $$ row$$. $$
+rows affected:
+  $$ row$$ affected. $$
+

data/lib/jrubysql/option_parser.rb

@@ -0,0 +1,124 @@
+require 'optparse'
+require 'highline'
+
+module JRubySQL
+module OptionParser
+  class << self
+    include JRubySQL::Messages
+  end
+
+  def self.parse argv
+    {:output => 'cterm'}.tap do |options|
+      opts = ::OptionParser.new { |opts|
+        opts.banner = 
+          [
+            "usage: jrubysql [options]",
+            "       jrubysql -t DBMS_TYPE -h HOSTNAME [-u USERNAME -p [PASSWORD] [-d DATABASE]] [-f FILENAME]",
+            "       jrubysql -c CLASSNAME -j JDBC_URL [-u USERNAME -p [PASSWORD] [-d DATABASE]] [-f FILENAME]"
+          ].join($/)
+        opts.separator ''
+
+        opts.on('-t', '--type DBMS_TYPE', 'Database type: mysql/oracle/postgres/sqlserver') do |v|
+          options[:type] = v.downcase.to_sym
+        end
+
+        opts.on('-h', '--host HOST', 'DBMS host address') do |v|
+          options[:host] = v
+        end
+
+        opts.separator ""
+      
+        opts.on('-c', '--class-name CLASSNAME', 'Class name of the JDBC driver') do |v|
+          options[:driver] = v
+        end
+
+        opts.on('-j', '--jdbc-url JDBC_URL', 'JDBC URL for the connection') do |v|
+          options[:url] = v
+        end
+
+        opts.separator ""
+      
+        opts.on('-u', '--user USERNAME', 'Username') do |v|
+          options[:user] = v
+        end
+      
+        opts.on('-p', '--password [PASSWORD]', 'Password') do |v|
+          options[:password] = v
+        end
+      
+        opts.on('-d', '--database DATABASE', 'Name of the database (optional)') do |v|
+          options[:database] = v
+        end
+
+        opts.separator ""
+
+        opts.on('-f', '--filename FILENAME', 'SQL script file') do |v|
+          options[:filename] = v
+        end
+
+        opts.on('-o', '--output OUTPUT_TYPE', 'Output type: cterm|term|csv (default: cterm)') do |v|
+          options[:output] = v
+        end
+      
+        opts.separator ""
+
+        opts.on_tail('--help', "Show this message") do
+          puts opts
+          exit
+        end
+
+        opts.on_tail('--version', "Show version") do
+          puts JRubySQL.name
+          exit
+        end
+      }
+      begin
+        opts.parse! argv
+        if options.has_key?(:password) && options[:password].nil?
+          options[:password] = ask_password
+        end
+
+        validate options
+      rescue SystemExit
+        exit 0
+      rescue Exception => e
+        puts e.to_s
+        puts '=' * e.to_s.length
+        puts opts
+        exit 1
+      end
+    end#tap
+  end
+
+private
+  def self.validate opts
+    unless %w[cterm term csv].include?(opts[:output])
+      raise ArgumentError.new m(:invalid_output)
+    end
+
+    if (!opts[:type] && !opts[:driver]) || (opts[:type] && opts[:driver])
+      raise ArgumentError.new m(:invalid_connection)
+    end
+
+    unless (opts[:type] && opts[:host]) || (opts[:driver] && opts[:url])
+      raise ArgumentError.new m(:invalid_connection)
+    end
+
+    if opts[:type] && !JRubySQL::Constants::SUPPORTED_DBMS_TYPES.include?(opts[:type])
+      raise ArgumentError.new m(:unsupported_database, opts[:type])
+    end
+
+    if opts[:filename] && !File.exists?(opts[:filename])
+      raise ArgumentError.new m(:file_not_found, opts[:filename])
+    end
+
+  end
+
+  def self.ask_password
+    HighLine.new.ask(m(:ask_password)) { |q| q.echo = "*" }
+  end
+
+end#OptionParser
+end#JRubySQL
+
+# p JRubySQL::OptionParser.parse %w[-h aa gg -t asdfa -p]

data/lib/jrubysql/output/csv.rb

@@ -0,0 +1,49 @@
+require 'csv'
+
+module JRubySQL
+module Output
+class CSV
+  def welcome!;   end
+  def info   msg
+    $stderr.puts "[I] #{msg}"
+  end
+  def result msg
+    $stderr.puts "[R] #{msg}"
+  end
+  def warn   msg
+    $stderr.puts "[W] #{msg}"
+  end
+  def error  msg
+    $stderr.puts "[E] #{msg}"
+  end
+  def print_help; end
+
+  def cursor empty
+    ''
+  end
+
+  def print_cursor empty; end
+
+  def print_result ret
+    # Footer
+    elapsed = "(#{'%.2f' % ret[:elapsed]} sec)"
+
+    if ret[:set?]
+      ret[:result].each_with_index do |row, idx|
+        puts ::CSV.generate_line(row.labels) if idx == 0
+        puts ::CSV.generate_line row.map { |col|
+          case col
+          when BigDecimal
+            col.to_s('F')
+          else
+            col
+          end
+        }
+      end
+    end
+  end
+
+end#CSV
+end#Output
+end#JRubySQL
+

data/lib/jrubysql/output/cterm.rb

@@ -0,0 +1,95 @@
+require 'ansi'
+require 'erubis'
+require 'bigdecimal'
+
+module JRubySQL
+module Output
+class CTerm < Term
+  include ANSI::Code
+  HELP = Erubis::Eruby.new(File.read File.join(File.dirname(__FILE__), '../doc/help.txt.erb')).result(binding)
+
+  def welcome!
+    puts bold + JRubySQL.name + reset
+  end
+
+  def cursor empty
+    if empty
+      wrap('jrubysql', bold) + 
+            wrap('> ', bold + green)
+    else
+      wrap('       -', bold + yellow) +
+            wrap('> ', bold + red)
+    end
+  end
+
+  def print_cursor empty
+    print cursor(empty)
+  end
+
+  def print_help
+    puts
+    puts wrap(HELP, blue + bold)
+    puts
+  end
+
+  def info message
+    col = blue + bold
+    puts wrap(message, col)
+  end
+
+  def result message
+    col = green
+    puts wrap(message, green)
+  end
+
+  def warn message
+    col = yellow + bold
+    puts wrap(message, col)
+  end
+
+  def error message
+    col = red + bold
+    puts wrap(message, col)
+  end
+
+private
+  def separator_for row
+    # 13-bytes for ANSI codes
+    # - bold/reset: 4
+    # - colors: 5
+    '+-' + row.map { |e| '-' * (e.length - 13) }.join('-+-') + '-+'
+  end
+
+  def decorate_label label
+    white + bold + label + reset
+  end
+
+  # This looks stupid though.
+  def decorate value
+    case value
+    when BigDecimal
+      cyan + value.to_s('F') + reset + reset
+    when Numeric
+      cyan + value.to_s + reset + reset
+    when String
+      yellow + value + reset + reset
+    when Time, Java::JavaSql::Timestamp
+      magenta + value.to_s + reset + reset
+    when NilClass
+      bold + red + '(null)' + reset
+    else
+      white + reset + value.to_s + reset
+    end
+  end
+
+  def cnow
+    green + now + reset
+  end
+
+  def wrap text, color
+    color + text + reset
+  end
+    
+end#CTerm
+end#Output
+end#JRubySQL

data/lib/jrubysql/output/term.rb

@@ -0,0 +1,122 @@
+require 'erubis'
+require 'tabularize'
+require 'java'
+
+module JRubySQL
+module Output
+class Term
+  include JRubySQL::Messages
+
+  HELP = Erubis::Eruby.new(File.read File.join(File.dirname(__FILE__), '../doc/help.txt.erb')).result(binding)
+
+  def initialize
+    # Make use of JLine included in JRuby
+    java_import 'jline.Terminal'
+    @terminal = Terminal.getTerminal
+    trap 'INT' do
+      Thread.main.raise Interrupt
+    end
+  end
+
+  def welcome!
+    puts JRubySQL.name
+  end
+
+  def cursor empty
+    if empty
+      'jrubysql> '
+    else
+      '       -> '
+    end
+  end
+
+  def print_cursor empty
+    print cursor(empty)
+  end
+
+  def print_help
+    puts
+    puts HELP
+    puts
+  end
+
+  def info message
+    puts "[I] #{message}"
+  end
+
+  def result message
+    puts "[R] #{message}"
+  end
+
+  def warn message
+    puts "[W] #{message}"
+  end
+
+  def error message
+    puts "[E] #{message}"
+  end
+
+  def print_result ret
+    # Footer
+    elapsed = "(#{'%.2f' % ret[:elapsed]} sec)"
+
+    if ret[:set?]
+      begin
+        cnt = print_table ret[:result]
+        result m(:rows_returned, cnt, cnt > 1 ? 's' : '', elapsed)
+      rescue Interrupt
+        warn m(:interrupted)
+      end
+    elsif ret[:result]
+      cnt = [0, ret[:result]].max
+      result m(:rows_affected, cnt, cnt > 1 ? 's' : '', elapsed)
+    else
+      result elapsed
+    end
+    puts
+  end
+
+private
+  def print_table ret
+    cnt = 0
+    lines = [(@terminal.getTerminalHeight rescue JRubySQL::Constants::MAX_SCREEN_ROWS) - 5, 
+             JRubySQL::Constants::MIN_SCREEN_ROWS].max
+    ret.each_slice(lines) do |slice|
+      cnt += slice.length
+
+      table = [slice.first.labels.map { |l| decorate_label l }] + 
+          slice.map { |row| row.to_a.map { |v| decorate v } }
+
+      output = Tabularize.it(table, :unicode_display => true)
+      separator = separator_for(output.first)
+      output_strs = output.map { |r| '| ' + r.join(' | ') + ' |' }
+      [0, 2, -1].each { |l| output_strs.insert l, separator }
+      puts output_strs
+    end
+    cnt
+  end
+
+  def separator_for row
+    '+-' + row.map { |e| '-' * e.length }.join('-+-') + '-+'
+  end
+
+  def decorate_label label
+    label
+  end
+
+  def decorate value
+    case value
+    when BigDecimal
+      value.to_s('F')
+    else
+      value.to_s
+    end
+  end
+
+  def now
+    Time.now.strftime('%Y/%m/%d %H:%M:%S')
+  end
+end#Term
+end#Output
+end#JRubySQL
+

data/lib/jrubysql/rdbms.rb

@@ -0,0 +1,82 @@
+require 'jdbc-helper'
+
+module JRubySQL
+class RDBMS
+  include JRubySQL::Messages
+
+  def self.get_type driver_or_type
+    case driver_or_type.to_s.downcase
+    when /oracle/
+      :oracle
+    when /mysql/
+      :mysql
+    when /postgres/
+      :postgres
+    when /sqlserver/
+      :sqlserver
+    else
+      :unknown
+    end
+  end
+
+  def initialize options
+    @conn = 
+      if options[:type]
+        case options[:type]
+        when :mysql
+          JDBCHelper::MySQLConnector.connect(
+            options[:host], options[:user], options[:password], options[:database])
+        when :oracle
+          host, svc = options[:host].split('/')
+          if svc.nil?
+            # FIXME
+            raise ArgumentError.new m(:oracle_service_name_required)
+          end
+          JDBCHelper::OracleConnector.connect(
+            host, options[:user], options[:password], svc)
+        when :postgres
+          JDBCHelper::PostgresConnector.connect(
+            options[:host], options[:user], options[:password], options[:database])
+        when :sqlserver
+          JDBCHelper::SqlServerConnector.connect(
+            options[:host], options[:user], options[:password], options[:database])
+        end
+      elsif options[:driver]
+        JDBCHelper::Connection.new(
+          {
+            :driver   => options[:driver],
+            :url      => options[:url],
+            :user     => options[:user],
+            :password => options[:password]
+        }.reject { |k, v| v.nil? }
+        )
+      else
+        raise ArgumentError.new m(:invalid_connection)
+      end
+  end
+
+  def autocommit
+    @conn.java_obj.get_auto_commit
+  end
+
+  def autocommit= ac
+    @conn.java_obj.set_auto_commit ac
+  end
+
+  def close
+    @conn.close
+  end
+
+  def execute sql
+    st = Time.now
+    result = @conn.execute sql
+    elapsed = Time.now - st
+
+    {
+      :set? => result.respond_to?(:each),
+      :result => result,
+      :elapsed => elapsed
+    }
+  end
+end#RDBMS
+end#JRubySQL

data/lib/jrubysql/version.rb

@@ -0,0 +1,3 @@
+module JRubySQL
+  VERSION = "0.1.1"
+end

data/lib/jrubysql.rb

@@ -0,0 +1,36 @@
+require "jrubysql/version"
+
+if RUBY_PLATFORM.match(/java/).nil?
+  puts 'Sorry. jrubysql only runs on JRuby.'
+  exit 1
+end
+
+require 'jrubysql/messages'
+require 'jrubysql/config'
+require 'jrubysql/constants'
+require 'jrubysql/rdbms'
+require 'jrubysql/option_parser'
+require 'jrubysql/input/input'
+require 'jrubysql/input/console'
+require 'jrubysql/input/file'
+require 'jrubysql/output/csv'
+require 'jrubysql/output/term'
+require 'jrubysql/output/cterm'
+require 'jrubysql/controller'
+
+module JRubySQL
+  def self.name
+    "JRubySQL #{JRubySQL::VERSION}"
+  end
+
+  def self.launch argv
+    if argv.empty?
+      JRubySQL::Controller.new(nil, nil).start
+    else
+      argv_str = argv.join(' ')
+      options = JRubySQL::OptionParser.parse argv
+      JRubySQL::Controller.new(options, argv_str).start
+    end
+  end
+end
+

data/test/test_jrubysql.rb

@@ -0,0 +1,9 @@
+require 'rubygems'
+require 'bundler'
+ENV['BUNDLE_GEMFILE'] = File.join(File.dirname(__FILE__), '..', 'Gemfile')
+Bundler.setup(:default, :development)
+require 'test/unit'
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+class TestJRubySQL < Test::Unit::TestCase
+end

metadata

@@ -0,0 +1,206 @@
+--- !ruby/object:Gem::Specification
+name: jrubysql
+version: !ruby/object:Gem::Version
+  prerelease:
+  version: 0.1.1
+platform: ruby
+authors:
+- Junegunn Choi
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2012-03-15 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: test-unit
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+    none: false
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+    none: false
+  prerelease: false
+  type: :development
+- !ruby/object:Gem::Dependency
+  name: jdbc-helper
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.7.2
+    none: false
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.7.2
+    none: false
+  prerelease: false
+  type: :runtime
+- !ruby/object:Gem::Dependency
+  name: insensitive_hash
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.2.3
+    none: false
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.2.3
+    none: false
+  prerelease: false
+  type: :runtime
+- !ruby/object:Gem::Dependency
+  name: tabularize
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.1.1
+    none: false
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.1.1
+    none: false
+  prerelease: false
+  type: :runtime
+- !ruby/object:Gem::Dependency
+  name: each_sql
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.3.1
+    none: false
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.3.1
+    none: false
+  prerelease: false
+  type: :runtime
+- !ruby/object:Gem::Dependency
+  name: highline
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.6.11
+    none: false
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.6.11
+    none: false
+  prerelease: false
+  type: :runtime
+- !ruby/object:Gem::Dependency
+  name: ansi
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.4.2
+    none: false
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.4.2
+    none: false
+  prerelease: false
+  type: :runtime
+- !ruby/object:Gem::Dependency
+  name: erubis
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.7.0
+    none: false
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.7.0
+    none: false
+  prerelease: false
+  type: :runtime
+description: An SQL client for any JDBC-compliant database. Written in JRuby.
+email:
+- junegunn.c@gmail.com
+executables:
+- jrubysql
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/jrubysql
+- jrubysql.gemspec
+- lib/jrubysql.rb
+- lib/jrubysql/config.rb
+- lib/jrubysql/constants.rb
+- lib/jrubysql/controller.rb
+- lib/jrubysql/doc/help.txt.erb
+- lib/jrubysql/input/console.rb
+- lib/jrubysql/input/file.rb
+- lib/jrubysql/input/input.rb
+- lib/jrubysql/messages.rb
+- lib/jrubysql/messages.yml
+- lib/jrubysql/option_parser.rb
+- lib/jrubysql/output/csv.rb
+- lib/jrubysql/output/cterm.rb
+- lib/jrubysql/output/term.rb
+- lib/jrubysql/rdbms.rb
+- lib/jrubysql/version.rb
+- test/test_cterm.rb
+- test/test_jrubysql.rb
+- test/test_option_parser.rb
+- test/test_term_output.rb
+homepage: https://github.com/junegunn/jrubysql
+licenses: []
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+  none: false
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+  none: false
+requirements: []
+rubyforge_project: jrubysql
+rubygems_version: 1.8.18
+signing_key:
+specification_version: 3
+summary: An SQL client for any JDBC-compliant database.
+test_files:
+- test/test_cterm.rb
+- test/test_jrubysql.rb
+- test/test_option_parser.rb
+- test/test_term_output.rb
+...