seyfert/packages/core/src/adapters/events.ts

import type {
	DiscordAutoModerationActionExecution,
	DiscordAutoModerationRule,
	DiscordChannel,
	DiscordChannelPinsUpdate,
	DiscordEmoji,
	DiscordGuild,
	DiscordGuildBanAddRemove,
	DiscordGuildEmojisUpdate,
	DiscordGuildMemberAdd,
	DiscordGuildMemberRemove,
	DiscordGuildMemberUpdate,
    DiscordGuildMembersChunk,
	DiscordGuildRoleCreate,
	DiscordGuildRoleDelete,
	DiscordGuildRoleUpdate,
	DiscordIntegration,
	DiscordIntegrationDelete,
	DiscordInteraction,
	DiscordInviteCreate,
	DiscordInviteDelete,
	DiscordMemberWithUser,
	DiscordMessage,
	DiscordMessageDelete,
	DiscordMessageReactionAdd,
	DiscordMessageReactionRemove,
	DiscordMessageReactionRemoveAll,
	DiscordMessageReactionRemoveEmoji,
	DiscordPresenceUpdate,
	DiscordReady,
	DiscordRole,
	DiscordScheduledEvent,
	DiscordScheduledEventUserAdd,
	DiscordScheduledEventUserRemove,
	DiscordThreadListSync,
	DiscordThreadMembersUpdate,
	DiscordThreadMemberUpdate,
	DiscordTypingStart,
	DiscordUser,
	DiscordWebhookUpdate,
    DiscordVoiceState,
    DiscordVoiceServerUpdate,
} from '@biscuitland/api-types';

import type { Session } from '../biscuit';
import type { Interaction } from '../structures/interactions';
import type { Snowflake } from '../snowflakes';

import {
	AutoModerationRule,
	AutoModerationExecution,
} from '../structures/automod';

import type { Channel } from '../structures/channels';
import {
	ChannelFactory,
	GuildChannel,
	ThreadChannel,
} from '../structures/channels';

import type { DiscordStageInstanceB } from '../structures/stage-instance';
import { StageInstance } from '../structures/stage-instance';
import { ScheduledEvent } from '../structures/scheduled-events';
import { Presence } from '../structures/presence';

import { Member, ThreadMember } from '../structures/members';

import { Message } from '../structures/message';
import { User } from '../structures/user';
import { Integration } from '../structures/integration';

import { Guild } from '../structures/guilds';
import { InteractionFactory } from '../structures/interactions';
import type { InviteCreate } from '../structures/invite';
import { NewInviteCreate } from '../structures/invite';

import type {
	MessageReactionAdd,
	MessageReactionRemove,
	MessageReactionRemoveAll,
	MessageReactionRemoveEmoji,
} from '../structures/message-reaction';

import { NewMessageReactionAdd } from '../structures/message-reaction';
import { Util, PartialMessage } from '../utils/util';

export type RawHandler<T> = (...args: [Session, number, T]) => void;
export type Handler<T extends [obj?: unknown, ddy?: unknown]> = (
	...args: T
) => unknown;

export const READY: RawHandler<DiscordReady> = (session, shardId, payload) => {
	session.applicationId = payload.application.id;
	session.botId = payload.user.id;
	session.events.emit(
		'ready',
		{ ...payload, user: new User(session, payload.user) },
		shardId
	);
};

export const MESSAGE_CREATE: RawHandler<DiscordMessage> = (
	session,
	_shardId,
	message
) => {
	session.events.emit('messageCreate', new Message(session, message));
};

export const MESSAGE_UPDATE: RawHandler<PartialMessage> = (
	session,
	_shardId,
	new_message
) => {
	// message is partial
	if (Util.isPartialMessage(new_message)) {
		const message = {
			session,
			id: new_message.id,
			guildId: new_message.guild_id,
			channelId: new_message.channel_id,
            fields: new_message,
		}; //satisfies Partial<Message>;

		// all methods of Message can run on partial messages
		// we aknowledge people that their callback could be partial but giving them all functions of Message
		Object.setPrototypeOf(message, Message.prototype);

		session.events.emit('messageUpdate', message);
		return;
	}

    if (Util.isFullMessage(new_message)) {
	    session.events.emit('messageUpdate', {
            ...new Message(session, new_message),
            fields: {} as PartialMessage
        });
    }
};

