Adds user permission check

README.md

@@ -10,7 +10,7 @@ From where you have the `composer.json` file run `composer require evanshunt/ser
 
 Run `dev/build`
 
-Then you can assign user groups the permission for `ServerInfoAccess`
+Then you can assign user groups the permission for `Access Serverinfo` under `Other` category
 ## Changelog:
 
 - 2021-01-09: code cleanup

code/ServerInfo.php

@@ -1,25 +1,26 @@
 <?php
 
-
-class ServerInfo extends LeftAndMain
+class ServerInfo extends LeftAndMain implements PermissionProvider
 {
-
     private static $url_segment = 'serverinfo';
     private static $menu_priority = -3;
 
     private static $allowed_actions = array(
 		'getinfo',
     );
 
-    public function init() {
-        parent::init();
-        if ( !Permission::check('ServerInfoAccess') ) {
-            CMSMenu::remove_menu_item('ServerInfo');
-        }
+    public function providePermissions() {
+        return array(
+            "CMS_ACCESS_ServerInfo" => "Access Serverinfo",
+        );
+    }
+
+    public function canView($member = null) {
+        return Permission::check('CMS_ACCESS_ServerInfo', 'any', $member);
     }
 
     public function getinfo(){
-        if ( Permission::check('ServerInfoAccess') ) {
+        if ( Permission::check('CMS_ACCESS_ServerInfo', 'any', Member::currentUser()) ) {
             ob_start();
             phpinfo();
             $info = ob_get_contents();
@@ -28,10 +29,4 @@ public function getinfo(){
         }
         return NULL;
     }
-
-}
-
-class ServerInfoAccess extends ModelAdmin
-{
-
 }