From cadedf43c795c99449fc132170f84304efe5b4ab Mon Sep 17 00:00:00 2001
From: bitpredator <67551273+bitpredator@users.noreply.github.com>
Date: Tue, 11 Jun 2024 11:00:48 +0200
Subject: [PATCH] convert: esx_datastore > bpt_datastore

---
 .../server/classes/datastore.lua              | 82 +++++++++++++++++++
 .../bpt_datastore/server/main.lua             | 80 ++++++++++++++++++
 2 files changed, 162 insertions(+)
 create mode 100644 server-data/resources/[bpt_addons]/bpt_datastore/server/classes/datastore.lua
 create mode 100644 server-data/resources/[bpt_addons]/bpt_datastore/server/main.lua

diff --git a/server-data/resources/[bpt_addons]/bpt_datastore/server/classes/datastore.lua b/server-data/resources/[bpt_addons]/bpt_datastore/server/classes/datastore.lua
new file mode 100644
index 000000000..79a103432
--- /dev/null
+++ b/server-data/resources/[bpt_addons]/bpt_datastore/server/classes/datastore.lua
@@ -0,0 +1,82 @@
+function stringsplit(inputstr, sep)
+    if sep == nil then
+        sep = "%s"
+    end
+
+    local t = {}
+    i = 1
+    for str in string.gmatch(inputstr, "([^" .. sep .. "]+)") do
+        t[i] = str
+        i = i + 1
+    end
+
+    return t
+end
+
+function CreateDataStore(name, owner, data)
+    local self = {}
+
+    self.name = name
+    self.owner = owner
+    self.data = type(data) == "string" and json.decode(data) or data
+
+    local timeoutCallbacks = {}
+
+    function self.set(key, val)
+        data[key] = val
+        self.save()
+    end
+
+    function self.get(key, i)
+        local path = stringsplit(key, ".")
+        local obj = self.data
+
+        for i = 1, #path, 1 do
+            obj = obj[path[i]]
+        end
+
+        if i == nil then
+            return obj
+        else
+            return obj[i]
+        end
+    end
+
+    function self.count(key, i)
+        local path = stringsplit(key, ".")
+        local obj = self.data
+
+        for i = 1, #path, 1 do
+            obj = obj[path[i]]
+        end
+
+        if i ~= nil then
+            obj = obj[i]
+        end
+
+        if obj == nil then
+            return 0
+        else
+            return #obj
+        end
+    end
+
+    function self.save()
+        for i = 1, #timeoutCallbacks, 1 do
+            ESX.ClearTimeout(timeoutCallbacks[i])
+            timeoutCallbacks[i] = nil
+        end
+
+        local timeoutCallback = ESX.SetTimeout(10000, function()
+            if self.owner == nil then
+                MySQL.update("UPDATE datastore_data SET data = ? WHERE name = ?", { json.encode(self.data), self.name })
+            else
+                MySQL.update("UPDATE datastore_data SET data = ? WHERE name = ? and owner = ?", { json.encode(self.data), self.name, self.owner })
+            end
+        end)
+
+        table.insert(timeoutCallbacks, timeoutCallback)
+    end
+
+    return self
+end
diff --git a/server-data/resources/[bpt_addons]/bpt_datastore/server/main.lua b/server-data/resources/[bpt_addons]/bpt_datastore/server/main.lua
new file mode 100644
index 000000000..4f38fae04
--- /dev/null
+++ b/server-data/resources/[bpt_addons]/bpt_datastore/server/main.lua
@@ -0,0 +1,80 @@
+local DataStores, DataStoresIndex, SharedDataStores = {}, {}, {}
+
+AddEventHandler("onResourceStart", function(resourceName)
+    if resourceName == GetCurrentResourceName() then
+        local dataStore = MySQL.query.await("SELECT * FROM datastore_data LEFT JOIN datastore ON datastore_data.name = datastore.name UNION SELECT * FROM datastore_data RIGHT JOIN datastore ON datastore_data.name = datastore.name")
+
+        local newData = {}
+        for i = 1, #dataStore do
+            local data = dataStore[i]
+            if data.shared == 0 then
+                if not DataStores[data.name] then
+                    DataStoresIndex[#DataStoresIndex + 1] = data.name
+                    DataStores[data.name] = {}
+                end
+                DataStores[data.name][#DataStores[data.name] + 1] = CreateDataStore(data.name, data.owner, json.decode(data.data))
+            else
+                if data.data then
+                    SharedDataStores[data.name] = CreateDataStore(data.name, nil, json.decode(data.data))
+                else
+                    newData[#newData + 1] = { data.name, "'{}'" }
+                end
+            end
+        end
+
+        if next(newData) then
+            MySQL.prepare("INSERT INTO datastore_data (name, data) VALUES (?, ?)", newData)
+            for i = 1, #newData do
+                local new = newData[i]
+                SharedDataStores[new[1]] = CreateDataStore(new[1], nil, {})
+            end
+        end
+    end
+end)
+
+function GetDataStore(name, owner)
+    for i = 1, #DataStores[name], 1 do
+        if DataStores[name][i].owner == owner then
+            return DataStores[name][i]
+        end
+    end
+end
+
+function GetDataStoreOwners(name)
+    local identifiers = {}
+
+    for i = 1, #DataStores[name], 1 do
+        table.insert(identifiers, DataStores[name][i].owner)
+    end
+
+    return identifiers
+end
+
+function GetSharedDataStore(name)
+    return SharedDataStores[name]
+end
+
+AddEventHandler("bpt_datastore:getDataStore", function(name, owner, cb)
+    cb(GetDataStore(name, owner))
+end)
+
+AddEventHandler("bpt_datastore:getDataStoreOwners", function(name, cb)
+    cb(GetDataStoreOwners(name))
+end)
+
+AddEventHandler("bpt_datastore:getSharedDataStore", function(name, cb)
+    cb(GetSharedDataStore(name))
+end)
+
+AddEventHandler("esx:playerLoaded", function(playerId, xPlayer)
+    for i = 1, #DataStoresIndex, 1 do
+        local name = DataStoresIndex[i]
+        local dataStore = GetDataStore(name, xPlayer.identifier)
+
+        if not dataStore then
+            MySQL.insert("INSERT INTO datastore_data (name, owner, data) VALUES (?, ?, ?)", { name, xPlayer.identifier, "{}" })
+
+            DataStores[name][#DataStores[name] + 1] = CreateDataStore(name, xPlayer.identifier, {})
+        end
+    end
+end)