add date function

Author: jonluca

electron-src/data/database.ts

@@ -303,6 +303,24 @@ export class SQLDatabase {
     return this.enhanceMessageResponses<typeof messages[number]>(allMessages);
   };
 
+  private addDateToMessage = (message: {
+    date?: number | null;
+    date_delivered?: number | null;
+    date_read?: number | null;
+    date_obj?: Date;
+    date_obj_delivered?: Date;
+    date_obj_read?: Date;
+  }) => {
+    if (message.date) {
+      message.date_obj = new Date(message.date / 1000000 + 978307200000);
+    }
+    if (message.date_delivered) {
+      message.date_obj_delivered = new Date(message.date_delivered / 1000000 + 978307200000);
+    }
+    if (message.date_read) {
+      message.date_obj_read = new Date(message.date_read / 1000000 + 978307200000);
+    }
+  };
   private enhanceMessageResponses = async <T extends JoinedMessageType = JoinedMessageType>(messages: T[]) => {
     type EnhancedMessage = T & {
       attachmentMessages?: EnhancedMessage[];
@@ -315,15 +333,7 @@ export class SQLDatabase {
 
     await Promise.all(
       enhancedMessages.map(async (message) => {
-        if (message.date) {
-          message.date_obj = new Date(message.date / 1000000 + 978307200000);
-        }
-        if (message.date_delivered) {
-          message.date_obj_delivered = new Date(message.date_delivered / 1000000 + 978307200000);
-        }
-        if (message.date_read) {
-          message.date_obj_read = new Date(message.date_read / 1000000 + 978307200000);
-        }
+        this.addDateToMessage(message);
         if (message.text) {
           message.text = message.text.replace(/[\u{FFFC}-\u{FFFD}]/gu, "");
         }
@@ -608,6 +618,16 @@ export class SQLDatabase {
       topEmojis,
     };
   };
+
+  getMessageDates = async (year: number, chatIds?: number[]) => {
+    const query = this.getMessageQueryByYear(year, chatIds).select("date");
+    const data = await query.execute();
+    for (const message of data) {
+      // maybe this is slow and its better not to pay serialization cost? idk
+      this.addDateToMessage(message);
+    }
+    return data[0];
+  };
 }
 
 const db = new SQLDatabase();

package.json

@@ -3,7 +3,7 @@
   "author": "JonLuca DeCaro <mimessage@jonlu.ca>",
   "main": "build/electron-src/index.js",
   "name": "Mimessage",
-  "version": "0.1.13",
+  "version": "0.1.14",
   "productName": "Mimessage",
   "description": "Apple Messages UI alternative, with export, search, and more.",
   "scripts": {

src/components/message/BaseAvatar.tsx

@@ -7,7 +7,6 @@ import type { AvatarClasses } from "@mui/material/Avatar/avatarClasses";
 import type { SxProps } from "@mui/system";
 import type { OverridableStringUnion } from "@mui/types";
 import type { OverrideProps } from "@mui/material/OverridableComponent";
-import { unstable_generateUtilityClasses as generateUtilityClasses } from "@mui/utils";
 import { useThemeProps } from "@mui/material/styles";
 const defaultGenerator = (componentName: string) => componentName;
 
@@ -64,15 +63,7 @@ function generateUtilityClass(componentName: string, slot: string, globalStatePr
 function getAvatarUtilityClass(slot: string) {
   return generateUtilityClass("MuiAvatar", slot);
 }
-const avatarClasses = generateUtilityClasses("MuiAvatar", [
-  "root",
-  "colorDefault",
-  "circular",
-  "rounded",
-  "square",
-  "img",
-  "fallback",
-]);
+
 export interface AvatarPropsVariantOverrides {}
 
 export interface AvatarTypeMap<P = object> {

src/components/wrapped/SelectedWrap.tsx

@@ -110,6 +110,7 @@ const TwoSidedSection = ({
 };
 const MessageCount = () => {
   const { data: wrappedStats } = useWrappedStats();
+  const chatId = useMimessage((state) => state.chatId);
 
   const chatMap = useChatMap();
   const contacts = React.useMemo(() => {
@@ -142,10 +143,12 @@ const MessageCount = () => {
           <SectionHeader>Received</SectionHeader>
           <Typography sx={sx}>{(wrappedStats?.messageCount?.received || 0).toLocaleString()}</Typography>
         </SectionWrapper>
-        <SectionWrapper sx={wrapperStyle}>
-          <SectionHeader>People</SectionHeader>
-          <Typography sx={sx}>{(uniqueContacts?.size || 0).toLocaleString()}</Typography>
-        </SectionWrapper>
+        {!chatId && (
+          <SectionWrapper sx={wrapperStyle}>
+            <SectionHeader>People</SectionHeader>
+            <Typography sx={sx}>{(uniqueContacts?.size || 0).toLocaleString()}</Typography>
+          </SectionWrapper>
+        )}
       </Box>
     </>
   );

src/hooks/dataHooks.ts

@@ -16,6 +16,7 @@ import type { GlobalSearchResponse } from "../interfaces";
 import { shallow } from "zustand/shallow";
 import type { SemanticSearchStats } from "../interfaces";
 import type { SlowWrappedStats } from "../interfaces";
+import type { MessageDates } from "../interfaces";
 
 const ipcRenderer = global.ipcRenderer;
 const useDbChatList = () => {
@@ -441,6 +442,23 @@ export const useSlowWrappedStats = () => {
     { enabled: isInWrapped },
   );
 };
+export const useWrappedDates = () => {
+  const wrappedYear = useMimessage((state) => state.wrappedYear);
+  const chatId = useMimessage((state) => state.chatId);
+  const chatMap = useChatMap();
+  const chat = chatMap?.get(chatId!);
+  const ids = chatId ? chat?.sameParticipantChatIds || [chatId] : null;
+  const isInWrapped = useMimessage((state) => state.isInWrapped);
+  return useQuery<MessageDates>(
+    ["getMessageDates", wrappedYear, ids],
+    async () => {
+      const resp = (await ipcRenderer.invoke("getMessageDates", wrappedYear, ids)) as MessageDates;
+      console.log(resp);
+      return resp;
+    },
+    { enabled: isInWrapped },
+  );
+};
 
 interface Permissions {
   contactsStatus: PermissionType | "not determined";

src/interfaces/index.ts

@@ -4,6 +4,7 @@ import type { Contact } from "electron-mac-contacts";
 export type ChatList = NonNullable<Awaited<ReturnType<SQLDatabase["getChatList"]>>>;
 export type WrappedStats = NonNullable<Awaited<ReturnType<SQLDatabase["calculateWrappedStats"]>>>;
 export type SlowWrappedStats = NonNullable<Awaited<ReturnType<SQLDatabase["calculateSlowWrappedStats"]>>>;
+export type MessageDates = NonNullable<Awaited<ReturnType<SQLDatabase["getMessageDates"]>>>;
 export type MessagesForChat = NonNullable<Awaited<ReturnType<SQLDatabase["getMessagesForChatId"]>>>;
 export type GlobalSearchResponse = NonNullable<Awaited<ReturnType<SQLDatabase["fullTextMessageSearch"]>>>;
 export type SemanticSearchStats = NonNullable<Awaited<ReturnType<SQLDatabase["calculateSemanticSearchStats"]>>>;