milkio 1.0.0-alpha.10 → 1.0.0-alpha.12

package/package.json

@@ -2,7 +2,7 @@
   "name": "milkio",
   "type": "module",
   "module": "index.ts",
-  "version": "1.0.0-alpha.10",
+  "version": "1.0.0-alpha.12",
   "peerDependencies": {
     "typescript": "^5.4.2"
   },

package/use/index.ts

@@ -1,12 +1,18 @@
-export const defineUse = <CreatorFn extends () => Promise<unknown> | unknown>(creatorFn: CreatorFn): (() => Promise<Awaited<ReturnType<CreatorFn>>>) => {
-  let use: any | undefined;
+import { $context } from "../context";
 
-  const getUse = async () => {
-    if (use === undefined) {
-      use = await creatorFn();
-    }
-    return use;
-  };
+export const defineUse = <Mode extends "dynamic" | "static", Use extends Mode extends "static" ? () => Promise<unknown> | unknown : (...args: [$context, ...Array<unknown>]) => Promise<unknown> | unknown>(mode: Mode, use: Use): ((...args: Parameters<Use>) => Promise<Awaited<ReturnType<Use>>>) => {
+  if (mode === "static") {
+    let useCreated: any | undefined;
 
-  return getUse;
+    const getUse = async () => {
+      if (useCreated === undefined) {
+        useCreated = await (use as any)();
+      }
+      return useCreated;
+    };
+
+    return getUse;
+  } else {
+    return use as any;
+  }
 };