src/derails/passive_aggressive/connection_adapters/postgresql/oid/cidr.rb
# frozen_string_literal: true

require "ipaddr"

module PassiveAggressive
  module ConnectionAdapters
    module PostgreSQL
      module OID # :nodoc:
        class Cidr < Type::Value # :nodoc:
          def type
            :cidr
          end

          def type_cast_for_schema(value)
            # If the subnet mask is equal to /32, don't output it
            if value.prefix == 32
              "\"#{value}\""
            else
              "\"#{value}/#{value.prefix}\""
            end
          end

          def serialize(value)
            if IPAddr === value
              "#{value}/#{value.prefix}"
            else
              value
            end
          end

          # TODO: Remove when IPAddr#== compares IPAddr#prefix. See
          # https://github.com/ruby/ipaddr/issues/21
          def changed?(old_value, new_value, _new_value_before_type_cast)
            !old_value.eql?(new_value) || !old_value.nil? && old_value.prefix != new_value.prefix
          end

          def cast_value(value)
            if value.nil?
              nil
            elsif String === value
              begin
                IPAddr.new(value)
              rescue ArgumentError
                nil
              end
            else
              value
            end
          end
        end
      end
    end
  end
end
