feat: offline db for polls wip

isekovanic · isekovanic · commit c99706814802 · 2024-11-07T15:11:16.000+01:00
diff --git a/package/src/store/QuickSqliteClient.ts b/package/src/store/QuickSqliteClient.ts
@@ -30,7 +30,7 @@ import type { PreparedQueries, Table } from './types';
  *
  */
 export class QuickSqliteClient {
-  static dbVersion = 6;
+  static dbVersion = 7;
 
   static dbName = DB_NAME;
   static dbLocation = DB_LOCATION;
diff --git a/package/src/store/apis/getChannelMessages.ts b/package/src/store/apis/getChannelMessages.ts
@@ -8,7 +8,8 @@ import type { DefaultStreamChatGenerics } from '../../types/types';
 import { isBlockedMessage } from '../../utils/utils';
 import { mapStorableToMessage } from '../mappers/mapStorableToMessage';
 import { QuickSqliteClient } from '../QuickSqliteClient';
-import type { TableRowJoinedUser } from '../types';
+import { createSelectQuery } from '../sqlite-utils/createSelectQuery';
+import type { TableRow, TableRowJoinedUser } from '../types';
 
 export const getChannelMessages = <
   StreamChatGenerics extends DefaultStreamChatGenerics = DefaultStreamChatGenerics,
@@ -35,6 +36,21 @@ export const getChannelMessages = <
     }
     messageIdVsReactions[reaction.messageId].push(reaction);
   });
+  const messageIdsVsPolls: Record<string, TableRow<'poll'>> = {};
+  const pollsById: Record<string, TableRow<'poll'>> = {};
+  const messagesWithPolls = messageRows.filter((message) => !!message.poll_id);
+  const polls = QuickSqliteClient.executeSql.apply(
+    null,
+    createSelectQuery('poll', ['*'], {
+      id: messagesWithPolls.map((message) => message.poll_id),
+    }),
+  );
+  polls.forEach((poll) => {
+    pollsById[poll.id] = poll;
+  });
+  messagesWithPolls.forEach((message) => {
+    messageIdsVsPolls[message.poll_id] = pollsById[message.poll_id];
+  });
 
   // Populate the messages.
   const cidVsMessages: Record<string, MessageResponse<StreamChatGenerics>[]> = {};
@@ -48,6 +64,7 @@ export const getChannelMessages = <
         mapStorableToMessage<StreamChatGenerics>({
           currentUserId,
           messageRow: m,
+          pollRow: messageIdsVsPolls[m.poll_id],
           reactionRows: messageIdVsReactions[m.id],
         }),
       );
diff --git a/package/src/store/apis/updatePollMessage.ts b/package/src/store/apis/updatePollMessage.ts
@@ -1,5 +1,6 @@
 import type { PollResponse } from 'stream-chat';
 
+import { mapPollToStorable } from '../mappers/mapPollToStorable';
 import { QuickSqliteClient } from '../QuickSqliteClient';
 import { createSelectQuery } from '../sqlite-utils/createSelectQuery';
 import { createUpdateQuery } from '../sqlite-utils/createUpdateQuery';
