# -----------------------------------------------------------------------------
#
# PostGIS adapter for ActiveRecord
#
# -----------------------------------------------------------------------------
# Copyright 2010-2012 Daniel Azuma
#
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice,
#   this list of conditions and the following disclaimer.
# * 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.
# * Neither the name of the copyright holder, nor the names of any other
#   contributors to this software, may be used to endorse or promote products
#   derived from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "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 COPYRIGHT OWNER OR CONTRIBUTORS 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.
# -----------------------------------------------------------------------------
;


module ActiveRecord  # :nodoc:

  module ConnectionAdapters  # :nodoc:


    # Extend JDBC's PostgreSQLAdapter implementation for compatibility with
    # ActiveRecord's default PostgreSQLAdapter.

    class PostgreSQLAdapter  # :nodoc:


      # Add `query` method for compatibility

      def query(*args)
        select_rows(*args)
      end


      # Backport from master, so PostGIS adapater will work with current stable
      # activerecord-jdbc-adapter gem.
      #
      # https://github.com/jruby/activerecord-jdbc-adapter/pull/200

      unless method_defined?(:schema_search_path=)
        def schema_search_path=(schema_csv)
          if schema_csv
            execute "SET search_path TO #{schema_csv}"
            @schema_search_path = schema_csv
          end
        end
      end


      # Backport from master, so PostGIS adapater will work with current stable
      # activerecord-jdbc-adapter gem.
      #
      # https://github.com/jruby/activerecord-jdbc-adapter/pull/200

      unless method_defined?(:schema_search_path)
        # Returns the active schema search path.
        def schema_search_path
          @schema_search_path ||= exec_query('SHOW search_path', 'SCHEMA')[0]['search_path']
        end
      end


      # For ActiveRecord 3.1 compatibility under activerecord-jdbc-adapter
      # 1.2.x: Add the "postgis" adapter to the matcher of jdbc-like adapters.
      if(defined?(ArJdbc::Version::VERSION) && ArJdbc::Version::VERSION.to_f < 1.3)
        def self.visitor_for(pool)
          config = pool.spec.config
          adapter = config[:adapter]
          adapter_spec = config[:adapter_spec] || self
          if adapter =~ /^(jdbc|jndi|postgis)$/
            adapter_spec.arel2_visitors(config).values.first.new(pool)
          else
            adapter_spec.arel2_visitors(config)[adapter].new(pool)
          end
        end
      end


    end


    module PostGISAdapter  # :nodoc:


      # Based on the default <tt>postgresql_connection</tt> definition from
      # activerecord-jdbc-adapter

      def self.create_jdbc_connection(context_, config_)
        begin
          require 'jdbc/postgres'
          ::Jdbc::Postgres.load_driver(:require) if defined?(::Jdbc::Postgres.load_driver)
        rescue LoadError # assuming driver.jar is on the class-path
        end
        require "arjdbc/postgresql"
        config_[:username] ||= ::Java::JavaLang::System.get_property("user.name")
        config_[:host] ||= "localhost"
        config_[:port] ||= 5432
        config_[:url] ||= "jdbc:postgresql://#{config_[:host]}:#{config_[:port]}/#{config_[:database]}"
        config_[:url] << config_[:pg_params] if config_[:pg_params]
        config_[:driver] ||= defined?(::Jdbc::Postgres.driver_name) ? ::Jdbc::Postgres.driver_name : 'org.postgresql.Driver'
        config_[:adapter_class] = ::ActiveRecord::ConnectionAdapters::PostGISAdapter::MainAdapter
        config_[:adapter_spec] = ::ArJdbc::PostgreSQL
        conn_ = context_.jdbc_connection(config_)
        conn_.execute("SET SEARCH_PATH TO #{config_[:schema_search_path]}") if config_[:schema_search_path]
        conn_
      end


    end


  end

end