module ActiveRecord
  module ConnectionAdapters
    module SchemaStatements
      def add_index_options_with_columns(table_name, column_name, options = {})
        columns = options.delete(:columns)
        index_name, index_type, index_columns, index_options, algorithm, using = add_index_options_without_columns(table_name, column_name, options)
        [index_name, index_type, columns || index_columns, index_options, algorithm, using]
      end

      alias_method_chain :add_index_options, :columns
    end

    module PostgreSQL
      module Quoting
        def quote_column_name_with_arrays(name)
          Array(name).map { |n| quote_column_name_without_arrays(n) }.join(", ")
        end

        alias_method_chain :quote_column_name, :arrays
      end

      module SchemaStatements
        def add_primary_key(table_name, column_name, _options = {})
          execute "ALTER TABLE #{quote_table_name(table_name)} ADD PRIMARY KEY (#{quote_column_name(column_name)})"
        end

        def remove_primary_key(table_name)
          execute "ALTER TABLE #{quote_table_name(table_name)} DROP PRIMARY KEY"
        end

        def alter_primary_key(table_name, new_columns)
          execute "ALTER TABLE #{quote_table_name(table_name)} DROP CONSTRAINT #{quote_table_name(table_name + "_pkey")}"
          execute "ALTER TABLE #{quote_table_name(table_name)} ADD PRIMARY KEY (#{quote_column_name(new_columns)})"
        end

        def create_enumeration(enumeration_name, values)
          execute "CREATE TYPE #{enumeration_name} AS ENUM ('#{values.join '\',\''}')"
        end

        def drop_enumeration(enumeration_name)
          execute "DROP TYPE #{enumeration_name}"
        end

        def rename_enumeration(old_name, new_name)
          execute "ALTER TYPE #{quote_table_name(old_name)} RENAME TO #{quote_table_name(new_name)}"
        end
      end
    end
  end
end