export const MESSAGE_DELETE: RawHandler<DiscordMessageDelete> = (
	session,
	_shardId,
	{ id, channel_id, guild_id }
) => {
	session.events.emit('messageDelete', {
		id,
		channelId: channel_id,
		guildId: guild_id,
	});
};

export const GUILD_CREATE: RawHandler<DiscordGuild> = (
	session,
	_shardId,
	guild
) => {
	session.events.emit('guildCreate', new Guild(session, guild));
};

export const GUILD_DELETE: RawHandler<DiscordGuild> = (
	session,
	_shardId,
	guild
) => {
	session.events.emit('guildDelete', { id: guild.id, unavailable: true });
};

export const GUILD_MEMBER_ADD: RawHandler<DiscordGuildMemberAdd> = (
	session,
	_shardId,
	member
) => {
	session.events.emit(
		'guildMemberAdd',
		new Member(session, member, member.guild_id)
	);
};

export const GUILD_MEMBER_UPDATE: RawHandler<DiscordGuildMemberUpdate> = (
	session,
	_shardId,
	member
) => {
	session.events.emit(
		'guildMemberUpdate',
		new Member(session, member, member.guild_id)
	);
};

export const GUILD_MEMBER_REMOVE: RawHandler<DiscordGuildMemberRemove> = (
	session,
	_shardId,
	member
) => {
	session.events.emit(
		'guildMemberRemove',
		new User(session, member.user),
		member.guild_id
	);
};

export const GUILD_BAN_ADD: RawHandler<DiscordGuildBanAddRemove> = (
	session,
	_shardId,
	data
) => {
	session.events.emit('guildBanAdd', {
		guildId: data.guild_id,
		user: data.user,
	});
};

export const GUILD_BAN_REMOVE: RawHandler<DiscordGuildBanAddRemove> = (
	session,
	_shardId,
	data
) => {
	session.events.emit('guildBanRemove', {
		guildId: data.guild_id,
		user: data.user,
	});
};

export const GUILD_EMOJIS_UPDATE: RawHandler<DiscordGuildEmojisUpdate> = (
	session,
	_shardId,
	data
) => {
	session.events.emit('guildEmojisUpdate', {
		guildId: data.guild_id,
		emojis: data.emojis,
	});
};

export const GUILD_ROLE_CREATE: RawHandler<DiscordGuildRoleCreate> = (
	session,
	_shardId,
	data
) => {
	session.events.emit('guildRoleCreate', {
		guildId: data.guild_id,
		role: data.role,
	});
};

export const GUILD_ROLE_UPDATE: RawHandler<DiscordGuildRoleUpdate> = (
	session,
	_shardId,
	data
) => {
	session.events.emit('guildRoleUpdate', {
		guildId: data.guild_id,
		role: data.role,
	});
};

export const GUILD_ROLE_DELETE: RawHandler<DiscordGuildRoleDelete> = (
	session,
	_shardId,
	data
) => {
	session.events.emit('guildRoleDelete', {
		guildId: data.guild_id,
		roleId: data.role_id,
	});
};

export const TYPING_START: RawHandler<DiscordTypingStart> = (
	session,
	_shardId,
	payload
) => {
	session.events.emit('typingStart', {
		channelId: payload.channel_id,
		guildId: payload.guild_id ? payload.guild_id : undefined,
		userId: payload.user_id,
		timestamp: payload.timestamp,
		member: payload.guild_id
			? new Member(
					session,
					payload.member as DiscordMemberWithUser,
					payload.guild_id
			)
			: undefined,
	});
};

export const INTERACTION_CREATE: RawHandler<DiscordInteraction> = (
	session,
	_shardId,
	interaction
) => {
	session.events.emit(
		'interactionCreate',
		InteractionFactory.from(session, interaction)
	);
};

export const CHANNEL_CREATE: RawHandler<DiscordChannel> = (
	session,
	_shardId,
	channel
) => {
	session.events.emit('channelCreate', ChannelFactory.from(session, channel));
};

