feat(webhook): call CreateOrUpdateCooccurrence to create cooccurrences

cofacts · Jan 3, 2024 · 8400d41 · 8400d41
1 parent cee0edc
commit 8400d41
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 14 deletions.
diff --git a/src/webhook/handlers/askingArticleSubmissionConsent.ts b/src/webhook/handlers/askingArticleSubmissionConsent.ts
@@ -27,6 +27,7 @@ import {
   searchText,
   searchMedia,
   createCooccurredSearchResultsCarouselContents,
+  setMostSimilarArticlesAsCooccurrence,
 } from './utils';
 
 // Input should be array of context.msgs idx. Empty if the user does not want to submit.
@@ -152,16 +153,17 @@ const askingArticleSubmissionConsent: ChatbotPostbackHandler = async ({
   );
 
   if (context.msgs.length > 1) {
-    // Continue with the rest of the messages
-    // FIXME: implement this
-
+    // Search again, this time all messages should be in the database.
+    // Most similar articles for each respective searched message will be set as cooccurrence.
+    //
     const searchResults = await Promise.all(
       context.msgs.map(async (msg) =>
         msg.type === 'text'
           ? searchText(msg.text)
           : searchMedia(getLineContentProxyURL(msg.id), userId)
       )
     );
+    await setMostSimilarArticlesAsCooccurrence(searchResults, userId);
 
     return {
       context,

diff --git a/src/webhook/handlers/askingCooccurrence.ts b/src/webhook/handlers/askingCooccurrence.ts
@@ -3,7 +3,6 @@ import { msgid, ngettext, t } from 'ttag';
 import { FlexSpan } from '@line/bot-sdk';
 
 import ga from 'src/lib/ga';
-import gql from 'src/lib/gql';
 import { ChatbotPostbackHandler } from 'src/types/chatbotState';
 
 import {
@@ -16,6 +15,7 @@ import {
   getLineContentProxyURL,
   createPostbackAction,
   createCooccurredSearchResultsCarouselContents,
+  setMostSimilarArticlesAsCooccurrence,
 } from './utils';
 
 const inputSchema = z.enum([POSTBACK_NO, POSTBACK_YES]);
@@ -26,16 +26,6 @@ const IN_DB_THRESHOLD = 0.8;
 /** Postback input type for ASKING_ARTICLE_SOURCE state handler */
 export type Input = z.infer<typeof inputSchema>;
 
-export function setCooccurrences(articleIds: string[]) {
-  return gql`
-    mutation SetCooccurrences($articleIds: [String!]!) {
-      CreateOrUpdateCooccurrence(articleIds: $articleIds) {
-        id
-      }
-    }
-  `({ articleIds });
-}
-
 const askingCooccurence: ChatbotPostbackHandler = async ({
   context,
   postbackData: { state, input: postbackInput },
@@ -178,6 +168,9 @@ const askingCooccurence: ChatbotPostbackHandler = async ({
         };
       }
 
+      // All messages in DB and thus can be set as cooccurrence.
+      await setMostSimilarArticlesAsCooccurrence(searchResults, userId);
+
       // Get first few search results for each message, and make at most 10 options
       //
 

diff --git a/src/webhook/handlers/utils.ts b/src/webhook/handlers/utils.ts
@@ -1275,3 +1275,26 @@ export function createCooccurredSearchResultsCarouselContents(
       )
   );
 }
+
+/**
+ * Mark the most similar item in the search of each searched messages as a cooccurrence
+ *
+ * @param searchResults - search results from searchMedia() or searchText(), with most similar item
+ *                        of each searched items in the first edge.
+ * @param userId - user that observes this cooccurrence
+ */
+export function setMostSimilarArticlesAsCooccurrence(
+  searchResults: (SearchMediaResult | SearchTextResult)[],
+  userId: string
+) {
+  const articleIds = searchResults.map(
+    (searchResult) => searchResult.edges[0].node.id
+  );
+  return gql`
+    mutation SetCooccurrences($articleIds: [String!]!) {
+      CreateOrUpdateCooccurrence(articleIds: $articleIds) {
+        id
+      }
+    }
+  `({ articleIds }, { userId });
+}