use crate::{
    db::{
        do_migrate,
        schema::*,
        POOL,
    },
    guild_id,
};
use anyhow::anyhow;
use diesel::{
    associations::HasTable,
    backend::Backend,
    pg::Pg,
    query_builder::AsQuery,
    ExpressionMethods,
};
use diesel_async::{
    scoped_futures::ScopedFutureExt,
    AsyncConnection,
    AsyncPgConnection,
    RunQueryDsl,
};

#[inline]
pub async fn connection_no_migrate()
-> anyhow::Result<diesel_async::pooled_connection::deadpool::Object<AsyncPgConnection>> {
    let conn = super::POOL.get().await?;
    Ok(conn)
}

pub async fn set_default_guild<Conn>(mut conn: Conn, guild_id: u64) -> anyhow::Result<()>
where
    Conn: AsyncConnection<Backend = Pg> + 'static,
{
    conn.transaction::<_, anyhow::Error, _>(|tx| {
        (async move {
            diesel::update(memes::table)
                .filter(memes::guild.is_null())
                .set(memes::guild.eq(guild_id as i64))
                .execute(tx)
                .await?;

            diesel::update(invocation_records::table)
                .filter(invocation_records::guild.is_null())
                .set(invocation_records::guild.eq(guild_id as i64))
                .execute(tx)
                .await?;

            diesel::update(tombstones::table)
                .filter(tombstones::guild.is_null())
                .set(tombstones::guild.eq(guild_id as i64))
                .execute(tx)
                .await?;

            Ok(())
        })
        .scope_boxed()
    })
    .await?;

    Ok(())
}