Contents

module ArJdbc
  module MSSQL
    # @note Lot of code kindly borrowed from **activerecord-sqlserver-adapter**.
    module Utils

      module_function

      GET_TABLE_NAME_INSERT_UPDATE_RE =
        /^\s*(INSERT|EXEC sp_executesql N'INSERT)\s+INTO\s+([^\(\s,]+)\s*|^\s*update\s+([^\(\s,]+)\s*/i

      GET_TABLE_NAME_FROM_RE = /\bFROM\s+([^\(\)\s,]+)\s*/i

      def get_table_name(sql, qualified = nil)
        if sql =~ GET_TABLE_NAME_INSERT_UPDATE_RE
          tn = $2 || $3
          qualified ? tn : unqualify_table_name(tn)
        elsif sql =~ GET_TABLE_NAME_FROM_RE
          qualified ? $1 : unqualify_table_name($1)
        else
          nil
        end
      end

      # protected

      # See "Delimited Identifiers": http://msdn.microsoft.com/en-us/library/ms176027.aspx
      def remove_identifier_delimiters(keyword)
        if /\A(\[|")(.*)/m.match(keyword)
          delim, rest = $1, $2
          if delim == '[' && rest =~ /]\z/ || delim == '"' && rest =~ /"\z/
            return rest.chop
          end
        end
        keyword
      end

      def unquote_table_name(table_name)
        remove_identifier_delimiters(table_name)
      end

      def unquote_column_name(column_name)
        remove_identifier_delimiters(column_name)
      end

      def unquote_string(string)
        string.to_s.gsub("''", "'")
      end

      def unqualify_table_name(table_name)
        remove_identifier_delimiters(table_name.to_s.split('.').last)
      end

      def unqualify_table_schema(table_name)
        remove_identifier_delimiters(table_name.to_s.split('.')[-2]) rescue nil
      end

      def unqualify_db_name(table_name)
        table_names = table_name.to_s.split('.')
        table_names.length == 3 ? remove_identifier_delimiters(table_names.first) : nil
      end

    end
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version	Path
activerecord-jdbc-adapter-1.3.9	lib/arjdbc/mssql/utils.rb
activerecord-jdbc-adapter-1.3.8	lib/arjdbc/mssql/utils.rb
activerecord-jdbc-adapter-1.3.7	lib/arjdbc/mssql/utils.rb
activerecord-jdbc-adapter-1.3.6	lib/arjdbc/mssql/utils.rb
activerecord-jdbc-adapter-1.3.5	lib/arjdbc/mssql/utils.rb
activerecord-jdbc-adapter-1.3.4	lib/arjdbc/mssql/utils.rb
activerecord-jdbc-adapter-1.3.3	lib/arjdbc/mssql/utils.rb
activerecord-jdbc-adapter-1.3.2	lib/arjdbc/mssql/utils.rb