createJSONStorage with superjson and Next.js throws "localStorage is not defined"

[0xaa55h]

Hello, I'm currently trying to use superjson instead of normal JSON class to serialize/deserialize my jsons, but for the following code:

// there is no "use client" or anything on top of the file(nor it does help)!
export const applicationProgress = atomWithStorage<
  ApplicationProgress | undefined
>(
  "application-progress",
  undefined,
  createJSONStorage(() => localStorage, {
    reviver: superjson.parse,
    replacer: superjson.stringify,
  }),
  {
    getOnInit: true,
  },
);

Results with following error:

I can't tell if this is an actual bug or my stupidness, therefore I chose Q&A and not bug report. Any idea how can this be fixed?

[dai-shi]

How about this?

- () => localStorage
+ () => typeof localStorage !== 'undefined' ? localStorage : undefined

[0xaa55h]

That doesn't work as the function can't return undefined/null, only AsyncStringStorage or SyncStringStorage

[dai-shi]

Oops, good catch.

-   createJSONStorage(() => localStorage, {
+   typeof localStorage === 'undefined' ? undefined : createJSONStorage(() => localStorage, {
      reviver: superjson.parse,
      replacer: superjson.stringify,
    }),

[0xaa55h]

Works like a charm, thanks so much!