composables.ts

import { useNuxtApp } from '#app'
import {
  defineStore as _defineStore,
  type Pinia,
  type StoreGeneric,
} from 'pinia'
export * from 'pinia'

export const usePinia = () => useNuxtApp().$pinia as Pinia | undefined

export const defineStore: typeof _defineStore =
  process.env.NODE_ENV === 'production' && !__TEST__
    ? _defineStore
    : (...args: [idOrOptions: any, setup?: any, setupOptions?: any]) => {
        if (!import.meta.server) {
          return _defineStore(...args)
        }

        const originalUseStore = _defineStore(...args)
        function useStore(
          pinia?: Pinia | null,
          hot?: StoreGeneric
        ): StoreGeneric {
          return originalUseStore(pinia || usePinia(), hot)
        }

        useStore.$id = originalUseStore.$id
        useStore._pinia = originalUseStore._pinia

        return useStore
      }