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: ArtifactUri,
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().parse().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!(serde_json::from_value::<MediaReference>(serde_json::json!({
80 "uri": "https://example.com/a"
81 }))
82 .is_err());
83 }
84}
85
86#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
88#[serde(try_from = "String", into = "String")]
89pub struct ArtifactUri(String);
90
91impl ArtifactUri {
92 pub fn as_str(&self) -> &str {
93 &self.0
94 }
95 pub fn reference(&self) -> ArtifactReference {
96 ArtifactReference::parse(&self.0).expect("ArtifactUri was validated at construction")
97 }
98}
99impl std::str::FromStr for ArtifactUri {
100 type Err = ArtifactReferenceError;
101 fn from_str(value: &str) -> Result<Self, Self::Err> {
102 let reference = ArtifactReference::parse(value)?;
103 Ok(Self(reference.uri()?))
104 }
105}
106impl TryFrom<String> for ArtifactUri {
107 type Error = ArtifactReferenceError;
108 fn try_from(value: String) -> Result<Self, Self::Error> {
109 value.parse()
110 }
111}
112impl From<ArtifactUri> for String {
113 fn from(value: ArtifactUri) -> String {
114 value.0
115 }
116}
117impl std::ops::Deref for ArtifactUri {
118 type Target = str;
119 fn deref(&self) -> &str {
120 self.as_str()
121 }
122}
123impl AsRef<str> for ArtifactUri {
124 fn as_ref(&self) -> &str {
125 self.as_str()
126 }
127}
128impl std::fmt::Display for ArtifactUri {
129 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
130 self.0.fmt(f)
131 }
132}