Aurora.3/code/datums/discord/bot.dm

#define CHAN_ADMIN		"channel_admin"
#define CHAN_CCIAA		"channel_cciaa"
#define CHAN_ANNOUNCE	"channel_announce"
#define CHAN_INVITE		"channel_invite"

#define SEND_OK			0
#define SEND_TIMEOUT	1
#define ERROR_PROC		2
#define ERROR_HTTP		4

var/datum/discord_bot/discord_bot = null

/hook/startup/proc/initialize_discord_bot()
	if (discord_bot)
		// This shouldn't be possible, but sure!
		return 0

	discord_bot = new()

	config.load("config/discord.txt", "discord")

	discord_bot.update_channels()

	return 1

/hook/roundstart/proc/alert_no_admins()
	var/admins_number = 0

	for (var/C in clients)
		var/client/cc = C
		if (cc.holder && (cc.holder.rights & (R_MOD|R_ADMIN)))
			admins_number++

	post_webhook_event(WEBHOOK_ROUNDSTART, list("playercount"=clients.len))
	if (!admins_number)
		post_webhook_event(WEBHOOK_ALERT_NO_ADMINS, list())
		if (!discord_bot)
			return 1
		discord_bot.send_to_admins("@here Round has started with no admins or mods online.")

	return 1

/datum/discord_bot
	var/list/channels_to_group = list()		// Group flag -> list of channel datums map.
	var/list/channels = list()				// Channel ID -> channel datum map. Will ensure that only one datum per channel ID exists.

	var/datum/discord_channel/invite = null	// The channel datum where the ingame Join Channel button will link to.

	var/active = 0
	var/auth_token = ""
	var/subscriber_role = ""

	var/robust_debug = 0

	// Lazy man's rate limiting vars
	var/list/queue = list()

	// Used to determine if BOREALIS should alert staff if the server is created
	// with world.visibility == 0.
	var/alert_visibility = 0

/*
 * Proc update_channels
 * Used to load channels from the database and construct them with the discord API.
 * Wipes all current channels and channel maps.
 *
 * @return num	- Error code. 0 upon success.
 */
/datum/discord_bot/proc/update_channels()
	if (!active)
		return 1

	if (!establish_db_connection(dbcon))
		log_debug("BOREALIS: Failed to update channels due to missing database.")
		return 2

	// Reset the channel lists.
	channels_to_group.Cut()
	channels.Cut()

	var/DBQuery/channel_query = dbcon.NewQuery("SELECT channel_group, channel_id, pin_flag, server_id FROM discord_channels")
	channel_query.Execute()

	while (channel_query.NextRow())
		// Create the channel map.
		if (isnull(channels_to_group[channel_query.item[1]]))
			channels_to_group[channel_query.item[1]] = list()

		var/datum/discord_channel/B = channels[channel_query.item[2]]

		// We don't have this channel datum yet.
		if (isnull(B))
			B = new(channel_query.item[2], channel_query.item[4], text2num(channel_query.item[3]))

			if (!B)
				log_debug("BOREALIS: Bad channel data during update channels. [jointext(channel_query.item, ", ")].")
				continue

			channels[channel_query.item[2]] = B

		if (text2num(channel_query.item[3]))
			B.pin_flag |= text2num(channel_query.item[3])

		// Add the channel to the required lists.
		channels_to_group[channel_query.item[1]] += B

	if (!isnull(channels_to_group[CHAN_INVITE]))
		invite = channels_to_group[CHAN_INVITE][1]
	else if (robust_debug)
		log_debug("BOREALIS: No invite channel designated.")

	log_debug("BOREALIS: Channels updated successfully.")
	return 0

/*
 * Proc send_message
 * Used to send a message to a specific channel group.
 *
 * @param text channel_group	- The name of the channel group which to target.
 * @param text message			- The message to send.
 */
/datum/discord_bot/proc/send_message(var/channel_group, var/message)
	if (!active || !auth_token)
		return

	if (!channel_group || !channels.len || isnull(channels_to_group[channel_group]))
		return

	if (!message)
		return

	if (length(message) > 2000)
		message = copytext(message, 1, 2001)

	// Let's run it through the proper JSON encoder, just in case of special characters.
	message = json_encode(list("content" = message))

	var/list/A = channels_to_group[channel_group]
	var/list/sent = list()
	for (var/B in A)
		var/datum/discord_channel/channel = B
		if (channel.send_message_to(auth_token, message) == SEND_TIMEOUT)
			// Whoopsies, rate limited.
			// Set up the queue.
			queue.Add(list(list(message, A - sent)))

			// Schedule a push.
			addtimer(CALLBACK(src, .proc/push_queue), 10 SECONDS, TIMER_UNIQUE)

			// And exit.
			return
		else
			sent += channel

	if (robust_debug)
		log_debug("BOREALIS: Message sent to [channel_group]. JSON body: '[message]'")

/*
 * Proc retreive_pins
 * Used to fetch a list of all pins from the designated channels.
 *
 * @return list	- A multilayered list of flags associated with pins. Structure looks like this:
 *		list("pin_flag" = list(list("author" = author name, "content" = content),
 *							   list("author" = author name, "content" = content)))
 */
