use sea_orm::entity::prelude::*;
use serde::{Deserialize, Serialize};

use crate::{cluster_stage::ClusterStage, icon::Icon, loader::GameLoader};

#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
#[sea_orm(table_name = "clusters")]
#[onelauncher_macro::specta]
pub struct Model {
	#[sea_orm(primary_key)]
	pub id: i32,
	#[sea_orm(column_type = "Text")]
	pub path: String,
	pub stage: ClusterStage,
	pub created_at: i32,
	pub updated_at: i32,
	pub group_id: Option<i32>,
	#[sea_orm(column_type = "Text")]
	pub name: String,
	#[sea_orm(column_type = "Text")]
	pub mc_version: String,
	pub mc_loader: GameLoader,
	#[sea_orm(column_type = "Text", nullable)]
	pub mc_loader_version: Option<String>,
	pub last_played: Option<i32>,
	pub overall_played: Option<i32>,
	#[sea_orm(column_type = "Text", nullable)]
	pub icon_url: Option<Icon>,
	#[sea_orm(column_type = "Text", nullable)]
	pub setting_profile_name: Option<String>,
	#[sea_orm(column_type = "Text", nullable)]
	pub linked_pack_id: Option<String>,
	pub linked_pack_version: Option<i32>,
}

#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
pub enum Relation {
	#[sea_orm(
		belongs_to = "super::cluster_groups::Entity",
		from = "Column::GroupId",
		to = "super::cluster_groups::Column::Id",
		on_update = "NoAction",
		on_delete = "NoAction"
	)]
	ClusterGroups,
	#[sea_orm(has_many = "super::cluster_packages::Entity")]
	ClusterPackages,
	#[sea_orm(
		belongs_to = "super::setting_profiles::Entity",
		from = "Column::SettingProfileName",
		to = "super::setting_profiles::Column::Name",
		on_update = "NoAction",
		on_delete = "NoAction"
	)]
	SettingProfiles,
}

impl Related<super::cluster_groups::Entity> for Entity {
	fn to() -> RelationDef {
		Relation::ClusterGroups.def()
	}
}

impl Related<super::cluster_packages::Entity> for Entity {
	fn to() -> RelationDef {
		Relation::ClusterPackages.def()
	}
}

impl Related<super::setting_profiles::Entity> for Entity {
	fn to() -> RelationDef {
		Relation::SettingProfiles.def()
	}
}

impl Related<super::packages::Entity> for Entity {
	fn to() -> RelationDef {
		super::cluster_packages::Relation::Packages.def()
	}
	fn via() -> Option<RelationDef> {
		Some(super::cluster_packages::Relation::Clusters.def().rev())
	}
}

impl ActiveModelBehavior for ActiveModel {}