fix: use cookie store for theme as well

apps/start/src/hooks/use-cookie-store.tsx

@@ -1,11 +1,14 @@
 import { useRouteContext } from '@tanstack/react-router';
 import { createServerFn, createServerOnlyFn } from '@tanstack/react-start';
 import { getCookies, setCookie } from '@tanstack/react-start/server';
+import { pick } from 'ramda';
 import { useMemo, useState } from 'react';
 import { z } from 'zod';
 
+const VALID_COOKIES = ['ui-theme', 'chartType', 'range'] as const;
+
 const setCookieFn = createServerFn({ method: 'POST' })
-  .inputValidator(z.object({ key: z.string(), value: z.string() }))
+  .inputValidator(z.object({ key: z.enum(VALID_COOKIES), value: z.string() }))
   .handler(({ data: { key, value } }) => {
     setCookie(key, value);
   });
@@ -13,10 +16,13 @@ const setCookieFn = createServerFn({ method: 'POST' })
 // Called in __root.tsx beforeLoad hook to get cookies from the server
 // And recieved with useRouteContext in the client
 export const getCookiesFn = createServerFn({ method: 'GET' }).handler(() =>
-  getCookies(),
+  pick(VALID_COOKIES, getCookies()),
 );
 
-export function useCookieStore<T>(key: string, defaultValue: T) {
+export function useCookieStore<T>(
+  key: (typeof VALID_COOKIES)[number],
+  defaultValue: T,
+) {
   const { cookies } = useRouteContext({ strict: false });
   const [value, setValue] = useState<T>((cookies?.[key] ?? defaultValue) as T);