updating the session refreshes queries

packages/houdini-react/src/runtime/hooks/useDocumentStore.ts

@@ -25,8 +25,7 @@ export function useDocumentStore<
 	...observeParams
 }: UseDocumentStoreParams<_Artifact, _Data, _Input>): [
 	QueryResult<_Data, _Input>,
-	DocumentStore<_Data, _Input>,
-	(store: DocumentStore<_Data, _Input>) => void
+	DocumentStore<_Data, _Input>
 ] {
 	const client = useClient()
 	const isMountedRef = useIsMountedRef()
@@ -43,6 +42,13 @@ export function useDocumentStore<
 
 	const box = React.useRef(observer.state)
 
+	// if the observer changes, we need to track the new one
+	if (obs && obs !== observer) {
+		box.current = obs.state
+		setObserver(obs)
+	}
+
+	// the function that registers a new subscription for the observer
 	const subscribe: any = React.useCallback(
 		(fn: () => void) => {
 			return observer.subscribe((val) => {
@@ -62,5 +68,5 @@ export function useDocumentStore<
 		() => box.current
 	)
 
-	return [storeValue!, observer, setObserver]
+	return [storeValue!, observer]
 }

packages/houdini-react/src/runtime/hooks/useDocumentSubscription.ts

@@ -20,12 +20,8 @@ export function useDocumentSubscription<
 	variables: _Input
 	disabled?: boolean
 	send?: Partial<SendParams>
-}): [
-	QueryResult<_Data, _Input> & { parent?: string | null },
-	DocumentStore<_Data, _Input>,
-	(store: DocumentStore<_Data, _Input>) => void
-] {
-	const [storeValue, observer, setObserver] = useDocumentStore<_Data, _Input>({
+}): [QueryResult<_Data, _Input> & { parent?: string | null }, DocumentStore<_Data, _Input>] {
+	const [storeValue, observer] = useDocumentStore<_Data, _Input>({
 		artifact,
 		...observeParams,
 	})
@@ -58,6 +54,5 @@ export function useDocumentSubscription<
 			...storeValue,
 		},
 		observer,
-		setObserver,
 	]
 }

packages/houdini-react/src/runtime/routing/Router.tsx

@@ -447,7 +447,7 @@ export function RouterContextProvider({
 				ssr_signals,
 				last_variables,
 				session,
-				setSession: (newSession) => setSession((old) => ({ ...old, newSession })),
+				setSession: (newSession) => setSession((old) => ({ ...old, ...newSession })),
 			}}
 		>
 			{children}