@atproto/bsky 0.0.274 → 0.0.276

package/dist/hydration/hydrator.js

@@ -270,6 +270,9 @@ export class Hydrator {
     //       - list basic
     async hydratePosts(refs, ctx, state = {}, options = {}) {
         const uris = refs.map((ref) => ref.uri);
+        const knownLikersPromise = options.knownLikers
+            ? this.hydrateKnownLikers(options.knownLikers, ctx)
+            : undefined;
         state.posts ??= new HydrationMap();
         const addPostsToHydrationState = (posts) => {
             posts.forEach((post, uri) => {
@@ -361,17 +364,31 @@ export class Hydrator {
         // post-author profile fetch below; any DIDs surfaced later by the
         // dataplane are picked up by the top-up after the parallel batch.
         const ssRefDids = siteStandardRefs.map((ref) => uriToDid(ref.uri));
+        // Collect known-liker profiles and author-liker block relationships.
+        const knownLikers = await knownLikersPromise;
+        const knownLikerDids = [];
+        const authorsAndKnownLikersRelationships = [];
+        knownLikers?.forEach((knownLiker, uri) => {
+            if (!knownLiker)
+                return;
+            knownLikerDids.push(...knownLiker.actors);
+            const authorDid = uriToDid(uri);
+            for (const likerDid of knownLiker.actors) {
+                authorsAndKnownLikersRelationships.push([authorDid, likerDid]);
+            }
+        });
         const knownProfileDids = dedupeStrs([
             ...allPostUris.map(didFromUri),
             ...ssRefDids,
+            ...knownLikerDids,
         ]);
-        const [postAggs, postViewers, labels, postBlocks, profileState, listState, feedGenState, labelerState, starterPackState, postgates, { documents: siteStandardDocuments, publications: siteStandardPublications, },] = await Promise.all([
+        const [postAggs, postViewers, labels, { postBlocks, additionalRelationshipsBidirectionalBlocks: authorsAndKnownLikersBidirectionalBlocks, }, profileState, listState, feedGenState, labelerState, starterPackState, postgates, { documents: siteStandardDocuments, publications: siteStandardPublications, },] = await Promise.all([
             this.feed.getPostAggregates(allRefs, ctx.viewer),
             ctx.viewer
                 ? this.feed.getPostViewerStates(threadRefs, ctx.viewer)
                 : undefined,
             this.label.getLabelsForSubjects([...allPostUris, ...siteStandardLabelSubjects], ctx.labelers),
-            this.hydratePostBlocks(posts, ctx),
+            this.hydratePostBlocks(posts, ctx, authorsAndKnownLikersRelationships),
             this.hydrateProfiles(knownProfileDids, ctx),
             this.hydrateLists([...nestedListUris, ...threadgateListUris], ctx),
             this.hydrateFeedGens(nestedFeedGenUris, ctx),
@@ -405,6 +422,8 @@ export class Hydrator {
             postAggs,
             postViewers,
             postBlocks,
+            knownLikers,
+            bidirectionalBlocks: authorsAndKnownLikersBidirectionalBlocks,
             labels,
             threadgates,
             postgates,
@@ -413,7 +432,7 @@ export class Hydrator {
             ctx,
         });
     }
-    async hydratePostBlocks(posts, ctx) {
+    async hydratePostBlocks(posts, ctx, additionalRelationships = []) {
         const postBlocks = new HydrationMap();
         const postBlocksPairs = new Map();
         const relationships = [];
@@ -446,6 +465,7 @@ export class Hydrator {
                 postBlockPairs.embed = pair;
             }
         }
+        relationships.push(...additionalRelationships);
         // replace embed/parent/root pairs with block state
         const blocks = await this.hydrateBidirectionalBlocks(pairsToMap(relationships), ctx);
         for (const [uri, { embed, parent, root }] of postBlocksPairs) {
@@ -455,7 +475,18 @@ export class Hydrator {
                 root: !!root && !!isBlocked(blocks, root),
             });
         }
-        return postBlocks;
+        const additionalRelationshipsBidirectionalBlocks = new HydrationMap();
+        for (const [source, targets] of pairsToMap(additionalRelationships)) {
+            const didBlocks = new HydrationMap();
+            for (const target of targets) {
+                didBlocks.set(target, blocks.get(source)?.get(target) ?? false);
+            }
+            additionalRelationshipsBidirectionalBlocks.set(source, didBlocks);
+        }
+        return {
+            postBlocks,
+            additionalRelationshipsBidirectionalBlocks,
+        };
     }
     // app.bsky.feed.defs#feedViewPost
     // - post (+ replies w/ reply parent author)
@@ -472,7 +503,7 @@ export class Hydrator {
     //     - list basic
     //   - post
     //     - ...
-    async hydrateFeedItems(items, ctx) {
+    async hydrateFeedItems(items, ctx, options = {}) {
         // get posts, collect reply refs
         const posts = await this.feed.getPosts(items.map((item) => item.post.uri), ctx.includeTakedowns, undefined, undefined, {
             includeOpThreadMetadata: ctx.features.checkGate(ctx.features.Gate.OpThreadMetadataEnable),
@@ -504,6 +535,8 @@ export class Hydrator {
         const [postState, repostProfileState, reposts] = await Promise.all([
             this.hydratePosts(postAndReplyRefs, ctx, {
                 posts: replies.merge(posts), // avoids refetches while preserving feed-item metadata
+            }, {
+                knownLikers: options.knownLikers,
             }),
             this.hydrateProfiles([...repostUris.map(didFromUri), ...replyParentAuthors], ctx),
             this.feed.getReposts(repostUris, ctx.includeTakedowns),
@@ -566,11 +599,12 @@ export class Hydrator {
     //   - feedgen
     //     - profile
     //       - list basic
-    async hydrateThreadPosts(refs, ctx) {
+    async hydrateThreadPosts(refs, ctx, options = {}) {
         const postsState = await this.hydratePosts(refs, ctx, undefined, {
             processDynamicTagsForView: ctx.features.checkGate(ctx.features.Gate.ThreadsReplyRankingExplorationEnable)
                 ? 'thread'
                 : undefined,
+            knownLikers: options.knownLikers,
         });
         const threadRefs = [];
         if (postsState.posts) {
@@ -587,6 +621,35 @@ export class Hydrator {
         const threadContexts = await this.feed.getThreadContexts(threadRefs);
         return mergeStates(postsState, { threadContexts });
     }
+    async hydrateKnownLikers(options, ctx) {
+        const { subjectUris } = options;
+        if (!ctx.viewer || subjectUris.length === 0)
+            return undefined;
+        // Fail open.
+        try {
+            const { results } = await this.dataplane.getKnownLikers({
+                actorDid: ctx.viewer,
+                subjectUris,
+                limit: 3,
+            }, { signal: AbortSignal.timeout(100) });
+            const knownLikers = new HydrationMap();
+            for (const result of results) {
+                const subjectUri = result.subjectUri;
+                const dids = result.dids;
+                if (result.count > 0) {
+                    knownLikers.set(subjectUri, {
+                        count: result.count,
+                        actors: dids,
+                    });
+                }
+            }
+            return knownLikers;
+        }
+        catch (err) {
+            hydrationLogger.error({ err }, 'Failed to hydrate known likers');
+            return undefined;
+        }
+    }
     // app.bsky.feed.defs#generatorView
     // - feedgen
     //   - profile
@@ -1235,6 +1298,7 @@ export const mergeStates = (stateA, stateB) => {
         labelerAggs: mergeMaps(stateA.labelerAggs, stateB.labelerAggs),
         labelerViewers: mergeMaps(stateA.labelerViewers, stateB.labelerViewers),
         knownFollowers: mergeMaps(stateA.knownFollowers, stateB.knownFollowers),
+        knownLikers: mergeMaps(stateA.knownLikers, stateB.knownLikers),
         activitySubscriptions: mergeMaps(stateA.activitySubscriptions, stateB.activitySubscriptions),
         bidirectionalBlocks: mergeNestedMaps(stateA.bidirectionalBlocks, stateB.bidirectionalBlocks),
         verifications: mergeMaps(stateA.verifications, stateB.verifications),