module SchemaPlus
  module ActiveRecord
    module ConnectionAdapters
      # Instances of this class are returned by the queries ActiveRecord::Base#foreign_keys and ActiveRecord::Base#reverse_foreign_keys (via AbstractAdapter#foreign_keys and AbstractAdapter#reverse_foreign_keys)
      #
      # The on_update and on_delete attributes can take on the following values:
      #   :cascade
      #   :restrict
      #   :set_null
      #   :set_default
      #   :no_action
      class ForeignKeyDefinition

        # The name of the foreign key constraint
        attr_reader :name

        # The name of the table the constraint is defined on
        attr_reader :table_name

        # The list of column names that are constrained (as strings).
        attr_reader :column_names

        # The foreign table that is referenced by the constraint
        attr_reader :references_table_name

        # The list of column names (as strings) of the foreign table that are referenced
        # by the constraint
        attr_reader :references_column_names

        # The ON_UPDATE behavior for the constraint.  See above for the
        # possible values.
        attr_reader :on_update

        # The ON_DELETE behavior for the constraint.  See above for the
        # possible values.
        attr_reader :on_delete

        # True if the constraint is deferrable
        attr_reader :deferrable

        # :enddoc:

        ACTIONS = { :cascade => "CASCADE", :restrict => "RESTRICT", :set_null => "SET NULL", :set_default => "SET DEFAULT", :no_action => "NO ACTION" }.freeze

        def initialize(name, table_name, column_names, references_table_name, references_column_names, on_update = nil, on_delete = nil, deferrable = nil)
          @name = name
          @table_name = unquote(table_name)
          @column_names = unquote(column_names)
          @references_table_name = unquote(references_table_name)
          @references_column_names = unquote(references_column_names)
          @on_update = on_update
          @on_delete = on_delete
          @deferrable = deferrable

          ACTIONS.has_key?(on_update) or raise(ArgumentError, "invalid :on_update action: #{on_update.inspect}") if on_update
          ACTIONS.has_key?(on_delete) or raise(ArgumentError, "invalid :on_delete action: #{on_delete.inspect}") if on_delete
          if ::ActiveRecord::Base.connection.adapter_name =~ /^mysql/i
            raise(NotImplementedError, "MySQL does not support ON UPDATE SET DEFAULT") if on_update == :set_default
            raise(NotImplementedError, "MySQL does not support ON DELETE SET DEFAULT") if on_delete == :set_default
          end
        end

        # Dumps a definition of foreign key.
        def to_dump(opts={})
          dump = (opts[:inline] ? "  t.foreign_key" : "add_foreign_key #{table_name.inspect},")
          dump << " [#{Array(column_names).collect{ |name| name.inspect }.join(', ')}]"
          dump << ", #{references_table_name.inspect}, [#{Array(references_column_names).collect{ |name| name.inspect }.join(', ')}]"
          dump << ", :on_update => :#{on_update}" if on_update
          dump << ", :on_delete => :#{on_delete}" if on_delete
          dump << ", :deferrable => #{deferrable}" if deferrable
          dump << ", :name => #{name.inspect}" if name
          dump << "\n"
          dump
        end

        def to_sql
          sql = name ? "CONSTRAINT #{name} " : ""
          sql << "FOREIGN KEY (#{quoted_column_names.join(", ")}) REFERENCES #{quoted_references_table_name} (#{quoted_references_column_names.join(", ")})"
          sql << " ON UPDATE #{ACTIONS[on_update]}" if on_update
          sql << " ON DELETE #{ACTIONS[on_delete]}" if on_delete
          sql << " DEFERRABLE" if deferrable
          sql
        end

        def quoted_column_names
          Array(column_names).collect { |name| ::ActiveRecord::Base.connection.quote_column_name(name) }
        end

        def quoted_references_column_names
          Array(references_column_names).collect { |name| ::ActiveRecord::Base.connection.quote_column_name(name) }
        end

        def quoted_references_table_name
          ::ActiveRecord::Base.connection.quote_table_name(references_table_name)
        end

        def unquote(names)
          if names.is_a?(Array)
            names.collect { |name| __unquote(name) }
          else
            __unquote(names)
          end
        end

        def __unquote(value)
          value.to_s.sub(/^["`](.*)["`]$/, '\1')
        end

        def self.auto_index_name(table_name, column_name)
          "fk__#{table_name}_#{Array.wrap(column_name).join('_and_')}"
        end

      end
    end
  end
end