fix(types): mapWritableState array

Fix #2014

diff --git a/packages/pinia/src/mapHelpers.ts b/packages/pinia/src/mapHelpers.ts
index 742e26f96370992a8f927858bb436209677a154a..6e1b5b704b3d59ddbc74a1e699b2e17b9a3af963 100644 (file)
--- a/packages/pinia/src/mapHelpers.ts
+++ b/packages/pinia/src/mapHelpers.ts
@@ -462,11 +462,12 @@ export function mapWritableState<
   Id extends string,
   S extends StateTree,
   G extends _GettersTree<S>,
-  A
+  A,
+  Keys extends keyof S
 >(
   useStore: StoreDefinition<Id, S, G, A>,
-  keys: Array<keyof S>
-): _MapWritableStateReturn<S>
+  keys: readonly Keys[]
+): { [K in Keys]: S[K] }
 /**
  * Allows using state and getters from one store without using the composition
  * API (`setup()`) by generating an object to be spread in the `computed` field

diff --git a/packages/pinia/test-dts/mapHelpers.test-d.ts b/packages/pinia/test-dts/mapHelpers.test-d.ts
index 991991fd1a299e9820420eefbab92d40b259eaf2..974e709459252a5b4e4d172e5ddc4984dc6ef13e 100644 (file)
--- a/packages/pinia/test-dts/mapHelpers.test-d.ts
+++ b/packages/pinia/test-dts/mapHelpers.test-d.ts
@@ -90,12 +90,9 @@ expectType<{
   newToggleA: () => void
 }>(mapActions(useStore, { newSetToggle: 'setToggle', newToggleA: 'toggleA' }))
 
-expectType<{
-  a: {
-    get: () => 'on' | 'off'
-    set: (v: 'on' | 'off') => any
-  }
-}>(mapWritableState(useStore, ['a']))
+expectType<{ a: 'on' | 'off' }>(mapWritableState(useStore, ['a']))
+// @ts-expect-error: only defined in array
+mapWritableState(useStore, ['a']).b
 
 expectType<{
   newA: {