1use relay_protocol::{Annotated, Array, Empty, FromValue, Getter, IntoValue, Val};
29
30use crate::processor::ProcessValue;
31use crate::protocol::{
32 AppContext, BrowserContext, ClientSdkInfo, Contexts, DefaultContext, DeviceContext, EventId,
33 LenientString, OsContext, ProfileContext, Request, ResponseContext, Tags, Timestamp,
34 TraceContext, User,
35};
36use uuid::Uuid;
37
38#[derive(Clone, Debug, Default, PartialEq, Empty, FromValue, IntoValue, ProcessValue)]
39#[metastructure(process_func = "process_replay", value_type = "Replay")]
40pub struct Replay {
41 pub event_id: Annotated<EventId>,
59
60 pub replay_id: Annotated<EventId>,
73
74 #[metastructure(max_chars = 64)]
86 pub replay_type: Annotated<String>,
87
88 pub segment_id: Annotated<u64>,
102
103 pub timestamp: Annotated<Timestamp>,
129
130 pub replay_start_timestamp: Annotated<Timestamp>,
132
133 #[metastructure(pii = "true", max_depth = 7, max_bytes = 8192)]
135 pub urls: Annotated<Array<String>>,
136
137 #[metastructure(max_depth = 5, max_bytes = 2048)]
139 pub error_ids: Annotated<Array<Uuid>>,
140
141 #[metastructure(max_depth = 5, max_bytes = 2048)]
143 pub trace_ids: Annotated<Array<Uuid>>,
144
145 #[metastructure(skip_serialization = "empty")]
147 pub contexts: Annotated<Contexts>,
148
149 #[metastructure(max_chars = 64)]
154 pub platform: Annotated<String>,
155
156 #[metastructure(
161 max_chars = 200,
162 required = false,
163 trim_whitespace = true,
164 nonempty = true,
165 skip_serialization = "empty"
166 )]
167 pub release: Annotated<LenientString>,
168
169 #[metastructure(
177 allow_chars = "a-zA-Z0-9_.-",
178 trim_whitespace = true,
179 required = false,
180 nonempty = true,
181 max_chars = 64
182 )]
183 pub dist: Annotated<String>,
184
185 #[metastructure(
191 max_chars = 64,
192 nonempty = true,
193 required = false,
194 trim_whitespace = true
195 )]
196 pub environment: Annotated<String>,
197
198 #[metastructure(skip_serialization = "empty", pii = "true")]
202 pub tags: Annotated<Tags>,
203
204 #[metastructure(field = "type", max_chars = 64)]
206 pub ty: Annotated<String>,
207
208 #[metastructure(skip_serialization = "empty")]
210 pub user: Annotated<User>,
211
212 #[metastructure(skip_serialization = "empty")]
214 pub request: Annotated<Request>,
215
216 #[metastructure(field = "sdk")]
218 #[metastructure(skip_serialization = "empty")]
219 pub sdk: Annotated<ClientSdkInfo>,
220}
221
222impl Replay {
223 pub fn context<C: DefaultContext>(&self) -> Option<&C> {
225 self.contexts.value()?.get()
226 }
227
228 pub fn user_agent(&self) -> Option<&str> {
233 let headers = self.request.value()?.headers.value()?;
234
235 for item in headers.iter() {
236 if let Some((ref o_k, ref v)) = item.value() {
237 if let Some(k) = o_k.as_str() {
238 if k.eq_ignore_ascii_case("user-agent") {
239 return v.as_str();
240 }
241 }
242 }
243 }
244
245 None
246 }
247}
248
249impl Getter for Replay {
250 fn get_value(&self, path: &str) -> Option<Val<'_>> {
251 Some(match path.strip_prefix("event.")? {
252 "release" => self.release.as_str()?.into(),
254 "dist" => self.dist.as_str()?.into(),
255 "environment" => self.environment.as_str()?.into(),
256 "platform" => self.platform.as_str().unwrap_or("other").into(),
257
258 "user.email" => or_none(&self.user.value()?.email)?.into(),
260 "user.id" => or_none(&self.user.value()?.id)?.into(),
261 "user.ip_address" => self.user.value()?.ip_address.as_str()?.into(),
262 "user.name" => self.user.value()?.name.as_str()?.into(),
263 "user.segment" => or_none(&self.user.value()?.segment)?.into(),
264 "user.geo.city" => self.user.value()?.geo.value()?.city.as_str()?.into(),
265 "user.geo.country_code" => self
266 .user
267 .value()?
268 .geo
269 .value()?
270 .country_code
271 .as_str()?
272 .into(),
273 "user.geo.region" => self.user.value()?.geo.value()?.region.as_str()?.into(),
274 "user.geo.subdivision" => self.user.value()?.geo.value()?.subdivision.as_str()?.into(),
275 "request.method" => self.request.value()?.method.as_str()?.into(),
276 "request.url" => self.request.value()?.url.as_str()?.into(),
277 "sdk.name" => self.sdk.value()?.name.as_str()?.into(),
278 "sdk.version" => self.sdk.value()?.version.as_str()?.into(),
279
280 "sentry_user" => self.user.value()?.sentry_user.as_str()?.into(),
282
283 "contexts.app.in_foreground" => {
285 self.context::<AppContext>()?.in_foreground.value()?.into()
286 }
287 "contexts.device.arch" => self.context::<DeviceContext>()?.arch.as_str()?.into(),
288 "contexts.device.battery_level" => self
289 .context::<DeviceContext>()?
290 .battery_level
291 .value()?
292 .into(),
293 "contexts.device.brand" => self.context::<DeviceContext>()?.brand.as_str()?.into(),
294 "contexts.device.charging" => self.context::<DeviceContext>()?.charging.value()?.into(),
295 "contexts.device.family" => self.context::<DeviceContext>()?.family.as_str()?.into(),
296 "contexts.device.model" => self.context::<DeviceContext>()?.model.as_str()?.into(),
297 "contexts.device.locale" => self.context::<DeviceContext>()?.locale.as_str()?.into(),
298 "contexts.device.online" => self.context::<DeviceContext>()?.online.value()?.into(),
299 "contexts.device.orientation" => self
300 .context::<DeviceContext>()?
301 .orientation
302 .as_str()?
303 .into(),
304 "contexts.device.name" => self.context::<DeviceContext>()?.name.as_str()?.into(),
305 "contexts.device.screen_density" => self
306 .context::<DeviceContext>()?
307 .screen_density
308 .value()?
309 .into(),
310 "contexts.device.screen_dpi" => {
311 self.context::<DeviceContext>()?.screen_dpi.value()?.into()
312 }
313 "contexts.device.screen_width_pixels" => self
314 .context::<DeviceContext>()?
315 .screen_width_pixels
316 .value()?
317 .into(),
318 "contexts.device.screen_height_pixels" => self
319 .context::<DeviceContext>()?
320 .screen_height_pixels
321 .value()?
322 .into(),
323 "contexts.device.simulator" => {
324 self.context::<DeviceContext>()?.simulator.value()?.into()
325 }
326 "contexts.os.build" => self.context::<OsContext>()?.build.as_str()?.into(),
327 "contexts.os.kernel_version" => {
328 self.context::<OsContext>()?.kernel_version.as_str()?.into()
329 }
330 "contexts.os.name" => self.context::<OsContext>()?.name.as_str()?.into(),
331 "contexts.os.version" => self.context::<OsContext>()?.version.as_str()?.into(),
332 "contexts.browser.name" => self.context::<BrowserContext>()?.name.as_str()?.into(),
333 "contexts.browser.version" => {
334 self.context::<BrowserContext>()?.version.as_str()?.into()
335 }
336 "contexts.profile.profile_id" => self
337 .context::<ProfileContext>()?
338 .profile_id
339 .value()?
340 .0
341 .into(),
342 "contexts.device.uuid" => self.context::<DeviceContext>()?.uuid.value()?.into(),
343 "contexts.trace.status" => self
344 .context::<TraceContext>()?
345 .status
346 .value()?
347 .as_str()
348 .into(),
349 "contexts.trace.op" => self.context::<TraceContext>()?.op.as_str()?.into(),
350 "contexts.response.status_code" => self
351 .context::<ResponseContext>()?
352 .status_code
353 .value()?
354 .into(),
355 "contexts.unreal.crash_type" => match self.contexts.value()?.get_key("unreal")? {
356 super::Context::Other(context) => context.get("crash_type")?.value()?.into(),
357 _ => return None,
358 },
359
360 path => {
362 if let Some(rest) = path.strip_prefix("tags.") {
363 self.tags.value()?.get(rest)?.into()
364 } else if let Some(rest) = path.strip_prefix("request.headers.") {
365 self.request
366 .value()?
367 .headers
368 .value()?
369 .get_header(rest)?
370 .into()
371 } else {
372 return None;
373 }
374 }
375 })
376 }
377}
378
379fn or_none(string: &Annotated<impl AsRef<str>>) -> Option<&str> {
380 match string.as_str() {
381 None | Some("") => None,
382 Some(other) => Some(other),
383 }
384}
385
386#[cfg(test)]
387mod tests {
388 use chrono::{TimeZone, Utc};
389
390 use crate::protocol::TagEntry;
391
392 use super::*;
393
394 #[test]
395 fn test_event_roundtrip() {
396 let json = r#"{
398 "event_id": "52df9022835246eeb317dbd739ccd059",
399 "replay_id": "52df9022835246eeb317dbd739ccd059",
400 "segment_id": 0,
401 "replay_type": "session",
402 "error_sample_rate": 0.5,
403 "session_sample_rate": 0.5,
404 "timestamp": 946684800.0,
405 "replay_start_timestamp": 946684800.0,
406 "urls": ["localhost:9000"],
407 "error_ids": ["52df9022835246eeb317dbd739ccd059"],
408 "trace_ids": ["52df9022835246eeb317dbd739ccd059"],
409 "platform": "myplatform",
410 "release": "myrelease",
411 "dist": "mydist",
412 "environment": "myenv",
413 "tags": [
414 [
415 "tag",
416 "value"
417 ]
418 ]
419}"#;
420
421 let replay = Annotated::new(Replay {
422 event_id: Annotated::new(EventId("52df9022835246eeb317dbd739ccd059".parse().unwrap())),
423 replay_id: Annotated::new(EventId("52df9022835246eeb317dbd739ccd059".parse().unwrap())),
424 replay_type: Annotated::new("session".to_string()),
425 segment_id: Annotated::new(0),
426 timestamp: Annotated::new(Utc.with_ymd_and_hms(2000, 1, 1, 0, 0, 0).unwrap().into()),
427 replay_start_timestamp: Annotated::new(
428 Utc.with_ymd_and_hms(2000, 1, 1, 0, 0, 0).unwrap().into(),
429 ),
430 urls: Annotated::new(vec![Annotated::new("localhost:9000".to_string())]),
431 error_ids: Annotated::new(vec![Annotated::new(
432 Uuid::parse_str("52df9022835246eeb317dbd739ccd059").unwrap(),
433 )]),
434 trace_ids: Annotated::new(vec![Annotated::new(
435 Uuid::parse_str("52df9022835246eeb317dbd739ccd059").unwrap(),
436 )]),
437 platform: Annotated::new("myplatform".to_string()),
438 release: Annotated::new("myrelease".to_string().into()),
439 dist: Annotated::new("mydist".to_string()),
440 environment: Annotated::new("myenv".to_string()),
441 tags: {
442 let items = vec![Annotated::new(TagEntry(
443 Annotated::new("tag".to_string()),
444 Annotated::new("value".to_string()),
445 ))];
446 Annotated::new(Tags(items.into()))
447 },
448 ..Default::default()
449 });
450
451 assert_eq!(replay, Annotated::from_json(json).unwrap());
452 }
453
454 #[test]
455 fn test_lenient_release() {
456 let input = r#"{"release":42}"#;
457 let output = r#"{"release":"42"}"#;
458 let event = Annotated::new(Replay {
459 release: Annotated::new("42".to_string().into()),
460 ..Default::default()
461 });
462
463 assert_eq!(event, Annotated::from_json(input).unwrap());
464 assert_eq!(output, event.to_json().unwrap());
465 }
466}