From bed2d75b03a94fb5a20958a9294ca27ca95ff127 Mon Sep 17 00:00:00 2001
From: phixxy <bottlecap@protonmail.com>
Date: Tue, 9 Jan 2024 12:34:10 -0800
Subject: [PATCH] added feed and affection commands, they need to be limited in
 the future

---
 sparkytron3000.py | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/sparkytron3000.py b/sparkytron3000.py
index 367e1f4..74992f7 100644
--- a/sparkytron3000.py
+++ b/sparkytron3000.py
@@ -1721,6 +1721,16 @@ async def pokemon(ctx, arg1=None, arg2=None, arg3=None, arg4=None):
         json_data = await get_json(url)
         return json_data
     
+    async def give_buddy_food(pkmn_data): #for now this will do the same as affection
+        return await give_buddy_affection(pkmn_data)
+
+    async def give_buddy_affection(pkmn_data):
+        print(pkmn_data['buddy_xp'])
+        xp = pkmn_data['buddy_xp']
+        level = await calc_pkmn_buddy_level(pkmn_data)
+        pkmn_data['buddy_xp'] += (3*level)
+        return pkmn_data
+    
     async def calc_pkmn_buddy_level(pkmn_json): #this uses the 'fast' xp rate
         buddy_xp = pkmn_json['buddy_xp']
         return min(math.floor(((5*buddy_xp)/4)**(1/3)),100)
@@ -1779,7 +1789,7 @@ async def pokemon(ctx, arg1=None, arg2=None, arg3=None, arg4=None):
             await ctx.channel.send("You already have a pokemon!")
             return
         
-    if arg1 == 'nick' or arg1 == 'nickname':
+    elif arg1 == 'nick' or arg1 == 'nickname':
         nickname = arg2
         json_data = await load_pokemon(ctx.author.id)
         json_data['nickname'] = nickname
@@ -1787,6 +1797,17 @@ async def pokemon(ctx, arg1=None, arg2=None, arg3=None, arg4=None):
         message = "You gave " + nickname + ' a new name!'
         await ctx.channel.send(message)
         return
+    
+    elif arg1 == 'feed' or arg1 == 'hug':
+        json_data = await load_pokemon(ctx.author.id)
+        json_data = await give_buddy_affection(json_data)
+        await save_pokemon(ctx.author.id, json_data)
+        if json_data['nickname']:
+            message = "You " + arg1 + ' ' + json_data['nickname']
+        else:
+            message = "You " + arg1 + ' ' + json_data['name']
+        await ctx.channel.send(message)
+        return
 
     #Default !pokemon behavior (Load and show pokemon embed)
     discord_id = ctx.author.id