use sea_orm_migration::{prelude::*, sea_query::extension::postgres::Type};
#[derive(DeriveMigrationName)]
pub struct Migration;
#[async_trait::async_trait]
impl MigrationTrait for Migration {
async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
manager
.create_type(
Type::create()
.as_enum(Role::Type)
.values(vec![Role::User, Role::Admin])
.to_owned(),
)
.await?;
manager
.create_table(
Table::create()
.table(Users::Table)
.col(
ColumnDef::new(Users::Id)
.string()
.not_null()
.unique_key()
.primary_key(),
)
.col(
ColumnDef::new(Users::Email)
.string_len(320)
.unique_key()
.not_null(),
)
.col(
ColumnDef::new(Users::Username)
.string_len(32)
.unique_key()
.not_null(),
)
.col(ColumnDef::new(Users::Password).string_len(128).not_null())
.col(
ColumnDef::new(Users::Verified)
.boolean()
.default(false)
.not_null(),
)
.col(
ColumnDef::new(Users::Role)
.custom(Role::Type)
.default("user")
.not_null(),
)
.to_owned(),
)
.await?;
manager
.create_table(
Table::create()
.table(Applications::Table)
.col(
ColumnDef::new(Applications::Id)
.string()
.not_null()
.unique_key()
.primary_key(),
)
.col(ColumnDef::new(Applications::UserId).string().not_null())
.col(ColumnDef::new(Applications::Name).string_len(16).not_null())
.col(
ColumnDef::new(Applications::LastAccessed)
.timestamp_with_time_zone()
.not_null(),
)
.foreign_key(
ForeignKey::create()
.from(Applications::Table, Applications::UserId)
.to(Users::Table, Users::Id),
)
.to_owned(),
)
.await?;
manager
.create_index(
Index::create()
.table(Applications::Table)
.col(Applications::UserId)
.col(Applications::Name)
.unique()
.to_owned(),
)
.await
}
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
manager
.drop_table(Table::drop().table(Users::Table).to_owned())
.await?;
manager
.drop_type(Type::drop().name(Role::Type).to_owned())
.await
}
}
#[derive(Iden)]
enum Role {
#[iden = "role"]
Type,
User,
Admin,
}
#[derive(Iden)]
enum Users {
Table,
Id,
Email,
Username,
Password,
Verified,
Role,
}
#[derive(Iden)]
enum Applications {
Table,
Id,
UserId,
Name,
LastAccessed,
}