as-model 0.1.4 → 0.1.5

package/dist/index.js

@@ -1159,18 +1159,26 @@
           return;
         }
         storeUnits.forEach(function(un, i) {
-          var key = keys[i];
-          if (!key) {
+          var keyLike = keys[i];
+          if (!keyLike) {
             return;
           }
+          var ifIsModelKey = validations.isModelKey(keyLike);
+          var key = ifIsModelKey ? keyLike : keyLike.key;
           Object.assign(un, {
             key
           });
           un.update({
-            model: typeof key === "function" ? key.source : key.key.source
+            model: ifIsModelKey ? keyLike.source : keyLike.key.source
           });
         });
       },
+      keys: function keys() {
+        return storeUnits.map(function(param) {
+          var key = param.key;
+          return key;
+        });
+      },
       destroy: function destroy2() {
         storeUnits.forEach(function(unit) {
           unit.destroy();

package/esm/key/index.js

@@ -1,7 +1,7 @@
 import { _ as _object_spread } from "@swc/helpers/_/_object_spread";
 import { _ as _object_spread_props } from "@swc/helpers/_/_object_spread_props";
 import { createStore, createPrimaryKey } from "../store";
-import { isModelKey } from "../validation";
+import { isModelKey, validations } from "../validation";
 function createKey(model) {
   var config = arguments.length > 1 && arguments[1] !== void 0 ? arguments[1] : {};
   var wrapModel = createPrimaryKey(model, config);
@@ -47,18 +47,26 @@ function createStores(modelKeys) {
         return;
       }
       storeUnits.forEach(function(un, i) {
-        var key = keys[i];
-        if (!key) {
+        var keyLike = keys[i];
+        if (!keyLike) {
           return;
         }
+        var ifIsModelKey = validations.isModelKey(keyLike);
+        var key = ifIsModelKey ? keyLike : keyLike.key;
         Object.assign(un, {
           key
         });
         un.update({
-          model: typeof key === "function" ? key.source : key.key.source
+          model: ifIsModelKey ? keyLike.source : keyLike.key.source
         });
       });
     },
+    keys: function keys() {
+      return storeUnits.map(function(param) {
+        var key = param.key;
+        return key;
+      });
+    },
     destroy: function destroy() {
       storeUnits.forEach(function(unit) {
         unit.destroy();

package/index.d.ts

@@ -138,6 +138,7 @@ export declare interface StoreCollection {
     key: Key<S, T, R> | StoreIndex<S, T, R>
   ) => Store<S, T, R> | null;
   update: (...keys: (ModelKey | StoreIndex)[]) => void;
+  keys: () => Key[];
   destroy: () => void;
 }
 

package/package.json

@@ -1,7 +1,7 @@
 {
   "private": false,
   "name": "as-model",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "description": "This is a model state management tool",
   "license": "MIT",
   "author": "Jimmy.Harding",