module NxtSchema
  class CallableOrValue
    def initialize(callee)
      @callee = callee

      if callee.is_a?(Symbol)
        self.type = :method
      elsif callee.respond_to?(:call)
        self.type = :proc
        self.context = callee.binding
      else
        self.type = :value
      end
    end

    def bind(execution_context = nil)
      self.context = execution_context
      ensure_context_not_missing
      self
    end

    # NOTE: Currently we only allow arguments! Not keyword args or **options
    # If we would allow **options and we would pass a hash as the only argument it would
    # automatically be parsed as the options!
    def call(*args)
      return callee if value?

      ensure_context_not_missing

      args = args.take(arity)

      if method?
        context.send(callee, *args)
      else
        context.instance_exec(*args, &callee)
      end
    end

    def arity
      if proc?
        callee.arity
      elsif method?
        method = context.send(:method, callee)
        method.arity
      else
        raise ArgumentError, "Can't resolve arity from #{callee}"
      end
    end

    private

    def proc?
      type == :proc
    end

    def method?
      type == :method
    end

    def value?
      type == :value
    end

    def ensure_context_not_missing
      return if context

      raise ArgumentError, "Missing context: #{context}"
    end

    attr_accessor :context, :callee, :type
  end
end