Gabriel Kaszewski
b83b7acf1c
- Updated user and thought models to utilize UUIDs for primary keys. - Modified persistence functions to accommodate UUIDs for user and thought IDs. - Implemented tag functionality with new Tag and ThoughtTag models. - Added migration scripts to create new tables for tags and thought-tag relationships. - Enhanced thought creation to parse hashtags and link them to thoughts. - Updated tests to reflect changes in user and thought ID types.
108 lines
3.7 KiB
Rust
108 lines
3.7 KiB
Rust
use super::m20240101_000001_init::User;
|
|
use sea_orm_migration::{prelude::*, schema::*};
|
|
|
|
#[derive(DeriveMigrationName)]
|
|
pub struct Migration;
|
|
|
|
#[async_trait::async_trait]
|
|
impl MigrationTrait for Migration {
|
|
async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
manager
|
|
.alter_table(
|
|
Table::alter()
|
|
.table(User::Table)
|
|
.add_column(string_null(UserExtension::Email).unique_key())
|
|
.add_column(string_null(UserExtension::DisplayName))
|
|
.add_column(string_null(UserExtension::Bio))
|
|
.add_column(text_null(UserExtension::AvatarUrl))
|
|
.add_column(text_null(UserExtension::HeaderUrl))
|
|
.add_column(text_null(UserExtension::CustomCss))
|
|
.add_column(
|
|
timestamp_with_time_zone(UserExtension::CreatedAt)
|
|
.not_null()
|
|
.default(Expr::current_timestamp()),
|
|
)
|
|
.add_column(
|
|
timestamp_with_time_zone(UserExtension::UpdatedAt)
|
|
.not_null()
|
|
.default(Expr::current_timestamp()),
|
|
)
|
|
.to_owned(),
|
|
)
|
|
.await?;
|
|
|
|
manager
|
|
.create_table(
|
|
Table::create()
|
|
.table(TopFriends::Table)
|
|
.if_not_exists()
|
|
.col(uuid(TopFriends::UserId).not_null())
|
|
.col(uuid(TopFriends::FriendId).not_null())
|
|
.col(small_integer(TopFriends::Position).not_null())
|
|
.primary_key(
|
|
Index::create()
|
|
.col(TopFriends::UserId)
|
|
.col(TopFriends::FriendId),
|
|
)
|
|
.foreign_key(
|
|
ForeignKey::create()
|
|
.name("fk_top_friends_user_id")
|
|
.from(TopFriends::Table, TopFriends::UserId)
|
|
.to(User::Table, User::Id)
|
|
.on_delete(ForeignKeyAction::Cascade),
|
|
)
|
|
.foreign_key(
|
|
ForeignKey::create()
|
|
.name("fk_top_friends_friend_id")
|
|
.from(TopFriends::Table, TopFriends::FriendId)
|
|
.to(User::Table, User::Id)
|
|
.on_delete(ForeignKeyAction::Cascade),
|
|
)
|
|
.to_owned(),
|
|
)
|
|
.await
|
|
}
|
|
|
|
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
manager
|
|
.drop_table(Table::drop().table(TopFriends::Table).to_owned())
|
|
.await?;
|
|
|
|
manager
|
|
.alter_table(
|
|
Table::alter()
|
|
.table(User::Table)
|
|
.drop_column(UserExtension::Email)
|
|
.drop_column(UserExtension::DisplayName)
|
|
.drop_column(UserExtension::Bio)
|
|
.drop_column(UserExtension::AvatarUrl)
|
|
.drop_column(UserExtension::HeaderUrl)
|
|
.drop_column(UserExtension::CustomCss)
|
|
.drop_column(UserExtension::CreatedAt)
|
|
.drop_column(UserExtension::UpdatedAt)
|
|
.to_owned(),
|
|
)
|
|
.await
|
|
}
|
|
}
|
|
|
|
#[derive(DeriveIden)]
|
|
enum UserExtension {
|
|
Email,
|
|
DisplayName,
|
|
Bio,
|
|
AvatarUrl,
|
|
HeaderUrl,
|
|
CustomCss,
|
|
CreatedAt,
|
|
UpdatedAt,
|
|
}
|
|
|
|
#[derive(DeriveIden)]
|
|
enum TopFriends {
|
|
Table,
|
|
UserId,
|
|
FriendId,
|
|
Position,
|
|
}
|