ydbi 0.5.0

data/lib/dbd/ODBC.rb

@@ -0,0 +1,89 @@
+#--
+# DBD::ODBC
+#
+# Copyright (c) 2001-2004 Michael Neumann <mneumann@ntecs.de>
+# 
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without 
+# modification, are permitted provided that the following conditions 
+# are met:
+# 1. Redistributions of source code must retain the above copyright 
+#    notice, this list of conditions and the following disclaimer.
+# 2. Redistributions in binary form must reproduce the above copyright 
+#    notice, this list of conditions and the following disclaimer in the 
+#    documentation and/or other materials provided with the distribution.
+# 3. The name of the author may not be used to endorse or promote products
+#    derived from this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
+# INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
+# AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
+# THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
+# OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
+# OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
+# ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+#
+# $Id$
+#++
+
+$:.delete(".") # FIXME oh for the love of pete..
+require "odbc"
+$: << "."
+
+begin
+    require 'rubygems'
+    gem 'dbi'
+rescue LoadError => e
+end
+
+require 'dbi'
+
+module DBI
+    module DBD
+        #
+        # DBD::ODBC - ODBC Database Driver
+        #
+        # Requires DBI and the 'ruby-odbc' package to work, and unixodbc (iodbc
+        # WILL NOT WORK).
+        #
+        # Only things that extend DBI's results are documented.
+        #
+        module ODBC
+
+            VERSION          = "0.2.5"
+            DESCRIPTION      = "ODBC DBI DBD"
+
+            def self.driver_name
+                "odbc"
+            end
+
+            DBI::TypeUtil.register_conversion(driver_name) do |obj|
+                newobj = if obj.nil?
+                             nil
+                         else
+                             case obj
+                             when ::Date
+                                 ::ODBC::Date.new(obj)
+                             when ::Time
+                                 ::ODBC::Time.new(obj)
+                             when ::DateTime
+                                 ::ODBC::Timestamp.new(obj)
+                             else
+                                 obj.to_s
+                             end
+                         end
+                [newobj, false]
+            end
+
+            ODBCErr = ::ODBC::Error
+        end # module ODBC
+    end # module DBD
+end # module DBI
+
+require 'dbd/odbc/driver'
+require 'dbd/odbc/database'
+require 'dbd/odbc/statement'

data/lib/dbd/Pg.rb

