You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
2.1 KiB
Rust
79 lines
2.1 KiB
Rust
use crate::file::File;
|
|
use mediarepo_core::error::RepoResult;
|
|
use mediarepo_database::entities::file;
|
|
use mediarepo_database::entities::hash;
|
|
use sea_orm::prelude::*;
|
|
use sea_orm::{DatabaseConnection, Set};
|
|
use std::fmt::Debug;
|
|
|
|
pub struct Hash {
|
|
db: DatabaseConnection,
|
|
model: hash::Model,
|
|
}
|
|
|
|
impl Hash {
|
|
#[tracing::instrument(level = "trace")]
|
|
pub(crate) fn new(db: DatabaseConnection, model: hash::Model) -> Self {
|
|
Self { db, model }
|
|
}
|
|
|
|
/// Searches for the hash by id
|
|
#[tracing::instrument(level = "debug", skip(db))]
|
|
pub async fn by_id(db: DatabaseConnection, id: i64) -> RepoResult<Option<Self>> {
|
|
let hash = hash::Entity::find_by_id(id)
|
|
.one(&db)
|
|
.await?
|
|
.map(|model| Self::new(db, model));
|
|
|
|
Ok(hash)
|
|
}
|
|
|
|
/// Returns the hash by value
|
|
#[tracing::instrument(level = "debug", skip(db))]
|
|
pub async fn by_value<S: AsRef<str> + Debug>(
|
|
db: DatabaseConnection,
|
|
value: S,
|
|
) -> RepoResult<Option<Self>> {
|
|
let hash = hash::Entity::find()
|
|
.filter(hash::Column::Value.eq(value.as_ref()))
|
|
.one(&db)
|
|
.await?
|
|
.map(|model| Self::new(db, model));
|
|
|
|
Ok(hash)
|
|
}
|
|
|
|
/// Adds a new hash to the database
|
|
#[tracing::instrument(level = "debug", skip(db))]
|
|
pub async fn add(db: DatabaseConnection, value: String) -> RepoResult<Self> {
|
|
let active_model = hash::ActiveModel {
|
|
value: Set(value),
|
|
..Default::default()
|
|
};
|
|
let model = active_model.insert(&db).await?;
|
|
|
|
Ok(Self::new(db, model))
|
|
}
|
|
|
|
pub fn id(&self) -> i64 {
|
|
self.model.id
|
|
}
|
|
|
|
pub fn value(&self) -> &String {
|
|
&self.model.value
|
|
}
|
|
|
|
/// Returns the file associated with the hash
|
|
#[tracing::instrument(level = "debug", skip(self))]
|
|
pub async fn file(&self) -> RepoResult<Option<File>> {
|
|
let file = self
|
|
.model
|
|
.find_related(file::Entity)
|
|
.one(&self.db)
|
|
.await?
|
|
.map(|file_model| File::new(self.db.clone(), file_model, self.model.clone()));
|
|
|
|
Ok(file)
|
|
}
|
|
}
|