mirror of
https://git.asonix.dog/asonix/pict-rs
synced 2024-12-22 11:21:24 +00:00
Enforce format on new details
This commit is contained in:
parent
f3c6239979
commit
521cdd5b9d
6 changed files with 67 additions and 59 deletions
|
@ -22,8 +22,7 @@ pub(crate) struct Details {
|
|||
frames: Option<u32>,
|
||||
content_type: Serde<mime::Mime>,
|
||||
created_at: MaybeHumanDate,
|
||||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
format: Option<InternalFormat>,
|
||||
format: InternalFormat,
|
||||
}
|
||||
|
||||
impl Details {
|
||||
|
@ -57,12 +56,8 @@ impl Details {
|
|||
Ok(Details::from_parts(format, width, height, frames))
|
||||
}
|
||||
|
||||
pub(crate) fn internal_format(&self) -> Option<InternalFormat> {
|
||||
if let Some(format) = self.format {
|
||||
return Some(format);
|
||||
}
|
||||
|
||||
InternalFormat::maybe_from_media_type(&self.content_type, self.frames.is_some())
|
||||
pub(crate) fn internal_format(&self) -> InternalFormat {
|
||||
self.format
|
||||
}
|
||||
|
||||
pub(crate) fn media_type(&self) -> mime::Mime {
|
||||
|
@ -74,15 +69,27 @@ impl Details {
|
|||
}
|
||||
|
||||
pub(crate) fn video_format(&self) -> Option<InternalVideoFormat> {
|
||||
if *self.content_type == crate::formats::mimes::video_mp4() {
|
||||
return Some(InternalVideoFormat::Mp4);
|
||||
match self.format {
|
||||
InternalFormat::Video(format) => Some(format),
|
||||
_ => None,
|
||||
}
|
||||
}
|
||||
|
||||
if *self.content_type == crate::formats::mimes::video_webm() {
|
||||
return Some(InternalVideoFormat::Webm);
|
||||
pub(crate) fn from_parts_full(
|
||||
format: InternalFormat,
|
||||
width: u16,
|
||||
height: u16,
|
||||
frames: Option<u32>,
|
||||
created_at: MaybeHumanDate,
|
||||
) -> Self {
|
||||
Self {
|
||||
width,
|
||||
height,
|
||||
frames,
|
||||
content_type: Serde::new(format.media_type()),
|
||||
created_at,
|
||||
format,
|
||||
}
|
||||
|
||||
None
|
||||
}
|
||||
|
||||
pub(crate) fn from_parts(
|
||||
|
@ -97,7 +104,7 @@ impl Details {
|
|||
frames,
|
||||
content_type: Serde::new(format.media_type()),
|
||||
created_at: MaybeHumanDate::HumanDate(OffsetDateTime::now_utc()),
|
||||
format: Some(format),
|
||||
format,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
@ -130,8 +130,8 @@ async fn process<S: Store + 'static>(
|
|||
|
||||
let input_format = input_details
|
||||
.internal_format()
|
||||
.and_then(|format| format.processable_format())
|
||||
.expect("Valid details should always have internal format");
|
||||
.processable_format()
|
||||
.expect("Already verified format is processable");
|
||||
|
||||
let Some(format) = input_format.process_to(output_format) else {
|
||||
return Err(UploadError::InvalidProcessExtension.into());
|
||||
|
|
24
src/lib.rs
24
src/lib.rs
|
@ -102,13 +102,7 @@ async fn ensure_details<S: Store + 'static>(
|
|||
return Err(UploadError::MissingAlias.into());
|
||||
};
|
||||
|
||||
let details = repo.details(&identifier).await?.and_then(|details| {
|
||||
if details.internal_format().is_some() {
|
||||
Some(details)
|
||||
} else {
|
||||
None
|
||||
}
|
||||
});
|
||||
let details = repo.details(&identifier).await?;
|
||||
|
||||
if let Some(details) = details {
|
||||
tracing::debug!("details exist");
|
||||
|
@ -784,13 +778,7 @@ async fn process<S: Store + 'static>(
|
|||
.transpose()?;
|
||||
|
||||
if let Some(identifier) = identifier_opt {
|
||||
let details = repo.details(&identifier).await?.and_then(|details| {
|
||||
if details.internal_format().is_some() {
|
||||
Some(details)
|
||||
} else {
|
||||
None
|
||||
}
|
||||
});
|
||||
let details = repo.details(&identifier).await?;
|
||||
|
||||
let details = if let Some(details) = details {
|
||||
tracing::debug!("details exist");
|
||||
|
@ -916,13 +904,7 @@ async fn process_head<S: Store + 'static>(
|
|||
.transpose()?;
|
||||
|
||||
if let Some(identifier) = identifier_opt {
|
||||
let details = repo.details(&identifier).await?.and_then(|details| {
|
||||
if details.internal_format().is_some() {
|
||||
Some(details)
|
||||
} else {
|
||||
None
|
||||
}
|
||||
});
|
||||
let details = repo.details(&identifier).await?;
|
||||
|
||||
let details = if let Some(details) = details {
|
||||
tracing::debug!("details exist");
|
||||
|
|
|
@ -398,14 +398,7 @@ where
|
|||
.details(identifier)
|
||||
.await
|
||||
.map_err(Error::from)
|
||||
.map_err(MigrateError::Details)?
|
||||
.and_then(|details| {
|
||||
if details.internal_format().is_some() {
|
||||
Some(details)
|
||||
} else {
|
||||
None
|
||||
}
|
||||
});
|
||||
.map_err(MigrateError::Details)?;
|
||||
|
||||
let details = if let Some(details) = details_opt {
|
||||
details
|
||||
|
|
|
@ -151,11 +151,7 @@ async fn do_migrate_hash_04<S: Store>(
|
|||
|
||||
let hash = old_hash[..].try_into().expect("Invalid hash size");
|
||||
|
||||
let hash = Hash::new(
|
||||
hash,
|
||||
size,
|
||||
hash_details.internal_format().expect("format exists"),
|
||||
);
|
||||
let hash = Hash::new(hash, size, hash_details.internal_format());
|
||||
|
||||
let _ = HashRepo::create(new_repo.as_ref(), hash.clone(), &identifier).await?;
|
||||
|
||||
|
@ -212,13 +208,7 @@ async fn fetch_or_generate_details<S: Store>(
|
|||
config: &Configuration,
|
||||
identifier: &S::Identifier,
|
||||
) -> Result<Details, Error> {
|
||||
let details_opt = old_repo.details(identifier).await?.and_then(|details| {
|
||||
if details.internal_format().is_some() {
|
||||
Some(details)
|
||||
} else {
|
||||
None
|
||||
}
|
||||
});
|
||||
let details_opt = old_repo.details(identifier).await?;
|
||||
|
||||
if let Some(details) = details_opt {
|
||||
Ok(details)
|
||||
|
|
|
@ -57,6 +57,41 @@ pub(crate) struct SledRepo {
|
|||
_db: Db,
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]
|
||||
pub(crate) struct OldDetails {
|
||||
width: u16,
|
||||
height: u16,
|
||||
frames: Option<u32>,
|
||||
content_type: crate::serde_str::Serde<mime::Mime>,
|
||||
created_at: crate::details::MaybeHumanDate,
|
||||
#[serde(skip_serializing_if = "Option::is_none")]
|
||||
format: Option<crate::formats::InternalFormat>,
|
||||
}
|
||||
|
||||
impl OldDetails {
|
||||
fn into_details(self) -> Option<Details> {
|
||||
let OldDetails {
|
||||
width,
|
||||
height,
|
||||
frames,
|
||||
content_type,
|
||||
created_at,
|
||||
format,
|
||||
} = self;
|
||||
|
||||
let format = format.or_else(|| {
|
||||
crate::formats::InternalFormat::maybe_from_media_type(
|
||||
&content_type,
|
||||
self.frames.is_some(),
|
||||
)
|
||||
})?;
|
||||
|
||||
Some(Details::from_parts_full(
|
||||
format, width, height, frames, created_at,
|
||||
))
|
||||
}
|
||||
}
|
||||
|
||||
impl SledRepo {
|
||||
#[tracing::instrument]
|
||||
pub(crate) fn build(path: PathBuf, cache_capacity: u64) -> color_eyre::Result<Option<Self>> {
|
||||
|
@ -135,11 +170,12 @@ impl IdentifierRepo for SledRepo {
|
|||
|
||||
let opt = b!(self.identifier_details, identifier_details.get(key));
|
||||
|
||||
opt.map(|ivec| serde_json::from_slice(&ivec))
|
||||
opt.map(|ivec| serde_json::from_slice::<OldDetails>(&ivec))
|
||||
.transpose()
|
||||
.map_err(SledError::from)
|
||||
.map_err(RepoError::from)
|
||||
.map_err(StoreError::from)
|
||||
.map(|opt| opt.and_then(OldDetails::into_details))
|
||||
}
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in a new issue