NgddlmZddlmZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZddlmZerddlZdd ZGdde ZdS)) annotations)datetime)Path) TYPE_CHECKINGIteratorListOptionalUnion)BaseChatLoader) ChatSession) HumanMessageN nanosecondsintreturnrc|dz }tddd}||z}tj|S)zConvert nanoseconds since 2001 to a datetime object. Args: nanoseconds (int): Nanoseconds since January 1, 2001. Returns: datetime: Datetime object. geAi)r timestamp fromtimestamp)rtimestamp_in_secondsreference_date_secondsactual_timestamps e/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/chat_loaders/imessage.py!nanoseconds_from_2001_to_datetimersQ',&dAq11;;==.0DD  !"2 3 33c\eZdZdZdddZedd Zedd ZddZddZ dS)IMessageChatLoaderaLoad chat sessions from the `iMessage` chat.db SQLite file. It only works on macOS when you have iMessage enabled and have the chat.db file. The chat.db file is likely located at ~/Library/Messages/chat.db. However, your terminal may not have permission to access this file. To resolve this, you can copy the file to a different location, change the permissions of the file, or grant full disk access for your terminal emulator in System Settings > Security and Privacy > Full Disk Access. NpathOptional[Union[str, Path]]cH|tjdz dz dz }t|tr|nt||_|jst d|jd ddl}dS#t$r}td|d}~wwxYw) a Initialize the IMessageChatLoader. Args: path (str or Path, optional): Path to the chat.db SQLite file. Defaults to None, in which case the default path ~/Library/Messages/chat.db will be used. NLibraryMessageszchat.dbzFile z not foundrzeThe sqlite3 module is required to load iMessage chats. Please install it with `pip install pysqlite3`)rhome isinstancedb_pathexistsFileNotFoundErrorsqlite3 ImportError)selfrr'es r__init__zIMessageChatLoader.__init__1s <9;;*Z7)CD)$55Ett4:: |""$$ F#$DDL$D$D$DEE E  NNNNN   A  s<B B! BB!attributed_bodybytesrstrc|dddd}|dd}}|ddkr%t|dddd}}||||zd d S) u Parse the attributedBody field of the message table for the text content of the message. The attributedBody field is a binary blob that contains the message content after the byte string b"NSString": 5 bytes 1-3 bytes `len` bytes ... | b"NSString" | preamble | `len` | contents | ... The 5 preamble bytes are always b"”„+" The size of `len` is either 1 byte or 3 bytes: - If the first byte in `len` is b"" then `len` is 3 bytes long. So the message length is the 2 bytes after, in little Endian. - Otherwise, the size of `len` is 1 byte, and the message length is that byte. Args: attributed_body (bytes): attributedBody field of the message table. Return: str: Text content of the message. sNSStringrNrlittlezutf-8ignore)errors)splitr from_bytesdecode)r,contentlengthstarts r_parse_attributed_bodyz)IMessageChatLoader._parse_attributed_bodyGs2"'' 44Q7; A 1:  NN71Q3<BBAEFuuv~-.55gh5OOOruse_chat_handle_tableboolc"d}d}|r|n|}d|dS)Nz JOIN chat_handle_join ON chat_message_join.chat_id = chat_handle_join.chat_id JOIN handle ON handle.ROWID = chat_handle_join.handle_idzE JOIN handle ON message.handle_id = handle.ROWID a@ SELECT message.date, handle.id, message.text, message.is_from_me, message.attributedBody FROM message JOIN chat_message_join ON message.ROWID = chat_message_join.message_id z` WHERE chat_message_join.chat_id = ? ORDER BY message.date ASC; )r=joins_w_chat_handlejoins_no_chat_handlejoinss r_get_session_queryz%IMessageChatLoader._get_session_queryfsC >  (=V##BV      rcursor'sqlite3.Cursor'chat_idrr c g}||}|||f|}|D]j\}}} } } | r| } n| r|| } n&|t || |t ||t| dkt|S)z Load a single chat session from the iMessage chat.db. Args: cursor: SQLite cursor object. chat_id (int): ID of the chat session to load. Returns: ChatSession: Loaded chat session. ) message_timemessage_time_as_datetimesender is_from_me)roler9additional_kwargs)messages) rDexecutefetchallr<appendr rr>r ) r)rEr=rGresultsqueryrOdaterKtextrLattributedBodyr9s r_load_single_chat_sessionz,IMessageChatLoader._load_single_chat_sessions')''(=>>uwj)))??$$>F   :D&$ N  55nEE NN#(,4U 55#)&*:&6&6 ''       G,,,,rIterator[ChatSession]c#Kddl} |j|j}n+#|j$r}t d|jd|d}~wwxYw|}d}|||}d}||d|D}|D]}| |||V| dS)z Lazy load the chat sessions from the iMessage chat.db and yield them in the required format. Yields: ChatSession: Loaded chat session. rNz Could not open iMessage DB file a,. Make sure your terminal emulator has disk access to this file. You can either copy the DB file to an accessible location or grant full disk access for your terminal emulator. You can grant full disk access for your terminal emulator in System Settings > Security and Privacy > Full Disk Access.zaSELECT name FROM sqlite_master WHERE type='table' AND name='chat_handle_join';zSELECT chat_id FROM message JOIN chat_message_join ON message.ROWID = chat_message_join.message_id GROUP BY chat_id ORDER BY MAX(date) DESC;cg|] }|d S)rr@).0rows r z0IMessageChatLoader.lazy_load..s888sCF888r) r'connectr$OperationalError ValueErrorrErPfetchonerQrXclose) r)r'connr*rErTis_chat_handle_join_existschat_idsrGs r lazy_loadzIMessageChatLoader.lazy_loadsD  "7?4<00DD'   Q4<QQQ  F u%+__%6%6"$ u88foo&7&7888  G002G     s AAA)N)rr)r,r-rr.)r=r>rr.)rErFr=r>rGrrr )rrY) __name__ __module__ __qualname____doc__r+ staticmethodr<rDrXrgr@rrrr%s  ,PPP\P<   \ <*-*-*-*-X,,,,,,rr)rrrr) __future__rrpathlibrtypingrrrr r langchain_core.chat_loadersr langchain_core.chat_sessionsr langchain_core.messagesr r'rrr@rrrss""""""AAAAAAAAAAAAAA666666444444000000NNN4444,xxxxxxxxxxr