export const CHANNEL_UPDATE: RawHandler<DiscordChannel> = (
	session,
	_shardId,
	channel
) => {
	session.events.emit('channelUpdate', ChannelFactory.from(session, channel));
};

export const CHANNEL_DELETE: RawHandler<DiscordChannel> = (
	session,
	_shardId,
	channel
) => {
	if (!channel.guild_id) {
		return;
	}

	session.events.emit(
		'channelDelete',
		new GuildChannel(session, channel, channel.guild_id)
	);
};

export const THREAD_CREATE: RawHandler<DiscordChannel> = (
	session,
	_shardId,
	channel
) => {
	if (!channel.guild_id) {
		return;
	}

	session.events.emit(
		'threadCreate',
		new ThreadChannel(session, channel, channel.guild_id)
	);
};

export const THREAD_UPDATE: RawHandler<DiscordChannel> = (
	session,
	_shardId,
	channel
) => {
	if (!channel.guild_id) {
		return;
	}

	session.events.emit(
		'threadUpdate',
		new ThreadChannel(session, channel, channel.guild_id)
	);
};

export const THREAD_DELETE: RawHandler<DiscordChannel> = (
	session,
	_shardId,
	channel
) => {
	if (!channel.guild_id) {
		return;
	}

	session.events.emit(
		'threadDelete',
		new ThreadChannel(session, channel, channel.guild_id)
	);
};

export const THREAD_MEMBER_UPDATE: RawHandler<DiscordThreadMemberUpdate> = (
	session,
	_shardId,
	payload
) => {
	session.events.emit('threadMemberUpdate', {
		guildId: payload.guild_id,
		id: payload.id,
		userId: payload.user_id,
		joinedAt: payload.joined_at,
		flags: payload.flags,
	});
};

export const THREAD_MEMBERS_UPDATE: RawHandler<DiscordThreadMembersUpdate> = (
	session,
	_shardId,
	payload
) => {
	session.events.emit('threadMembersUpdate', {
		memberCount: payload.member_count,
		addedMembers: payload.added_members
			? payload.added_members.map(tm => new ThreadMember(session, tm))
			: undefined,
		removedMemberIds: payload.removed_member_ids
			? payload.removed_member_ids
			: undefined,
		guildId: payload.guild_id,
		id: payload.id,
	});
};

export const THREAD_LIST_SYNC: RawHandler<DiscordThreadListSync> = (
	session,
	_shardId,
	payload
) => {
	session.events.emit('threadListSync', {
		guildId: payload.guild_id,
		channelIds: payload.channel_ids ?? [],
		threads: payload.threads.map(
			channel => new ThreadChannel(session, channel, payload.guild_id)
		),
		members: payload.members.map(
			member => new ThreadMember(session, member)
		),
	});
};

export const CHANNEL_PINS_UPDATE: RawHandler<DiscordChannelPinsUpdate> = (
	session,
	_shardId,
	payload
) => {
	session.events.emit('channelPinsUpdate', {
		guildId: payload.guild_id,
		channelId: payload.channel_id,
		lastPinTimestamp: payload.last_pin_timestamp
			? Date.parse(payload.last_pin_timestamp)
			: undefined,
	});
};

export const USER_UPDATE: RawHandler<DiscordUser> = (
	session,
	_shardId,
	payload
) => {
	session.events.emit('userUpdate', new User(session, payload));
};

export const PRESENCE_UPDATE: RawHandler<DiscordPresenceUpdate> = (
	session,
	_shardId,
	payload
) => {
	session.events.emit('presenceUpdate', new Presence(session, payload));
};

export const WEBHOOKS_UPDATE: RawHandler<DiscordWebhookUpdate> = (
	session,
	_shardId,
	webhook
) => {
	session.events.emit('webhooksUpdate', {
		guildId: webhook.guild_id,
		channelId: webhook.channel_id,
	});
};

export const INTEGRATION_CREATE: RawHandler<
	DiscordIntegration & { guildId?: Snowflake }
> = (session, _shardId, payload) => {
	session.events.emit('integrationCreate', new Integration(session, payload));
};

