module AR
  module Check
    module SchemaDumper
      def indexes(table, stream)
        super
        check_constraints(table, stream)
      end

      def check_constraints(table, stream)
        constraints = @connection.check_constraints(table)
        return if constraints.empty?

        constraints.each do |constraint|
          statement = [
            "add_check",
            ":#{constraint["table"]},",
            ":#{constraint["name"].gsub("_on_#{table}", "")},",
            constraint["expression"][1..-2].inspect
          ].join(" ")

          stream.puts "  #{statement}"
        end

        stream.puts
      end
    end
  end
end