add support for entitlements

include/sporks/bot.h

@@ -136,6 +136,9 @@ class Bot {
 	void onVoiceStateUpdate (const dpp::voice_state_update_t &event);
 	void onVoiceServerUpdate (const dpp::voice_server_update_t &event);
 	void onWebhooksUpdate (const dpp::webhooks_update_t &event);
+	void onEntitlementDelete(const dpp::entitlement_delete_t& ed);
+	void onEntitlementCreate(const dpp::entitlement_create_t& ed);
+	void onEntitlementUpdate(const dpp::entitlement_update_t& ed);
 
 	static std::string GetConfig(const std::string &name);
 

include/sporks/modules.h

@@ -66,6 +66,9 @@ enum Implementation
 	I_OnVoiceStateUpdate,
 	I_OnVoiceServerUpdate,
 	I_OnWebhooksUpdate,
+	I_OnEntitlementCreate,
+	I_OnEntitlementUpdate,
+	I_OnEntitlementDelete,
 	I_END
 };
 
@@ -232,6 +235,9 @@ class Module {
 	virtual bool OnVoiceStateUpdate(const dpp::voice_state_update_t &obj);
 	virtual bool OnVoiceServerUpdate(const dpp::voice_server_update_t &obj);
 	virtual bool OnWebhooksUpdate(const dpp::webhooks_update_t &obj);
+	virtual bool OnEntitlementCreate(const dpp::entitlement_create_t &obj);
+	virtual bool OnEntitlementUpdate(const dpp::entitlement_update_t &obj);
+	virtual bool OnEntitlementDelete(const dpp::entitlement_delete_t &obj);
 
 	/* Emit a simple text only embed to a channel, many modules use this for error reporting */
 	void EmbedSimple(const std::string &message, int64_t channelID);

modules/trivia/entitlement.cpp

@@ -0,0 +1,36 @@
+#include "trivia.h"
+#include <sporks/database.h>
+#include <dpp/dpp.h>
+
+bool TriviaModule::OnEntitlementCreate(const dpp::entitlement_create_t& entitlement)
+{
+	db::query(
+		"INSERT INTO premium_credits (user_id, guild_id, subscription_id, active, since, plan_id, payment_failed, created_at, updated_at)"
+		"VALUES(?, ?, ?, 1, now(), 'ssod-monthly', 0, now(), now()) ON DUPLICATE KEY UPDATE subscription_id = ?, active = 1",
+		{ entitlement.created.user_id, entitlement.created.guild_id, entitlement.created.subscription_id, entitlement.created.subscription_id }
+	);
+	return true;
+}
+
+bool TriviaModule::OnEntitlementUpdate(const dpp::entitlement_update_t& entitlement)
+{
+	db::query(
+		"UPDATE premium_credits SET active = ?, updated_at = now() WHERE user_id = ? AND subscription_id = ?",
+		{
+			entitlement.updating_entitlement.is_deleted() || entitlement.updating_entitlement.ends_at < time(nullptr) ? 0 : 1,
+			entitlement.updating_entitlement.user_id,
+			entitlement.updating_entitlement.id
+		}
+	);
+	return true;
+}
+
+bool TriviaModule::OnEntitlementDelete(const dpp::entitlement_delete_t& entitlement)
+{
+	db::query(
+		"UPDATE premium_credits SET active = 0, cancel_date = now(), updated_at = now() WHERE user_id = ? AND subscription_id = ?",
+		{ entitlement.deleted.user_id, entitlement.deleted.subscription_id }
+	);
+	return true;
+}
+

modules/trivia/trivia.cpp

@@ -49,7 +49,16 @@ TriviaModule::TriviaModule(Bot* instigator, ModuleLoader* ml) : Module(instigato
 	srand(time(NULL) * time(NULL));
 
 	/* Attach D++ events to module */
-	ml->Attach({ I_OnMessage, I_OnPresenceUpdate, I_OnChannelDelete, I_OnGuildDelete, I_OnAllShardsReady, I_OnGuildCreate }, this);
+	ml->Attach({ I_OnMessage,
+		     I_OnPresenceUpdate,
+		     I_OnChannelDelete,
+		     I_OnGuildDelete,
+		     I_OnAllShardsReady,
+		     I_OnGuildCreate,
+		     I_OnEntitlementCreate,
+		     I_OnEntitlementDelete,
+		     I_OnEntitlementUpdate
+	}, this);
 
 	/* Various regular expressions */
 	number_tidy_dollars = new PCRE("^([\\d\\,]+)\\s+dollars$");

modules/trivia/trivia.h

@@ -145,6 +145,10 @@ class TriviaModule : public Module
 	virtual bool OnAllShardsReady();
 	virtual bool OnChannelDelete(const dpp::channel_delete_t &cd);
 	virtual bool OnGuildDelete(const dpp::guild_delete_t &gd);
+	virtual bool OnEntitlementCreate(const dpp::entitlement_create_t& entitlement);
+	virtual bool OnEntitlementUpdate(const dpp::entitlement_update_t& entitlement);
+	virtual bool OnEntitlementDelete(const dpp::entitlement_delete_t& entitlement);
+
 
 	/* Returns a local count */
 	uint64_t GetActiveLocalGames();

src/main.cpp

@@ -228,6 +228,18 @@ void Bot::onServerDelete(const dpp::guild_delete_t& gd) {
 	FOREACH_MOD(I_OnGuildDelete, OnGuildDelete(gd));
 }
 
+void Bot::onEntitlementDelete(const dpp::entitlement_delete_t& ed) {
+	FOREACH_MOD(I_OnEntitlementDelete, OnEntitlementDelete(ed));
+}
+
+void Bot::onEntitlementCreate(const dpp::entitlement_create_t& ed) {
+	FOREACH_MOD(I_OnEntitlementCreate, OnEntitlementCreate(ed));
+}
+
+void Bot::onEntitlementUpdate(const dpp::entitlement_update_t& ed) {
+	FOREACH_MOD(I_OnEntitlementUpdate, OnEntitlementUpdate(ed));
+}
+
 int main(int argc, char** argv) {
 
 	int dev = 0;	/* Note: getopt expects ints, this is actually treated as bool */
@@ -323,11 +335,13 @@ int main(int argc, char** argv) {
 
 		/* Set cache policy for D++ library
 		 * --------------------------------
-		 * User caching:  none
-		 * Emoji caching: none
-		 * Role caching:  none (WAS: aggressive)
+		 * User caching:    none
+		 * Emoji caching:   none
+		 * Role caching:    none (WAS: aggressive)
+		 * Channel caching: aggressive
+		 * Guild caching:   aggressive
 		*/
-		dpp::cache_policy_t cp = { dpp::cp_none, dpp::cp_none, dpp::cp_none };
+		dpp::cache_policy_t cp = { dpp::cp_none, dpp::cp_none, dpp::cp_none, dpp::cp_aggressive, dpp::cp_aggressive };
 		/* Construct cluster */
 		dpp::cluster bot(token, intents, dev ? 1 : from_string<uint32_t>(Bot::GetConfig("shardcount"), std::dec), clusterid, maxclusters, true, cp);
 
@@ -370,6 +384,9 @@ int main(int argc, char** argv) {
 		bot.on_guild_member_add(std::bind(&Bot::onMember, &client, std::placeholders::_1));
 		bot.on_guild_create(std::bind(&Bot::onServer, &client, std::placeholders::_1));
 		bot.on_guild_delete(std::bind(&Bot::onServerDelete, &client, std::placeholders::_1));
+		bot.on_entitlement_create(std::bind(&Bot::onEntitlementCreate, &client, std::placeholders::_1));
+		bot.on_entitlement_update(std::bind(&Bot::onEntitlementUpdate, &client, std::placeholders::_1));
+		bot.on_entitlement_delete(std::bind(&Bot::onEntitlementDelete, &client, std::placeholders::_1));
 
 		bot.set_websocket_protocol(dpp::ws_etf);
 

src/modules.cpp

@@ -75,6 +75,9 @@ const char* StringNames[I_END + 1] = {
 	"I_OnVoiceStateUpdate",
 	"I_OnVoiceServerUpdate",
 	"I_OnWebhooksUpdate",
+	"I_OnEntitlementCreate",
+	"I_OnEntitlementUpdate",
+	"I_OnEntitlementDelete",
 	"I_END"
 };
 
@@ -352,6 +355,21 @@ bool Module::OnPresenceUpdate()
 	return true;
 }
 
+bool Module::OnEntitlementDelete(const dpp::entitlement_delete_t& ed)
+{
+	return true;
+}
+
+bool Module::OnEntitlementCreate(const dpp::entitlement_create_t& ed)
+{
+	return true;
+}
+
+bool Module::OnEntitlementUpdate(const dpp::entitlement_update_t& ed)
+{
+	return true;
+}
+
 bool Module::OnTypingStart(const dpp::typing_start_t &obj)
 {
 	return true;