feat: improve internal implementation

src/index.ts

@@ -2,4 +2,4 @@ export * from './spy'
 export * from './spyOn'
 export * from './restoreAll'
 export { createInternalSpy, spies, getInternalState } from './internal'
-export type { Spy, SpyImpl } from './internal'
+export type { Spy, SpyImpl, SpyInternal, SpyInternalImpl } from './internal'

src/internal.ts

@@ -1,6 +1,11 @@
 import { assert, define, defineValue, isPromise, isType } from './utils'
 import { S } from './constants'
 
+interface GetState {
+  <A extends any[], R>(spy: SpyInternalImpl<A, R>): SpyInternalImplState<A, R>
+  <A extends any[], R>(spy: SpyInternal<A, R>): SpyInternalState<A, R>
+}
+
 export let spies = new Set<SpyImpl>()
 
 let reset = (state: SpyInternalState) => {
@@ -13,8 +18,8 @@ let defineState = (spy: SpyInternal) => {
   define(spy, S, { value: { reset: () => reset(spy[S]) } })
   return spy[S]
 }
-export let getInternalState = <A extends any[], R>(spy: SpyInternal<A, R>) => {
-  return spy[S] || defineState(spy)
+export let getInternalState: GetState = (spy) => {
+  return (spy[S] || defineState(spy)) as any
 }
 
 type ReturnError = ['error', any]

src/spyOn.ts

@@ -23,7 +23,7 @@ type Constructors<T> = {
 let getDescriptor = (obj: any, method: string | symbol | number) =>
   Object.getOwnPropertyDescriptor(obj, method)
 
-export function internalSpyOn<T extends object, K extends string & keyof T>(
+export function internalSpyOn<T, K extends string & keyof T>(
   obj: T,
   methodName: K | { getter: K } | { setter: K },
   mock?: Procedure

src/utils.ts

@@ -1,4 +1,4 @@
-export function assert(condition: any, message: string) {
+export function assert(condition: any, message: string): asserts condition {
   if (!condition) {
     throw new Error(message)
   }