use crate::models::Emoji;
use crate::{AbstractEmoji, Result};

use super::super::MongoDb;

static COL: &str = "emojis";

#[async_trait]
impl AbstractEmoji for MongoDb {
    /// Fetch an emoji by its id
    async fn fetch_emoji(&self, id: &str) -> Result<Emoji> {
        self.find_one_by_id(COL, id).await
    }

    /// Fetch emoji by their ids
    async fn fetch_emoji_by_parent_id(&self, parent_id: &str) -> Result<Vec<Emoji>> {
        self.find(
            COL,
            doc! {
                "parent.id": parent_id
            },
        )
        .await
    }

    /// Fetch emoji by their parent ids
    async fn fetch_emoji_by_parent_ids(&self, parent_ids: &[String]) -> Result<Vec<Emoji>> {
        self.find(
            COL,
            doc! {
                "parent.id": {
                    "$in": parent_ids
                }
            },
        )
        .await
    }

    /// Insert emoji into database.
    async fn insert_emoji(&self, emoji: &Emoji) -> Result<()> {
        self.insert_one(COL, emoji).await.map(|_| ())
    }

    /// Delete an emoji by its id
    async fn delete_emoji(&self, emoji: &Emoji) -> Result<()> {
        self.delete_one_by_id(COL, &emoji.id).await.map(|_| ())
    }
}