From 46cd6306b5405ebef3eb63eb2c60eb5edb2faba5 Mon Sep 17 00:00:00 2001
From: Eduardo San Martin Morote <posva13@gmail.com>
Date: Tue, 1 Jun 2021 15:40:14 +0200
Subject: [PATCH] refactor: better types

---
 src/store.ts | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/store.ts b/src/store.ts
index aa19b71e..63b91f9d 100644
--- a/src/store.ts
+++ b/src/store.ts
@@ -11,6 +11,7 @@ import {
   DebuggerEvent,
   WatchOptions,
   UnwrapRef,
+  markRaw,
 } from 'vue'
 import {
   StateTree,
@@ -114,14 +115,16 @@ function initStore<
   // const state: Ref<S> = toRef(_p.state.value, $id)
 
   let isListening = true
-  let subscriptions: SubscriptionCallback<S>[] = []
-  let actionSubscriptions: StoreOnActionListener<Id, S, G, A>[] = []
+  let subscriptions: SubscriptionCallback<S>[] = markRaw([])
+  let actionSubscriptions: StoreOnActionListener<Id, S, G, A>[] = markRaw([])
   let debuggerEvents: DebuggerEvent[] | DebuggerEvent
 
-  function $patch(stateMutation: (state: S) => void): void
-  function $patch(partialState: DeepPartial<S>): void
+  function $patch(stateMutation: (state: UnwrapRef<S>) => void): void
+  function $patch(partialState: DeepPartial<UnwrapRef<S>>): void
   function $patch(
-    partialStateOrMutator: DeepPartial<S> | ((state: S) => void)
+    partialStateOrMutator:
+      | DeepPartial<UnwrapRef<S>>
+      | ((state: UnwrapRef<S>) => void)
   ): void {
     let subscriptionMutation: SubscriptionCallbackMutation<S>
     isListening = false
@@ -131,7 +134,7 @@ function initStore<
       debuggerEvents = []
     }
     if (typeof partialStateOrMutator === 'function') {
-      partialStateOrMutator(pinia.state.value[$id])
+      partialStateOrMutator(pinia.state.value[$id] as UnwrapRef<S>)
       subscriptionMutation = {
         type: MutationType.patchFunction,
         storeName: $id,
@@ -361,7 +364,7 @@ function buildStoreToUse<
 
   // add getters for devtools
   if (__DEV__ && IS_CLIENT) {
-    store._getters = Object.keys(getters)
+    store._getters = markRaw(Object.keys(getters))
   }
 
   // apply all plugins
-- 
2.47.2