Set dev mode auth headers

graphql/client.tsx

@@ -1,4 +1,10 @@
-import React, { ReactNode, createContext, useContext, useState } from "react";
+import React, {
+	ReactNode,
+	createContext,
+	useContext,
+	useState,
+	useMemo,
+} from "react";
 import {
 	ApolloClient,
 	ApolloLink,
@@ -13,31 +19,33 @@ type Props = {
 	children: ReactNode;
 };
 
-const AuthHeaderContext = createContext();
-
-export const AuthHeaderProvider = ({ children }) => {
-	const [authHeader, setAuthHeader] = useState({ key: "", value: "" });
-
-	return (
-		<AuthHeaderContext.Provider value={{ authHeader, setAuthHeader }}>
-			{children}
-		</AuthHeaderContext.Provider>
-	);
-};
+export const AuthHeaderContext = createContext<
+	| {
+			authHeader: { key: string; value: string };
+			setAuthHeader: React.Dispatch<
+				React.SetStateAction<{ key: string; value: string }>
+			>;
+	  }
+	| undefined
+>(undefined);
 
 // Hook to use the auth header context
-export const useAuthHeader = () => useContext(AuthHeaderContext);
+export const useAuthHeader = () => {
+	const context = useContext(AuthHeaderContext);
+	if (!context) {
+		throw new Error("useAuthHeader must be used within a ClientProvider");
+	}
+	return context;
+};
 
-const ClientProvider: React.FC<Props> = ({ children }) => {
+export const ClientProvider: React.FC<Props> = ({ children }) => {
+	const [authHeader, setAuthHeader] = useState({ key: "", value: "" });
 	const httpLink = new HttpLink({
-		uri: API_URI,
+		uri: API_URI || "https://api-dev.railbird.ai/graphql",
 	});
 	const cache = new InMemoryCache({});
 
-	const { authHeader } = useAuthHeader();
-
 	const authMiddleware = new ApolloLink((operation, forward) => {
-		// Retrieve auth key and value from context
 		const { key, value } = authHeader;
 
 		if (key && value) {
@@ -55,13 +63,16 @@ const ClientProvider: React.FC<Props> = ({ children }) => {
 	const client = useMemo(
 		() =>
 			new ApolloClient({
-				link: from([authMiddleware, httpLink]), // Chain the middleware with the httpLink
+				// Chain the middleware with the httpLink
+				link: from([authMiddleware, httpLink]),
 				cache: new InMemoryCache(),
 			}),
 		[authHeader],
 	);
 
-	return <ApolloProvider client={client}>{children}</ApolloProvider>;
+	return (
+		<AuthHeaderContext.Provider value={{ authHeader, setAuthHeader }}>
+			<ApolloProvider client={client}>{children}</ApolloProvider>
+		</AuthHeaderContext.Provider>
+	);
 };
-
-export default ClientProvider;