statsig_rust/user/
statsig_user_internal.rs

1use super::StatsigUserLoggable;
2use crate::evaluation::dynamic_value::DynamicValue;
3use crate::StatsigUser;
4use crate::{evaluation::dynamic_string::DynamicString, Statsig};
5
6pub type FullUserKey = (
7    u64,      // app_version
8    u64,      // country
9    u64,      // email
10    u64,      // ip
11    u64,      // locale
12    u64,      // user_agent
13    u64,      // user_id
14    Vec<u64>, // custom_ids
15    Vec<u64>, // custom
16    Vec<u64>, // private_attributes
17    Vec<u64>, // statsig_env
18);
19
20#[derive(Clone)]
21pub struct StatsigUserInternal<'statsig, 'user> {
22    pub user_ref: &'user StatsigUser,
23    pub statsig_instance: Option<&'statsig Statsig>,
24}
25
26impl<'statsig, 'user> StatsigUserInternal<'statsig, 'user> {
27    pub fn new(user: &'user StatsigUser, statsig_instance: Option<&'statsig Statsig>) -> Self {
28        Self {
29            user_ref: user,
30            statsig_instance,
31        }
32    }
33
34    pub fn get_unit_id(&self, id_type: &DynamicString) -> Option<&DynamicValue> {
35        if id_type.lowercased_value.eq("userid") {
36            return self.user_ref.data.user_id.as_ref();
37        }
38
39        let custom_ids = self.user_ref.data.custom_ids.as_ref()?;
40
41        if let Some(custom_id) = custom_ids.get(&id_type.value) {
42            return Some(custom_id);
43        }
44
45        custom_ids.get(&id_type.lowercased_value)
46    }
47
48    pub fn get_user_value(&self, field: &Option<DynamicString>) -> Option<&DynamicValue> {
49        let field = field.as_ref()?;
50
51        let lowered_field = &field.lowercased_value;
52
53        let str_value = match lowered_field as &str {
54            "userid" => &self.user_ref.data.user_id,
55            "email" => &self.user_ref.data.email,
56            "ip" => &self.user_ref.data.ip,
57            "country" => &self.user_ref.data.country,
58            "locale" => &self.user_ref.data.locale,
59            "appversion" => &self.user_ref.data.app_version,
60            "useragent" => &self.user_ref.data.user_agent,
61            _ => &None,
62        };
63
64        if str_value.is_some() {
65            return str_value.as_ref();
66        }
67
68        if let Some(custom) = &self.user_ref.data.custom {
69            if let Some(found) = custom.get(&field.value) {
70                return Some(found);
71            }
72            if let Some(lowered_found) = custom.get(lowered_field) {
73                return Some(lowered_found);
74            }
75        }
76
77        if let Some(instance) = &self.statsig_instance {
78            if let Some(val) = instance.get_value_from_global_custom_fields(&field.value) {
79                return Some(val);
80            }
81
82            if let Some(val) = instance.get_value_from_global_custom_fields(&field.lowercased_value)
83            {
84                return Some(val);
85            }
86        }
87
88        if let Some(private_attributes) = &self.user_ref.data.private_attributes {
89            if let Some(found) = private_attributes.get(&field.value) {
90                return Some(found);
91            }
92            if let Some(lowered_found) = private_attributes.get(lowered_field) {
93                return Some(lowered_found);
94            }
95        }
96
97        let str_value_alt = match lowered_field as &str {
98            "user_id" => &self.user_ref.data.user_id,
99            "app_version" => &self.user_ref.data.app_version,
100            "user_agent" => &self.user_ref.data.user_agent,
101            _ => &None,
102        };
103
104        if str_value_alt.is_some() {
105            return str_value_alt.as_ref();
106        }
107
108        None
109    }
110
111    pub fn get_value_from_environment(
112        &self,
113        field: &Option<DynamicString>,
114    ) -> Option<DynamicValue> {
115        let field = field.as_ref()?;
116
117        if let Some(result) = self.statsig_instance?.get_from_statsig_env(&field.value) {
118            return Some(result);
119        }
120
121        self.statsig_instance?
122            .get_from_statsig_env(&field.lowercased_value)
123    }
124
125    pub fn to_loggable(&self) -> StatsigUserLoggable {
126        let (environment, global_custom) = match self.statsig_instance {
127            Some(statsig) => (
128                statsig.use_statsig_env(|e| e.cloned()),
129                statsig.use_global_custom_fields(|gc| gc.cloned()),
130            ),
131            None => (None, None),
132        };
133
134        StatsigUserLoggable::new(&self.user_ref.data, environment, global_custom)
135    }
136}
statsig_rust/user/statsig_user_internal.rs

statsig_rust/user/
statsig_user_internal.rs
