1use crate::{ArtifactKind, ArtifactReference, ArtifactReferenceError};
3use serde::{Deserialize, Serialize};
4
5#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
6#[serde(deny_unknown_fields)]
7pub struct MediaReference {
8 pub uri: String,
9}
10
11impl MediaReference {
12 pub fn reference(
13 &self,
14 kind: ArtifactKind,
15 ) -> Result<ArtifactReference, ArtifactReferenceError> {
16 let reference = ArtifactReference::parse(&self.uri)?;
17 if reference.metadata().kind() != kind {
18 return Err(super::invalid(
19 "media artifact kind does not match the field type",
20 ));
21 }
22 Ok(reference)
23 }
24
25 pub fn validate(
26 &self,
27 tenant: &str,
28 scope: &str,
29 kind: ArtifactKind,
30 ) -> Result<ArtifactReference, ArtifactReferenceError> {
31 let reference = self.reference(kind)?;
32 reference.ensure_scope(tenant, scope)?;
33 Ok(reference)
34 }
35}
36
37#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
39#[serde(rename_all = "camelCase", deny_unknown_fields)]
40pub struct ImportArtifactRequest {
41 pub kind: ArtifactKind,
42 pub source_url: String,
43 #[serde(default, skip_serializing_if = "Option::is_none")]
44 pub mime_type: Option<String>,
45}
46
47#[cfg(test)]
48mod tests {
49 use super::*;
50 #[test]
51 fn media_rejects_legacy_fields_and_wrong_scope_or_kind() {
52 for value in [
53 serde_json::json!({"url":"https://example.com/a"}),
54 serde_json::json!({"uri":"x","base64":"eA=="}),
55 serde_json::json!({"uri":"x","mimeType":"image/png"}),
56 serde_json::json!({"uri":"x","createdAt":1}),
57 ] {
58 assert!(serde_json::from_value::<MediaReference>(value).is_err());
59 }
60 let reference = ArtifactReference::new(
61 "tenant",
62 "scope",
63 "a".repeat(64),
64 crate::ArtifactMetadata::audio("audio/mpeg", 10, None).unwrap(),
65 )
66 .unwrap();
67 let media = MediaReference {
68 uri: reference.uri().unwrap(),
69 };
70 assert!(media
71 .validate("tenant", "scope", ArtifactKind::Audio)
72 .is_ok());
73 assert!(media
74 .validate("tenant", "other", ArtifactKind::Audio)
75 .is_err());
76 assert!(media
77 .validate("tenant", "scope", ArtifactKind::Image)
78 .is_err());
79 assert!(MediaReference {
80 uri: "https://example.com/a".into()
81 }
82 .reference(ArtifactKind::Audio)
83 .is_err());
84 }
85}