export const INTEGRATION_UPDATE: RawHandler<
	DiscordIntegration & { guildId?: Snowflake }
> = (session, _shardId, payload) => {
	session.events.emit('integrationCreate', new Integration(session, payload));
};

export const INTEGRATION_DELETE: RawHandler<DiscordIntegrationDelete> = (
	session,
	_shardId,
	payload
) => {
	session.events.emit('integrationDelete', {
		id: payload.id,
		guildId: payload.guild_id,
		applicationId: payload.application_id,
	});
};

export const AUTO_MODERATION_RULE_CREATE: RawHandler<
	DiscordAutoModerationRule
> = (session, _shardId, payload) => {
	session.events.emit(
		'autoModerationRuleCreate',
		new AutoModerationRule(session, payload)
	);
};

export const AUTO_MODERATION_RULE_UPDATE: RawHandler<
	DiscordAutoModerationRule
> = (session, _shardId, payload) => {
	session.events.emit(
		'autoModerationRuleUpdate',
		new AutoModerationRule(session, payload)
	);
};

export const AUTO_MODERATION_RULE_DELETE: RawHandler<
	DiscordAutoModerationRule
> = (session, _shardId, payload) => {
	session.events.emit(
		'autoModerationRuleDelete',
		new AutoModerationRule(session, payload)
	);
};

export const AUTO_MODERATION_ACTION_EXECUTE: RawHandler<
	DiscordAutoModerationActionExecution
> = (session, _shardId, payload) => {
	session.events.emit(
		'autoModerationActionExecution',
		new AutoModerationExecution(session, payload)
	);
};

export const MESSAGE_REACTION_ADD: RawHandler<DiscordMessageReactionAdd> = (
	session,
	_shardId,
	reaction
) => {
	session.events.emit(
		'messageReactionAdd',
		NewMessageReactionAdd(session, reaction)
	);
};

export const MESSAGE_REACTION_REMOVE: RawHandler<
	DiscordMessageReactionRemove
> = (session, _shardId, reaction) => {
	session.events.emit(
		'messageReactionRemove',
		NewMessageReactionAdd(session, reaction)
	);
};

export const MESSAGE_REACTION_REMOVE_ALL: RawHandler<
	DiscordMessageReactionRemoveAll
> = (session, _shardId, reaction) => {
	session.events.emit(
		'messageReactionRemoveAll',
		NewMessageReactionAdd(session, reaction as DiscordMessageReactionAdd)
	);
};

export const MESSAGE_REACTION_REMOVE_EMOJI: RawHandler<
	DiscordMessageReactionRemoveEmoji
> = (session, _shardId, reaction) => {
	session.events.emit(
		'messageReactionRemoveEmoji',
		NewMessageReactionAdd(session, reaction as DiscordMessageReactionAdd)
	);
};

export const INVITE_CREATE: RawHandler<DiscordInviteCreate> = (
	session,
	_shardId,
	invite
) => {
	session.events.emit('inviteCreate', NewInviteCreate(session, invite));
};

export const INVITE_DELETE: RawHandler<DiscordInviteDelete> = (
	session,
	_shardId,
	data
) => {
	session.events.emit('inviteDelete', {
		channelId: data.channel_id,
		guildId: data.guild_id,
		code: data.code,
	});
};

export const STAGE_INSTANCE_CREATE: RawHandler<DiscordStageInstanceB> = (
	session,
	_shardId,
	payload
) => {
	session.events.emit(
		'stageInstanceCreate',
		new StageInstance(session, payload)
	);
};

export const STAGE_INSTANCE_UPDATE: RawHandler<DiscordStageInstanceB> = (
	session,
	_shardId,
	payload
) => {
	session.events.emit(
		'stageInstanceUpdate',
		new StageInstance(session, payload)
	);
};

export const STAGE_INSTANCE_DELETE: RawHandler<DiscordStageInstanceB> = (
	session,
	_shardId,
	payload
) => {
	session.events.emit(
		'stageInstanceDelete',
		new StageInstance(session, payload)
	);
};

