mirror of
https://git.asonix.dog/asonix/pict-rs
synced 2024-12-22 19:31:35 +00:00
59 lines
1.6 KiB
Rust
59 lines
1.6 KiB
Rust
use crate::{
|
|
error_code::ErrorCode,
|
|
process::{Process, ProcessError, ProcessRead},
|
|
};
|
|
use actix_web::web::Bytes;
|
|
|
|
#[derive(Debug, thiserror::Error)]
|
|
pub(crate) enum ExifError {
|
|
#[error("Error in process")]
|
|
Process(#[source] ProcessError),
|
|
|
|
#[error("Invalid media file provided")]
|
|
CommandFailed(ProcessError),
|
|
}
|
|
|
|
impl From<ProcessError> for ExifError {
|
|
fn from(value: ProcessError) -> Self {
|
|
match value {
|
|
e @ ProcessError::Status(_, _) => Self::CommandFailed(e),
|
|
otherwise => Self::Process(otherwise),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl ExifError {
|
|
pub(crate) const fn error_code(&self) -> ErrorCode {
|
|
match self {
|
|
Self::Process(e) => e.error_code(),
|
|
Self::CommandFailed(_) => ErrorCode::COMMAND_FAILURE,
|
|
}
|
|
}
|
|
pub(crate) fn is_client_error(&self) -> bool {
|
|
// if exiftool bails we probably have bad input
|
|
match self {
|
|
Self::CommandFailed(_) => true,
|
|
Self::Process(e) => e.is_client_error(),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip(input))]
|
|
pub(crate) async fn needs_reorienting(timeout: u64, input: Bytes) -> Result<bool, ExifError> {
|
|
let buf = Process::run("exiftool", &["-n", "-Orientation", "-"], &[], timeout)?
|
|
.bytes_read(input)
|
|
.into_string()
|
|
.await?;
|
|
|
|
Ok(!buf.is_empty())
|
|
}
|
|
|
|
#[tracing::instrument(level = "trace", skip(input))]
|
|
pub(crate) fn clear_metadata_bytes_read(
|
|
timeout: u64,
|
|
input: Bytes,
|
|
) -> Result<ProcessRead, ExifError> {
|
|
let process = Process::run("exiftool", &["-all=", "-", "-out", "-"], &[], timeout)?;
|
|
|
|
Ok(process.bytes_read(input))
|
|
}
|