Files
fluxer/fluxer_api/src/api/guild/services/member/GuildMemberOperationsService.ts
T

985 lines
35 KiB
TypeScript

// SPDX-License-Identifier: AGPL-3.0-or-later
import {requireEmailVerified} from '@app/api/auth/EmailVerificationUtils';
import {
createChannelID,
createRoleID,
type GuildID,
type InviteCode,
type RoleID,
type UserID,
} from '@app/api/BrandedTypes';
import type {ChannelService} from '@app/api/channel/services/ChannelService';
import {assertMutableUserId} from '@app/api/constants/Core';
import type {GuildMemberRow} from '@app/api/database/types/GuildTypes';
import type {GuildAuditLogService} from '@app/api/guild/GuildAuditLogService';
import {resolveMaxGuildMembersLimit} from '@app/api/guild/GuildMemberLimitUtils';
import {mapGuildMemberToResponse} from '@app/api/guild/GuildModel';
import type {GuildRepository} from '@app/api/guild/repositories/GuildRepository';
import {assertNotBannedFromGuild} from '@app/api/guild/services/GuildBanChecks';
import {getGuildAuth} from '@app/api/guild/services/GuildMfaEnforcement';
import type {GuildMemberEventService} from '@app/api/guild/services/member/GuildMemberEventService';
import type {GuildMemberSearchIndexService} from '@app/api/guild/services/member/GuildMemberSearchIndexService';
import type {GuildMemberValidationService} from '@app/api/guild/services/member/GuildMemberValidationService';
import {emitActivity} from '@app/api/infrastructure/activity/ActivityEvents';
import type {JoinSource} from '@app/api/infrastructure/activity/Contract.generated';
import {emitGuildMemberRemoved} from '@app/api/infrastructure/activity/GuildEvents';
import {contentModerationService} from '@app/api/infrastructure/ContentModerationService';
import type {EntityAssetService, PreparedAssetUpload} from '@app/api/infrastructure/EntityAssetService';
import type {IGatewayService} from '@app/api/infrastructure/IGatewayService';
import type {UserCacheService} from '@app/api/infrastructure/UserCacheService';
import {Logger} from '@app/api/Logger';
import type {LimitConfigService} from '@app/api/limits/LimitConfigService';
import {resolveLimitSafe} from '@app/api/limits/LimitConfigUtils';
import {createLimitMatchContext} from '@app/api/limits/LimitMatchContextBuilder';
import {profileSubstringBlocklistCache} from '@app/api/middleware/ProfileSubstringBlocklistCache';
import type {RequestCache} from '@app/api/middleware/RequestCacheMiddleware';
import type {GuildMember} from '@app/api/models/GuildMember';
import type {User} from '@app/api/models/User';
import type {UserGuildSettings} from '@app/api/models/UserGuildSettings';
import type {UserSettings} from '@app/api/models/UserSettings';
import {assertAccountNotLimited} from '@app/api/user/AccountLimit';
import type {UserRepository} from '@app/api/user/repositories/UserRepository';
import {dispatchUserGuildSettingsUpdate} from '@app/api/user/UserGuildSettingsThreadView';
import {isProfileSubstringExempt} from '@app/api/user/UserHelpers';
import {mapUserSettingsToResponse} from '@app/api/user/UserMappers';
import {addGuildToUncategorizedFolder, removeGuildFromUserFolders} from '@app/api/user/utils/GuildFolderUtils';
import {AuditLogActionType} from '@fluxer/constants/src/AuditLogActionType';
import {Permissions} from '@fluxer/constants/src/ChannelConstants';
import {
GuildFeatures,
type JoinSourceType,
JoinSourceTypes,
SystemChannelFlags,
} from '@fluxer/constants/src/GuildConstants';
import {
DEFAULT_GUILD_FOLDER_ICON,
type MentionReplyPreference,
UserNotificationSettings,
} from '@fluxer/constants/src/UserConstants';
import {ValidationErrorCodes} from '@fluxer/constants/src/ValidationErrorCodes';
import {ContentBlockedError} from '@fluxer/errors/src/domains/content/ContentBlockedError';
import {InputValidationError} from '@fluxer/errors/src/domains/core/InputValidationError';
import {MissingPermissionsError} from '@fluxer/errors/src/domains/core/MissingPermissionsError';
import {MaxGuildMembersError} from '@fluxer/errors/src/domains/guild/MaxGuildMembersError';
import {MaxGuildsError} from '@fluxer/errors/src/domains/guild/MaxGuildsError';
import {UnknownGuildError} from '@fluxer/errors/src/domains/guild/UnknownGuildError';
import {UnknownGuildMemberError} from '@fluxer/errors/src/domains/guild/UnknownGuildMemberError';
import {CommunicationDisabledError} from '@fluxer/errors/src/domains/moderation/CommunicationDisabledError';
import {UserNotInVoiceError} from '@fluxer/errors/src/domains/user/UserNotInVoiceError';
import {DEFAULT_STOCK_LIMITS} from '@fluxer/limits/src/LimitDefaults';
import type {GuildMemberResponse} from '@fluxer/schema/src/domains/guild/GuildMemberSchemas';
import type {GuildMemberUpdateRequest} from '@fluxer/schema/src/domains/guild/GuildRequestSchemas';
import type {IRateLimitService} from '@pkgs/rate_limit/src/IRateLimitService';
import {ms} from 'itty-time';
const MEMBERSHIP_METADATA_TTL_SECONDS = 365 * 24 * 60 * 60;
function isMemberCurrentlyTimedOut(member: GuildMember): boolean {
return member.communicationDisabledUntil !== null && member.communicationDisabledUntil.getTime() > Date.now();
}
const EMAIL_VERIFICATION_REQUIRED_SELF_PROFILE_FIELDS = [
'nick',
'avatar',
'banner',
'bio',
'pronouns',
'accent_color',
'profile_flags',
] as const;
interface MemberUpdateData {
nick?: string | null;
role_ids?: Set<RoleID>;
avatar_hash?: string | null;
banner_hash?: string | null;
bio?: string | null;
pronouns?: string | null;
accent_color?: number | null;
profile_flags?: number | null;
mention_flags?: MentionReplyPreference | null;
mute?: boolean;
deaf?: boolean;
communication_disabled_until?: Date | null;
}
interface PreparedMemberAssets {
avatar: PreparedAssetUpload | null;
banner: PreparedAssetUpload | null;
}
interface VoiceAuditLogMetadataParams {
newChannelId: bigint | null;
previousChannelId: string | null;
}
function buildVoiceAuditLogMetadata(params: VoiceAuditLogMetadataParams): Record<string, string> | null {
const channelId = params.newChannelId !== null ? params.newChannelId.toString() : (params.previousChannelId ?? null);
if (!channelId) {
return null;
}
return {
channel_id: channelId,
count: '1',
};
}
function hasSelfProfileCustomizationUpdate(
data: GuildMemberUpdateRequest | Omit<GuildMemberUpdateRequest, 'roles'>,
): boolean {
return EMAIL_VERIFICATION_REQUIRED_SELF_PROFILE_FIELDS.some((field) => data[field] !== undefined);
}
function joinSourceName(type: JoinSourceType): JoinSource {
switch (type) {
case JoinSourceTypes.CREATOR:
return 'creator';
case JoinSourceTypes.INSTANT_INVITE:
return 'invite';
case JoinSourceTypes.VANITY_URL:
return 'vanity';
case JoinSourceTypes.BOT_INVITE:
return 'bot_invite';
case JoinSourceTypes.ADMIN_FORCE_ADD:
return 'admin_force_add';
case JoinSourceTypes.DISCOVERY:
return 'discovery';
default:
return 'other';
}
}
export class GuildMemberOperationsService {
constructor(
private readonly guildRepository: GuildRepository,
private readonly channelService: ChannelService,
private readonly userCacheService: UserCacheService,
private readonly gatewayService: IGatewayService,
private readonly entityAssetService: EntityAssetService,
private readonly userRepository: UserRepository,
private readonly rateLimitService: IRateLimitService,
private readonly validationService: GuildMemberValidationService,
private readonly guildAuditLogService: GuildAuditLogService,
private readonly limitConfigService: LimitConfigService,
private readonly searchIndexService: GuildMemberSearchIndexService,
) {}
async getMembers(params: {
userId: UserID;
guildId: GuildID;
limit?: number;
after?: UserID;
}): Promise<Array<GuildMemberResponse>> {
const {userId, guildId, limit = 1, after} = params;
await getGuildAuth(this.gatewayService, this.userRepository, {userId, guildId});
const cursorResult = await this.gatewayService.listGuildMembersCursor({
guildId,
limit,
after,
});
return cursorResult.members;
}
private async recordVoiceAuditLog(params: {
guildId: GuildID;
userId: UserID;
targetId: UserID;
newChannelId: bigint | null;
previousChannelId: string | null;
connectionId: string | null;
auditLogReason?: string | null;
}): Promise<void> {
if (
params.previousChannelId !== null &&
params.newChannelId !== null &&
params.previousChannelId === params.newChannelId.toString()
) {
return;
}
const action = params.newChannelId === null ? AuditLogActionType.MEMBER_DISCONNECT : AuditLogActionType.MEMBER_MOVE;
const previousSnapshot = params.previousChannelId !== null ? {channel_id: params.previousChannelId} : null;
const nextSnapshot = params.newChannelId !== null ? {channel_id: params.newChannelId.toString()} : null;
const voiceChanges = this.guildAuditLogService.computeChanges(previousSnapshot, nextSnapshot);
const changes = voiceChanges.length > 0 ? voiceChanges : null;
const metadata = buildVoiceAuditLogMetadata({
newChannelId: params.newChannelId,
previousChannelId: params.previousChannelId,
});
await this.guildAuditLogService.record({
guildId: params.guildId,
userId: params.userId,
action,
targetId: params.targetId,
auditLogReason: params.auditLogReason,
changes,
metadata: metadata ?? undefined,
});
}
private async fetchCurrentChannelId(guildId: GuildID, userId: UserID): Promise<string | null> {
const voiceState = await this.gatewayService.getVoiceState({guildId, userId});
return voiceState?.channel_id ?? null;
}
async getMember(params: {
userId: UserID;
targetId: UserID;
guildId: GuildID;
requestCache: RequestCache;
}): Promise<GuildMemberResponse> {
const {userId, targetId, guildId, requestCache} = params;
await getGuildAuth(this.gatewayService, this.userRepository, {userId, guildId});
const member = await this.guildRepository.getMember(guildId, targetId);
if (!member) throw new UnknownGuildMemberError();
return await mapGuildMemberToResponse(member, this.userCacheService, requestCache);
}
async updateMember(params: {
userId: UserID;
targetId: UserID;
guildId: GuildID;
data: GuildMemberUpdateRequest | Omit<GuildMemberUpdateRequest, 'roles'>;
requestCache: RequestCache;
auditLogReason?: string | null;
}): Promise<GuildMemberResponse> {
const {userId, targetId, guildId, data, requestCache} = params;
const {guildData, canManageRoles, hasPermission, checkTargetMember} = await getGuildAuth(
this.gatewayService,
this.userRepository,
{
userId,
guildId,
},
);
const effectiveData: GuildMemberUpdateRequest | Omit<GuildMemberUpdateRequest, 'roles'> = {...data};
const updateData: MemberUpdateData = {};
if (effectiveData.nick !== undefined) {
if (userId === targetId) {
const canChangeNick = await hasPermission(Permissions.CHANGE_NICKNAME);
if (!canChangeNick) {
effectiveData.nick = undefined;
}
} else {
const hasManageNicknames = await hasPermission(Permissions.MANAGE_NICKNAMES);
if (!hasManageNicknames) throw new MissingPermissionsError();
await checkTargetMember(targetId);
}
}
const moderationCtx = {
userId: targetId,
guildId,
channelId: null,
messageId: null,
surface: 'profile_field' as const,
};
contentModerationService.scanText(effectiveData.nick ?? null, moderationCtx);
contentModerationService.scanText(effectiveData.bio ?? null, moderationCtx);
contentModerationService.scanText(effectiveData.pronouns ?? null, moderationCtx);
if (effectiveData.communication_disabled_until !== undefined) {
updateData.communication_disabled_until = await this.validateTimeout({
userId,
targetId,
guildId,
rawTimeout: effectiveData.communication_disabled_until,
hasPermission,
checkTargetMember,
});
}
const targetMember = await this.guildRepository.getMember(guildId, targetId);
if (!targetMember) throw new UnknownGuildMemberError();
if (userId === targetId && effectiveData.nick !== undefined && isMemberCurrentlyTimedOut(targetMember)) {
throw new CommunicationDisabledError();
}
const targetUser = await this.userRepository.findUnique(targetId);
if (!targetUser) {
throw new UnknownGuildMemberError();
}
const preparedAssets: PreparedMemberAssets = {avatar: null, banner: null};
if (effectiveData.nick !== undefined) {
this.enforceProfileSubstringBlocklist(targetUser, 'nickname', effectiveData.nick);
updateData.nick = effectiveData.nick;
}
if ('roles' in effectiveData && effectiveData.roles !== undefined) {
const roleIds = await this.validationService.validateAndGetRoleIds({
userId,
guildId,
guildData,
targetId,
targetMember,
newRoles: Array.from(effectiveData.roles).map(createRoleID),
hasPermission,
canManageRoles,
});
updateData.role_ids = new Set(roleIds);
}
if (userId === targetId) {
try {
await this.updateSelfProfile({
userId,
targetId,
guildId,
targetUser,
targetMember,
data: effectiveData,
updateData,
preparedAssets,
});
} catch (error) {
await this.rollbackPreparedAssets(preparedAssets);
throw error;
}
}
await this.updateVoiceAndChannel({
userId,
targetId,
guildId,
targetMember,
data: effectiveData,
updateData,
hasPermission,
checkTargetMember,
auditLogReason: params.auditLogReason,
});
const isAssigningRoles = updateData.role_ids !== undefined && updateData.role_ids.size > 0;
const shouldRemoveTemporaryStatus = targetMember.isTemporary && isAssigningRoles;
const updatedMemberData = this.buildMemberUpdateRow({targetMember, updateData, shouldRemoveTemporaryStatus});
let updatedMember: GuildMember;
try {
updatedMember = await this.guildRepository.upsertMember(updatedMemberData);
} catch (error) {
Logger.error(
{error, guildId, userId: targetId},
'Guild member update failed with unknown commit status; retaining uploaded assets',
);
throw error;
}
await this.entityAssetService.commitAssetChanges([preparedAssets.avatar, preparedAssets.banner]);
if (shouldRemoveTemporaryStatus) {
await this.gatewayService.removeTemporaryGuild({userId: targetId, guildId});
}
return await mapGuildMemberToResponse(updatedMember, this.userCacheService, requestCache);
}
async removeMember(params: {userId: UserID; targetId: UserID; guildId: GuildID}): Promise<void> {
const {userId, targetId, guildId} = params;
const {guildData, checkTargetMember, checkPermission} = await getGuildAuth(
this.gatewayService,
this.userRepository,
{
userId,
guildId,
},
);
await checkPermission(Permissions.KICK_MEMBERS);
const targetMember = await this.guildRepository.getMember(guildId, targetId);
if (!targetMember) throw new UnknownGuildMemberError();
if (targetMember.userId === userId || guildData.owner_id === targetId.toString()) {
throw new UnknownGuildMemberError();
}
await checkTargetMember(targetId);
await this.snapshotMembershipMetadata(targetMember);
const guild = await this.guildRepository.findUnique(guildId);
await this.guildRepository.deleteMember(guildId, targetId);
if (guild) {
await this.guildRepository.upsertPartial(
guildId,
{member_count: Math.max(0, guild.memberCount - 1)},
guild.toRow(),
);
}
await this.gatewayService.leaveGuild({userId: targetId, guildId});
await emitGuildMemberRemoved({guildId, userId: targetId, actorId: userId, kind: 'kicked'});
}
async addUserToGuild(
params: {
userId: UserID;
guildId: GuildID;
sendJoinMessage?: boolean;
skipGuildLimitCheck?: boolean;
skipBanCheck?: boolean;
skipAccountLimitCheck?: boolean;
isTemporary?: boolean;
joinSourceType?: JoinSourceType;
sourceInviteCode?: InviteCode;
inviterId?: UserID;
requestCache: RequestCache;
initiatorId?: UserID;
},
eventService: GuildMemberEventService,
): Promise<GuildMember> {
let succeeded = false;
try {
const {
userId,
guildId,
sendJoinMessage = true,
skipGuildLimitCheck = false,
skipBanCheck = false,
skipAccountLimitCheck = false,
isTemporary = false,
joinSourceType = JoinSourceTypes.INSTANT_INVITE,
sourceInviteCode = null,
inviterId = null,
requestCache,
} = params;
const initiatorId = params.initiatorId ?? userId;
const guild = await this.guildRepository.findUnique(guildId);
if (!guild) throw new UnknownGuildError();
const existingMember = await this.guildRepository.getMember(guildId, userId);
if (existingMember) return existingMember;
assertMutableUserId(userId);
const user = await this.userRepository.findUnique(userId);
if (!user) throw new UnknownGuildError();
if (!skipAccountLimitCheck) {
assertAccountNotLimited(user);
}
if (!skipBanCheck) {
await assertNotBannedFromGuild(this.guildRepository, this.userRepository, {userId, guildId});
}
const userGuildsCount = await this.guildRepository.countUserGuilds(userId);
if (!skipGuildLimitCheck) {
await this.enforceGuildLimit(user, userGuildsCount);
}
const maxGuildMembers = resolveMaxGuildMembersLimit({
guildFeatures: guild.features,
snapshot: this.limitConfigService.getConfigSnapshot(),
});
if (guild.memberCount >= maxGuildMembers) {
throw new MaxGuildMembersError(maxGuildMembers);
}
const metadata = await this.guildRepository.getMembershipMetadata(guildId, userId);
const restoredTimeout =
metadata?.communication_disabled_until && metadata.communication_disabled_until.getTime() > Date.now()
? metadata.communication_disabled_until
: null;
const guildMember = await this.guildRepository.upsertMember({
guild_id: guildId,
user_id: userId,
joined_at: new Date(),
nick: null,
avatar_hash: null,
banner_hash: null,
bio: null,
pronouns: null,
accent_color: null,
join_source_type: joinSourceType,
source_invite_code: sourceInviteCode,
inviter_id: inviterId,
deaf: false,
mute: false,
communication_disabled_until: restoredTimeout,
role_ids: null,
is_premium_sanitized: null,
temporary: isTemporary,
profile_flags: null,
mention_flags: null,
version: 1,
});
await this.guildRepository.upsertPartial(guildId, {member_count: guild.memberCount + 1}, guild.toRow());
await this.applyJoinUserSettings({userId, guildId, user});
await eventService.dispatchGuildMemberAdd({member: guildMember, requestCache});
await this.gatewayService.joinGuild({userId, guildId});
const includeDefault = guild.membersIndexedAt != null;
if (includeDefault) {
void this.searchIndexService.indexMember(guildMember, user, {includeDefault});
}
if (sendJoinMessage && !(guild.systemChannelFlags & SystemChannelFlags.SUPPRESS_JOIN_NOTIFICATIONS)) {
await this.channelService.messages.system.sendJoinSystemMessage({guildId, userId, requestCache});
}
void emitActivity('guild_joined', userId.toString(), {
user_id: userId.toString(),
guild_id: guildId.toString(),
member_count: guild.memberCount + 1,
discoverable: guild.features.has(GuildFeatures.DISCOVERABLE),
invite_code: sourceInviteCode,
inviter_id: inviterId ? inviterId.toString() : null,
join_source: joinSourceName(joinSourceType),
});
if (user.isBot) {
await this.guildAuditLogService.record({
guildId,
userId: initiatorId,
action: AuditLogActionType.BOT_ADD,
targetId: userId,
metadata: {
temporary: isTemporary ? 'true' : 'false',
},
});
}
succeeded = true;
return guildMember;
} finally {
if (!succeeded) {
}
}
}
async leaveGuild(params: {userId: UserID; guildId: GuildID}): Promise<void> {
const {userId, guildId} = params;
const guildData = await this.gatewayService.getGuildData({guildId, userId});
if (!guildData) throw new UnknownGuildError();
if (guildData.owner_id === userId.toString()) {
throw InputValidationError.fromCode('guild_id', ValidationErrorCodes.CANNOT_LEAVE_GUILD_AS_OWNER);
}
const user = await this.userRepository.findUnique(userId);
const guild = await this.guildRepository.findUnique(guildId);
const leavingMember = await this.guildRepository.getMember(guildId, userId);
if (leavingMember) {
await this.snapshotMembershipMetadata(leavingMember);
}
await this.guildRepository.deleteMember(guildId, userId);
if (guild) {
const newMemberCount = Math.max(0, guild.memberCount - 1);
await this.guildRepository.upsertPartial(guildId, {member_count: newMemberCount}, guild.toRow());
}
if (user && !user.isBot) {
await removeGuildFromUserFolders({
userId,
guildId,
userRepository: this.userRepository,
gatewayService: this.gatewayService,
});
}
await this.gatewayService.leaveGuild({userId, guildId});
await emitGuildMemberRemoved({guildId, userId, actorId: null, kind: 'left'});
}
private async validateTimeout(params: {
userId: UserID;
targetId: UserID;
guildId: GuildID;
rawTimeout: string | null;
hasPermission: (permission: bigint) => Promise<boolean>;
checkTargetMember: (targetId: UserID) => Promise<void>;
}): Promise<Date | null> {
const {userId, targetId, guildId, rawTimeout, hasPermission, checkTargetMember} = params;
if (userId === targetId) {
throw new MissingPermissionsError();
}
const hasModerateMembers = await hasPermission(Permissions.MODERATE_MEMBERS);
if (!hasModerateMembers) throw new MissingPermissionsError();
const targetPermissions = await this.gatewayService.getUserPermissions({guildId, userId: targetId});
if ((targetPermissions & Permissions.ADMINISTRATOR) === Permissions.ADMINISTRATOR) {
throw new MissingPermissionsError();
}
await checkTargetMember(targetId);
if (rawTimeout === null) {
return null;
}
const parsedTimeout = new Date(rawTimeout);
if (Number.isNaN(parsedTimeout.getTime())) {
throw InputValidationError.fromCode('communication_disabled_until', ValidationErrorCodes.INVALID_TIMEOUT_VALUE);
}
const diffMs = parsedTimeout.getTime() - Date.now();
if (diffMs <= 0) {
return null;
}
if (diffMs > ms('1 year')) {
throw InputValidationError.fromCode(
'communication_disabled_until',
ValidationErrorCodes.TIMEOUT_CANNOT_EXCEED_365_DAYS,
);
}
return parsedTimeout;
}
private buildMemberUpdateRow(params: {
targetMember: GuildMember;
updateData: MemberUpdateData;
shouldRemoveTemporaryStatus: boolean;
}): GuildMemberRow {
const {targetMember, updateData, shouldRemoveTemporaryStatus} = params;
return {
...targetMember.toRow(),
nick: updateData.nick !== undefined ? updateData.nick : targetMember.nickname,
role_ids: updateData.role_ids ?? targetMember.roleIds,
avatar_hash: updateData.avatar_hash !== undefined ? updateData.avatar_hash : targetMember.avatarHash,
banner_hash: updateData.banner_hash !== undefined ? updateData.banner_hash : targetMember.bannerHash,
bio: updateData.bio !== undefined ? updateData.bio : targetMember.bio,
pronouns: updateData.pronouns !== undefined ? updateData.pronouns : targetMember.pronouns,
accent_color: updateData.accent_color !== undefined ? updateData.accent_color : targetMember.accentColor,
profile_flags: updateData.profile_flags !== undefined ? updateData.profile_flags : targetMember.profileFlags,
mention_flags: updateData.mention_flags !== undefined ? updateData.mention_flags : targetMember.mentionFlags,
mute: updateData.mute !== undefined ? updateData.mute : targetMember.isMute,
deaf: updateData.deaf !== undefined ? updateData.deaf : targetMember.isDeaf,
communication_disabled_until:
updateData.communication_disabled_until !== undefined
? updateData.communication_disabled_until
: targetMember.communicationDisabledUntil,
temporary: shouldRemoveTemporaryStatus ? false : targetMember.isTemporary,
};
}
private async enforceGuildLimit(user: User, currentGuildCount: number): Promise<void> {
const ctx = createLimitMatchContext({user});
const maxGuilds = resolveLimitSafe(
this.limitConfigService.getConfigSnapshot(),
ctx,
'max_guilds',
DEFAULT_STOCK_LIMITS.max_guilds,
);
if (currentGuildCount >= maxGuilds) throw new MaxGuildsError(maxGuilds);
}
private async applyJoinUserSettings(params: {userId: UserID; guildId: GuildID; user: User}): Promise<void> {
const {userId, guildId, user} = params;
const userSettings = await this.userRepository.findSettings(userId);
if (!userSettings) {
return;
}
let needsUpdate = false;
const settingsRow = userSettings.toRow();
if (user.isBot && userSettings.botDefaultGuildsRestricted) {
const updatedBotRestrictedGuilds = new Set(userSettings.botRestrictedGuilds);
updatedBotRestrictedGuilds.add(guildId);
settingsRow.bot_restricted_guilds = updatedBotRestrictedGuilds;
needsUpdate = true;
} else if (!user.isBot && userSettings.defaultGuildsRestricted) {
const updatedRestrictedGuilds = new Set(userSettings.restrictedGuilds);
updatedRestrictedGuilds.add(guildId);
settingsRow.restricted_guilds = updatedRestrictedGuilds;
needsUpdate = true;
}
if (!user.isBot) {
const {folders: nextFolders, modified} = addGuildToUncategorizedFolder({
folders: settingsRow.guild_folders ?? [],
guildId,
defaultIcon: DEFAULT_GUILD_FOLDER_ICON,
});
if (modified) {
settingsRow.guild_folders = nextFolders;
needsUpdate = true;
}
}
if (needsUpdate) {
const updatedSettings = await this.userRepository.upsertSettings(settingsRow);
await this.dispatchUserSettingsUpdate({userId, settings: updatedSettings});
}
if (!user.isBot && userSettings.defaultHideMutedChannels) {
const existingGuildSettings = await this.userRepository.findGuildSettings(userId, guildId);
const guildSettingsRow = existingGuildSettings
? {...existingGuildSettings.toRow(), hide_muted_channels: true}
: {
user_id: userId,
guild_id: guildId,
message_notifications: UserNotificationSettings.INHERIT,
muted: false,
mute_config: null,
mobile_push: true,
suppress_everyone: false,
suppress_roles: false,
hide_muted_channels: true,
channel_overrides: null,
unread_badges: null,
version: 1,
};
const updatedGuildSettings = await this.userRepository.upsertGuildSettings(guildSettingsRow);
await this.dispatchUserGuildSettingsUpdate({userId, settings: updatedGuildSettings});
}
}
private async updateSelfProfile(params: {
userId: UserID;
targetId: UserID;
guildId: GuildID;
targetUser: User;
targetMember: GuildMember;
data: GuildMemberUpdateRequest | Omit<GuildMemberUpdateRequest, 'roles'>;
updateData: MemberUpdateData;
preparedAssets: PreparedMemberAssets;
}): Promise<void> {
const {targetId, guildId, targetUser, targetMember, data, updateData, preparedAssets} = params;
if (hasSelfProfileCustomizationUpdate(data)) {
requireEmailVerified(targetUser, 'profile');
assertAccountNotLimited(targetUser);
}
const ctx = createLimitMatchContext({user: targetUser});
const hasGuildProfileCustomization = resolveLimitSafe(
this.limitConfigService.getConfigSnapshot(),
ctx,
'feature_per_guild_profiles',
0,
);
if (hasGuildProfileCustomization === 0) {
if (data.avatar !== undefined) {
data.avatar = undefined;
}
if (data.banner !== undefined) {
data.banner = undefined;
}
if (data.bio !== undefined) {
data.bio = undefined;
}
if (data.accent_color !== undefined) {
data.accent_color = undefined;
}
}
if (data.profile_flags !== undefined) {
updateData.profile_flags = data.profile_flags;
}
if (data.mention_flags !== undefined) {
updateData.mention_flags = data.mention_flags;
}
if (data.avatar !== undefined) {
const avatarRateLimit = await this.rateLimitService.checkLimit({
identifier: `guild_avatar_change:${guildId}:${targetId}`,
maxAttempts: 25,
windowMs: ms('30 minutes'),
});
if (!avatarRateLimit.allowed) {
const minutes = Math.ceil((avatarRateLimit.retryAfter || 0) / 60);
throw InputValidationError.fromCode('avatar', ValidationErrorCodes.AVATAR_CHANGED_TOO_MANY_TIMES, {minutes});
}
const prepared = await this.entityAssetService.prepareAssetUpload({
assetType: 'avatar',
entityType: 'guild_member',
entityId: targetId,
guildId,
previousHash: targetMember.avatarHash,
base64Image: data.avatar,
errorPath: 'avatar',
});
preparedAssets.avatar = prepared;
if (prepared.newHash !== targetMember.avatarHash) {
updateData.avatar_hash = prepared.newHash;
}
}
if (data.banner !== undefined) {
const bannerRateLimit = await this.rateLimitService.checkLimit({
identifier: `guild_banner_change:${guildId}:${targetId}`,
maxAttempts: 25,
windowMs: ms('30 minutes'),
});
if (!bannerRateLimit.allowed) {
const minutes = Math.ceil((bannerRateLimit.retryAfter || 0) / 60);
throw InputValidationError.fromCode('banner', ValidationErrorCodes.BANNER_CHANGED_TOO_MANY_TIMES, {minutes});
}
const prepared = await this.entityAssetService.prepareAssetUpload({
assetType: 'banner',
entityType: 'guild_member',
entityId: targetId,
guildId,
previousHash: targetMember.bannerHash,
base64Image: data.banner,
errorPath: 'banner',
});
preparedAssets.banner = prepared;
if (prepared.newHash !== targetMember.bannerHash) {
updateData.banner_hash = prepared.newHash;
}
}
if (data.bio !== undefined) {
if (data.bio !== targetMember.bio) {
const bioRateLimit = await this.rateLimitService.checkLimit({
identifier: `guild_bio_change:${guildId}:${targetId}`,
maxAttempts: 25,
windowMs: ms('30 minutes'),
});
if (!bioRateLimit.allowed) {
const minutes = Math.ceil((bioRateLimit.retryAfter || 0) / 60);
throw InputValidationError.fromCode('bio', ValidationErrorCodes.BIO_CHANGED_TOO_MANY_TIMES, {minutes});
}
this.enforceProfileSubstringBlocklist(targetUser, 'bio', data.bio);
updateData.bio = data.bio;
}
}
if (data.accent_color !== undefined) {
if (data.accent_color !== targetMember.accentColor) {
const accentColorRateLimit = await this.rateLimitService.checkLimit({
identifier: `guild_accent_color_change:${guildId}:${targetId}`,
maxAttempts: 25,
windowMs: ms('30 minutes'),
});
if (!accentColorRateLimit.allowed) {
const minutes = Math.ceil((accentColorRateLimit.retryAfter || 0) / 60);
throw InputValidationError.fromCode(
'accent_color',
ValidationErrorCodes.ACCENT_COLOR_CHANGED_TOO_MANY_TIMES,
{minutes},
);
}
updateData.accent_color = data.accent_color;
}
}
if (data.pronouns !== undefined) {
if (data.pronouns !== targetMember.pronouns) {
const pronounsRateLimit = await this.rateLimitService.checkLimit({
identifier: `guild_pronouns_change:${guildId}:${targetId}`,
maxAttempts: 25,
windowMs: ms('30 minutes'),
});
if (!pronounsRateLimit.allowed) {
const minutes = Math.ceil((pronounsRateLimit.retryAfter || 0) / 60);
throw InputValidationError.fromCode('pronouns', ValidationErrorCodes.PRONOUNS_CHANGED_TOO_MANY_TIMES, {
minutes,
});
}
this.enforceProfileSubstringBlocklist(targetUser, 'pronouns', data.pronouns);
updateData.pronouns = data.pronouns;
}
}
}
private enforceProfileSubstringBlocklist(
user: User,
scope: 'nickname' | 'bio' | 'pronouns',
value: string | null | undefined,
): void {
if (
value &&
!isProfileSubstringExempt(user) &&
profileSubstringBlocklistCache.containsBannedSubstring(scope, value)
) {
throw new ContentBlockedError();
}
}
private async updateVoiceAndChannel(params: {
userId: UserID;
targetId: UserID;
guildId: GuildID;
targetMember: GuildMember;
data: GuildMemberUpdateRequest | Omit<GuildMemberUpdateRequest, 'roles'>;
updateData: MemberUpdateData;
hasPermission: (permission: bigint) => Promise<boolean>;
checkTargetMember: (targetId: UserID) => Promise<void>;
auditLogReason?: string | null;
}): Promise<void> {
const {
userId,
targetId,
guildId,
targetMember,
data,
updateData,
hasPermission,
checkTargetMember,
auditLogReason,
} = params;
if (data.mute !== undefined || data.deaf !== undefined || data.channel_id !== undefined) {
if (userId !== targetId) {
await checkTargetMember(targetId);
}
if (data.mute !== undefined) {
if (!(await hasPermission(Permissions.MUTE_MEMBERS))) {
throw new MissingPermissionsError();
}
}
if (data.deaf !== undefined) {
if (!(await hasPermission(Permissions.DEAFEN_MEMBERS))) {
throw new MissingPermissionsError();
}
}
if (data.channel_id !== undefined) {
if (!(await hasPermission(Permissions.MOVE_MEMBERS))) {
throw new MissingPermissionsError();
}
const previousChannelId = await this.fetchCurrentChannelId(guildId, targetId);
const result = await this.gatewayService.moveMember({
guildId,
moderatorId: userId,
userId: targetId,
channelId: data.channel_id !== null ? createChannelID(data.channel_id) : null,
connectionId: data.connection_id ?? null,
});
if (result.error) {
switch (result.error) {
case 'user_not_in_voice':
case 'connection_not_found':
throw new UserNotInVoiceError();
case 'channel_not_found':
throw InputValidationError.fromCode('channel_id', ValidationErrorCodes.CHANNEL_DOES_NOT_EXIST);
case 'channel_not_voice':
throw InputValidationError.fromCode('channel_id', ValidationErrorCodes.CHANNEL_MUST_BE_VOICE);
case 'target_missing_connect':
case 'moderator_missing_connect':
throw new MissingPermissionsError();
default:
throw new UserNotInVoiceError();
}
} else {
await this.recordVoiceAuditLog({
guildId,
userId,
targetId,
newChannelId: data.channel_id,
previousChannelId,
connectionId: data.connection_id ?? null,
auditLogReason,
});
}
}
if (data.mute !== undefined || data.deaf !== undefined) {
try {
await this.gatewayService.updateMemberVoice({
guildId,
userId: targetId,
mute: data.mute ?? targetMember.isMute,
deaf: data.deaf ?? targetMember.isDeaf,
});
if (data.mute !== undefined) {
updateData.mute = data.mute;
}
if (data.deaf !== undefined) {
updateData.deaf = data.deaf;
}
} catch (error) {
Logger.error({error, userId: targetId, guildId}, 'Failed to get user voice state, user not in voice');
throw new UserNotInVoiceError();
}
}
}
}
private async dispatchUserSettingsUpdate({
userId,
settings,
}: {
userId: UserID;
settings: UserSettings;
}): Promise<void> {
await this.gatewayService.dispatchPresence({
userId,
event: 'USER_SETTINGS_UPDATE',
data: mapUserSettingsToResponse({settings}),
});
}
private async dispatchUserGuildSettingsUpdate({
userId,
settings,
}: {
userId: UserID;
settings: UserGuildSettings;
}): Promise<void> {
await dispatchUserGuildSettingsUpdate(this.gatewayService, userId, settings);
}
private async rollbackPreparedAssets(preparedAssets: PreparedMemberAssets): Promise<void> {
const rollbackPromises: Array<Promise<void>> = [];
if (preparedAssets.avatar) {
rollbackPromises.push(this.entityAssetService.rollbackAssetUpload(preparedAssets.avatar));
}
if (preparedAssets.banner) {
rollbackPromises.push(this.entityAssetService.rollbackAssetUpload(preparedAssets.banner));
}
await Promise.all(rollbackPromises);
}
private async snapshotMembershipMetadata(member: GuildMember): Promise<void> {
const existingMetadata = await this.guildRepository.getMembershipMetadata(member.guildId, member.userId);
await this.guildRepository.upsertMembershipMetadata(
{
guild_id: member.guildId,
user_id: member.userId,
communication_disabled_until: member.communicationDisabledUntil,
first_joined_at: existingMetadata?.first_joined_at ?? member.joinedAt,
last_left_at: new Date(),
},
MEMBERSHIP_METADATA_TTL_SECONDS,
);
}
}