use sea_orm::Schema;
use sea_orm_migration::prelude::*;

use crate::entity::prelude::*;

#[derive(DeriveMigrationName)]
pub struct Migration;

#[async_trait::async_trait]
impl MigrationTrait for Migration {
    async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
        let db = manager.get_connection();
        let schema = Schema::new(manager.get_database_backend());

        let stmt = schema.create_table_from_entity(BoardData);
        let stmt = schema.create_table_from_entity(BoardProject);
        let stmt = schema.create_table_from_entity(BoardDataMedia);
        let stmt = schema.create_table_from_entity(CommentData);
        let stmt = schema.create_table_from_entity(Contacts);
        let stmt = schema.create_table_from_entity(BoardMediaJunction);
        let stmt = schema.create_table_from_entity(BoardTagJunction);
        let stmt = schema.create_table_from_entity(Messages);
        let stmt = schema.create_table_from_entity(Reading);
        let stmt = schema.create_table_from_entity(RoomMember);
        let stmt = schema.create_table_from_entity(Users);
        let stmt = schema.create_table_from_entity(UsersUranus);

        db.execute(db.get_database_backend().build(&stmt)).await?;
        Ok(())
    }

    async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
        let db = manager.get_connection();
        let backend = manager.get_database_backend();

        let stmt = Table::drop().table(BoardData).if_exists().to_owned();

        db.execute(backend.build(&stmt)).await?;

        Ok(())
    }
}