From d0abd6105c2891f194322eeb6b8291e6a226eaf8 Mon Sep 17 00:00:00 2001
From: Freeman <project@gnuhacker.org>
Date: Thu, 30 May 2024 21:24:54 +0200
Subject: [PATCH] added set_emeralds(player) func

---
 functions.lua | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/functions.lua b/functions.lua
index 80a7b6c..a3b1619 100644
--- a/functions.lua
+++ b/functions.lua
@@ -26,6 +26,38 @@ function emeraldbank.get_emeralds(name)
 	return atm.balance[name]
 end
 
+-- Accepts a player object or player name as input now
+function emeraldbank.set_emeralds(player, num)
+    if not player then return false end
+
+    local name
+    if type(player) == "string" then
+        name = player
+    else
+        name = player:get_player_name()
+    end
+
+    atm.read_account(name)
+
+    -- Check if atm.balance[name] exists
+    if atm.balance[name] == nil then
+        return false
+    end
+
+    if num then
+        atm.balance[name] = math.floor(num)
+
+        -- Update actionbar only if 'player' is a player object
+        if type(player) ~= "string" then
+            mcl_title.set(player, "actionbar", {text=S("Emeralds in Bank: @1", atm.balance[name]), color="yellow"})
+        end
+
+        atm.save_account(name)
+        return true
+    end
+    return false
+end
+
 -- Accepts a player object or player name as input now
 function emeraldbank.add_emeralds(player, num)
     if not player then return false end