From 82e4908dcfca89c2c749f57cc21df02415632482 Mon Sep 17 00:00:00 2001
From: HadiSharifi <sharifihadi002@gmail.com>
Date: Sun, 10 May 2026 12:29:07 +0330
Subject: [PATCH] implement xp/level logic

---
 Java-Knight/src/main/java/org/project/Main.java | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/Java-Knight/src/main/java/org/project/Main.java b/Java-Knight/src/main/java/org/project/Main.java
index 30b3e74..78c5c7d 100644
--- a/Java-Knight/src/main/java/org/project/Main.java
+++ b/Java-Knight/src/main/java/org/project/Main.java
@@ -1,9 +1,6 @@
 package org.project;
 
-import org.project.entity.enemies.Enemy;
-import org.project.entity.enemies.Goblin;
-import org.project.entity.enemies.Skeleton;
-import org.project.entity.enemies.Vampire;
+import org.project.entity.enemies.*;
 import org.project.entity.players.Assassin;
 import org.project.entity.players.Knight;
 import org.project.entity.players.Player;
@@ -187,7 +184,10 @@ public class Main {
         }
         else {
             System.out.println("You Won!");
-            if (enemy.dropKey()) {
+            player.gainXP(Player.xpRewardFor(enemy));
+            player.setHP(player.getMaxHP());
+            player.setMP(player.getMaxMP());
+            if (!(enemy instanceof Dragon) && enemy.dropKey()) {
                 System.out.println(enemy.getClass().getSimpleName() + "'s key dropped!");
                 player.achieveKey(enemy);
             }