Support ContentController::getMenu (if CMS is installed)

changelog.md

@@ -27,4 +27,5 @@ Minor update to:
 
 # 1.1.1
 
-* Allow setting of password restrictions on registration
+* Allow setting of password restrictions on registration
+* Add out of the box support for ContentController::getMenu

code/control/Users_Account_Controller.php

@@ -133,6 +133,24 @@ public function RelativeLink($action = null)
         );
     }
 
+    /**
+     * If content controller exists, return it's menu function
+     * @param int $level Menu level to return.
+     * @return ArrayList
+     */
+    public function getMenu($level = 1)
+    {
+        if (class_exists(ContentController::class)) {
+            $controller = Injector::inst()->get(ContentController::class);
+            return $controller->getMenu($level);
+        }
+    }
+
+    public function Menu($level)
+    {
+        return $this->getMenu();
+    }
+
     /**
      * Display the currently outstanding orders for the current user
      *

code/control/Users_Register_Controller.php

@@ -88,6 +88,24 @@ public function RelativeLink($action = null)
         );
     }
 
+    /**
+     * If content controller exists, return it's menu function
+     * @param int $level Menu level to return.
+     * @return ArrayList
+     */
+    public function getMenu($level = 1)
+    {
+        if (class_exists(ContentController::class)) {
+            $controller = Injector::inst()->get(ContentController::class);
+            return $controller->getMenu($level);
+        }
+    }
+
+    public function Menu($level)
+    {
+        return $this->getMenu();
+    }
+
     /**
      * Default action this controller will deal with
      *