From 2a5e3419e8ee20d26ec0224b413f14d72e278179 Mon Sep 17 00:00:00 2001
From: garrettmills <shout@garrettmills.dev>
Date: Tue, 13 Oct 2020 09:58:36 -0500
Subject: [PATCH] Include child nodes in menu tree (Noded/frontend#17)

---
 app/controllers/api/v1/Menu.controller.js |  9 ++++-
 app/models/api/Page.model.js              | 46 +++++++++++++++++++++++
 2 files changed, 54 insertions(+), 1 deletion(-)

diff --git a/app/controllers/api/v1/Menu.controller.js b/app/controllers/api/v1/Menu.controller.js
index f34ef03..2202a13 100644
--- a/app/controllers/api/v1/Menu.controller.js
+++ b/app/controllers/api/v1/Menu.controller.js
@@ -24,6 +24,7 @@ class Menu extends Controller {
             children: nodes,
             noDelete: true,
             virtual: true,
+            type: 'branch',
         })
 
         // Get view only shared trees
@@ -36,6 +37,7 @@ class Menu extends Controller {
                 name: tree.Name,
                 children: await this._build_secure_menu_object(tree, req.user),
                 level: await tree.access_level_for(req.user),
+                type: 'page',
             })
         }
 
@@ -49,6 +51,7 @@ class Menu extends Controller {
                 name: tree.Name,
                 children: await this._build_secure_menu_object(tree, req.user),
                 level: await tree.access_level_for(req.user),
+                type: 'page',
             })
         }
 
@@ -62,6 +65,7 @@ class Menu extends Controller {
                 name: tree.Name,
                 children: await this._build_secure_menu_object(tree, req.user),
                 level: await tree.access_level_for(req.user),
+                type: 'page',
             })
         }
 
@@ -71,6 +75,7 @@ class Menu extends Controller {
             children: [...view_only_nodes, ...update_nodes, ...manage_nodes],
             noDelete: true,
             virtual: true,
+            type: 'branch',
         })
 
         return res.api(menu)
@@ -84,7 +89,8 @@ class Menu extends Controller {
                     id: child.UUID,
                     name: child.is_shared() ? child.Name + ' ⁽ˢʰᵃʳᵉᵈ⁾' : child.Name,
                     shared: child.is_shared(),
-                    children: await this._build_menu_object(child),
+                    children: [...(await child.get_menu_items()), ...(await this._build_menu_object(child))],
+                    type: 'page',
                 })
             }
         }
@@ -102,6 +108,7 @@ class Menu extends Controller {
                     name: child.Name,
                     children: await this._build_secure_menu_object(child, user),
                     level: await child.access_level_for(user),
+                    type: 'page',
                 })
             }
         }
diff --git a/app/models/api/Page.model.js b/app/models/api/Page.model.js
index 1a07b46..5202d16 100644
--- a/app/models/api/Page.model.js
+++ b/app/models/api/Page.model.js
@@ -221,6 +221,52 @@ class Page extends Model {
 		await user.save()
 	}
 
+	async get_menu_items() {
+		// {
+		// 	id: child.UUID,
+		// 		name: child.is_shared() ? child.Name + ' ⁽ˢʰᵃʳᵉᵈ⁾' : child.Name,
+		// 	shared: child.is_shared(),
+		// 	children: await this._build_menu_object(child),
+		// 	type: 'page',
+		// }
+
+		// Databases & Code Snips
+		const children = []
+
+		const Database = this.models.get('api:db:Database')
+		const dbs = await Database.find({
+			Active: true,
+			PageId: this.UUID,
+		})
+
+		for ( const db of dbs ) {
+			children.push({
+				id: db.PageId,
+				node_id: db.NodeId,
+				children: [],
+				type: 'db',
+				name: db.Name,
+			})
+		}
+
+		const Codium = this.models.get('api:Codium')
+		const codiums = await Codium.find({
+			PageId: this.UUID,
+		})
+
+		for ( const codium of codiums ) {
+			children.push({
+				id: codium.PageId,
+				node_id: codium.NodeId,
+				children: [],
+				type: 'code',
+				name: codium.code.slice(0, 25) + (codium.code.length > 25 ? '...' : ''),
+			})
+		}
+
+		return children
+	}
+
 }
 
 module.exports = exports = Page;