/datum/discord_bot/proc/retreive_pins()
	if (!active || !auth_token)
		return list()

	if (!channels.len || isnull(channels_to_group["channel_pins"]))
		if (robust_debug)
			log_debug("BOREALIS: No pins channel group.")
		return list()

	var/list/output = list()

	if (robust_debug)
		log_debug("BOREALIS: Acquiring pins.")

	for (var/A in channels_to_group["channel_pins"])
		var/datum/discord_channel/channel = A
		if (isnull(output["[channel.pin_flag]"]))
			output["[channel.pin_flag]"] = list()

		var/ch_pins = channel.get_pins(auth_token)
		if (length(ch_pins))
			output["[channel.pin_flag]"] += ch_pins

	if (robust_debug)
		log_debug("BOREALIS: Finished acquiring pins.")

	return output

/*
 * Proc retreive_invite
 * Used to retreive the invite to the invite channel.
 * One will be created if none exist.
 *
 * @return text	- The invite URL to the designated invite channel.
 */
/datum/discord_bot/proc/retreive_invite()
	set background = 1
	if (!active || !auth_token)
		return ""

	if (!invite)
		return ""

	var/res = invite.get_invite(auth_token)
	return isnum(res) ? "" : res

/*
 * Proc send_to_admin
 * Forwards a message to the admin channels.
 */
/datum/discord_bot/proc/send_to_admins(message)
	set background = 1
	send_message(CHAN_ADMIN, message)

/*
 * Proc send_to_cciaa
 * Forwards a message to the CCIAA channels.
 */
/datum/discord_bot/proc/send_to_cciaa(message)
	set background = 1
	send_message(CHAN_CCIAA, message)

/*
 * Proc send_to_announce
 * Forwards a message to the announcements channels.
 */
/datum/discord_bot/proc/send_to_announce(message, prepend_role = 0)
	set background = 1
	if (prepend_role && subscriber_role)
		message = "<@&[subscriber_role]> " + message
	send_message(CHAN_ANNOUNCE, message)

/*
 * Proc push_queue
 * Handles the queue pushing for the bot. If there is no need to reschedule (all messages get successfully
 * pushed), then it deletes push_task and sets it back to null. Otherwise, it simply reschedules it.
 */
/datum/discord_bot/proc/push_queue()
	// What facking queue.
	if (!queue || !queue.len)
		if (robust_debug)
			log_debug("BOREALIS: Attempted to push a null length queue.")
		return

	// A[1] - message body.
	// A[2] - list of channels to send to.
	var/message
	var/list/destinations
	for (var/list/A in queue)
		message = A[1]
		destinations = A[2]

		for (var/B in destinations)
			var/datum/discord_channel/channel = B
			if (channel.send_message_to(auth_token, message) == SEND_TIMEOUT)
				addtimer(CALLBACK(src, .proc/push_queue), 10 SECONDS, TIMER_UNIQUE)

				return
			else
				destinations.Remove(channel)

		queue.Remove(A)

/**
 * Will alert the staff on Discord if the server is initialized in invisible mode.
 * Can be toggled via config.
 */
/datum/discord_bot/proc/alert_server_visibility()
	set background = 1
	if (alert_visibility && !world.visibility)
		send_to_admins("Server started as invisible!")


// A holder class for channels.
/datum/discord_channel
	var/id = ""
	var/server_id = ""
	var/pin_flag = 0
	var/invite_url = ""

/*
 * Constructor
 *
 * @param text _id	- the discord API id of the channel, as a string.
 * @param text _sid	- the discord API server id for the channel, as a string.
 * @param num _pin	- the bitflags of admin permissions which have access to the pins from this channel.
 */
/datum/discord_channel/New(var/_id, var/_sid, var/_pin)
	id = _id
	server_id = _sid

	if (_pin)
		pin_flag = _pin

/*
 * Proc send_message_to
 * Sends a message to the channel.
 *
 * @param text token	- the authorization token to be used for the requests.
 * @param text message	- the sanitized message content to be sent.
 * @return num			- a specific return code for the discord_bot to handle.
 */
/datum/discord_channel/proc/send_message_to(var/token, var/message)
	if (!token || !message)
		return ERROR_PROC

	var/datum/http_request/req = http_create_post("https://discordapp.com/api/channels/[id]/messages", message, list("Authorization" = "Bot [token]", "Content-Type" = "application/json"))

	req.begin_async()
	UNTIL(req.is_complete())

	var/datum/http_response/res = req.into_response()

	if (res.errored)
		log_debug("BOREALIS: library error during HTTP query. [res.error]")
		return ERROR_PROC

	switch (res.status_code)
		if (200)
			return SEND_OK

		if (429)
			return SEND_TIMEOUT

		else
			log_debug("BOREALIS: HTTP error while forwarding message to Discord API: [res]. Channel: [id]. Message body: [message].")
			return ERROR_HTTP

/*
 * Proc get_pins
 * Retreives a list of pinned messages from the channel.
 *
 * @param text token	- the authorization token to be used for the requests.
 * @return mixed		- 2d array of content upon success, in format: list(list("author" = text, "content" = text)).
 *						  Num upon failure.
 */
