Skip to content

Commit

Permalink
fix(ThreadManager): add members and conditionally include hasMore (
Browse files Browse the repository at this point in the history
…#9164)

* fix(ThreadManager): conditionally include `hasMore`

* types: fix tests

---------

Co-authored-by: kodiakhq[bot] <49736102+kodiakhq[bot]@users.noreply.github.com>
  • Loading branch information
Jiralite and kodiakhq[bot] committed Mar 22, 2023
1 parent cc57563 commit e9a8eb3
Show file tree
Hide file tree
Showing 4 changed files with 54 additions and 20 deletions.
9 changes: 8 additions & 1 deletion packages/discord.js/src/managers/GuildChannelManager.js
Original file line number Diff line number Diff line change
Expand Up @@ -451,7 +451,14 @@ class GuildChannelManager extends CachedManager {
}

/**
* Obtains all active thread channels in the guild from Discord
* Data returned from fetching threads.
* @typedef {Object} FetchedThreads
* @property {Collection<Snowflake, ThreadChannel>} threads The threads that were fetched
* @property {Collection<Snowflake, ThreadMember>} members The thread members in the received threads
*/

/**
* Obtains all active thread channels in the guild.
* @param {boolean} [cache=true] Whether to cache the fetched data
* @returns {Promise<FetchedThreads>}
* @example
Expand Down
38 changes: 21 additions & 17 deletions packages/discord.js/src/managers/ThreadManager.js
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,8 @@ class ThreadManager extends CachedManager {
* ThreadChannelResolvable then the specified thread will be fetched. Fetches all active threads if `undefined`
* @param {BaseFetchOptions} [cacheOptions] Additional options for this fetch. <warn>The `force` field gets ignored
* if `options` is not a {@link ThreadChannelResolvable}</warn>
* @returns {Promise<?(ThreadChannel|FetchedThreads)>}
* @returns {Promise<?(ThreadChannel|FetchedThreads|FetchedThreadsMore)>}
* {@link FetchedThreads} if active & {@link FetchedThreadsMore} if archived.
* @example
* // Fetch a thread by its id
* channel.threads.fetch('831955138126104859')
Expand Down Expand Up @@ -124,9 +125,8 @@ class ThreadManager extends CachedManager {
*/

/**
* The data returned from a thread fetch that returns multiple threads.
* @typedef {Object} FetchedThreads
* @property {Collection<Snowflake, ThreadChannel>} threads The threads that were fetched, with any members returned
* Data returned from fetching multiple threads.
* @typedef {FetchedThreads} FetchedThreadsMore
* @property {?boolean} hasMore Whether there are potentially additional threads that require a subsequent call
*/

Expand All @@ -136,7 +136,7 @@ class ThreadManager extends CachedManager {
* in the parent channel.</info>
* @param {FetchArchivedThreadOptions} [options] The options to fetch archived threads
* @param {boolean} [cache=true] Whether to cache the new thread objects if they aren't already
* @returns {Promise<FetchedThreads>}
* @returns {Promise<FetchedThreadsMore>}
*/
async fetchArchived({ type = 'public', fetchAll = false, before, limit } = {}, cache = true) {
let path = Routes.channelThreads(this.channel.id, type);
Expand Down Expand Up @@ -171,15 +171,13 @@ class ThreadManager extends CachedManager {
}

/**
* Obtains the accessible active threads from Discord.
* <info>This method requires the {@link PermissionFlagsBits.ReadMessageHistory} permission
* in the parent channel.</info>
* @param {boolean} [cache=true] Whether to cache the new thread objects if they aren't already
* Obtains all active thread channels in the guild.
* This internally calls {@link GuildChannelManager#fetchActiveThreads}.
* @param {boolean} [cache=true] Whether to cache the fetched data
* @returns {Promise<FetchedThreads>}
*/
async fetchActive(cache = true) {
const raw = await this.client.rest.get(Routes.guildActiveThreads(this.channel.guild.id));
return this.constructor._mapThreads(raw, this.client, { parent: this.channel, cache });
fetchActive(cache = true) {
return this.channel.guild.channels.fetchActiveThreads(cache);
}

static _mapThreads(rawThreads, client, { parent, guild, cache }) {
Expand All @@ -188,12 +186,18 @@ class ThreadManager extends CachedManager {
if (parent && thread.parentId !== parent.id) return coll;
return coll.set(thread.id, thread);
}, new Collection());

// Discord sends the thread id as id in this object
for (const rawMember of rawThreads.members) client.channels.cache.get(rawMember.id)?.members._add(rawMember);
return {
threads,
hasMore: rawThreads.has_more ?? false,
};
const threadMembers = rawThreads.members.reduce(
(coll, raw) => coll.set(raw.user_id, threads.get(raw.id).members._add(raw)),
new Collection(),
);

const response = { threads, members: threadMembers };

// The GET `/guilds/{guild.id}/threads/active` route does not return `has_more`.
if ('has_more' in rawThreads) response.hasMore = rawThreads.has_more;
return response;
}
}

Expand Down
12 changes: 10 additions & 2 deletions packages/discord.js/typings/index.d.ts
Original file line number Diff line number Diff line change
Expand Up @@ -4109,8 +4109,12 @@ export class ThreadManager<Forum extends boolean = boolean> extends CachedManage
protected constructor(channel: TextChannel | NewsChannel | ForumChannel, iterable?: Iterable<RawThreadChannelData>);
public channel: If<Forum, ForumChannel, TextChannel | NewsChannel>;
public fetch(options: ThreadChannelResolvable, cacheOptions?: BaseFetchOptions): Promise<AnyThreadChannel | null>;
public fetch(
options: FetchThreadsOptions & { archived: FetchArchivedThreadOptions },
cacheOptions?: { cache?: boolean },
): Promise<FetchedThreadsMore>;
public fetch(options?: FetchThreadsOptions, cacheOptions?: { cache?: boolean }): Promise<FetchedThreads>;
public fetchArchived(options?: FetchArchivedThreadOptions, cache?: boolean): Promise<FetchedThreads>;
public fetchArchived(options?: FetchArchivedThreadOptions, cache?: boolean): Promise<FetchedThreadsMore>;
public fetchActive(cache?: boolean): Promise<FetchedThreads>;
}

Expand Down Expand Up @@ -5148,7 +5152,11 @@ export interface FetchChannelOptions extends BaseFetchOptions {

export interface FetchedThreads {
threads: Collection<Snowflake, AnyThreadChannel>;
hasMore?: boolean;
members: Collection<Snowflake, ThreadMember>;
}

export interface FetchedThreadsMore extends FetchedThreads {
hasMore: boolean;
}

export interface FetchGuildOptions extends BaseFetchOptions {
Expand Down
15 changes: 15 additions & 0 deletions packages/discord.js/typings/index.test-d.ts
Original file line number Diff line number Diff line change
Expand Up @@ -160,6 +160,9 @@ import {
PublicThreadChannel,
GuildMemberManager,
GuildMemberFlagsBitField,
ThreadManager,
FetchedThreads,
FetchedThreadsMore,
} from '.';
import { expectAssignable, expectNotAssignable, expectNotType, expectType } from 'tsd';
import type { ContextMenuCommandBuilder, SlashCommandBuilder } from '@discordjs/builders';
Expand Down Expand Up @@ -1479,6 +1482,18 @@ declare const guildChannelManager: GuildChannelManager;
expectType<null>(message.mentions.members);
}

declare const threadManager: ThreadManager;
{
expectType<Promise<AnyThreadChannel | null>>(threadManager.fetch('12345678901234567'));
expectType<Promise<AnyThreadChannel | null>>(threadManager.fetch('12345678901234567', { cache: true, force: false }));
expectType<Promise<FetchedThreads>>(threadManager.fetch());
expectType<Promise<FetchedThreads>>(threadManager.fetch({}));
expectType<Promise<FetchedThreadsMore>>(threadManager.fetch({ archived: { limit: 4 } }));

// @ts-expect-error The force option has no effect here.
threadManager.fetch({ archived: {} }, { force: true });
}

declare const guildForumThreadManager: GuildForumThreadManager;
expectType<ForumChannel>(guildForumThreadManager.channel);

Expand Down

2 comments on commit e9a8eb3

@vercel
Copy link

@vercel vercel bot commented on e9a8eb3 Mar 22, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@vercel
Copy link

@vercel vercel bot commented on e9a8eb3 Mar 22, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please sign in to comment.