knockapi 0.6.1 → 1.1.0

data/lib/knockapi/internal/type/union.rb

@@ -0,0 +1,226 @@
+# frozen_string_literal: true
+
+module Knockapi
+  module Internal
+    module Type
+      # @api private
+      #
+      # @example
+      #   # `recipient` is a `Knockapi::Recipient`
+      #   case recipient
+      #   when Knockapi::User
+      #     puts(recipient.id)
+      #   when Knockapi::Object
+      #     puts(recipient._typename)
+      #   else
+      #     puts(recipient)
+      #   end
+      module Union
+        include Knockapi::Internal::Type::Converter
+        include Knockapi::Internal::Util::SorbetRuntimeSupport
+
+        # @api private
+        #
+        # All of the specified variant info for this union.
+        #
+        # @return [Array<Array(Symbol, Proc)>]
+        private def known_variants = (@known_variants ||= [])
+
+        # @api private
+        #
+        # @return [Array<Array(Symbol, Object)>]
+        protected def derefed_variants
+          known_variants.map { |key, variant_fn| [key, variant_fn.call] }
+        end
+
+        # All of the specified variants for this union.
+        #
+        # @return [Array<Object>]
+        def variants = derefed_variants.map(&:last)
+
+        # @api private
+        #
+        # @param property [Symbol]
+        private def discriminator(property)
+          case property
+          in Symbol
+            @discriminator = property
+          end
+        end
+
+        # @api private
+        #
+        # @param key [Symbol, Hash{Symbol=>Object}, Proc, Knockapi::Internal::Type::Converter, Class]
+        #
+        # @param spec [Hash{Symbol=>Object}, Proc, Knockapi::Internal::Type::Converter, Class] .
+        #
+        #   @option spec [NilClass, TrueClass, FalseClass, Integer, Float, Symbol] :const
+        #
+        #   @option spec [Proc] :enum
+        #
+        #   @option spec [Proc] :union
+        #
+        #   @option spec [Boolean] :"nil?"
+        private def variant(key, spec = nil)
+          variant_info =
+            case key
+            in Symbol
+              [key, Knockapi::Internal::Type::Converter.type_info(spec)]
+            in Proc | Knockapi::Internal::Type::Converter | Class | Hash
+              [nil, Knockapi::Internal::Type::Converter.type_info(key)]
+            end
+
+          known_variants << variant_info
+        end
+
+        # @api private
+        #
+        # @param value [Object]
+        #
+        # @return [Knockapi::Internal::Type::Converter, Class, nil]
+        private def resolve_variant(value)
+          case [@discriminator, value]
+          in [_, Knockapi::Internal::Type::BaseModel]
+            value.class
+          in [Symbol, Hash]
+            key = value.fetch(@discriminator) do
+              value.fetch(@discriminator.to_s, Knockapi::Internal::OMIT)
+            end
+
+            return nil if key == Knockapi::Internal::OMIT
+
+            key = key.to_sym if key.is_a?(String)
+            known_variants.find { |k,| k == key }&.last&.call
+          else
+            nil
+          end
+        end
+
+        # rubocop:disable Style/HashEachMethods
+        # rubocop:disable Style/CaseEquality
+
+        # @api public
+        #
+        # @param other [Object]
+        #
+        # @return [Boolean]
+        def ===(other)
+          known_variants.any? do |_, variant_fn|
+            variant_fn.call === other
+          end
+        end
+
+        # @api public
+        #
+        # @param other [Object]
+        #
+        # @return [Boolean]
+        def ==(other)
+          Knockapi::Internal::Type::Union === other && other.derefed_variants == derefed_variants
+        end
+
+        # @api public
+        #
+        # @return [Integer]
+        def hash = variants.hash
+
+        # @api private
+        #
+        # @param value [Object]
+        #
+        # @param state [Hash{Symbol=>Object}] .
+        #
+        #   @option state [Boolean, :strong] :strictness
+        #
+        #   @option state [Hash{Symbol=>Object}] :exactness
+        #
+        #   @option state [Integer] :branched
+        #
+        # @return [Object]
+        def coerce(value, state:)
+          if (target = resolve_variant(value))
+            return Knockapi::Internal::Type::Converter.coerce(target, value, state: state)
+          end
+
+          strictness = state.fetch(:strictness)
+          exactness = state.fetch(:exactness)
+          state[:strictness] = strictness == :strong ? true : strictness
+
+          alternatives = []
+          known_variants.each do |_, variant_fn|
+            target = variant_fn.call
+            exact = state[:exactness] = {yes: 0, no: 0, maybe: 0}
+            state[:branched] += 1
+
+            coerced = Knockapi::Internal::Type::Converter.coerce(target, value, state: state)
+            yes, no, maybe = exact.values
+            if (no + maybe).zero? || (!strictness && yes.positive?)
+              exact.each { exactness[_1] += _2 }
+              state[:exactness] = exactness
+              return coerced
+            elsif maybe.positive?
+              alternatives << [[-yes, -maybe, no], exact, coerced]
+            end
+          end
+
+          case alternatives.sort_by(&:first)
+          in []
+            exactness[:no] += 1
+            if strictness == :strong
+              message = "no possible conversion of #{value.class} into a variant of #{target.inspect}"
+              raise ArgumentError.new(message)
+            end
+            value
+          in [[_, exact, coerced], *]
+            exact.each { exactness[_1] += _2 }
+            coerced
+          end
+            .tap { state[:exactness] = exactness }
+        ensure
+          state[:strictness] = strictness
+        end
+
+        # @api private
+        #
+        # @param value [Object]
+        #
+        # @param state [Hash{Symbol=>Object}] .
+        #
+        #   @option state [Boolean] :can_retry
+        #
+        # @return [Object]
+        def dump(value, state:)
+          if (target = resolve_variant(value))
+            return Knockapi::Internal::Type::Converter.dump(target, value, state: state)
+          end
+
+          known_variants.each do
+            target = _2.call
+            return Knockapi::Internal::Type::Converter.dump(target, value, state: state) if target === value
+          end
+
+          super
+        end
+
+        # rubocop:enable Style/CaseEquality
+        # rubocop:enable Style/HashEachMethods
+
+        # @api private
+        #
+        # @param depth [Integer]
+        #
+        # @return [String]
+        def inspect(depth: 0)
+          if depth.positive?
+            return is_a?(Module) ? super() : self.class.name
+          end
+
+          members = variants.map { Knockapi::Internal::Type::Converter.inspect(_1, depth: depth.succ) }
+          prefix = is_a?(Module) ? name : self.class.name
+
+          "#{prefix}[#{members.join(' | ')}]"
+        end
+      end
+    end
+  end
+end

