feat: add gacha and init cms

imphnen-gacha/src/v1/gacha_roll/gacha_roll_repository.rs

@@ -1,6 +1,10 @@
+use super::GachaRollQueryDto;
 use super::GachaRollSchema;
-use crate::{AppState, ResourceEnum};
+use crate::{AppState, DetailQueryBuilder, ResourceEnum};
 use anyhow::{Result, bail};
+use rand::prelude::*;
+use rand::rng;
+use rand_distr::weighted::WeightedIndex;
 
 pub struct GachaRollRepository<'a> {
 	state: &'a AppState,
@@ -11,11 +15,18 @@ impl<'a> GachaRollRepository<'a> {
 		Self { state }
 	}
 
-	pub async fn query_gacha_roll_by_id(&self, id: String) -> Result<GachaRollSchema> {
+	pub async fn query_gacha_roll_by_id(
+		&self,
+		id: String,
+	) -> Result<GachaRollQueryDto> {
 		let db = &self.state.surrealdb_ws;
-		let result: Option<GachaRollSchema> = db
-			.select((ResourceEnum::GachaRolls.to_string(), id.clone()))
-			.await?;
+		let builder = DetailQueryBuilder::new(ResourceEnum::GachaRolls.to_string())
+			.with_id(id.clone())
+			.with_select_fields(vec!["*"])
+			.with_fetch("item");
+		let sql = builder.build();
+		let result: Option<GachaRollQueryDto> =
+			builder.apply_bindings(db.query(sql)).await?.take(0)?;
 		match result {
 			Some(roll) if !roll.is_deleted => Ok(roll),
 			_ => bail!("Gacha Roll not found"),
@@ -36,4 +47,34 @@ impl<'a> GachaRollRepository<'a> {
 			None => bail!("Failed to create Gacha Roll"),
 		}
 	}
+
+	pub async fn query_all_active_rolls(&self) -> Result<Vec<GachaRollQueryDto>> {
+		let db = &self.state.surrealdb_ws;
+		let sql = DetailQueryBuilder::new(ResourceEnum::GachaRolls.to_string())
+			.with_where("is_deleted")
+			.where_value("false")
+			.with_select_fields(vec!["*"])
+			.with_fetch("item")
+			.build();
+		let result: Vec<GachaRollQueryDto> = db.query(sql).await?.take(0)?;
+		Ok(result)
+	}
+
+	pub fn roll_once(rolls: &[GachaRollQueryDto]) -> Option<GachaRollQueryDto> {
+		let filtered: Vec<_> = rolls
+			.iter()
+			.filter(|r| !r.is_deleted && r.quantity > 0)
+			.collect();
+		let weights: Vec<f32> = filtered
+			.iter()
+			.map(|r| r.weight * r.quantity as f32)
+			.collect();
+		if weights.iter().all(|&w| w <= 0.0) {
+			return None;
+		}
+		let dist = WeightedIndex::new(&weights).ok()?;
+		let mut rng = rng();
+		let index = dist.sample(&mut rng);
+		Some(filtered[index].clone())
+	}
 }