use chrono::TimeDelta;
use sqlx::sqlite::SqlitePool;

use crate::{
    clock::DateTime,
    events::{broadcaster::Broadcaster, repo::message::Provider as _, types::ChannelEvent},
    repo::channel::{Channel, Provider as _},
};

pub struct Channels<'a> {
    db: &'a SqlitePool,
    events: &'a Broadcaster,
}

impl<'a> Channels<'a> {
    pub const fn new(db: &'a SqlitePool, events: &'a Broadcaster) -> Self {
        Self { db, events }
    }

    pub async fn create(&self, name: &str, created_at: &DateTime) -> Result<Channel, CreateError> {
        let mut tx = self.db.begin().await?;
        let channel = tx
            .channels()
            .create(name, created_at)
            .await
            .map_err(|err| CreateError::from_duplicate_name(err, name))?;
        tx.commit().await?;

        self.events
            .broadcast(&ChannelEvent::created(channel.clone()));

        Ok(channel)
    }

    pub async fn all(&self) -> Result<Vec<Channel>, InternalError> {
        let mut tx = self.db.begin().await?;
        let channels = tx.channels().all().await?;
        tx.commit().await?;

        Ok(channels)
    }

    pub async fn expire(&self, relative_to: &DateTime) -> Result<(), sqlx::Error> {
        // Somewhat arbitrarily, expire after 90 days.
        let expire_at = relative_to.to_owned() - TimeDelta::days(90);

        let mut tx = self.db.begin().await?;
        let expired = tx.channels().expired(&expire_at).await?;

        let mut events = Vec::with_capacity(expired.len());
        for channel in expired {
            let sequence = tx.message_events().assign_sequence(&channel).await?;
            let event = tx
                .channels()
                .delete_expired(&channel, sequence, relative_to)
                .await?;
            events.push(event);
        }

        tx.commit().await?;

        for event in events {
            self.events.broadcast(&event);
        }

        Ok(())
    }
}

#[derive(Debug, thiserror::Error)]
pub enum CreateError {
    #[error("channel named {0} already exists")]
    DuplicateName(String),
    #[error(transparent)]
    DatabaseError(#[from] sqlx::Error),
}

impl CreateError {
    fn from_duplicate_name(error: sqlx::Error, name: &str) -> Self {
        if let Some(error) = error.as_database_error() {
            if error.is_unique_violation() {
                return Self::DuplicateName(name.into());
            }
        }

        Self::from(error)
    }
}

#[derive(Debug, thiserror::Error)]
pub enum InternalError {
    #[error(transparent)]
    DatabaseError(#[from] sqlx::Error),
}