From 1e6e4fcd2482a641daa529892b898cb21ff41561 Mon Sep 17 00:00:00 2001
From: holzmaster <holzmaster@outlook.com>
Date: Sun, 8 Sep 2024 21:59:01 +0200
Subject: [PATCH] PKV -> 1x more AU

---
 src/service/loot.ts | 25 ++++++++++++++++++++-----
 1 file changed, 20 insertions(+), 5 deletions(-)

diff --git a/src/service/loot.ts b/src/service/loot.ts
index 04ec9d7a..2864811c 100644
--- a/src/service/loot.ts
+++ b/src/service/loot.ts
@@ -619,6 +619,10 @@ export async function getUserLootsById(userId: Snowflake, lootTypeId: number) {
     return await loot.getUserLootsById(userId, lootTypeId);
 }
 
+export async function getUserLootCountById(userId: Snowflake, lootTypeId: number): Promise<number> {
+    return (await loot.getUserLootsById(userId, lootTypeId)).length;
+}
+
 export async function getLootsByKindId(lootTykeId: LootTypeId) {
     return await loot.getLootsByKindId(lootTykeId);
 }
@@ -648,20 +652,31 @@ async function getDropWeightAdjustments(
     user: User,
     weights: readonly number[],
 ): Promise<AdjustmentResult> {
-    const waste = await getUserLootsById(user.id, LootTypeId.RADIOACTIVE_WASTE);
+    const waste = await getUserLootCountById(user.id, LootTypeId.RADIOACTIVE_WASTE);
     const messages = [];
 
     let wasteFactor = 1;
-    if (waste.length > 0) {
+    if (waste > 0) {
         const wasteDropPenalty = 1.05;
-        wasteFactor = Math.min(2, waste.length ** wasteDropPenalty);
+        wasteFactor = Math.min(2, waste ** wasteDropPenalty);
         messages.push(
-            `Du hast ${waste.length} Tonnen radioaktiven Müll, deshalb ist die Chance auf ein Geschenk geringer.`,
+            `Du hast ${waste} Tonnen radioaktiven Müll, deshalb ist die Chance auf ein Geschenk geringer.`,
         );
     }
 
+    const pkv = await getUserLootCountById(user.id, LootTypeId.PKV);
+    let pkvFactor = 1;
+    if (pkv > 0) {
+        pkvFactor = 2;
+        messages.push("Da du privat versichert bist, hast du die doppelte Chance auf eine AU.");
+    }
+
+    const newWeights = [...weights];
+    newWeights[LootTypeId.NICHTS] = Math.ceil(weights[LootTypeId.NICHTS] * wasteFactor) | 0;
+    newWeights[LootTypeId.KRANKSCHREIBUNG] = (weights[LootTypeId.KRANKSCHREIBUNG] * pkvFactor) | 0;
+
     return {
         messages,
-        weights: [Math.ceil(weights[0] * wasteFactor) | 0, ...weights.slice(1)],
+        weights: newWeights,
     };
 }