|
| 1 | +# frozen_string_literal: true |
| 2 | + |
| 3 | +require_relative 'mappers/environment' |
| 4 | +require_relative 'mappers/identity' |
| 5 | +require_relative 'mappers/segments' |
| 6 | + |
| 7 | +module Flagsmith |
| 8 | + module Engine |
| 9 | + # Mappers for converting between models and evaluation contexts |
| 10 | + module Mappers |
| 11 | + STRONGEST_PRIORITY = Float::INFINITY |
| 12 | + WEAKEST_PRIORITY = -Float::INFINITY |
| 13 | + |
| 14 | + # @param environment [Flagsmith::Engine::Environment] The environment model |
| 15 | + # @param identity [Flagsmith::Engine::Identity, nil] Optional identity model |
| 16 | + # @param override_traits [Array<Flagsmith::Engine::Identities::Trait>, nil] Optional override traits |
| 17 | + # @return [Hash] Evaluation context with environment, features, segments, and optionally identity |
| 18 | + def self.get_evaluation_context(environment, identity = nil, override_traits = nil) |
| 19 | + context = map_environment_model_to_evaluation_context(environment) |
| 20 | + context[:identity] = map_identity_model_to_identity_context(identity, override_traits) if identity |
| 21 | + context |
| 22 | + end |
| 23 | + |
| 24 | + # Maps environment model to evaluation context |
| 25 | + # |
| 26 | + # @param environment [Flagsmith::Engine::Environment] The environment model |
| 27 | + # @return [Hash] Context with :environment, :features, and :segments keys |
| 28 | + def self.map_environment_model_to_evaluation_context(environment) |
| 29 | + context = { |
| 30 | + environment: Environment.build_environment_context(environment), |
| 31 | + features: Environment.build_features_context(environment.feature_states), |
| 32 | + segments: Segments.build_segments_context(environment.project.segments) |
| 33 | + } |
| 34 | + |
| 35 | + context[:segments].merge!(Identity.map_overrides_to_segments(environment.identity_overrides)) if environment.identity_overrides&.any? |
| 36 | + |
| 37 | + context |
| 38 | + end |
| 39 | + |
| 40 | + # Maps identity model to identity context |
| 41 | + # |
| 42 | + # @param identity [Flagsmith::Engine::Identity] The identity model |
| 43 | + # @param override_traits [Array<Flagsmith::Engine::Identities::Trait>, nil] Optional override traits |
| 44 | + # @return [Hash] Identity context with :identifier, :key, and :traits |
| 45 | + def self.map_identity_model_to_identity_context(identity, override_traits = nil) |
| 46 | + Identity.build_environment_context(identity, override_traits) |
| 47 | + end |
| 48 | + |
| 49 | + # Maps segment rule model to rule hash |
| 50 | + # |
| 51 | + # @param rule [Flagsmith::Engine::Segments::Rule] The segment rule model |
| 52 | + # @return [Hash] Mapped rule with :type, :conditions, and :rules |
| 53 | + def self.map_segment_rule_model_to_rule(rule) |
| 54 | + Segments.map_rule(rule) |
| 55 | + end |
| 56 | + |
| 57 | + # Maps identity overrides to segments |
| 58 | + # |
| 59 | + # @param identity_overrides [Array<Flagsmith::Engine::Identity>] Array of identity override models |
| 60 | + # @return [Hash] Segments hash for identity overrides |
| 61 | + def self.map_identity_overrides_to_segments(identity_overrides) |
| 62 | + Identity.map_overrides_to_segments(identity_overrides) |
| 63 | + end |
| 64 | + end |
| 65 | + end |
| 66 | +end |
0 commit comments