postgress

imphnen-backend/src/bin/seed_permissions.rs

@@ -1,77 +1,81 @@
-use imphnen_iam::PermissionsEnum;
-use imphnen_utils::{get_iso_date};
-use serde_json::json;
-use std::error::Error;
-use surrealdb::engine::any;
-use surrealdb::opt::auth::Root;
-
-#[tokio::main]
-async fn main() -> Result<(), Box<dyn Error>> {
-	let env = &imphnen_libs::environment::ENV;
-	let db = any::connect(&env.surrealdb_url).await?;
-	db.signin(Root {
-		username: &env.surrealdb_username,
-		password: &env.surrealdb_password,
-	})
-	.await?;
-	db.use_ns(env.surrealdb_namespace.clone())
-		.use_db(env.surrealdb_dbname.clone())
-		.await?;
-
-	for permission in [
-		PermissionsEnum::ReadListUsers,
-		PermissionsEnum::ReadDetailUsers,
-		PermissionsEnum::CreateUsers,
-		PermissionsEnum::DeleteUsers,
-		PermissionsEnum::UpdateUsers,
-		PermissionsEnum::ActivateUsers,
-		PermissionsEnum::ReadListRoles,
-		PermissionsEnum::ReadDetailRoles,
-		PermissionsEnum::CreateRoles,
-		PermissionsEnum::DeleteRoles,
-		PermissionsEnum::UpdateRoles,
-		PermissionsEnum::ReadListPermissions,
-		PermissionsEnum::ReadDetailPermissions,
-		PermissionsEnum::CreatePermissions,
-		PermissionsEnum::DeletePermissions,
-		PermissionsEnum::UpdatePermissions,
-		PermissionsEnum::CreateGachaClaims,
-		PermissionsEnum::ReadDetailGachaClaims,
-		PermissionsEnum::ReadListGachaItems,
-		PermissionsEnum::ReadDetailGachaItems,
-		PermissionsEnum::CreateGachaItems,
-		PermissionsEnum::DeleteGachaItems,
-		PermissionsEnum::UpdateGachaItems,
-		PermissionsEnum::ReadDetailGachaRolls,
-		PermissionsEnum::CreateGachaRolls,
-		PermissionsEnum::ExecuteGachaRolls,
-		PermissionsEnum::ReadListMentors,
-		PermissionsEnum::ReadDetailMentors,
-		PermissionsEnum::RegisterMentors,
-		PermissionsEnum::ReadOwnMentorProfile,
-		PermissionsEnum::UpdateOwnMentorProfile,
-		PermissionsEnum::ReadOwnMentorStatus,
-		PermissionsEnum::UpdateMentors,
-		PermissionsEnum::VerifyMentors,
-		PermissionsEnum::DeleteMentors,
-		PermissionsEnum::Administrator,
-	] {
-		db.query("CREATE type::thing('app_permissions', $id) CONTENT $data")
-			.bind(("id", permission.id()))
-			.bind((
-				"data",
-				json!({
-					"name": permission.to_string(),
-					"is_deleted": false,
-					"created_at": get_iso_date(),
-					"updated_at": get_iso_date()
-				}),
-			))
-			.await?;
-		println!("✅ Inserted: {permission}");
-	}
-
-	println!("✅ All Permissions seeded");
-
-	Ok(())
-}
+#![allow(clippy::all)]
+
+use imphnen_iam::PermissionsEnum;
+use std::error::Error;
+use imphnen_libs::postgres::{PostgresConfig, PostgresConnection};
+use imphnen_entities::seaorm::auth::permissions::ActiveModel as PermissionActiveModel;
+use imphnen_entities::seaorm::auth::permissions::Entity as PermissionEntity;
+use sea_orm::ActiveValue::Set;
+use sea_orm::{ActiveModelTrait};
+use uuid::Uuid;
+use chrono::Utc;
+
+#[tokio::main]
+async fn main() -> Result<(), Box<dyn Error>> {
+	let config = PostgresConfig::from_env()?;
+	let pg_conn = PostgresConnection::new(config).await?;
+	let db = &pg_conn.conn;
+
+	for permission in [
+		PermissionsEnum::ReadListUsers,
+		PermissionsEnum::ReadDetailUsers,
+		PermissionsEnum::CreateUsers,
+		PermissionsEnum::DeleteUsers,
+		PermissionsEnum::UpdateUsers,
+		PermissionsEnum::ActivateUsers,
+		PermissionsEnum::ReadListRoles,
+		PermissionsEnum::ReadDetailRoles,
+		PermissionsEnum::CreateRoles,
+		PermissionsEnum::DeleteRoles,
+		PermissionsEnum::UpdateRoles,
+		PermissionsEnum::ReadListPermissions,
+		PermissionsEnum::ReadDetailPermissions,
+		PermissionsEnum::CreatePermissions,
+		PermissionsEnum::DeletePermissions,
+		PermissionsEnum::UpdatePermissions,
+		PermissionsEnum::CreateGachaClaims,
+		PermissionsEnum::ReadDetailGachaClaims,
+		PermissionsEnum::ReadListGachaItems,
+		PermissionsEnum::ReadDetailGachaItems,
+		PermissionsEnum::CreateGachaItems,
+		PermissionsEnum::DeleteGachaItems,
+		PermissionsEnum::UpdateGachaItems,
+		PermissionsEnum::ReadDetailGachaRolls,
+		PermissionsEnum::CreateGachaRolls,
+		PermissionsEnum::ExecuteGachaRolls,
+		PermissionsEnum::ReadListMentors,
+		PermissionsEnum::ReadDetailMentors,
+		PermissionsEnum::RegisterMentors,
+		PermissionsEnum::ReadOwnMentorProfile,
+		PermissionsEnum::UpdateOwnMentorProfile,
+		PermissionsEnum::ReadOwnMentorStatus,
+		PermissionsEnum::UpdateMentors,
+		PermissionsEnum::VerifyMentors,
+		PermissionsEnum::DeleteMentors,
+		PermissionsEnum::Administrator,
+	] {
+		// permission.id() returns a string, try parse to uuid
+		let parsed_id = Uuid::parse_str(&permission.id()).unwrap_or_else(|_| Uuid::new_v4());
+
+		// Check if permission already exists
+		let existing = PermissionEntity::find_by_id(parsed_id).one(db).await?;
+		if existing.is_some() {
+			println!("ℹ️  Skipping (already exists): {permission}");
+			continue;
+		}
+
+		// Insert permission using active model
+		let mut perm_model: PermissionActiveModel = Default::default();
+		perm_model.id = Set(parsed_id);
+		perm_model.name = Set(permission.to_string());
+		perm_model.is_deleted = Set(false);
+		perm_model.created_at = Set(Utc::now());
+		perm_model.updated_at = Set(Utc::now());
+		perm_model.insert(db).await?;
+		println!("✅ Inserted: {permission}");
+	}
+
+	println!("✅ All Permissions seeded");
+
+	Ok(())
+}