@travetto/model-mongo 3.1.3 → 3.1.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@travetto/model-mongo",
-  "version": "3.1.3",
+  "version": "3.1.4",
   "description": "Mongo backing for the travetto model module.",
   "keywords": [
     "mongo",
@@ -26,8 +26,8 @@
   },
   "dependencies": {
     "@travetto/config": "^3.1.2",
-    "@travetto/model": "^3.1.5",
-    "@travetto/model-query": "^3.1.3",
+    "@travetto/model": "^3.1.6",
+    "@travetto/model-query": "^3.1.4",
     "mongodb": "^5.0.1"
   },
   "peerDependencies": {

package/src/internal/util.ts

@@ -134,7 +134,7 @@ export class MongoUtil {
   /**
    * Convert `'a.b.c'` to `{ a: { b: { c: ... }}}`
    */
-  static extractSimple<T>(o: T, path: string = ''): Record<string, unknown> {
+  static extractSimple<T>(o: T, path: string = '', recursive: boolean = true): Record<string, unknown> {
     const out: Record<string, unknown> = {};
     // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
     const sub = o as Record<string, unknown>;
@@ -150,7 +150,11 @@ export class MongoUtil {
         const isPlain = v && ObjectUtil.isPlainObject(v);
         const firstKey = isPlain ? Object.keys(v)[0] : '';
         if ((isPlain && !firstKey.startsWith('$')) || v?.constructor?.Ⲑid) {
-          Object.assign(out, this.extractSimple(v, `${subpath}.`));
+          if (recursive) {
+            Object.assign(out, this.extractSimple(v, `${subpath}.`));
+          } else {
+            out[subpath] = v;
+          }
         } else {
           if (firstKey === '$gt' || firstKey === '$lt' || firstKey === '$gte' || firstKey === '$lte') {
             for (const [sk, sv] of Object.entries(v)) {

package/src/service.ts

@@ -234,7 +234,7 @@ export class MongoModelService implements
 
     let final: Record<string, unknown> = item;
 
-    const items = MongoUtil.extractSimple(final);
+    const items = MongoUtil.extractSimple(final, undefined, false);
     final = Object
       .entries(items)
       .reduce<Record<string, unknown>>((acc, [k, v]) => {
@@ -258,7 +258,7 @@ export class MongoModelService implements
     );
 
     if (!res.value) {
-      new NotFoundError(cls, id);
+      throw new NotFoundError(cls, id);
     }
 
     return this.get(cls, id);
@@ -497,6 +497,19 @@ export class MongoModelService implements
   }
 
   // Query Crud
+  async updateOneWithQuery<T extends ModelType>(cls: Class<T>, data: T, query: ModelQuery<T>): Promise<T> {
+    const col = await this.getStore(cls);
+    const item = await ModelCrudUtil.preStore(cls, data, this);
+    query = ModelQueryUtil.getQueryWithId(cls, data, query);
+
+    const { filter } = MongoUtil.prepareQuery(cls, query);
+    const res = await col.replaceOne(filter, item);
+    if (res.matchedCount === 0) {
+      throw new NotFoundError(cls, item.id);
+    }
+    return item;
+  }
+
   async deleteByQuery<T extends ModelType>(cls: Class<T>, query: ModelQuery<T>): Promise<number> {
     const col = await this.getStore(cls);
     const { filter } = MongoUtil.prepareQuery(cls, query, false);