Files
exchange/database/seeds/RolesAndPermissionsSeeder.php
T
Jack Goh 5df307a165 make all test pass
updated seeder to be informative
fix seeder error on test
2018-08-01 15:30:19 +08:00

189 lines
5.3 KiB
PHP

<?php
/*
|--------------------------------------------------------------------------
| Roles & Permissions Seeder
|--------------------------------------------------------------------------
|
| This Seeder class allows you to update and create Roles & Permissions
| for the Laravel Entrust package.
|
| USE -> php artisan db:seed --class=RolesAndPermissionsSeeder
|
| https://github.com/thomasfw/RolesAndPermissionsSeeder
|
|--------------------------------------------------------------------------
| Make sure you update the namespaces for your User & Entrust models
|--------------------------------------------------------------------------
*/
use App\User as User;
use App\Role as Role;
use App\Permission as Permission;
use Illuminate\Database\Seeder;
use Illuminate\Database\Eloquent\Model;
class RolesAndPermissionsSeeder extends Seeder {
protected $roles = [
'admin' => [
],
'member' => [
],
];
protected $permissions = [
// Add your Permissions here
];
/**
* Roles
*
* @return array()
*/
public function roles()
{
return $this->roles;
}
/**
* Permissions
*
* @param $name
* @return array()
*/
public function permissions($name = '')
{
$single = (array_key_exists($name,$this->permissions) ? array($name =>$this->permissions[$name]) : false );
return ($name ? $single : $this->permissions);
}
/**
* Run the Seeder
*
* @return void
*/
public function run()
{
DB::table(Config::get('entrust.permissions_table'))->delete();
foreach ($this->roles() as $key => $val) {
$this->command->info(" ");
$this->command->info('Creating/updating the \''.$key.'\' role');
$this->command->info('-----------------------------------------');
$val['name'] = $key;
$this->reset($val);
}
$this->cleanup();
}
/**
* Reset Role, Permissions & Users
*
* @param $role
* @return void
*/
public function reset($role)
{
$commandBullet = ' -> ';
// The Old Role
$originalRole = Role::where('name',$role['name'])->first();
if($originalRole) Role::where('id',$originalRole->id)->update(['name' => $role['name'].'__remove']);
// The New Role
$newRole = new Role();
$newRole->name = $role['name'];
if(isset($role['display_name'])) $newRole->display_name = $role['display_name']; // optional
if(isset($role['description'])) $newRole->description = $role['description']; // optional
$newRole->save();
$this->command->info($commandBullet."Created $role[name] role");
// Set the Permissions (if they exist)
$pcount = 0;
if(!empty($role['permissions']))
{
foreach ($role['permissions'] as $permission_name) {
$permission = $this->permissions($permission_name);
if($permission === false || (!$permission_name)) {
$this->command->error($commandBullet."Failed to attach permission '$permission_name'. It does not exist");
continue;
}
$newPermission = \Permission::where('name',$permission_name)->first();
if (!$newPermission) {
$newPermission = new Permission();
$newPermission->name = key($permission);
if(isset($permission['display_name'])) $newPermission->display_name = $permission['display_name']; // optional
if(isset($permission['description'])) $newPermission->description = $permission['description']; // optional
$newPermission->save();
}
$newRole->attachPermission($newPermission);
$pcount++;
}
}
$this->command->info($commandBullet."Attached $pcount permissions to $role[name] role");
// Update old records : Temporary disabled
// if ($originalRole)
// {
// $userCount = 0;
// $RoleUsers = DB::table(Config::get('entrust.role_user_table'))->where('role_id',$originalRole->id)->get();
// foreach ($RoleUsers as $user) {
// $u = User::where('id',$user->user_id)->first();
// $u->attachRole($newRole);
// $userCount++;
// }
// $this->command->info($commandBullet."Updated role attachment for $userCount users");
// Role::where('id',$originalRole->id)->delete(); // will also remove old role_user records
// $this->command->info($commandBullet."Removed the original $role[name] role");
// }
}
/**
* Cleanup()
* Remove any roles & permissions that have been removed
* @return void
*/
public function cleanup()
{
$commandBullet = ' -> ';
$this->command->info(" ");
$this->command->info('Cleaning up roles & permissions:');
$this->command->info('--------------------------------');
$storedRoles = Role::all();
if(!empty($storedRoles)) {
$definedRoles = $this->roles();
foreach ($storedRoles as $role) {
if ( !array_key_exists($role->name,$definedRoles) ) {
Role::where('name',$role->name)->delete();
$this->command->info($commandBullet.'The \''.$role->name.'\' role was removed');
}
}
}
$storedPerms = DB::table(Config::get('entrust.permissions_table'))->get();
if(!empty($storedPerms)) {
$definedPerms = $this->permissions();
foreach ($storedPerms as $perm) {
if ( !array_key_exists($perm->name,$definedPerms) ) {
DB::table(Config::get('entrust.permissions_table'))->where('name',$perm->name)->delete();
$this->command->info($commandBullet.'The \''.$perm->name.'\' permission was removed');
}
}
}
$this->command->info($commandBullet.'Done');
$this->command->info(" ");
}
}