fix/ensemble-response-mapping (#969)

fix ensemble response mapping with screenData

Author: sagardspeed2

.changeset/smooth-parents-type.md

@@ -0,0 +1,6 @@
+---
+"@ensembleui/react-framework": patch
+"@ensembleui/react-runtime": patch
+---
+
+fix ensemble api response mapping with screendata

packages/framework/src/api/data.ts

@@ -39,8 +39,7 @@ export const invokeAPI = async (
     // Now, because the API exists, set its state to loading
     set(update, api.name, {
       isLoading: true,
-      isError: false,
-      isSuccess: false,
+      statusCode: undefined,
     });
     setter(screenDataAtom, update);
   }

packages/framework/src/hooks/useScreenContext.tsx

@@ -110,7 +110,7 @@ export const useScreenContext = ():
   const screenContext = useAtomValue(screenAtom);
   const setDataAtom = useSetAtom(screenDataAtom);
   const setData = useCallback(
-    (name: string, response: Response | WebSocketConnection) => {
+    (name: string, response: Partial<Response> | WebSocketConnection) => {
       const data = screenContext.data;
       data[name] = response;
       setDataAtom(clone(data));

packages/framework/src/hooks/useScreenData.ts

@@ -16,7 +16,10 @@ export const useScreenData = (): {
   apis?: EnsembleAPIModel[];
   sockets?: EnsembleSocketModel[];
   data: ScreenContextData;
-  setData: (name: string, response: Response | WebSocketConnection) => void;
+  setData: (
+    name: string,
+    response: Partial<Response> | WebSocketConnection,
+  ) => void;
   mockResponses: {
     [apiName: string]: EnsembleMockResponse | string | undefined;
   };
@@ -41,7 +44,7 @@ export const useScreenData = (): {
   const mockResponses = useEvaluate(apiMockResponses);
 
   const setData = useCallback(
-    (name: string, response: Response | WebSocketConnection) => {
+    (name: string, response: Partial<Response> | WebSocketConnection) => {
       if (isEqual(data[name], response)) {
         return;
       }

packages/framework/src/state/screen.ts

@@ -15,12 +15,15 @@ export interface ScreenContextDefinition {
 }
 
 export interface ScreenContextData {
-  [key: string]: Response | WebSocketConnection;
+  [key: string]: Partial<Response> | WebSocketConnection;
 }
 
 export interface ScreenContextActions {
   setWidget: (id: string, state: WidgetState) => void;
-  setData: (name: string, response: Response | WebSocketConnection) => void;
+  setData: (
+    name: string,
+    response: Partial<Response> | WebSocketConnection,
+  ) => void;
   setCustom: (id: string, data: unknown) => void;
 }
 

packages/runtime/src/runtime/hooks/useEnsembleAction.tsx

@@ -195,15 +195,13 @@ export const useInvokeAPI: EnsembleActionHook<InvokeAPIAction> = (action) => {
       // Set initial loading state
       setData(currentApi.name, {
         isLoading: true,
-        isSuccess: false,
-        isError: false,
+        statusCode: undefined,
       });
 
       if (action.id) {
         setData(action.id, {
           isLoading: true,
-          isSuccess: false,
-          isError: false,
+          statusCode: undefined,
         });
       }