/datum/discord_channel/proc/get_pins(var/token)
	var/datum/http_request/req = http_create_get("https://discordapp.com/api/channels/[id]/pins", headers = list("Authorization" = "Bot [token]"))

	req.begin_async()
	UNTIL(req.is_complete())

	var/datum/http_response/res = req.into_response()

	if (res.errored)
		log_debug("BOREALIS: Proc error while fetching pins: [res.error]")
		return
	else if (res.status_code != 200)
		log_debug("BOREALIS: HTTP error while fetching pins: [res.status_code].")
		return
	else
		var/list/A = json_decode(res.body)
		var/list/pinned_messages = list()

		// Begin processing the list structure delivered back to us from send_get_request.
		for (var/list/B in A)
			var/content = B["content"]

			// We have mentions to take care of.
			if (!isnull(B["mentions"]))
				var/list/mentions = B["mentions"]

				for (var/list/C in mentions)
					content = replacetextEx(content, "<@[C["id"]]>", C["username"])

			// Role mentions are up next.
			if (!isnull(B["mention_roles"]))
				var/list/mentions = B["mention_roles"]

				for (var/C in mentions)
					content = replacetextEx(content, "<@&[C]>", "@SomeRole")

			pinned_messages += list(list("author" = B["author"]["username"], "content" = content))

		return pinned_messages

/*
 * Proc get_invite
 * Retreives (or creates if none found) an invite URL to the specific channel.
 *
 * @param text token	- the authorization token to be used for the requests.
 * @return mixed		- text upon success, the link to the invite; num upon failure.
 */
/datum/discord_channel/proc/get_invite(var/token)
	if (invite_url)
		return invite_url

	var/datum/http_request/req = http_create_get("https://discordapp.com/api/channels/[id]/invites", headers = list("Authorization" = "Bot [token]"))

	req.begin_async()
	UNTIL(req.is_complete())

	var/datum/http_response/res = req.into_response()

	if (res.errored)
		log_debug("BOREALIS: Proc error while fetching invite: [res.error]")
		return
	else if (res.status_code != 200)
		log_debug("BOREALIS: HTTP error while fetching invite: [res.status_code].")
		return
	else
		var/list/A = json_decode(res.body)

		// No length to return data, but a valid 200 return header.
		// So we simply have no invites active. Make one!
		if (!A?.len)
			return create_invite(token)

		var/best_age = A[1]["max_age"]
		var/code = A[1]["code"]

		// Find the best invite.
		// Why? Because round time expires are lame, I guess.
		if (best_age != 0)
			for (var/i = 2, i < A.len, i++)
				if (A[i]["max_age"] == 0)
					code = A[i]["code"]
					break

				if (best_age < A[i]["max_age"])
					best_age = A[i]["max_age"]
					code = A[i]["code"]

		// Sanity check for debug I guess.
		if (!code)
			log_debug("BOREALIS: Retreived an empty invite. This should not happen. Response object: [json_encode(A)]")
			return

		// Save the URL for later retreival.
		invite_url = "https://discord.gg/[code]"
		return invite_url

/*
 * Proc create_invite
 * Creates a permanent invite to a channel and returns it, under the assumption that
 * there are no other invites for this channel.
 *
 * @param text token	- the authorization token to be used for the requests.
 * @return mixed		- String upon success - the URL of the newly generated invite.
 *						  Num upon failure.
 */
/datum/discord_channel/proc/create_invite(var/token)
	var/data = list("max_age" = 0, "max_uses" = 0)
	var/datum/http_request/req = http_create_post("https://discordapp.com/api/channels/[id]/invites", json_encode(data), list("Authorization" = "Bot [token]", "Content-Type" = "application/json"))

	req.begin_async()
	UNTIL(req.is_complete())

	var/datum/http_response/res = req.into_response()

	if (!res.errored && res.status_code == 200)
		var/list/r_data = json_decode(res.body)

		// The first index should now exist. So we just use that!
		invite_url = "https://discord.gg/[r_data["code"]]"
		return invite_url

/proc/discord_escape(var/input, var/remove_everyone = TRUE, var/remove_mentions = FALSE)
	. = replace_characters(input, list("`" = "\\`", "*" = "\\*", "_" = "\\_", "~" = "\\~"))

	if(remove_everyone)
		var/regex/evReg = regex("@everyone", "gi")
		. = evReg.Replace(., "@ everyone")
		var/regex/hereReg = regex("@here", "gi")
		. = hereReg.Replace(., "@ here")

	if(remove_mentions)
		var/regex/menReg = regex("<@\[\\d]+>", "g")
		. = menReg.Replace(., "\[mention]")
		var/regex/roleReg = regex("<@&\[\\d]+>", "g")
		. = roleReg.Replace(., "\[role mention]")

#undef CHAN_ADMIN
#undef CHAN_CCIAA
#undef CHAN_ANNOUNCE
#undef CHAN_INVITE

#undef SEND_OK
#undef SEND_TIMEOUT
#undef ERROR_PROC
#undef ERROR_HTTP