feat: polls rewrite

src/channel.ts

@@ -1079,6 +1079,8 @@ export class Channel<StreamChatGenerics extends ExtendableGenerics = DefaultGene
       }),
     };
 
+    this.getClient().polls.hydratePollCache(messageSet.messages, true);
+
     const areCapabilitiesChanged =
       [...(state.channel.own_capabilities || [])].sort().join() !==
       [...(Array.isArray(this.data?.own_capabilities) ? (this.data?.own_capabilities as string[]) : [])].sort().join();
@@ -1462,31 +1464,6 @@ export class Channel<StreamChatGenerics extends ExtendableGenerics = DefaultGene
           };
         }
         break;
-      case 'poll.updated':
-        if (event.poll) {
-          channelState.updatePoll(event.poll, event.message?.id || '');
-        }
-        break;
-      case 'poll.vote_casted':
-        if (event.poll_vote && event.poll) {
-          channelState.addPollVote(event.poll_vote, event.poll, event.message?.id || '');
-        }
-        break;
-      case 'poll.vote_changed':
-        if (event.poll_vote && event.poll) {
-          channelState.updatePollVote(event.poll_vote, event.poll, event.message?.id || '');
-        }
-        break;
-      case 'poll.vote_removed':
-        if (event.poll_vote && event.poll) {
-          channelState.removePollVote(event.poll_vote, event.poll, event.message?.id || '');
-        }
-        break;
-      case 'poll.closed':
-        if (event.message) {
-          channelState.addMessageSorted(event.message, false, false);
-        }
-        break;
       case 'reaction.new':
         if (event.message && event.reaction) {
           event.message = channelState.addReaction(event.reaction, event.message);

src/channel_state.ts

@@ -9,8 +9,6 @@ import {
   MessageSet,
   MessageSetType,
   PendingMessageResponse,
-  PollResponse,
-  PollVote,
   ReactionResponse,
   UserResponse,
 } from './types';
@@ -493,96 +491,6 @@ export class ChannelState<StreamChatGenerics extends ExtendableGenerics = Defaul
     return { removed: result.length < msgArray.length, result };
   };
 
-  // this handles the case when vote on poll is changed
-  updatePollVote = (
-    pollVote: PollVote<StreamChatGenerics>,
-    poll: PollResponse<StreamChatGenerics>,
-    messageId: string,
-  ) => {
-    const message = this.findMessage(messageId);
-    if (!message) return;
-
-    if (message.poll_id !== pollVote.poll_id) return;
-
-    const updatedPoll = { ...poll };
-    let ownVotes = [...(message.poll?.own_votes || [])];
-
-    if (pollVote.user_id === this._channel.getClient().userID) {
-      if (pollVote.option_id && poll.enforce_unique_vote) {
-        // remove all previous votes where option_id is not empty
-        ownVotes = ownVotes.filter((vote) => !vote.option_id);
-      } else if (pollVote.answer_text) {
-        // remove all previous votes where option_id is empty
-        ownVotes = ownVotes.filter((vote) => vote.answer_text);
-      }
-
-      ownVotes.push(pollVote);
-    }
-
-    updatedPoll.own_votes = ownVotes as PollVote<StreamChatGenerics>[];
-    const newMessage = { ...message, poll: updatedPoll };
-
-    this.addMessageSorted((newMessage as unknown) as MessageResponse<StreamChatGenerics>, false, false);
-  };
-
-  addPollVote = (pollVote: PollVote<StreamChatGenerics>, poll: PollResponse<StreamChatGenerics>, messageId: string) => {
-    const message = this.findMessage(messageId);
-    if (!message) return;
-
-    if (message.poll_id !== pollVote.poll_id) return;
-
-    const updatedPoll = { ...poll };
-    const ownVotes = [...(message.poll?.own_votes || [])];
-
-    if (pollVote.user_id === this._channel.getClient().userID) {
-      ownVotes.push(pollVote);
-    }
-
-    updatedPoll.own_votes = ownVotes as PollVote<StreamChatGenerics>[];
-    const newMessage = { ...message, poll: updatedPoll };
-
-    this.addMessageSorted((newMessage as unknown) as MessageResponse<StreamChatGenerics>, false, false);
-  };
-
-  removePollVote = (
-    pollVote: PollVote<StreamChatGenerics>,
-    poll: PollResponse<StreamChatGenerics>,
-    messageId: string,
-  ) => {
-    const message = this.findMessage(messageId);
-    if (!message) return;
-
-    if (message.poll_id !== pollVote.poll_id) return;
-
-    const updatedPoll = { ...poll };
-    const ownVotes = [...(message.poll?.own_votes || [])];
-    if (pollVote.user_id === this._channel.getClient().userID) {
-      const index = ownVotes.findIndex((vote) => vote.option_id === pollVote.option_id);
-      if (index > -1) {
-        ownVotes.splice(index, 1);
-      }
-    }
-
-    updatedPoll.own_votes = ownVotes as PollVote<StreamChatGenerics>[];
-
-    const newMessage = { ...message, poll: updatedPoll };
-    this.addMessageSorted((newMessage as unknown) as MessageResponse<StreamChatGenerics>, false, false);
-  };
-
-  updatePoll = (poll: PollResponse<StreamChatGenerics>, messageId: string) => {
-    const message = this.findMessage(messageId);
-    if (!message) return;
-
-    const updatedPoll = {
-      ...poll,
-      own_votes: [...(message.poll?.own_votes || [])],
-    };
-
-    const newMessage = { ...message, poll: updatedPoll };
-
-    this.addMessageSorted((newMessage as unknown) as MessageResponse<StreamChatGenerics>, false, false);
-  };
-
   /**
    * Updates the message.user property with updated user object, for messages.
    *