@@ -14,29 +15,28 @@ export const updatePollMessage = ({
 }) => {
   const queries: PreparedQueries[] = [];
 
-  const messagesWithPoll = QuickSqliteClient.executeSql.apply(
+  const pollsFromDB = QuickSqliteClient.executeSql.apply(
     null,
-    createSelectQuery('messages', ['*'], {
-      poll_id: poll.id,
+    createSelectQuery('poll', ['*'], {
+      id: poll.id,
     }),
   );
 
-  for (const message of messagesWithPoll) {
-    const { latest_votes, own_votes } = JSON.parse(message.poll);
-    const storablePoll = JSON.stringify({
-      ...poll,
-      latest_votes,
-      own_votes,
+  for (const pollFromDB of pollsFromDB) {
+    const { latest_votes, own_votes } = pollFromDB;
+    console.log(own_votes);
+    const storablePoll = mapPollToStorable({
+      ...pollFromDB,
+      latest_votes: latest_votes ? JSON.parse(latest_votes) : [],
+      own_votes: own_votes ? JSON.parse(own_votes) : [],
     });
-    const storableMessage = { ...message, poll: storablePoll };
 
     queries.push(
-      createUpdateQuery('messages', storableMessage, {
-        id: message.id,
+      createUpdateQuery('poll', storablePoll, {
+        id: poll.id,
       }),
     );
     QuickSqliteClient.logger?.('info', 'updatePoll', {
-      message: storableMessage,
       poll: storablePoll,
     });
   }
diff --git a/package/src/store/apis/upsertMessages.ts b/package/src/store/apis/upsertMessages.ts
@@ -1,6 +1,7 @@
 import type { MessageResponse } from 'stream-chat';
 
 import { mapMessageToStorable } from '../mappers/mapMessageToStorable';
+import { mapPollToStorable } from '../mappers/mapPollToStorable';
 import { mapReactionToStorable } from '../mappers/mapReactionToStorable';
 import { mapUserToStorable } from '../mappers/mapUserToStorable';
 import { QuickSqliteClient } from '../QuickSqliteClient';
@@ -16,6 +17,7 @@ export const upsertMessages = ({
   const storableMessages: Array<ReturnType<typeof mapMessageToStorable>> = [];
   const storableUsers: Array<ReturnType<typeof mapUserToStorable>> = [];
   const storableReactions: Array<ReturnType<typeof mapReactionToStorable>> = [];
+  const storablePolls: Array<ReturnType<typeof mapPollToStorable>> = [];
 
   messages?.forEach((message: MessageResponse) => {
     storableMessages.push(mapMessageToStorable(message));
@@ -28,6 +30,9 @@ export const upsertMessages = ({
       }
       storableReactions.push(mapReactionToStorable(r));
     });
+    if (message.poll) {
+      storablePolls.push(mapPollToStorable(message.poll));
+    }
   });
 
   const finalQueries = [
@@ -36,11 +41,13 @@ export const upsertMessages = ({
     ...storableReactions.map((storableReaction) =>
       createUpsertQuery('reactions', storableReaction),
     ),
+    ...storablePolls.map((storablePoll) => createUpsertQuery('poll', storablePoll)),
   ];
 
   QuickSqliteClient.logger?.('info', 'upsertMessages', {
     flush,
     messages: storableMessages,
+    polls: storablePolls,
     reactions: storableReactions,
     users: storableUsers,
   });
diff --git a/package/src/store/mappers/mapMessageToStorable.ts b/package/src/store/mappers/mapMessageToStorable.ts
@@ -2,6 +2,8 @@ import type { FormatMessageResponse, MessageResponse } from 'stream-chat';
 
 import { mapDateTimeToStorable } from './mapDateTimeToStorable';
 
+import { mapPollToStorable } from './mapPollToStorable';
+
 import type { TableRow } from '../types';
 
 export const mapMessageToStorable = (
@@ -28,6 +30,8 @@ export const mapMessageToStorable = (
     ...extraData
   } = message;
 
+  const pollInMessage = poll && mapPollToStorable(poll);
+
   return {
     attachments: JSON.stringify(attachments),
     cid: cid || '',
@@ -36,12 +40,12 @@ export const mapMessageToStorable = (
     extraData: JSON.stringify(extraData),
     id,
     messageTextUpdatedAt: mapDateTimeToStorable(message_text_updated_at),
-    poll: JSON.stringify(poll),
     poll_id: poll_id || '',
     reactionGroups: JSON.stringify(reaction_groups),
     text,
     type,
     updatedAt: mapDateTimeToStorable(updated_at),
     userId: user?.id,
+    ...(pollInMessage ? { poll: pollInMessage } : {}),
   };
 };
diff --git a/package/src/store/mappers/mapPollToStorable.ts b/package/src/store/mappers/mapPollToStorable.ts
@@ -0,0 +1,53 @@
+import type { PollResponse } from 'stream-chat';
+
+import { mapDateTimeToStorable } from './mapDateTimeToStorable';
+
+import type { TableRow } from '../types';
+
+export const mapPollToStorable = (poll: PollResponse): TableRow<'poll'> => {
+  const {
+    allow_answers,
+    allow_user_suggested_options,
+    answers_count,
+    created_at,
+    created_by,
+    created_by_id,
+    description,
+    enforce_unique_vote,
+    id,
+    is_closed,
+    latest_answers,
+    latest_votes_by_option,
+    max_votes_allowed,
+    name,
+    options,
+    own_votes,
+    updated_at,
+    vote_count,
+    vote_counts_by_option,
+    voting_visibility,
+  } = poll;
+
+  return {
+    allow_answers,
+    allow_user_suggested_options,
+    answers_count,
+    created_at: mapDateTimeToStorable(created_at),
+    created_by: JSON.stringify(created_by), // decouple the users from the actual poll
+    created_by_id,
+    description,
+    enforce_unique_vote,
+    id,
+    is_closed,
+    latest_answers: JSON.stringify(latest_answers),
+    latest_votes_by_option: JSON.stringify(latest_votes_by_option),
+    max_votes_allowed,
+    name,
+    options: JSON.stringify(options),
+    own_votes: JSON.stringify(own_votes),
+    updated_at,
+    vote_count,
+    vote_counts_by_option: JSON.stringify(vote_counts_by_option),
+    voting_visibility,
+  };
+};
diff --git a/package/src/store/mappers/mapStorableToMessage.ts b/package/src/store/mappers/mapStorableToMessage.ts
@@ -1,30 +1,32 @@
 import type { MessageResponse } from 'stream-chat';
 
+import { mapStorableToPoll } from './mapStorableToPoll';
 import { mapStorableToReaction } from './mapStorableToReaction';
 
 import { mapStorableToUser } from './mapStorableToUser';
 
 import type { DefaultStreamChatGenerics } from '../../types/types';
 
-import type { TableRowJoinedUser } from '../types';
+import type { TableRow, TableRowJoinedUser } from '../types';
 
 export const mapStorableToMessage = <
   StreamChatGenerics extends DefaultStreamChatGenerics = DefaultStreamChatGenerics,
 >({
   currentUserId,
   messageRow,
+  pollRow,
   reactionRows,
 }: {
   currentUserId: string;
   messageRow: TableRowJoinedUser<'messages'>;
+  pollRow: TableRow<'poll'>;
   reactionRows: TableRowJoinedUser<'reactions'>[];
 }): MessageResponse<StreamChatGenerics> => {
   const {
     createdAt,
     deletedAt,
     extraData,
     messageTextUpdatedAt,
-    poll,
     poll_id,
     reactionGroups,
     updatedAt,
@@ -44,11 +46,11 @@ export const mapStorableToMessage = <
     latest_reactions: latestReactions,
     message_text_updated_at: messageTextUpdatedAt,
     own_reactions: ownReactions,
-    poll: JSON.parse(poll),
     poll_id,
     reaction_groups: reactionGroups ? JSON.parse(reactionGroups) : {},
     updated_at: updatedAt,
     user: mapStorableToUser(user),
+    ...(pollRow ? { poll: mapStorableToPoll(pollRow) } : {}),
     ...(extraData ? JSON.parse(extraData) : {}),
   };
 };
diff --git a/package/src/store/mappers/mapStorableToPoll.ts b/package/src/store/mappers/mapStorableToPoll.ts
@@ -0,0 +1,56 @@
+import { PollResponse, VotingVisibility } from 'stream-chat';
+
+import type { DefaultStreamChatGenerics } from '../../types/types';
+import type { TableRow } from '../types';
+
+export const mapStorableToPoll = <
+  StreamChatGenerics extends DefaultStreamChatGenerics = DefaultStreamChatGenerics,
+>(
+  pollRow: TableRow<'poll'>,
+): PollResponse<StreamChatGenerics> => {
+  const {
+    allow_answers,
+    allow_user_suggested_options,
+    answers_count,
+    created_at,
+    created_by,
+    created_by_id,
+    description,
+    enforce_unique_vote,
+    id,
+    is_closed,
+    latest_answers,
+    latest_votes_by_option,
+    max_votes_allowed,
+    name,
+    options,
+    own_votes,
+    updated_at,
+    vote_count,
+    vote_counts_by_option,
+    voting_visibility,
+  } = pollRow;
+
+  return {
+    allow_answers,
+    allow_user_suggested_options,
+    answers_count,
+    created_at,
+    created_by: JSON.parse(created_by),
+    created_by_id,
+    description,
+    enforce_unique_vote,
+    id,
+    is_closed,
+    latest_answers: JSON.parse(latest_answers),
+    latest_votes_by_option: JSON.parse(latest_votes_by_option),
+    max_votes_allowed,
+    name,
+    options: JSON.parse(options),
+    own_votes: own_votes ? JSON.parse(own_votes) : [],
+    updated_at,
+    vote_count,
+    vote_counts_by_option: JSON.parse(vote_counts_by_option),
+    voting_visibility: voting_visibility as VotingVisibility | undefined,
+  };
+};
diff --git a/package/src/store/schema.ts b/package/src/store/schema.ts
@@ -139,6 +139,38 @@ export const tables: Tables = {
       type: 'TEXT',
     },
   },
+  poll: {
+    columns: {
+      allow_answers: 'BOOLEAN DEFAULT FALSE',
+      allow_user_suggested_options: 'BOOLEAN DEFAULT FALSE',
+      answers_count: 'INTEGER DEFAULT 0',
+      created_at: 'TEXT',
+      created_by: 'TEXT',
+      created_by_id: 'TEXT',
+      description: 'TEXT',
+      enforce_unique_vote: 'BOOLEAN DEFAULT FALSE',
+      id: 'TEXT NOT NULL',
+      is_closed: 'BOOLEAN DEFAULT FALSE',
+      latest_answers: 'TEXT',
+      latest_votes_by_option: 'TEXT',
+      max_votes_allowed: 'INTEGER DEFAULT 1',
+      name: 'TEXT',
+      options: 'TEXT',
+      own_votes: 'TEXT',
+      updated_at: 'TEXT',
+      vote_count: 'INTEGER DEFAULT 0',
+      vote_counts_by_option: 'TEXT',
+      voting_visibility: 'TEXT',
+    },
+    indexes: [
+      {
+        columns: ['id'],
+        name: 'index_poll',
+        unique: false,
+      },
+    ],
+    primaryKey: ['id'],
+  },
   reactions: {
     columns: {
       createdAt: 'TEXT',
@@ -283,6 +315,28 @@ export type Schema = {
     payload: string;
     type: ValueOf<PendingTaskTypes>;
   };
+  poll: {
+    answers_count: number;
+    created_at: string;
+    created_by: string;
+    created_by_id: string;
+    enforce_unique_vote: boolean;
+    id: string;
+    latest_answers: string;
+    latest_votes_by_option: string;
+    max_votes_allowed: number;
+    name: string;
+    options: string;
+    updated_at: string;
+    vote_count: number;
+    vote_counts_by_option: string;
+    allow_answers?: boolean;
+    allow_user_suggested_options?: boolean;
+    description?: string;
+    is_closed?: boolean;
+    own_votes?: string;
+    voting_visibility?: string;
+  };
   reactions: {
     createdAt: string;
     messageId: string;

Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@ import type { PreparedQueries, Table } from './types';`
`30`	`30`	`*`
`31`	`31`	`*/`
`32`	`32`	`export class QuickSqliteClient {`
`33`		`- static dbVersion = 6;`
	`33`	`+ static dbVersion = 7;`
`34`	`34`
`35`	`35`	`static dbName = DB_NAME;`
`36`	`36`	`static dbLocation = DB_LOCATION;`