systemprompt_analytics/services/
providers.rs

1use async_trait::async_trait;
2use chrono::Utc;
3use http::{HeaderMap, Uri};
4use systemprompt_identifiers::{SessionId, UserId};
5use systemprompt_traits::{
6    AnalyticsProvider, AnalyticsProviderError, AnalyticsResult, AnalyticsSession,
7    CreateSessionInput, FingerprintProvider, SessionAnalytics as TraitSessionAnalytics,
8    SessionAnalyticsProvider, SessionAnalyticsProviderError, SessionAnalyticsResult,
9};
10
11use super::SessionAnalytics;
12use super::service::AnalyticsService;
13use crate::repository::{FingerprintRepository, SessionRepository};
14
15#[async_trait]
16impl AnalyticsProvider for AnalyticsService {
17    fn extract_analytics(&self, headers: &HeaderMap, uri: Option<&Uri>) -> TraitSessionAnalytics {
18        let local = Self::extract_analytics(self, headers, uri);
19        TraitSessionAnalytics {
20            ip_address: local.ip_address.clone(),
21            user_agent: local.user_agent.clone(),
22            device_type: local.device_type.clone(),
23            browser: local.browser.clone(),
24            os: local.os.clone(),
25            fingerprint_hash: local.fingerprint_hash.clone(),
26            referer: local.referrer_url.clone(),
27            referrer_url: local.referrer_url.clone(),
28            referrer_source: local.referrer_source.clone(),
29            accept_language: local.preferred_locale.clone(),
30            preferred_locale: local.preferred_locale.clone(),
31            screen_width: None,
32            screen_height: None,
33            timezone: None,
34            page_url: local.entry_url.clone(),
35            landing_page: local.landing_page.clone(),
36            entry_url: local.entry_url,
37            country: local.country.clone(),
38            region: local.region.clone(),
39            city: local.city,
40            utm_source: local.utm_source,
41            utm_medium: local.utm_medium,
42            utm_campaign: local.utm_campaign,
43            utm_content: local.utm_content,
44            utm_term: local.utm_term,
45        }
46    }
47
48    async fn create_session(&self, input: CreateSessionInput<'_>) -> AnalyticsResult<()> {
49        let local_analytics = SessionAnalytics {
50            ip_address: input.analytics.ip_address.clone(),
51            user_agent: input.analytics.user_agent.clone(),
52            device_type: input.analytics.device_type.clone(),
53            browser: input.analytics.browser.clone(),
54            os: input.analytics.os.clone(),
55            fingerprint_hash: input.analytics.fingerprint_hash.clone(),
56            referrer_url: input
57                .analytics
58                .referrer_url
59                .clone()
60                .or_else(|| input.analytics.referer.clone()),
61            referrer_source: input.analytics.referrer_source.clone(),
62            preferred_locale: input
63                .analytics
64                .preferred_locale
65                .clone()
66                .or_else(|| input.analytics.accept_language.clone()),
67            landing_page: input.analytics.landing_page.clone(),
68            entry_url: input
69                .analytics
70                .entry_url
71                .clone()
72                .or_else(|| input.analytics.page_url.clone()),
73            country: input.analytics.country.clone(),
74            region: input.analytics.region.clone(),
75            city: input.analytics.city.clone(),
76            utm_source: input.analytics.utm_source.clone(),
77            utm_medium: input.analytics.utm_medium.clone(),
78            utm_campaign: input.analytics.utm_campaign.clone(),
79            utm_content: input.analytics.utm_content.clone(),
80            utm_term: input.analytics.utm_term.clone(),
81        };
82
83        let local_input = super::service::CreateAnalyticsSessionInput {
84            session_id: input.session_id,
85            user_id: input.user_id,
86            analytics: &local_analytics,
87            session_source: input.session_source,
88            is_bot: input.is_bot,
89            expires_at: input.expires_at,
90        };
91
92        self.create_analytics_session(local_input)
93            .await
94            .map_err(|e| AnalyticsProviderError::Internal(e.to_string()))
95    }
96
97    async fn find_recent_session_by_fingerprint(
98        &self,
99        fingerprint: &str,
100        max_age_seconds: i64,
101    ) -> AnalyticsResult<Option<AnalyticsSession>> {
102        let result = Self::find_recent_session_by_fingerprint(self, fingerprint, max_age_seconds)
103            .await
104            .map_err(|e| AnalyticsProviderError::Internal(e.to_string()))?;
105
106        Ok(result.map(|r| AnalyticsSession {
107            session_id: r.session_id,
108            user_id: r.user_id,
109            fingerprint: Some(fingerprint.to_string()),
110            created_at: Utc::now(),
111        }))
112    }
113
114    async fn find_session_by_id(
115        &self,
116        session_id: &SessionId,
117    ) -> AnalyticsResult<Option<AnalyticsSession>> {
118        let result = self
119            .session_repo()
120            .find_by_id(session_id)
121            .await
122            .map_err(|e| AnalyticsProviderError::Internal(e.to_string()))?;
123
124        Ok(result.map(|r| AnalyticsSession {
125            session_id: r.session_id,
126            user_id: r.user_id,
127            fingerprint: r.fingerprint_hash,
128            created_at: r.started_at.unwrap_or_else(Utc::now),
129        }))
130    }
131
132    async fn migrate_user_sessions(
133        &self,
134        from_user_id: &UserId,
135        to_user_id: &UserId,
136    ) -> AnalyticsResult<u64> {
137        self.session_repo()
138            .migrate_user_sessions(from_user_id, to_user_id)
139            .await
140            .map_err(|e| AnalyticsProviderError::Internal(e.to_string()))
141    }
142
143    async fn mark_session_converted(&self, session_id: &SessionId) -> AnalyticsResult<()> {
144        self.session_repo()
145            .mark_converted(session_id)
146            .await
147            .map_err(|e| AnalyticsProviderError::Internal(e.to_string()))
148    }
149}
150
151#[async_trait]
152impl FingerprintProvider for FingerprintRepository {
153    async fn count_active_sessions(&self, fingerprint: &str) -> AnalyticsResult<i64> {
154        self.count_active_sessions(fingerprint)
155            .await
156            .map(i64::from)
157            .map_err(|e| AnalyticsProviderError::Internal(e.to_string()))
158    }
159
160    async fn find_reusable_session(&self, fingerprint: &str) -> AnalyticsResult<Option<String>> {
161        self.find_reusable_session(fingerprint)
162            .await
163            .map_err(|e| AnalyticsProviderError::Internal(e.to_string()))
164    }
165
166    async fn upsert_fingerprint(
167        &self,
168        fingerprint: &str,
169        ip_address: Option<&str>,
170        user_agent: Option<&str>,
171        _screen_info: Option<&str>,
172    ) -> AnalyticsResult<()> {
173        self.upsert_fingerprint(fingerprint, ip_address, user_agent, None)
174            .await
175            .map(|_| ())
176            .map_err(|e| AnalyticsProviderError::Internal(e.to_string()))
177    }
178}
179
180#[async_trait]
181impl SessionAnalyticsProvider for SessionRepository {
182    async fn increment_task_count(&self, session_id: &SessionId) -> SessionAnalyticsResult<()> {
183        Self::increment_task_count(self, session_id)
184            .await
185            .map_err(|e| SessionAnalyticsProviderError::Internal(e.to_string()))
186    }
187
188    async fn increment_message_count(&self, session_id: &SessionId) -> SessionAnalyticsResult<()> {
189        Self::increment_message_count(self, session_id)
190            .await
191            .map_err(|e| SessionAnalyticsProviderError::Internal(e.to_string()))
192    }
193}
systemprompt_analytics/services/providers.rs

systemprompt_analytics/services/
providers.rs
