Add heatsync and support hot reloading

This commit is contained in:
Papa 2023-03-05 09:05:35 -07:00
commit 11e5cd7f77
7 changed files with 155 additions and 106 deletions

View file

@ -1,7 +1,11 @@
const { SnowTransfer } = require("snowtransfer")
const { Client: CloudStorm } = require("cloudstorm")
let wasReadyBefore = false
const passthrough = require("../passthrough")
const { sync } = passthrough
/** @type {typeof import("./DiscordUtils")} */
const dUtils = sync.require("./DiscordUtils")
class DiscordClient {
/**
@ -24,82 +28,21 @@ class DiscordClient {
encoding: "json"
}
})
/** @type {import("discord-typings").User} */
this.ready = false
/** @type {import("discord-api-types/v10").APIUser} */
// @ts-ignore avoid setting as or null because we know we need to wait for ready anyways
this.user = null
/** @type {import("discord-typings").Application} */
/** @type {Pick<import("discord-api-types/v10").APIApplication, "id" | "flags">} */
// @ts-ignore
this.application = null
/** @type {Map<string, import("discord-typings").Channel>} */
/** @type {Map<string, import("discord-api-types/v10").APIChannel>} */
this.channels = new Map()
/** @type {Map<string, import("discord-typings").Guild>} */
/** @type {Map<string, import("discord-api-types/v10").APIGuild>} */
this.guilds = new Map()
/**
* @type {Map<string, Array<string>>}
* @private
*/
/** @type {Map<string, Array<string>>} */
this.guildChannelMap = new Map()
this.cloud.on("event", this.onPacket.bind(this))
}
/**
* @param {import("cloudstorm").IGatewayMessage} message
* @private
*/
onPacket(message) {
if (message.t === "READY") {
if (wasReadyBefore) return
wasReadyBefore = true
/** @type {import("discord-typings").ReadyPayload} */
const typed = message.d
this.user = typed.user
this.application = typed.application
console.log(`Discord logged in as ${this.user.username}#${this.user.discriminator} (${this.user.id})`)
} else if (message.t === "GUILD_CREATE") {
/** @type {import("discord-typings").Guild} */
const typed = message.d
this.guilds.set(typed.id, typed)
const arr = []
this.guildChannelMap.set(typed.id, arr)
for (const channel of typed.channels || []) {
arr.push(channel.id)
this.channels.set(channel.id, channel)
}
} else if (message.t === "GUILD_DELETE") {
/** @type {import("discord-typings").Guild} */
const typed = message.d
this.guilds.delete(typed.id)
const channels = this.guildChannelMap.get(typed.id)
if (channels) {
for (const id of channels) this.channels.delete(id)
}
this.guildChannelMap.delete(typed.id)
} else if (message.t === "CHANNEL_CREATE" || message.t === "CHANNEL_DELETE") {
/** @type {import("discord-typings").Channel} */
const typed = message.d
if (message.t === "CHANNEL_CREATE") {
this.channels.set(typed.id, typed)
if (typed["guild_id"]) { // obj[prop] notation can be used to access a property without typescript complaining that it doesn't exist on all values something can have
const channels = this.guildChannelMap.get(typed["guild_id"])
if (channels && !channels.includes(typed.id)) channels.push(typed.id)
}
} else {
this.channels.delete(typed.id)
if (typed["guild_id"]) {
const channels = this.guildChannelMap.get(typed["guild_id"])
if (channels) {
const previous = channels.indexOf(typed.id)
if (previous !== -1) channels.splice(previous, 1)
}
}
}
}
this.cloud.on("event", message => dUtils.onPacket(this, message))
this.cloud.on("error", console.error)
}
}

View file

@ -2,10 +2,11 @@ module.exports = {
/**
* Process Discord messages and convert to a message Matrix can understand
*
* @param {import("discord-typings").Message} message
* @param {import("./DiscordClient")} client
* @param {import("discord-api-types/v10").APIMessage} message
* @returns {import("../types").MatrixMessage}
*/
onMessageCreate: message => {
onMessageCreate(client, message) {
return {}
}
}

63
modules/DiscordUtils.js Normal file
View file

@ -0,0 +1,63 @@
const passthrough = require("../passthrough")
const { sync } = passthrough
/** @type {typeof import("./DiscordEvents")} */
const DiscordEvents = sync.require("./DiscordEvents")
const utils = {
/**
* @param {import("./DiscordClient")} client
* @param {import("cloudstorm").IGatewayMessage} message
*/
onPacket(client, message) {
if (message.t === "READY") {
if (client.ready) return
client.ready = true
client.user = message.d.user
client.application = message.d.application
console.log(`Discord logged in as ${client.user.username}#${client.user.discriminator} (${client.user.id})`)
} else if (message.t === "GUILD_CREATE") {
client.guilds.set(message.d.id, message.d)
const arr = []
client.guildChannelMap.set(message.d.id, arr)
for (const channel of message.d.channels || []) {
arr.push(channel.id)
client.channels.set(channel.id, channel)
}
} else if (message.t === "GUILD_DELETE") {
client.guilds.delete(message.d.id)
const channels = client.guildChannelMap.get(message.d.id)
if (channels) {
for (const id of channels) client.channels.delete(id)
}
client.guildChannelMap.delete(message.d.id)
} else if (message.t === "CHANNEL_CREATE" || message.t === "CHANNEL_DELETE") {
if (message.t === "CHANNEL_CREATE") {
client.channels.set(message.d.id, message.d)
if (message.d["guild_id"]) { // obj[prop] notation can be used to access a property without typescript complaining that it doesn't exist on all values something can have
const channels = client.guildChannelMap.get(message.d["guild_id"])
if (channels && !channels.includes(message.d.id)) channels.push(message.d.id)
}
} else {
client.channels.delete(message.d.id)
if (message.d["guild_id"]) {
const channels = client.guildChannelMap.get(message.d["guild_id"])
if (channels) {
const previous = channels.indexOf(message.d.id)
if (previous !== -1) channels.splice(previous, 1)
}
}
}
} else if (message.t === "MESSAGE_CREATE") DiscordEvents.onMessageCreate(client, message.d)
}
}
module.exports = utils