Merge b0c7ffd into 86e647e

src/Exceptions/InvalidPermissionModel.php

@@ -0,0 +1,11 @@
+<?php
+
+namespace Maklad\Permission\Exceptions;
+
+/**
+ * Class InvalidPermissionModel
+ * @package Maklad\Permission\Exceptions
+ */
+class InvalidPermissionModel extends MakladException
+{
+}

src/Exceptions/InvalidRoleModel.php

@@ -0,0 +1,11 @@
+<?php
+
+namespace Maklad\Permission\Exceptions;
+
+/**
+ * Class InvalidRoleModel
+ * @package Maklad\Permission\Exceptions
+ */
+class InvalidRoleModel extends MakladException
+{
+}

src/Helpers.php

@@ -79,4 +79,20 @@ public function getRoleDoesNotExistMessage(string $name, string $guardName): str
     {
         return "There is no role named `{$name}` for guard `{$guardName}`.";
     }
+
+    /**
+     * @return string
+     */
+    public function getInvalidRoleModelMessage(): string
+    {
+        return 'Invalid Role model in the configuration file.';
+    }
+
+    /**
+     * @return string
+     */
+    public function getInvalidPermissionModelMessage(): string
+    {
+        return 'Invalid Permission model in the configuration file.';
+    }
 }

src/Models/Permission.php

@@ -7,6 +7,9 @@
 use Jenssegers\Mongodb\Eloquent\Model;
 use Jenssegers\Mongodb\Relations\BelongsToMany;
 use Maklad\Permission\Contracts\PermissionInterface;
+use Maklad\Permission\Contracts\RoleInterface;
+use Maklad\Permission\Exceptions\InvalidPermissionModel;
+use Maklad\Permission\Exceptions\InvalidRoleModel;
 use Maklad\Permission\Exceptions\PermissionAlreadyExists;
 use Maklad\Permission\Exceptions\PermissionDoesNotExist;
 use Maklad\Permission\Helpers;
@@ -42,6 +45,7 @@ public function __construct(array $attributes = [])
 
     /**
      * Create new Permission
+     *
      * @param array $attributes
      *
      * @return $this|\Illuminate\Database\Eloquent\Model
@@ -55,9 +59,9 @@ public static function create(array $attributes = [])
             'guard_name',
             $attributes['guard_name']
         )->first()) {
-            $name = $attributes['name'];
+            $name      = $attributes['name'];
             $guardName = $attributes['guard_name'];
-            $helpers = new Helpers();
+            $helpers   = new Helpers();
             throw new PermissionAlreadyExists($helpers->getPermissionAlreadyExistsMessage($name, $guardName));
         }
 
@@ -71,13 +75,23 @@ public static function create(array $attributes = [])
     /**
      * A permission can be applied to roles.
      * @return BelongsToMany
+     * @throws \Maklad\Permission\Exceptions\InvalidPermissionModel
+     * @throws \Maklad\Permission\Exceptions\InvalidRoleModel
      */
     public function roles(): BelongsToMany
     {
-        return $this->belongsToMany(
-            \config('permission.models.role'),
-            \config('permission.collection_names.role_has_permissions')
-        );
+        $roleModel       = \config('permission.models.role');
+        $permissionModel = \config('permission.models.permission');
+        $helpers = new Helpers();
+
+        if ($roleModel instanceof RoleInterface) {
+            throw new InvalidRoleModel($helpers->getInvalidRoleModelMessage());
+        }
+        if ($permissionModel instanceof PermissionInterface) {
+            throw new InvalidPermissionModel($helpers->getInvalidPermissionModelMessage());
+        }
+
+        return $this->belongsToMany($roleModel, $permissionModel);
     }
 
     /**
@@ -104,7 +118,7 @@ public static function findByName(string $name, $guardName = null): PermissionIn
 
         $permission = static::getPermissions()->where('name', $name)->where('guard_name', $guardName)->first();
 
-        if (! $permission) {
+        if ( ! $permission) {
             $helpers = new Helpers();
             throw new PermissionDoesNotExist($helpers->getPermissionDoesNotExistMessage($name, $guardName));
         }