fix: all underlying offline store issues with polls

Author: isekovanic

package/src/components/Chat/hooks/handleEventToSyncDB.ts

@@ -233,11 +233,13 @@ export const handleEventToSyncDB = <
       'poll.vote_removed',
     ].includes(type)
   ) {
-    const poll = event.poll;
+    const { poll, poll_vote } = event;
     if (poll) {
       return updatePollMessage({
+        client,
         flush,
         poll,
+        poll_vote,
       });
     }
   }

package/src/store/apis/updatePollMessage.ts

@@ -1,18 +1,25 @@
-import type { PollResponse } from 'stream-chat';
+import { isVoteAnswer, PollAnswer, PollResponse, PollVote, StreamChat } from 'stream-chat';
 
+import { DefaultStreamChatGenerics } from '../../types/types';
 import { mapPollToStorable } from '../mappers/mapPollToStorable';
 import { mapStorableToPoll } from '../mappers/mapStorableToPoll';
 import { QuickSqliteClient } from '../QuickSqliteClient';
 import { createSelectQuery } from '../sqlite-utils/createSelectQuery';
 import { createUpdateQuery } from '../sqlite-utils/createUpdateQuery';
 import type { PreparedQueries } from '../types';
 
-export const updatePollMessage = ({
+export const updatePollMessage = <
+  StreamChatGenerics extends DefaultStreamChatGenerics = DefaultStreamChatGenerics,
+>({
+  client,
   flush = true,
   poll,
+  poll_vote,
 }: {
-  poll: PollResponse;
+  client: StreamChat<StreamChatGenerics>;
+  poll: PollResponse<StreamChatGenerics>;
   flush?: boolean;
+  poll_vote?: PollVote<StreamChatGenerics> | PollAnswer<StreamChatGenerics>;
 }) => {
   const queries: PreparedQueries[] = [];
 
@@ -25,12 +32,19 @@ export const updatePollMessage = ({
 
   for (const pollFromDB of pollsFromDB) {
     const serializedPoll = mapStorableToPoll(pollFromDB);
-    const { latest_votes, own_votes } = serializedPoll;
-    console.log(own_votes);
+    const { latest_answers = [], own_votes = [] } = serializedPoll;
+    const newOwnVotes =
+      poll_vote && poll_vote.user?.id === client.userID
+        ? [poll_vote, ...own_votes.filter((vote) => vote.id !== poll_vote.id)]
+        : own_votes;
+    const newLatestAnswers =
+      poll_vote && isVoteAnswer(poll_vote)
+        ? [poll_vote, ...latest_answers.filter((answer) => answer.id !== poll_vote?.id)]
+        : latest_answers;
     const storablePoll = mapPollToStorable({
       ...poll,
-      // latest_votes: latest_votes ?? [],
-      // own_votes: own_votes ?? [],
+      latest_answers: newLatestAnswers,
+      own_votes: newOwnVotes,
     });
 
     console.log('STORABLE POLL: ', storablePoll);