data/lib/knockapi/internal/type/unknown.rb

@@ -0,0 +1,67 @@
+# frozen_string_literal: true
+
+module Knockapi
+  module Internal
+    module Type
+      # @api private
+      #
+      # @abstract
+      #
+      # When we don't know what to expect for the value.
+      class Unknown
+        extend Knockapi::Internal::Type::Converter
+
+        # rubocop:disable Lint/UnusedMethodArgument
+
+        private_class_method :new
+
+        # @api public
+        #
+        # @param other [Object]
+        #
+        # @return [Boolean]
+        def self.===(other) = true
+
+        # @api public
+        #
+        # @param other [Object]
+        #
+        # @return [Boolean]
+        def self.==(other) = other.is_a?(Class) && other <= Knockapi::Internal::Type::Unknown
+
+        class << self
+          # @api private
+          #
+          # @param value [Object]
+          #
+          # @param state [Hash{Symbol=>Object}] .
+          #
+          #   @option state [Boolean, :strong] :strictness
+          #
+          #   @option state [Hash{Symbol=>Object}] :exactness
+          #
+          #   @option state [Integer] :branched
+          #
+          # @return [Object]
+          def coerce(value, state:)
+            state.fetch(:exactness)[:yes] += 1
+            value
+          end
+
+          # @!method dump(value, state:)
+          #   @api private
+          #
+          #   @param value [Object]
+          #
+          #   @param state [Hash{Symbol=>Object}] .
+          #
+          #     @option state [Boolean] :can_retry
+          #
+          #   @return [Object]
+        end
+
+        # rubocop:enable Lint/UnusedMethodArgument
+      end
+    end
+  end
+end