feat: Key file memoization cache on path and args

Author: effigies

src/utils/memoize.ts

@@ -1,5 +1,6 @@
 export type WithCache<T> = T & { cache: Map<string, any> }
-interface HasParent {
+interface FileLike {
+  path: string,
   parent: { path: string }
 }
 
@@ -14,20 +15,21 @@ export const memoize = <T>(
   return cached
 }
 
-export function filememoizeAsync<F extends HasParent, T>(
+export function filememoizeAsync<F extends FileLike, T>(
   fn: (file: F, ...args: any[]) => Promise<T>,
 ): WithCache<(file: F, ...args: any[]) => Promise<T>> {
-  const cache = new Map<string, Map<F, T>>()
+  const cache = new Map<string, Map<string, T>>()
   const cached = async function (this: any, file: F, ...args: any[]): Promise<T> {
     let subcache = cache.get(file.parent.path)
     if (!subcache) {
       subcache = new Map()
       cache.set(file.parent.path, subcache)
     }
-    let val = subcache.get(file)
+    const key = `${file.path}:${args.join(',')}`
+    let val = subcache.get(key)
     if (!val) {
       val = await fn.call(this, file, ...args)
-      subcache.set(file, val)
+      subcache.set(key, val)
     }
     return val
   }