1 files changed, 96 insertions, 36 deletions

diff --git a/gateway/src/opcodes/LazyRequest.ts b/gateway/src/opcodes/LazyRequest.ts
index f5fd561a..c304dfe7 100644
--- a/gateway/src/opcodes/LazyRequest.ts
+++ b/gateway/src/opcodes/LazyRequest.ts
@@ -1,46 +1,55 @@
 import {
+	EVENTEnum,
+	EventOpts,
 	getPermission,
+	listenEvent,
 	Member,
-	PublicMemberProjection,
 	Role,
 } from "@fosscord/util";
 import { LazyRequest } from "../schema/LazyRequest";
 import { Send } from "../util/Send";
 import { OPCODES } from "../util/Constants";
-import { WebSocket, Payload } from "@fosscord/gateway";
+import { WebSocket, Payload, handlePresenceUpdate } from "@fosscord/gateway";
 import { check } from "./instanceOf";
 import "missing-native-js-functions";
+import { getRepository } from "typeorm";
+import "missing-native-js-functions";
 
-// TODO: check permission and only show roles/members that have access to this channel
+// TODO: only show roles/members that have access to this channel
 // TODO: config: to list all members (even those who are offline) sorted by role, or just those who are online
 // TODO: rewrite typeorm
 
-export async function onLazyRequest(this: WebSocket, { d }: Payload) {
-	// TODO: check data
-	check.call(this, LazyRequest, d);
-	const { guild_id, typing, channels, activities } = d as LazyRequest;
-
-	const permissions = await getPermission(this.user_id, guild_id);
-	permissions.hasThrow("VIEW_CHANNEL");
-
-	var members = await Member.find({
-		where: { guild_id: guild_id },
-		relations: ["roles", "user"],
-		select: PublicMemberProjection,
-	});
+async function getMembers(guild_id: string, range: [number, number]) {
+	if (!Array.isArray(range) || range.length !== 2) {
+		throw new Error("range is not a valid array");
+	}
+	// TODO: wait for typeorm to implement ordering for .find queries https://github.com/typeorm/typeorm/issues/2620
 
-	const roles = await Role.find({
-		where: { guild_id: guild_id },
-		order: {
-			position: "DESC",
-		},
-	});
+	let members = await getRepository(Member)
+		.createQueryBuilder("member")
+		.where("member.guild_id = :guild_id", { guild_id })
+		.leftJoinAndSelect("member.roles", "role")
+		.leftJoinAndSelect("member.user", "user")
+		.leftJoinAndSelect("user.sessions", "session")
+		.addSelect(
+			"CASE WHEN session.status = 'offline' THEN 0 ELSE 1 END",
+			"_status"
+		)
+		.orderBy("role.position", "DESC")
+		.addOrderBy("_status", "DESC")
+		.addOrderBy("user.username", "ASC")
+		.offset(Number(range[0]) || 0)
+		.limit(Number(range[1]) || 100)
+		.getMany();
 
 	const groups = [] as any[];
-	var member_count = 0;
 	const items = [];
+	const member_roles = members
+		.map((m) => m.roles)
+		.flat()
+		.unique((r) => r.id);
 
-	for (const role of roles) {
+	for (const role of member_roles) {
 		// @ts-ignore
 		const [role_members, other_members] = partition(members, (m: Member) =>
 			m.roles.find((r) => r.id === role.id)
@@ -54,35 +63,86 @@ export async function onLazyRequest(this: WebSocket, { d }: Payload) {
 		groups.push(group);
 
 		for (const member of role_members) {
-			member.roles = member.roles.filter((x: Role) => x.id !== guild_id);
+			const roles = member.roles
+				.filter((x: Role) => x.id !== guild_id)
+				.map((x: Role) => x.id);
+
+			const session = member.user.sessions.first();
+
+			// TODO: properly mock/hide offline/invisible status
 			items.push({
 				member: {
 					...member,
-					roles: member.roles.map((x: Role) => x.id),
+					roles,
+					user: { ...member.user, sessions: undefined },
+					presence: {
+						...session,
+						activities: session?.activities || [],
+						user: { id: member.user.id },
+					},
 				},
 			});
 		}
 		members = other_members;
-		member_count += role_members.length;
 	}
 
+	return {
+		items,
+		groups,
+		range,
+		members: items.map((x) => x.member).filter((x) => x),
+	};
+}
+
+export async function onLazyRequest(this: WebSocket, { d }: Payload) {
+	// TODO: check data
+	check.call(this, LazyRequest, d);
+	const { guild_id, typing, channels, activities } = d as LazyRequest;
+
+	const channel_id = Object.keys(channels || {}).first();
+	if (!channel_id) return;
+
+	const permissions = await getPermission(this.user_id, guild_id, channel_id);
+	permissions.hasThrow("VIEW_CHANNEL");
+
+	const ranges = channels![channel_id];
+	if (!Array.isArray(ranges)) throw new Error("Not a valid Array");
+
+	const member_count = await Member.count({ guild_id });
+	const ops = await Promise.all(ranges.map((x) => getMembers(guild_id, x)));
+
+	// TODO: unsubscribe member_events that are not in op.members
+
+	ops.forEach((op) => {
+		op.members.forEach(async (member) => {
+			if (this.events[member.user.id]) return; // already subscribed as friend
+			if (this.member_events[member.user.id]) return; // already subscribed in member list
+			this.member_events[member.user.id] = await listenEvent(
+				member.user.id,
+				handlePresenceUpdate.bind(this),
+				this.listen_options
+			);
+		});
+	});
+
 	return Send(this, {
 		op: OPCODES.Dispatch,
 		s: this.sequence++,
 		t: "GUILD_MEMBER_LIST_UPDATE",
 		d: {
-			ops: [
-				{
-					range: [0, 99],
-					op: "SYNC",
-					items,
-				},
-			],
-			online_count: member_count, // TODO count online count
+			ops: ops.map((x) => ({
+				items: x.items,
+				op: "SYNC",
+				range: x.range,
+			})),
+			online_count: member_count,
 			member_count,
 			id: "everyone",
 			guild_id,
-			groups,
+			groups: ops
+				.map((x) => x.groups)
+				.flat()
+				.unique(),
 		},
 	});
 }