2b-rs/bot-database/migration/src/m20220101_000001_create_table.rs

use sea_query::Table;
use sea_schema::migration::prelude::*;

pub struct Migration;

#[derive(Iden)]
pub enum EphemeralMessages {
    Table,
    ChannelId,
    MessageId,
    Timeout,
}

#[derive(Iden)]
pub enum GuildPlaylists {
    Table,
    GuildId,
    Name,
    Url,
}

#[derive(Iden)]
pub enum GuildSettings {
    Table,
    GuildId,
    Key,
    Value,
}

#[derive(Iden)]
pub enum Media {
    Table,
    Id,
    Category,
    Name,
    Url,
}

#[derive(Iden)]
pub enum Statistics {
    Table,
    Id,
    Version,
    Command,
    ExecutedAt,
    Success,
    ErrorMsg,
}

#[derive(Iden)]
pub enum YoutubeSongs {
    Table,
    Id,
    SpotifyId,
    Artist,
    Title,
    Album,
    Url,
    Score,
}

impl MigrationName for Migration {
    fn name(&self) -> &str {
        "m20220101_000001_create_table"
    }
}

#[async_trait::async_trait]
impl MigrationTrait for Migration {
    async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
        manager.create_table(ephemeral_messages()).await?;
        manager.create_table(guild_playlists()).await?;
        manager.create_table(guild_settings()).await?;
        manager.create_table(media()).await?;
        manager.create_table(statistics()).await?;
        manager.create_table(youtube_songs()).await?;

        Ok(())
    }

    async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
        manager
            .drop_table(Table::drop().table(EphemeralMessages::Table).to_owned())
            .await?;
        manager
            .drop_table(Table::drop().table(GuildPlaylists::Table).to_owned())
            .await?;
        manager
            .drop_table(Table::drop().table(GuildSettings::Table).to_owned())
            .await?;
        manager
            .drop_table(Table::drop().table(Media::Table).to_owned())
            .await?;
        manager
            .drop_table(Table::drop().table(Statistics::Table).to_owned())
            .await?;
        manager
            .drop_table(Table::drop().table(YoutubeSongs::Table).to_owned())
            .await?;

        Ok(())
    }
}

fn ephemeral_messages() -> TableCreateStatement {
    Table::create()
        .table(EphemeralMessages::Table)
        .if_not_exists()
        .col(
            ColumnDef::new(EphemeralMessages::ChannelId)
                .big_integer()
                .not_null(),
        )
        .col(
            ColumnDef::new(EphemeralMessages::MessageId)
                .big_integer()
                .not_null(),
        )
        .col(
            ColumnDef::new(EphemeralMessages::Timeout)
                .timestamp()
                .not_null(),
        )
        .primary_key(
            Index::create()
                .col(EphemeralMessages::ChannelId)
                .col(EphemeralMessages::MessageId),
        )
        .to_owned()
}

fn guild_playlists() -> TableCreateStatement {
    Table::create()
        .table(GuildPlaylists::Table)
        .if_not_exists()
        .col(
            ColumnDef::new(GuildPlaylists::GuildId)
                .big_integer()
                .not_null(),
        )
        .col(
            ColumnDef::new(GuildPlaylists::Name)
                .string_len(255)
                .not_null(),
        )
        .col(
            ColumnDef::new(GuildPlaylists::Url)
                .string_len(1204)
                .not_null(),
        )
        .primary_key(
            Index::create()
                .col(GuildPlaylists::GuildId)
                .col(GuildPlaylists::Name),
        )
        .to_owned()
}

fn guild_settings() -> TableCreateStatement {
    Table::create()
        .table(GuildSettings::Table)
        .if_not_exists()
        .col(
            ColumnDef::new(GuildSettings::GuildId)
                .big_integer()
                .not_null(),
        )
        .col(
            ColumnDef::new(GuildSettings::Key)
                .string_len(255)
                .not_null(),
        )
        .col(
            ColumnDef::new(GuildSettings::Value)
                .string_len(1024)
                .not_null(),
        )
        .primary_key(
            Index::create()
                .col(GuildSettings::GuildId)
                .col(GuildSettings::Key),
        )
        .to_owned()
}

fn media() -> TableCreateStatement {
    Table::create()
        .table(Media::Table)
        .if_not_exists()
        .col(
            ColumnDef::new(Media::Id)
                .big_integer()
                .auto_increment()
                .not_null()
                .primary_key(),
        )
        .col(ColumnDef::new(Media::Category).string_len(128))
        .col(ColumnDef::new(Media::Name).string_len(128))
        .col(ColumnDef::new(Media::Url).string_len(128))
        .index(
            Index::create()
                .unique()
                .col(Media::Category)
                .col(Media::Name),
        )
        .to_owned()
}

fn statistics() -> TableCreateStatement {
    Table::create()
        .table(Statistics::Table)
        .if_not_exists()
        .col(
            ColumnDef::new(Statistics::Id)
                .big_integer()
                .auto_increment()
                .primary_key(),
        )
        .col(
            ColumnDef::new(Statistics::Version)
                .string_len(32)
                .not_null(),
        )
        .col(
            ColumnDef::new(Statistics::Command)
                .string_len(255)
                .not_null(),
        )
        .col(
            ColumnDef::new(Statistics::ExecutedAt)
                .timestamp()
                .not_null(),
        )
        .col(
            ColumnDef::new(Statistics::Success)
                .boolean()
                .not_null()
                .default(true),
        )
        .col(ColumnDef::new(Statistics::ErrorMsg).string())
        .to_owned()
}

fn youtube_songs() -> TableCreateStatement {
    Table::create()
        .table(YoutubeSongs::Table)
        .if_not_exists()
        .col(
            ColumnDef::new(YoutubeSongs::Id)
                .big_integer()
                .primary_key()
                .auto_increment(),
        )
        .col(
            ColumnDef::new(YoutubeSongs::SpotifyId)
                .string_len(255)
                .not_null(),
        )
        .col(
            ColumnDef::new(YoutubeSongs::Artist)
                .string_len(128)
                .not_null(),
        )
        .col(
            ColumnDef::new(YoutubeSongs::Title)
                .string_len(255)
                .not_null(),
        )
        .col(
            ColumnDef::new(YoutubeSongs::Album)
                .string_len(255)
                .not_null(),
        )
        .col(ColumnDef::new(YoutubeSongs::Url).string_len(128).not_null())
        .col(
            ColumnDef::new(YoutubeSongs::Score)
                .integer()
                .default(0)
                .not_null(),
        )
        .index(
            Index::create()
                .unique()
                .col(YoutubeSongs::SpotifyId)
                .col(YoutubeSongs::Url),
        )
        .to_owned()
}