chingor13/json_api_client

require 'bigdecimal'
module JsonApiClient
  class Schema
    module Types

      class Integer
        def self.cast(value, _)
          value.to_i
        end
      end

      class String
        def self.cast(value, _)
          value.to_s
        end
      end

      class Float
        def self.cast(value, _)
          value.to_f
        end
      end

      class Time
        def self.cast(value, _)
          value.is_a?(::Time) ? value : ::Time.parse(value)
        end
      end

      class Decimal
        def self.cast(value, _)
          BigDecimal(value)
        end
      end

      class Boolean
        def self.cast(value, default)
          case value
            when "false", "0", 0, false
              false
            when "true", "1", 1, true
              true
            else
              # if it's unknown, use the default value
              default
          end
        end
      end

    end

    class TypeFactory
      @@types = {}
      # Register a new type key or keys with appropriate classes
      #
      # eg:
      #
      #   require 'money'
      #
      #   class MyMoneyCaster
      #      def self.cast(value, default)
      #         begin
      #           Money.new(value, "USD")
      #         rescue ArgumentError
      #           default
      #         end
      #      end
      #   end
      #
      #   JsonApiClient::Schema::TypeFactory.register money: MyMoneyCaster
      #
      # You can setup several at once:
      #
      #   JsonApiClient::Schema::TypeFactory.register money: MyMoneyCaster,
      #                                         date: MyJsonDateTypeCaster
      #
      #
      #
      #
      def self.register(type_hash)
        @@types.merge!(type_hash)
      end

      def self.type_for(type)
        @@types[type]
      end

      self.register int: Types::Integer,
                    integer: Types::Integer,
                    string: Types::String,
                    float: Types::Float,
                    time: Types::Time,
                    decimal: Types::Decimal,
                    boolean: Types::Boolean

    end

    Property = Struct.new(:name, :type, :default) do
      def cast(value)
        return nil if value.nil?
        return value if type.nil?
        type_caster = TypeFactory.type_for(type)
        return value if type_caster.nil?
        type_caster.cast(value, default)
      end
    end

    def initialize
      @properties = {}
    end

    # Add a property to the schema
    #
    # @param name [Symbol] the name of the property
    # @param options [Hash] property options
    # @option options [Symbol] :type The property type
    # @option options [Symbol] :default The default value for the property
    # @return [void]
    def add(name, options)
      @properties[name.to_sym] = Property.new(name.to_sym, options[:type], options[:default])
    end

    # How many properties are defined
    #
    # @return [Fixnum] the number of defined properties
    def size
      @properties.size
    end

    alias_method :length, :size

    def each_property(&block)
      @properties.values.each(&block)
    end

    alias_method :each, :each_property

    # Look up a property by name
    #
    # @param property_name [String] the name of the property
    # @return [Property, nil] the property definition for property_name or nil
    def find(property_name)
      @properties[property_name.to_sym]
    end

    alias_method :[], :find

    class << self
      def register(type_hash)
        TypeFactory.register(type_hash)
      end
    end
  end
end