Loadout restrictions (removal and donation tier)

SQL/bandastation/bandastation_update.sql

@@ -38,3 +38,20 @@ CREATE TABLE `admin_wl` (
 	PRIMARY KEY (`id`),
 	KEY `ckey` (`ckey`)
 ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
+
+--
+-- Table structure for table `budget`
+--
+DROP TABLE IF EXISTS `budget`;
+CREATE TABLE `budget` (
+    `id` INT(11) NOT NULL AUTO_INCREMENT,
+    `date` DATETIME NOT NULL DEFAULT current_timestamp(),
+    `ckey` VARCHAR(32) NULL DEFAULT NULL COLLATE 'utf8mb4_unicode_ci',
+    `amount` INT(10) UNSIGNED NOT NULL,
+    `source` VARCHAR(32) NOT NULL COLLATE 'utf8mb4_general_ci',
+    `date_start` DATETIME NOT NULL DEFAULT current_timestamp(),
+    `date_end` DATETIME NULL DEFAULT (current_timestamp() + interval 1 month),
+    `is_valid` TINYINT(1) NOT NULL DEFAULT '1',
+    `discord_id` bigint(20) DEFAULT NULL,
+    PRIMARY KEY (`id`) USING BTREE
+) COLLATE='utf8mb4_general_ci' ENGINE=InnoDB;

SQL/bandastation/database_changelog.md

@@ -2,17 +2,35 @@ Any time you make a change to the schema files, remember to increment the databa
 
 Make sure to also update `DB_MAJOR_VERSION_220` and `DB_MINOR_VERSION_220`, which can be found in `code/modular_bandastation/_defines220/code/defines/subsystems.dm`.
 
-The latest database version is 1.1; The query to update the schema revision table is:
+The latest database version is 1.2; The query to update the schema revision table is:
 
 ```sql
-INSERT INTO `schema_revision_220` (`major`, `minor`) VALUES (1, 1);
+INSERT INTO `schema_revision_220` (`major`, `minor`) VALUES (1, 2);
 ```
 or
 
 ```sql
-INSERT INTO `SS13_schema_revision_220` (`major`, `minor`) VALUES (1, 1);
+INSERT INTO `SS13_schema_revision_220` (`major`, `minor`) VALUES (1, 2);
 ```
 
+-----------------------------------------------------
+Version 1.2, 22 July 2024, by larentoun
+Created the table: budget
+
+```sql
+CREATE TABLE `budget` (
+    `id` INT(11) NOT NULL AUTO_INCREMENT,
+    `date` DATETIME NOT NULL DEFAULT current_timestamp(),
+    `ckey` VARCHAR(32) NULL DEFAULT NULL COLLATE 'utf8mb4_unicode_ci',
+    `amount` INT(10) UNSIGNED NOT NULL,
+    `source` VARCHAR(32) NOT NULL COLLATE 'utf8mb4_general_ci',
+    `date_start` DATETIME NOT NULL DEFAULT current_timestamp(),
+    `date_end` DATETIME NULL DEFAULT (current_timestamp() + interval 1 month),
+    `is_valid` TINYINT(1) NOT NULL DEFAULT '1',
+    `discord_id` bigint(20) DEFAULT NULL,
+    PRIMARY KEY (`id`) USING BTREE
+) COLLATE='utf8mb4_general_ci' ENGINE=InnoDB;
+```
 -----------------------------------------------------
 Version 1.1, 17 April 2024, by larentoun
 Created the tables: schema_revision220, ckey_whitelist, admin_wl

modular_bandastation/_defines220/code/defines/subsystems.dm

@@ -15,4 +15,4 @@
  *
  * make sure you add an update to the schema_version stable in the db changelog
  */
-#define DB_MINOR_VERSION_220 1
+#define DB_MINOR_VERSION_220 2

modular_bandastation/loadout/_loadout.dm

@@ -0,0 +1,4 @@
+/datum/modpack/loadout
+	name = "Loadout"
+	desc = "Изменения в loadout."
+	author = "larentoun"

modular_bandastation/loadout/_loadout.dme

@@ -0,0 +1,7 @@
+#include "_loadout.dm"
+
+#include "code/client.dm"
+#include "code/loadout_categories.dm"
+#include "code/loadout_items.dm"
+#include "code/loadout_preference.dm"
+#include "code/categories/accessories.dm"

modular_bandastation/loadout/code/categories/accessories.dm

@@ -0,0 +1,5 @@
+/datum/loadout_item/accessory/pride/is_available()
+	return FALSE
+
+/datum/loadout_item/accessory/maid_apron
+	donator_level = 1

modular_bandastation/loadout/code/client.dm

@@ -0,0 +1,69 @@
+#define DONATION_TIER_1 220
+#define DONATION_TIER_2 440
+#define DONATION_TIER_3 1000
+#define DONATION_TIER_4 2220
+#define DONATION_TIER_5 10000
+
+/client
+	/// Call `proc/get_donator_level()` instead to get a value when possible.
+	var/donator_level = 0
+	COOLDOWN_DECLARE(db_check_cooldown)
+
+// For unit-tests
+/datum/client_interface
+	var/donator_level = 0
+
+/datum/client_interface/proc/get_donator_level()
+	return donator_level
+
+/client/proc/get_donator_level()
+	donator_level = max(donator_level, get_donator_level_from_db(), get_donator_level_from_admin())
+	return donator_level
+
+/client/proc/get_donator_level_from_admin()
+	if(!holder)
+		return 0
+	var/rank_flags = holder.rank_flags()
+	if(rank_flags & R_EVERYTHING)
+		return 5
+	if(rank_flags & R_ADMIN)
+		return 3
+	return 0
+
+/client/proc/get_donator_level_from_db()
+	if(!COOLDOWN_FINISHED(src, db_check_cooldown))
+		return 0
+	COOLDOWN_START(src, db_check_cooldown, 15 SECONDS)
+	var/datum/db_query/query_get_donator_level = SSdbcore.NewQuery({"
+		SELECT CAST(SUM(amount) as UNSIGNED INTEGER) FROM budget
+		WHERE ckey=:ckey
+			AND is_valid=true
+			AND date_start <= NOW()
+			AND (NOW() < date_end OR date_end IS NULL)
+		GROUP BY ckey
+	"}, list("ckey" = ckey))
+
+	var/amount = 0
+	if(query_get_donator_level.warn_execute() && length(query_get_donator_level.rows))
+		query_get_donator_level.NextRow()
+		amount = query_get_donator_level.item[1]
+	qdel(query_get_donator_level)
+
+	switch(amount)
+		if(DONATION_TIER_1 to (DONATION_TIER_2 - 1))
+			return 1
+		if(DONATION_TIER_2 to (DONATION_TIER_3 - 1))
+			return 2
+		if(DONATION_TIER_3 to (DONATION_TIER_4 - 1))
+			return 3
+		if(DONATION_TIER_4 to (DONATION_TIER_5 - 1))
+			return 4
+		if(DONATION_TIER_5 to INFINITY)
+			return 5
+	return 0
+
+#undef DONATION_TIER_1
+#undef DONATION_TIER_2
+#undef DONATION_TIER_3
+#undef DONATION_TIER_4
+#undef DONATION_TIER_5

modular_bandastation/loadout/code/loadout_categories.dm

@@ -0,0 +1,6 @@
+/datum/loadout_category/get_items()
+	. = ..()
+	for(var/datum/loadout_item/item as anything in .)
+		if(item.is_available())
+			continue
+		. -= item

modular_bandastation/loadout/code/loadout_items.dm

@@ -0,0 +1,11 @@
+/datum/loadout_item
+	var/donator_level = 0
+
+/datum/loadout_item/get_item_information()
+	. = ..()
+	if(donator_level)
+		. += "Tier [donator_level]"
+
+// Removes item from the preferences menu, period. Use it only to remove stuff for ALL players.
+/datum/loadout_item/proc/is_available()
+	return TRUE

modular_bandastation/loadout/code/loadout_preference.dm

@@ -0,0 +1,25 @@
+/datum/preferences/load_preferences()
+	. = ..()
+	parent.get_donator_level()
+
+// Handles selecting from the preferences UI
+/datum/preference_middleware/loadout/select_item(datum/loadout_item/selected_item)
+	var/donator_level = preferences.parent.get_donator_level()
+	if(donator_level >= selected_item.donator_level)
+		return ..()
+	to_chat(preferences.parent.mob, span_warning("У вас недостаточный уровень доната, чтобы взять [selected_item.name]!"))
+
+// Removes donator_level items from the user if their donator_level is insufficient
+/datum/preference/loadout/deserialize(input, datum/preferences/preferences)
+	. = ..()
+	// For loadout purposes, donator_level is updated in middleware on select
+	var/donator_level = preferences.parent.donator_level
+	var/removed_items = list()
+	for(var/path in .)
+		var/datum/loadout_item/item = GLOB.all_loadout_datums[path]
+		if(donator_level >= item.donator_level)
+			continue
+		. -= path
+		removed_items += item.name
+	if(length(removed_items) && preferences.parent.mob)
+		to_chat(preferences.parent.mob, span_warning("У вас недостаточный уровень доната, чтобы взять: [english_list(removed_items, and_text = " и ")]!"))

modular_bandastation/modular_bandastation.dme

@@ -20,6 +20,7 @@
 #include "examine_panel/_examine_panel.dme"
 #include "gunhud/_gunhud.dme"
 #include "keybinding/_keybinding.dme"
+#include "loadout/_loadout.dme"
 #include "mapping/_mapping.dme"
 #include "pixel_shift/_pixel_shift.dme"
 #include "ru_jobs/_ru_jobs.dme"