use music_player_types::types::Song;
use sea_orm::{entity::prelude::*, ActiveValue};
use serde::{Deserialize, Serialize};
#[derive(Clone, Debug, Default, PartialEq, DeriveEntityModel, Serialize, Deserialize)]
#[sea_orm(table_name = "album")]
pub struct Model {
#[sea_orm(primary_key, auto_increment = false)]
pub id: String,
pub title: String,
pub artist: String,
pub artist_id: Option<String>,
pub year: Option<u32>,
pub cover: Option<String>,
#[sea_orm(ignore)]
pub tracks: Vec<super::track::Model>,
}
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
pub enum Relation {
#[sea_orm(has_many = "super::track::Entity")]
Track,
#[sea_orm(
belongs_to = "super::artist::Entity",
from = "Column::ArtistId",
to = "super::artist::Column::Id"
)]
Artist,
}
impl Related<super::track::Entity> for Entity {
fn to() -> RelationDef {
Relation::Track.def()
}
}
impl Related<super::artist::Entity> for Entity {
fn to() -> RelationDef {
Relation::Artist.def()
}
}
impl ActiveModelBehavior for ActiveModel {}
impl From<&Song> for ActiveModel {
fn from(song: &Song) -> Self {
let id = format!("{:x}", md5::compute(format!("{}", song.album)));
Self {
id: ActiveValue::set(id),
title: ActiveValue::Set(song.album.clone()),
artist: ActiveValue::Set(song.album_artist.clone()),
artist_id: ActiveValue::Set(Some(format!(
"{:x}",
md5::compute(song.album_artist.to_owned())
))),
year: ActiveValue::Set(song.year),
cover: ActiveValue::Set(song.cover.clone()),
}
}
}