fix type issue

Psiphon-Inc · Sep 19, 2024 · 61dfc51 · 61dfc51
1 parent 787ee87
commit 61dfc51
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/src/psiphon/inproxy.ts b/src/psiphon/inproxy.ts
@@ -45,7 +45,7 @@ export const InProxyErrorSchema = z.object({
     action: z.enum(["inProxyMustUpgrade"]),
 });
 
-const InProxyStatusSchema = z.object({
+export const InProxyStatusSchema = z.object({
     status: z.enum(["running", "stopped", "unknown"]),
 });
 

diff --git a/src/psiphon/mockContext.tsx b/src/psiphon/mockContext.tsx
@@ -6,6 +6,7 @@ import {
     InProxyActivityStats,
     InProxyParameters,
     InProxyStatus,
+    InProxyStatusSchema,
     getDefaultInProxyParameters,
     getZeroedInProxyActivityStats,
 } from "@/src/psiphon/inproxy";
@@ -252,13 +253,17 @@ export function InProxyProvider({ children }: { children: React.ReactNode }) {
     }
 
     const getInProxyStatus = React.useCallback(() => {
+        let state;
         if (!inProxyState.synced) {
-            return { status: "unknown" };
+            state = { status: "unknown" };
         } else {
-            return inProxyState.running
-                ? { status: "running" }
-                : { status: "stopped" };
+            if (inProxyState.running) {
+                state = { status: "running" };
+            } else {
+                state = { status: "stopped" };
+            }
         }
+        return InProxyStatusSchema.parse(state);
     }, [inProxyState]);
 
     async function sendFeedback(): Promise<void> {