From 38ac520b97f17395acf98a73c43b4bd444f7b7c6 Mon Sep 17 00:00:00 2001
From: Robert Soriano <sorianorobertc@gmail.com>
Date: Tue, 24 May 2022 10:01:19 -0700
Subject: [PATCH] fix: server storage and incorrect imports

---
 playground/pages/index.vue |  4 +---
 src/module.ts              |  2 +-
 src/runtime/plugin.ts      | 10 +++-------
 3 files changed, 5 insertions(+), 11 deletions(-)

diff --git a/playground/pages/index.vue b/playground/pages/index.vue
index adeef95..74a0144 100644
--- a/playground/pages/index.vue
+++ b/playground/pages/index.vue
@@ -1,10 +1,8 @@
 <script setup lang="ts">
 const client = useClient()
-const headers = useClientHeader({ authorization: 'asdada' })
+const headers = useClientHeaders()
 const { data: todos, pending, error, refresh } = await useAsyncQuery(['getTodos'])
 
-console.log(headers.value)
-
 const addHeader = () => {
   // headers.value.cookie = 'counter=69'
   console.log(headers.value)
diff --git a/src/module.ts b/src/module.ts
index 1ede328..a7f8ec6 100644
--- a/src/module.ts
+++ b/src/module.ts
@@ -38,7 +38,7 @@ export default defineNuxtModule<ModuleOptions>({
       imports.push(
         { name: 'useClient', from: join(runtimeDir, 'client') },
         { name: 'useAsyncQuery', from: join(runtimeDir, 'client') },
-        { name: 'useClientHeader', from: join(runtimeDir, 'client') },
+        { name: 'useClientHeaders', from: join(runtimeDir, 'client') },
       )
     })
 
diff --git a/src/runtime/plugin.ts b/src/runtime/plugin.ts
index e7ada9b..d6a8b46 100644
--- a/src/runtime/plugin.ts
+++ b/src/runtime/plugin.ts
@@ -1,4 +1,5 @@
 import * as trpc from '@trpc/client'
+import { useClientHeaders } from './client'
 import { defineNuxtPlugin, useRequestHeaders, useRuntimeConfig } from '#app'
 import type { router } from '~/server/trpc'
 
@@ -7,17 +8,12 @@ declare type AppRouter = typeof router
 export default defineNuxtPlugin((nuxtApp) => {
   const config = useRuntimeConfig().public.trpc
   const headers = useRequestHeaders()
+  const otherHeaders = useClientHeaders()
   const client = trpc.createTRPCClient<AppRouter>({
     url: `${config.baseURL}${config.trpcURL}`,
     headers: () => {
-      let otherHeaders = {}
-      if (!process.server) {
-        const key = 'trpc-nuxt-header'
-        otherHeaders = JSON.parse(localStorage.getItem(key) || JSON.stringify({}))
-      }
-
       return {
-        ...otherHeaders,
+        ...unref(otherHeaders),
         ...headers,
       }
     },