@@ -0,0 +1,188 @@
+#--
+# DBD::Pg
+#
+# Copyright (c) 2001, 2002, 2003 Jim Weirich, Michael Neumann <mneumann@ntecs.de>
+# Copyright (c) 2008 Erik Hollensbe, Christopher Maujean
+#
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions
+# are met:
+# 1. Redistributions of source code must retain the above copyright
+#    notice, this list of conditions and the following disclaimer.
+# 2. Redistributions in binary form must reproduce the above copyright
+#    notice, this list of conditions and the following disclaimer in the
+#    documentation and/or other materials provided with the distribution.
+# 3. The name of the author may not be used to endorse or promote products
+#    derived from this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
+# INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
+# AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
+# THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
+# OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
+# OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
+# ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+#++
+
+begin
+    require 'rubygems'
+    gem 'pg'
+    gem 'dbi'
+rescue Exception => e
+end
+
+require 'dbi'
+require 'pg'
+
+module DBI
+    module DBD
+        #
+        # DBD::Pg - Database Driver for the PostgreSQL database system.
+        #
+        # Requires DBI and the 'pg' gem or package to work.
+        #
+        # Only things that extend DBI's results are documented.
+        #
+        module Pg
+            VERSION          = "0.3.9"
+            DESCRIPTION      = "PostgreSQL DBI DBD"
+
+            #
+            # returns 'Pg'
+            #
+            # See DBI::TypeUtil#convert for more information.
+            #
+            def self.driver_name
+                "Pg"
+            end
+
+            #
+            # This method takes a ruby Array and converts it to PostgreSQL array syntax.
+            #
+            def self.generate_array(obj)
+                # XXX yarr, there be recursion here, and it's probably not a good idea.
+                output = "{"
+                obj.each do |item|
+                    case item
+                    when ::Array
+                        output += generate_array(item)
+                    else
+                        generated = DBI::TypeUtil.convert(driver_name, item)
+                        generated = case item
+                                    when String
+                                        # in strings, escapes are doubled and the quotes are different.
+                                        # this gets *really* ugly and needs to be well-tested
+                                        "\"#{generated.gsub(/\\/) { "\\\\" }}\""
+                                    when Fixnum
+                                        generated.to_s
+                                    end
+                        output += generated
+                    end
+                    output += "," # FIXME technically, delimiters are variable
+                end
+
+                output.sub(/,$/, '}')
+            end
+
+            #
+            # A quote helper, this uses the new syntax in PostgreSQL 8.2 and up.
+            #
+            def self.quote(value)
+                "E'#{ value.gsub(/\\/){ '\\\\' }.gsub(/'/){ '\\\'' } }'"
+            end
+
+            #
+            # Parse a postgresql type. Returns a hash with these fields (as Symbol)
+            #
+            # * ftype: the full type, as passed in to this method.
+            # * type: the type stripped of all attribute information.
+            # * size: the LHS of the attribute information, typically the precision.
+            # * decimal: the RHS of the attribute information, typically the scale.
+            # * array: true if this type is actually an array of that type.
+            #
+            def self.parse_type(ftype)
+                type = ftype
+                pos = ftype.index('(')
+                decimal = nil
+                size = nil
+                array_of_type = nil
+
+                if pos != nil
+                    type = ftype[0..pos-1]
+                    size = ftype[pos+1..-2]
+                    pos = size.index(',')
+                    if pos != nil
+                        size, decimal = size.split(',', 2)
+                        size = size.to_i
+                        decimal = decimal.to_i
+                    else
+                        size = size.to_i
+                    end
+                end
+
+                if type =~ /\[\]$/
+                    type.sub!(/\[\]$/, '')
+                    array_of_type = true
+                end
+
+                return {
+                    :ftype   => ftype.dup,
+                    :type    => type,
+                    :size    => size,
+                    :decimal => decimal,
+                    :array   => array_of_type
+                }
+            end
+
+            #
+            # See DBI::BaseDriver.
+            #
+            class Driver < DBI::BaseDriver
+                def initialize
+                    super("0.4.0")
+                end
+
+                ## List of datasources for this database.
+                def data_sources
+                    []
+                end
+
+                ## Connect to a database.
+                def connect(dbname, user, auth, attr)
+                    Database.new(dbname, user, auth, attr)
+                end
+            end
+        end # module Pg
+    end # module DBD
+end # module DBI
+
+require 'dbd/pg/type'
+require 'dbd/pg/database'
+require 'dbd/pg/statement'
+require 'dbd/pg/tuples'
+require 'dbd/pg/exec'
+
+pg = DBI::DBD::Pg
+
+DBI::TypeUtil.register_conversion(pg.driver_name) do |obj|
+    newobj = case obj
+             when ::DateTime
+                 obj.strftime("%Y-%m-%dT%H:%M:%S.%N")
+             when ::Time
+                 ::DateTime.parse(obj.to_s).strftime("%H:%M:%S.%N")
+             when ::Date
+                 obj.strftime("%Y-%m-%d")
+             when ::Array
+                 pg.generate_array(obj)
+             when DBI::DBD::Pg::Type::ByteA
+                 obj.escaped
+             else
+                 obj
+             end
+    [newobj, false]
+end

data/lib/dbd/SQLite.rb

