Contents

require 'ostruct'

module Datacaster
  module Runtimes
    class UserContext < Base
      class ContextStruct
        def self.context_has_key?(context, key)
          context.respond_to?(:key?) && context.key?(key) || context.to_h.key?(key.to_sym)
        end

        def initialize(context, node)
          @context = context
          @node = node
        end

        def method_missing(m, *args)
          if args.length > 1 || block_given?
            return super
          end

          if self.class.context_has_key?(@context, m) && args.empty?
            return @context[m]
          end

          if m =~ /\A.+=\z/ && args.length == 1
            return @context[m[0..-2].to_sym] = args[0]
          end

          begin
            @node.class.send_to_parent(@node, :context).public_send(m, *args)
          rescue NoMethodError
            raise NoMethodError.new("Key #{m.inspect} is not found in the context")
          end
        end

        def has_key?(key)
          self.class.context_has_key?(@context, key) || @node.class.send_to_parent(@node, :context).has_key?(key)
        rescue NoMethodError
          false
        end
      end

      def initialize(parent, user_context)
        super(parent)
        @context_struct = ContextStruct.new(user_context, self)

        @reserved_instance_variables += instance_variables
      end

      def context
        @context_struct
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
datacaster-4.2.0	lib/datacaster/runtimes/user_context.rb
datacaster-4.1.0	lib/datacaster/runtimes/user_context.rb