mediarepo/mediarepo-daemon/mediarepo-model/src/content_descriptor.rs

use crate::file::File;
use mediarepo_core::error::RepoResult;
use mediarepo_database::entities::content_descriptor;
use mediarepo_database::entities::file;
use sea_orm::prelude::*;
use sea_orm::{DatabaseConnection, Set};
use std::fmt::Debug;

pub struct Hash {
    db: DatabaseConnection,
    model: content_descriptor::Model,
}

impl Hash {
    #[tracing::instrument(level = "trace")]
    pub(crate) fn new(db: DatabaseConnection, model: content_descriptor::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 = content_descriptor::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<D: AsRef<[u8]> + Debug>(
        db: DatabaseConnection,
        descriptor: D,
    ) -> RepoResult<Option<Self>> {
        let cid = content_descriptor::Entity::find()
            .filter(content_descriptor::Column::Descriptor.eq(descriptor.as_ref()))
            .one(&db)
            .await?
            .map(|model| Self::new(db, model));

        Ok(cid)
    }

    /// Adds a new hash to the database
    #[tracing::instrument(level = "debug", skip(db))]
    pub async fn add(db: DatabaseConnection, descriptor: Vec<u8>) -> RepoResult<Self> {
        let active_model = content_descriptor::ActiveModel {
            descriptor: Set(descriptor),
            ..Default::default()
        };
        let model = active_model.insert(&db).await?;

        Ok(Self::new(db, model))
    }

    pub fn id(&self) -> i64 {
        self.model.id
    }

    pub fn descriptor(&self) -> &[u8] {
        &self.model.descriptor[..]
    }

    /// 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)
    }
}