export const GUILD_SCHEDULED_EVENT_CREATE: RawHandler<DiscordScheduledEvent> = (
	session,
	_shardId,
	payload
) => {
	session.events.emit(
		'guildScheduledEventCreate',
		new ScheduledEvent(session, payload)
	);
};

export const GUILD_SCHEDULED_EVENT_UPDATE: RawHandler<DiscordScheduledEvent> = (
	session,
	_shardId,
	payload
) => {
	session.events.emit(
		'guildScheduledEventUpdate',
		new ScheduledEvent(session, payload)
	);
};

export const GUILD_SCHEDULED_EVENT_DELETE: RawHandler<DiscordScheduledEvent> = (
	session,
	_shardId,
	payload
) => {
	session.events.emit(
		'guildScheduledEventDelete',
		new ScheduledEvent(session, payload)
	);
};

export const GUILD_SCHEDULED_EVENT_USER_ADD: RawHandler<
	DiscordScheduledEventUserAdd
> = (session, _shardId, payload) => {
	session.events.emit('guildScheduledEventUserAdd', {
		scheduledEventId: payload.guild_scheduled_event_id,
		userId: payload.user_id,
		guildId: payload.guild_id,
	});
};

export const GUILD_SCHEDULED_EVENT_USER_REMOVE: RawHandler<
	DiscordScheduledEventUserRemove
> = (session, _shardId, payload) => {
	session.events.emit('guildScheduledEventUserRemove', {
		scheduledEventId: payload.guild_scheduled_event_id,
		userId: payload.user_id,
		guildId: payload.guild_id,
	});
};

export const VOICE_STATE_UPDATE: RawHandler<DiscordVoiceState> = (session, _shardId, payload) => {
    if (!payload.guild_id) { return; }
    session.events.emit('voiceStateUpdate', payload);
};

export const VOICE_SERVER_UPDATE: RawHandler<DiscordVoiceServerUpdate> = (session, _shardId, payload) => {
    session.events.emit('voiceServerUpdate', {
        token: payload.token,
        guildId: payload.guild_id,
        endpoint: payload.endpoint ? payload.endpoint : undefined
    });
};

export const GUILD_MEMBERS_CHUNK: RawHandler<DiscordGuildMembersChunk> = (session, _shardId, payload) => {
    session.events.emit('guildMembersChunk', {
        guildId: payload.guild_id,
        members: new Map(payload.members.map(m => [m.user.id, new Member(session, m, payload.guild_id)])),
        chunkIndex: payload.chunk_index,
        chunkCount: payload.chunk_count,
        notFound: payload.not_found,
        presences: payload.presences?.map(p => new Presence(session, p)) ?? [],
    });
};

export const raw: RawHandler<unknown> = (session, shardId, data) => {
	session.events.emit('raw', data as { t: string; d: unknown }, shardId);
};

export interface Ready extends Omit<DiscordReady, 'user'> {
	user: User;
}

export interface GuildMembersChunk {
    guildId: Snowflake;
    members: Map<Snowflake, Member>;
    chunkIndex: number;
    chunkCount: number;
    notFound?: string[];
    presences: Presence[];
    nonce?: string;
}

/**
 * Alias of `keyof Events`. List of all events that can be emitted by the session.
 * Ex: ready, messageCreate, guildCreate, etc.
 * @see {@link Events}
*/
export type AllEvents = keyof Events;

