feat: let tree opts be optional

webpod · Apr 6, 2024 · 7e25163 · 7e25163
1 parent 048db11
commit 7e25163
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@webpod/ps",
-  "version": "0.0.0-beta.2",
+  "version": "0.0.0-beta.3",
   "description": "A process lookup utility",
   "publishConfig": {
     "access": "public"

diff --git a/src/main/ts/ps.ts b/src/main/ts/ps.ts
@@ -128,14 +128,17 @@ export const pickTree = (list: TPsLookupEntry[], pid: string | number, recursive
   ]
 }
 
-export const tree = async (opts: string | number | TPsTreeOpts, cb: TPsLookupCallback = noop): Promise<TPsLookupEntry[]> => {
+export const tree = async (opts?: string | number | TPsTreeOpts | undefined, cb: TPsLookupCallback = noop): Promise<TPsLookupEntry[]> => {
   if (typeof opts === 'string' || typeof opts === 'number') {
     return tree({ pid: opts }, cb)
   }
 
   try {
+    const all = await lookup()
+    if (opts === undefined) return all
+
     const {pid, recursive = false} = opts
-    const list = pickTree(await lookup(), pid, recursive)
+    const list = pickTree(all, pid, recursive)
 
     cb(null, list)
     return list

diff --git a/src/test/ts/ps.test.ts b/src/test/ts/ps.test.ts
@@ -83,4 +83,9 @@ describe('tree()', () => {
 
     assert.equal((await lookup({ arguments: marker })).length, 0)
   })
+
+  it('returns all ps list if no opts provided', async () => {
+    const list = await tree()
+    assert.ok(list.length > 0)
+  })
 })