@@ -0,0 +1,97 @@
+#--
+###############################################################################
+#
+# DBD::SQLite - a DBD for SQLite for versions < 3
+#
+# Uses Jamis Buck's 'sqlite-ruby' driver to interface with SQLite directly
+#
+# (c) 2008 Erik Hollensbe & Christopher Maujean.
+#
+# TODO
+#
+# fetch_scroll implementation?
+# columns and column_info differ too much and have too much copied code, refactor
+# there are probably some edge cases with transactions
+#
+################################################################################
+#++
+
+begin
+    require 'rubygems'
+    gem 'sqlite-ruby'
+    gem 'dbi'
+rescue Exception => e
+end
+
+require 'dbi'
+require 'sqlite'
+
+module DBI
+    module DBD
+        #
+        # DBD::SQLite - Database Driver for SQLite versions 2.x and lower.
+        #
+        # Requires DBI and the 'sqlite-ruby' gem to work.
+        #
+        # Only things that extend DBI's results are documented.
+        #
+        class SQLite
+            VERSION = "0.1.2"
+            DESCRIPTION = "SQLite 2.x DBI DBD"
+
+            #
+            # returns 'SQLite'
+            #
+            # See DBI::TypeUtil#convert for more information.
+            #
+            def self.driver_name
+                "SQLite"
+            end
+
+            #
+            # Validates that the SQL has no literal NUL characters. (ASCII 0)
+            #
+            # SQLite apparently really hates it when you do that.
+            #
+            # It will raise DBI::DatabaseError should it find any.
+            #
+            def self.check_sql(sql)
+                # XXX I'm starting to think this is less of a problem with SQLite
+                # and more with the old C DBD
+                raise DBI::DatabaseError, "Bad SQL: SQL cannot contain nulls" if sql =~ /\0/
+            end
+
+            #
+            # Split a type definition into parts via String#match and return the whole result.
+            #
+            def self.parse_type(type_name)
+                type_name.match(/^([^\(]+)(\((\d+)(,(\d+))?\))?$/)
+            end
+
+            #
+            # See DBI::BaseDriver.
+            #
+            class Driver < DBI::BaseDriver
+                def initialize
+                    super "0.4.0"
+                end
+
+                def connect(dbname, user, auth, attr_hash)
+                    return Database.new(dbname, user, auth, attr_hash)
+                end
+            end
+        end
+    end
+end
+
+require 'dbd/sqlite/database'
+require 'dbd/sqlite/statement'
+
+DBI::TypeUtil.register_conversion(DBI::DBD::SQLite.driver_name) do |obj|
+    case obj
+    when ::NilClass
+        ["NULL", false]
+    else 
+        [obj, true]
+    end
+end

data/lib/dbd/SQLite3.rb

@@ -0,0 +1,124 @@
+#--
+# DBD::SQLite3
+#
+# copyright (c) 2005 Jun Mukai <mukai@jmuk.org>
+# Compatibility patches by Erik Hollensbe <erik@hollensbe.org>
+#
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions
+# are met:
+# 1. Redistributions of source code must retain the above copyright
+#    notice, this list of conditions and the following disclaimer.
+# 2. Redistributions in binary form must reproduce the above copyright
+#    notice, this list of conditions and the following disclaimer in the
+#    documentation and/or other materials provided with the distribution.
+# 3. The name of the author may not be used to endorse or promote products
+#    derived from this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
+# INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
+# AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
+# THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
+# OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
+# OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
+# ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+#++
+
+begin
+  require 'rubygems'
+  gem 'sqlite3-ruby'
+  gem 'dbi'
+rescue LoadError
+end
+
+require 'dbi'
+require 'sqlite3'
+require 'sqlite3/version'
+
+module DBI
+    module DBD
+        #
+        # DBD::SQLite3 - Database Driver for SQLite versions 3.x
+        #
+        # Requires DBI and the 'sqlite3-ruby' gem to work.
+        #
+        # Only things that extend DBI's results are documented.
+        #
+        module SQLite3
+            VERSION = "1.2.5"
+            DESCRIPTION = "SQLite 3.x DBD for DBI"
+
+            #
+            # returns 'SQLite3'
+            #
+            # See DBI::TypeUtil#convert for more information.
+            #
+            def self.driver_name
+                "SQLite3"
+            end
+
+            #
+            # Validates that the SQL has no literal NUL characters. (ASCII 0)
+            #
+            # SQLite apparently really hates it when you do that.
+            #
+            # It will raise DBI::DatabaseError should it find any.
+            #
+            def self.parse_type(type_name)
+                # FIXME plucked from SQLite driver, this needs to be in DBI proper
+                return ['varchar'] unless type_name
+                type_name.match(/^([^\(\s]+)\s*(\(\s*(\d+)\s*(,\s*(\d+))?\s*\))?$/)
+            end
+
+            #
+            # See DBI::BaseDriver.
+            #
+            class Driver < DBI::BaseDriver
+                def initialize
+                    @dbs = []
+                    super "0.4.0"
+                end
+
+                def connect(dbname, user, auth, attr)
+                    raise DBI::InterfaceError, "dbname must be a string" unless dbname.kind_of? String
+                    raise DBI::InterfaceError, "dbname must have some length" unless dbname.length > 0
+                    raise DBI::InterfaceError, "attrs must be a hash" unless attr.kind_of? Hash
+                    db = DBI::DBD::SQLite3::Database.new(dbname, attr)
+                    @dbs.push(db)
+                    db
+                end
+
+                def disconnect_all()
+                    @dbs.each{|db| db.disconnect()}
+                end
+            end
+        end
+    end
+end
+
+require 'dbd/sqlite3/database'
+require 'dbd/sqlite3/statement'
+
+DBI::TypeUtil.register_conversion(DBI::DBD::SQLite3.driver_name) do |obj|
+    newobj = case obj
+             when ::TrueClass
+                '1'
+             when ::FalseClass
+                '0'
+             else
+                 # SQLite3 is managing its own conversion right now, until I'm happy let's keep it that way
+                 obj.dup rescue obj
+             end
+    if newobj.kind_of?(::Symbol)
+        newobj.to_s
+    elsif newobj.object_id == obj.object_id
+        [newobj, true]
+    else
+        [newobj, false]
+    end
+end