fixes after PR

axotion · Dec 12, 2022 · 4029ebc · 4029ebc
1 parent 8fe1137
commit 4029ebc
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "mikroorm-nested-set",
-  "version": "1.1.6",
+  "version": "1.1.7",
   "types": "src/index.d.ts",
   "description": "Package for MikroORM nested set feature",
   "main": "src/index.js",

diff --git a/src/operator/exception/root-not-found.exception.ts b/src/operator/exception/root-not-found.exception.ts
@@ -0,0 +1 @@
+export class RootNotFoundException extends Error {}
diff --git a/src/repository/nested-set-subject.repository.ts b/src/repository/nested-set-subject.repository.ts
@@ -4,6 +4,7 @@ import { Query } from "@mikro-orm/core/typings";
 import {SqlEntityManager} from "@mikro-orm/knex/SqlEntityManager";
 import {EntityName} from "@mikro-orm/core";
 import {NestedSetNodeOperator} from "../operator";
+import { RootNotFoundException } from "../operator/exception/root-not-found.exception";
 
 export abstract class NestedSetSubjectRepository<T extends NestedSetSubjectAbstract<T>> extends EntityRepository<T>{
 
@@ -33,9 +34,10 @@ export abstract class NestedSetSubjectRepository<T extends NestedSetSubjectAbstr
     root?: NestedSetSubjectAbstract<T> | T
     } = {}) : Promise<NestedSetSubjectAbstract<T>> {
 
-    let root = options.root;
+    const root = options.root ? options.root : await this.findRoot()
+
     if(!root) {
-      root = this.findRoot()
+      throw new RootNotFoundException(`Root of the tree was not found`)
     }
 
     let query = {