build role adapter

Author: daniwebdev

migrations/1699245515021-AddColumnRoleIdToUsersTable.ts

@@ -0,0 +1,17 @@
+import { MigrationInterface, QueryRunner, TableColumn } from "typeorm"
+
+export class AddColumnRoleIdToUsersTable1699245515021 implements MigrationInterface {
+
+    public async up(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.addColumn('users', new TableColumn({
+            name: "role_id",
+            type: "int8",
+            default: 1, //admin
+        }))
+    }
+
+    public async down(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.dropColumn('users', 'role_id');
+    }
+
+}

src/admin/misc/misc.controller.ts

@@ -1,32 +1,46 @@
-import { Controller, Get, Header } from '@nestjs/common';
+import { Controller, Get, Header, Req } from '@nestjs/common';
+import { Request } from 'express';
+import { UseJwtGuard } from 'src/common/guards/jwt.guard';
+import { MiscService } from './misc.service';
 
 @Controller('misc')
 export class MiscController {
 
+    constructor(
+        private miscService: MiscService,
+    ) {}
+
     @Get('/permissions')
-    @Header('content-type', 'text/plain')
-    getPermission() {
+    @UseJwtGuard()
+    // @Header('content-type', 'text/plain')
+    async getPermission(
+        @Req() req: Request
+    ) {
+        console.log(req.user);
+
+        return await this.miscService.getUserPermission(req.user.id);
+
         return `
-p, admin, posts, (list)|(create)
-p, admin, posts/*, (edit)|(show)|(delete)
-p, admin, posts/*, field
+            p, admin, posts, (list)|(create)
+            p, admin, posts/*, (edit)|(show)|(delete)
+            p, admin, posts/*, field
 
-p, admin, users, (list)|(create)
-p, admin, users/*, (edit)|(show)|(delete)
+            p, admin, users, (list)|(create)
+            p, admin, users/*, (edit)|(show)|(delete)
 
-p, admin, roles, (list)|(create)
-p, admin, roles/*, (edit)|(show)|(delete)
+            p, admin, roles, (list)|(create)
+            p, admin, roles/*, (edit)|(show)|(delete)
 
-p, admin, access_control
+            p, admin, access_control
 
-p, admin, permissions, (list)|(create)
-p, admin, permissions/*, (edit)|(show)|(delete)
+            p, admin, permissions, (list)|(create)
+            p, admin, permissions/*, (edit)|(show)|(delete)
 
-p, editor, posts, (list)|(create)
-p, editor, posts/*, (edit)|(show)
-p, editor, posts/hit, field, deny
+            p, editor, posts, (list)|(create)
+            p, editor, posts/*, (edit)|(show)
+            p, editor, posts/hit, field, deny
 
-p, editor, categories, list
+            p, editor, categories, list
         `;
     }
 }

src/admin/misc/misc.module.ts

@@ -1,8 +1,13 @@
 import { Module } from '@nestjs/common';
 import { MiscController } from './misc.controller';
 import { MiscService } from './misc.service';
+import { TypeOrmModule } from '@nestjs/typeorm';
+import { User } from 'src/entities/user.entity';
 
 @Module({
+  imports: [
+    TypeOrmModule.forFeature([User]),
+  ],
   controllers: [MiscController],
   providers: [MiscService]
 })

src/admin/misc/misc.service.ts

@@ -1,4 +1,46 @@
 import { Injectable } from '@nestjs/common';
+import { InjectRepository } from '@nestjs/typeorm';
+import { RoleToPermission } from 'src/entities/role-to-permission.entity';
+import { User } from 'src/entities/user.entity';
+import { Repository } from 'typeorm';
 
 @Injectable()
-export class MiscService {}
+export class MiscService {
+
+    constructor(
+        @InjectRepository(User)
+        private userRepository: Repository<User>,
+    ) {}
+
+
+    async getUserPermission(userId: number) {
+        const user = await this.userRepository.findOne({
+            where: {
+                id: userId
+            },
+            relations: {
+                role: {
+                    permissions: {
+                        permission: true
+                    },
+                }
+            }
+        })
+
+        const permissions = user.role.permissions;
+        let buildPermissionCasbin = '';
+        permissions.forEach((item: RoleToPermission) => {
+            console.log(item.permission);
+            let actions = item.permission.actions.map((x: string) => `(${x})`).join('|');
+            buildPermissionCasbin += `p, ${user.role.key}, ${item.permission.path}, ${actions}${"\n"}`
+        })
+
+        console.log(buildPermissionCasbin.trim());
+        return {
+            role: user.role.key,
+            permissions,
+            casbin_permission_adapter: buildPermissionCasbin.trim()
+        }
+
+    }
+}

src/entities/user.entity.ts

@@ -1,6 +1,7 @@
 import { strRandom } from 'src/common/utils';
-import { Entity, Column, PrimaryGeneratedColumn, BeforeInsert, UpdateDateColumn, CreateDateColumn } from 'typeorm';
+import { Entity, Column, PrimaryGeneratedColumn, BeforeInsert, UpdateDateColumn, CreateDateColumn, OneToOne, JoinColumn } from 'typeorm';
 import { v4 } from 'uuid'
+import { Role } from './role.entity';
 @Entity({
     name: 'users'
 })
@@ -26,6 +27,9 @@ export class User {
   @Column()
   name: string;
 
+  @Column()
+  role_id: string;
+
   @Column()
   avatar: string;
 
@@ -54,6 +58,10 @@ export class User {
   @CreateDateColumn()
   created_at: Date
 
+  @OneToOne(type => Role)
+  @JoinColumn({ name: "role_id" })
+  role: Role;
+
   @BeforeInsert()
   beforeInsert() {
     this.code = strRandom(6)