Handle invisible connections and return userids of visible connected folks

Author: Perlkonig

api/abstractplay.ts

@@ -173,6 +173,7 @@ type MeData = {
     standingChallenges?: FullChallenge[];
     realStanding?: StandingChallenge[];
     connections: number;
+    connected: string[]
 }
 
 type Rating = {
@@ -1779,7 +1780,7 @@ async function me(claim: PartialClaims, pars: { size: string, vars: string, upda
     }));
 
     // Get number of current connections
-    const connections = await countByPk("wsConnections");
+    const { totalCount: connections, visibleUserIds: connected } = await countAndVisibleUserIds("wsConnections");
 
     let data = null;
     console.log(`Fetching challenges`);
@@ -1847,6 +1848,7 @@ async function me(claim: PartialClaims, pars: { size: string, vars: string, upda
           "standingChallenges": (data[3] as any[]).map(d => d.Item),
           "realStanding": realStanding,
           "connections": connections,
+          "connected": connected,
         } as MeData, Set_toJSON),
         headers
       };
@@ -1869,6 +1871,7 @@ async function me(claim: PartialClaims, pars: { size: string, vars: string, upda
           palettes,
           "realStanding": realStanding,
           "connections": connections,
+          "connected": connected,
         } as MeData, Set_toJSON),
         headers
       }
@@ -8718,28 +8721,70 @@ const getAllUsers = async (): Promise<FullUser[]> => {
 /**
  * Count all items for a given partition key, paginating until complete.
  */
-async function countByPk(pkValue: string): Promise<number> {
-  let total = 0;
+// async function countByPk(pkValue: string): Promise<number> {
+//   let total = 0;
+//   let lastKey: Record<string, any> | undefined = undefined;
+
+//   do {
+//     const params: QueryCommandInput = {
+//       TableName: process.env.ABSTRACT_PLAY_TABLE,
+//       KeyConditionExpression: `pk = :pk`,
+//       ExpressionAttributeValues: {
+//         ":pk": pkValue,
+//       },
+//       Select: "COUNT",
+//       ExclusiveStartKey: lastKey,
+//     };
+
+//     const result = await ddbDocClient.send(new QueryCommand(params));
+
+//     total += result.Count ?? 0;
+//     lastKey = result.LastEvaluatedKey;
+//   } while (lastKey);
+
+//   return total;
+// }
+
+async function countAndVisibleUserIds(pkValue: string): Promise<{
+  totalCount: number;
+  visibleUserIds: string[];
+}> {
+  let totalCount = 0;
+  const visibleUserIds: string[] = [];
   let lastKey: Record<string, any> | undefined = undefined;
 
   do {
     const params: QueryCommandInput = {
       TableName: process.env.ABSTRACT_PLAY_TABLE,
-      KeyConditionExpression: `pk = :pk`,
+      KeyConditionExpression: "pk = :pk",
       ExpressionAttributeValues: {
         ":pk": pkValue,
       },
-      Select: "COUNT",
+      // Only fetch what we need
+      ProjectionExpression: "userId, invisible",
       ExclusiveStartKey: lastKey,
     };
 
     const result = await ddbDocClient.send(new QueryCommand(params));
+    const items = result.Items ?? [];
+
+    // Count all items
+    totalCount += items.length;
+
+    // Collect userIds where invisible is missing or false
+    for (const item of items) {
+      if (item.invisible !== true) {
+        // invisible is undefined OR false
+        if (item.userId) {
+          visibleUserIds.push(item.userId);
+        }
+      }
+    }
 
-    total += result.Count ?? 0;
     lastKey = result.LastEvaluatedKey;
   } while (lastKey);
 
-  return total;
+  return { totalCount, visibleUserIds };
 }
 
 async function wsBroadcast (verb: string, payload: any, exclude?: string[]): Promise<SendMessageCommandOutput> {

api/sockets/authHandler.ts

@@ -44,6 +44,7 @@ export const handler = async (event: WebSocketEvent) => {
    // Parse the incoming message body
     const body = JSON.parse(event.body ?? "{}");
     const token: string | undefined = body.token;
+    const invisible: boolean = body.invisible ?? false;
 
   if (!token) {
      console.error("Missing token in auth message");
@@ -69,6 +70,7 @@ export const handler = async (event: WebSocketEvent) => {
 
                 connectionId,
                 userId,
+                invisible,
 
                 // Optional TTL for auto-cleanup
                 ttl: Math.floor(Date.now() / 1000) + 3600,