export interface Events {
	ready: Handler<[Ready, number]>;
	messageCreate: Handler<[Message]>;
	messageUpdate: Handler<[Partial<Message> & { fields: PartialMessage }]>;
	messageDelete: Handler<
		[{ id: Snowflake; channelId: Snowflake; guildId?: Snowflake }]
	>;
	messageReactionAdd: Handler<[MessageReactionAdd]>;
	messageReactionRemove: Handler<[MessageReactionRemove]>;
	messageReactionRemoveAll: Handler<[MessageReactionRemoveAll]>;
	messageReactionRemoveEmoji: Handler<[MessageReactionRemoveEmoji]>;
	guildCreate: Handler<[Guild]>;
	guildDelete: Handler<[{ id: Snowflake; unavailable: boolean }]>;
	guildMemberAdd: Handler<[Member]>;
	guildMemberUpdate: Handler<[Member]>;
	guildMemberRemove: Handler<[User, Snowflake]>;
    guildMembersChunk: Handler<[GuildMembersChunk]>;
	guildBanAdd: Handler<[{ guildId: Snowflake; user: DiscordUser }]>;
	guildBanRemove: Handler<[{ guildId: Snowflake; user: DiscordUser }]>;
	guildEmojisUpdate: Handler<
		[{ guildId: Snowflake; emojis: DiscordEmoji[] }]
	>;
	guildRoleCreate: Handler<[{ guildId: Snowflake; role: DiscordRole }]>;
	guildRoleUpdate: Handler<[{ guildId: Snowflake; role: DiscordRole }]>;
	guildRoleDelete: Handler<[{ guildId: Snowflake; roleId: Snowflake }]>;
	typingStart: Handler<
		[
			{
				channelId: Snowflake;
				guildId?: Snowflake;
				userId: Snowflake;
				timestamp: number;
				member?: Member;
			}
		]
	>;
	channelCreate: Handler<[Channel]>;
	channelUpdate: Handler<[Channel]>;
	channelDelete: Handler<[GuildChannel]>;
	channelPinsUpdate: Handler<
		[
			{
				guildId?: Snowflake;
				channelId: Snowflake;
				lastPinTimestamp?: number;
			}
		]
	>;
	threadCreate: Handler<[ThreadChannel]>;
	threadUpdate: Handler<[ThreadChannel]>;
	threadDelete: Handler<[ThreadChannel]>;
	threadListSync: Handler<
		[
			{
				guildId: Snowflake;
				channelIds: Snowflake[];
				threads: ThreadChannel[];
				members: ThreadMember[];
			}
		]
	>;
	threadMemberUpdate: Handler<
		[
			{
				id: Snowflake;
				userId: Snowflake;
				guildId: Snowflake;
				joinedAt: string;
				flags: number;
			}
		]
	>;
	threadMembersUpdate: Handler<
		[
			{
				id: Snowflake;
				memberCount: number;
				addedMembers?: ThreadMember[];
				guildId: Snowflake;
				removedMemberIds?: Snowflake[];
			}
		]
	>;
	interactionCreate: Handler<[Interaction]>;
	integrationCreate: Handler<[Integration]>;
	integrationUpdate: Handler<[Integration]>;
	integrationDelete: Handler<
		[{ id: Snowflake; guildId?: Snowflake; applicationId?: Snowflake }]
	>;
	inviteCreate: Handler<[InviteCreate]>;
	inviteDelete: Handler<
		[{ channelId: string; guildId?: string; code: string }]
	>;
	autoModerationRuleCreate: Handler<[AutoModerationRule]>;
	autoModerationRuleUpdate: Handler<[AutoModerationRule]>;
	autoModerationRuleDelete: Handler<[AutoModerationRule]>;
	autoModerationActionExecution: Handler<[AutoModerationExecution]>;
	stageInstanceCreate: Handler<[StageInstance]>;
	stageInstanceUpdate: Handler<[StageInstance]>;
	stageInstanceDelete: Handler<[StageInstance]>;
	guildScheduledEventCreate: Handler<[ScheduledEvent]>;
	guildScheduledEventUpdate: Handler<[ScheduledEvent]>;
	guildScheduledEventDelete: Handler<[ScheduledEvent]>;
	guildScheduledEventUserAdd: Handler<
		[{ scheduledEventId: Snowflake; userId: Snowflake; guildId: Snowflake }]
	>;
	guildScheduledEventUserRemove: Handler<
		[{ scheduledEventId: Snowflake; userId: Snowflake; guildId: Snowflake }]
	>;
	raw: Handler<[{ t: string; d: unknown }, number]>;
	webhooksUpdate: Handler<[{ guildId: Snowflake; channelId: Snowflake }]>;
	userUpdate: Handler<[User]>;
	presenceUpdate: Handler<[Presence]>;
	debug: Handler<[string]>;
    voiceStateUpdate: Handler<[DiscordVoiceState]>;
    voiceServerUpdate: Handler<
        [{ token: string; guildId: Snowflake; endpoint?: string }]
    >;
}