Make useXXXContext return all states & actions for backward compatibility

Author: AhyoungRyu

src/modules/GroupChannel/context/GroupChannelProvider.tsx

@@ -1,4 +1,4 @@
-import React, { useMemo, useEffect, useRef, createContext, useContext } from 'react';
+import React, { useMemo, useEffect, useRef, createContext } from 'react';
 import {
   ReplyType as ChatReplyType,
 } from '@sendbird/chat/message';
@@ -336,12 +336,10 @@ const useGroupChannelStore = () => {
 };
 /**
  * Keep this function for backward compatibility.
- * @returns {ReturnType<typeof createStore<GroupChannelState>>}
  */
 const useGroupChannelContext = () => {
-  const store = useContext(GroupChannelContext);
-  if (!store) throw new Error('GroupChannelContext not found. Use within the GroupChannel module.');
-  return store.getState();
+  const { state, actions } = useGroupChannel();
+  return { ...state, ...actions };
 };
 
 export {

src/modules/MessageSearch/context/MessageSearchProvider.tsx

@@ -1,4 +1,4 @@
-import React, { createContext, useRef, useContext, useCallback, useEffect } from 'react';
+import React, { createContext, useRef, useCallback, useEffect } from 'react';
 import type { GroupChannel } from '@sendbird/chat/groupChannel';
 import { MessageSearchQuery } from '@sendbird/chat/message';
 import { ClientSentMessages } from '../../../types';
@@ -14,6 +14,7 @@ import useSearchStringEffect from './hooks/useSearchStringEffect';
 import { CoreMessageType } from '../../../utils';
 import { createStore } from '../../../utils/storeManager';
 import { useStore } from '../../../hooks/useStore';
+import useMessageSearch from './hooks/useMessageSearch';
 
 export interface MessageSearchProviderProps {
   channelUrl: string;
@@ -164,12 +165,10 @@ const MessageSearchProvider: React.FC<MessageSearchProviderProps> = ({
 
 /**
  * Keep this function for backward compatibility.
- * @returns {ReturnType<typeof createStore<MessageSearchState>>}
  */
 const useMessageSearchContext = () => {
-  const store = useContext(MessageSearchContext);
-  if (!store) throw new Error('MessageSearchContext not found. Use within the MessageSearch module.');
-  return store.getState();
+  const { state, actions } = useMessageSearch();
+  return { ...state, ...actions };
 };
 
 export {

src/modules/MessageSearch/context/__test__/MessageSearchProvider.spec.tsx

@@ -3,7 +3,7 @@ import { waitFor, act } from '@testing-library/react';
 import { renderHook } from '@testing-library/react-hooks';
 import { MessageSearchQuery } from '@sendbird/chat/message';
 
-import { MessageSearchProvider, useMessageSearchContext } from '../MessageSearchProvider';
+import { MessageSearchProvider } from '../MessageSearchProvider';
 import useMessageSearch from '../hooks/useMessageSearch';
 
 jest.mock('../../../../hooks/useSendbirdStateContext', () => ({
@@ -71,28 +71,38 @@ describe('MessageSearchProvider', () => {
       </MessageSearchProvider>
     );
 
-    const { result } = renderHook(() => useMessageSearchContext(), { wrapper });
+    const { result } = renderHook(() => useMessageSearch(), { wrapper });
 
-    expect(result.current.getState()).toEqual(expect.objectContaining(initialState));
+    expect(result.current.state).toEqual(expect.objectContaining(initialState));
   });
 
   it('updates state correctly when props change', async () => {
-    const wrapper = ({ children }) => (
-      <MessageSearchProvider channelUrl="test-channel">
+    const initialUrl = 'test-channel';
+    const newUrl = 'new-channel';
+
+    const wrapper = ({ children, channelUrl }) => (
+      <MessageSearchProvider channelUrl={channelUrl}>
         {children}
       </MessageSearchProvider>
     );
 
-    const { result } = renderHook(() => useMessageSearchContext(), { wrapper });
+    const { result, rerender } = renderHook(
+      () => useMessageSearch(),
+      {
+        wrapper,
+        initialProps: { channelUrl: initialUrl, children: null },
+      },
+    );
 
-    expect(result.current.getState().channelUrl).toBe('test-channel');
+    expect(result.current.state.channelUrl).toBe(initialUrl);
 
     await act(async () => {
-      result.current.setState({ channelUrl: 'new-channel' });
+      rerender({ channelUrl: newUrl, children: null });
+
       await waitFor(() => {
-        const newState = result.current.getState();
-        expect(newState.channelUrl).toBe('new-channel');
         // Verify other states remain unchanged
+        const newState = result.current.state;
+        expect(newState.channelUrl).toBe(newUrl);
         expect(newState.allMessages).toEqual(initialState.allMessages);
         expect(newState.loading).toBe(initialState.loading);
         expect(newState.isQueryInvalid).toBe(initialState.isQueryInvalid);