FEAT: rls rules routes

src/modules/catalog/catalog.controller.ts

@@ -31,6 +31,7 @@ import { DadosferaLogger } from '@dadosfera/dadosfera-logs';
 import { PackTheMetadata } from 'src/utils/ PackTheMetadata';
 import { RequestUser, User } from 'src/decorators/user.decorator';
 import {
+  BatchRemoveRlsRulesRequest,
   GetDatasetCatalogTaskRes,
   ICatalogAllRequest,
   ICatalogAllResponse,
@@ -49,6 +50,11 @@ import { GrpcToHttpExceptionFilter } from 'src/error/grpc-to-http-exception.filt
 import { Language } from 'src/decorators/language.decorator';
 import { LanguageEnum } from 'src/utils/languages.enum';
 import { ApiInternalOnlyEndpoint } from 'src/decorators/swagger.decorator';
+import {
+  AddRlsRuleRequest,
+  GetNimbusDashboardsRequest,
+  GetRlsRulesRequest,
+} from '@dadosfera/protospack-v2/dist/lib/Catalog/interfaces/messages';
 
 @ApiTags('Catalog')
 @ApiHeaders([{ name: 'dadosfera-lang', enum: LanguageEnum, required: false }])
@@ -635,8 +641,8 @@ export class CatalogController {
   }
 
   @Post('rls-rule')
-  @ApiOkResponse({ type: GetDatasetCatalogTaskRes })
-  async addRlsRule(@User() user: RequestUser, @Body() body) {
+  @RequireAllPermissions(PERMISSIONS_GROUPS.CATALOG.permissions.DATA_MANAGER)
+  async addRlsRule(@User() user: RequestUser, @Body() body: AddRlsRuleRequest) {
     const { customer_id, customer_name, user_id, username } = user;
     const metadata = PackTheMetadata({
       customer_id,
@@ -644,9 +650,97 @@ export class CatalogController {
       user_id,
       username,
     });
-    console.log(body);
     const response = await this.catalogService.addRlsRule(body, metadata);
 
     return response;
   }
+
+  @Get('rls-rule/:id')
+  @RequireAllPermissions(PERMISSIONS_GROUPS.CATALOG.permissions.DATA_MANAGER)
+  async getOneRlsRule(@Param('id') id: string, @User() user: RequestUser) {
+    const { customer_id, customer_name, user_id, username } = user;
+    const metadata = PackTheMetadata({
+      customer_id,
+      customer_name,
+      user_id,
+      username,
+    });
+    const idInt = parseInt(id);
+    const response = await this.catalogService.getOneRlsRule(idInt, metadata);
+
+    return response;
+  }
+
+  @Get('rls-rule')
+  @RequireAllPermissions(PERMISSIONS_GROUPS.CATALOG.permissions.DATA_MANAGER)
+  async getRlsRules(
+    @User() user: RequestUser,
+    @Query() query: GetRlsRulesRequest,
+  ) {
+    const { customer_id, customer_name, user_id, username } = user;
+    const metadata = PackTheMetadata({
+      customer_id,
+      customer_name,
+      user_id,
+      username,
+    });
+    const response = await this.catalogService.getRlsRules(query, metadata);
+
+    return response;
+  }
+
+  @Delete('rls-rule/:id')
+  @RequireAllPermissions(PERMISSIONS_GROUPS.CATALOG.permissions.DATA_MANAGER)
+  async removeRlsRule(@Param('id') id: string, @User() user: RequestUser) {
+    const { customer_id, customer_name, user_id, username } = user;
+    const metadata = PackTheMetadata({
+      customer_id,
+      customer_name,
+      user_id,
+      username,
+    });
+    const idInt = parseInt(id);
+    const response = await this.catalogService.removeRlsRule(idInt, metadata);
+
+    return response;
+  }
+
+  @Delete('rls-rule')
+  @RequireAllPermissions(PERMISSIONS_GROUPS.CATALOG.permissions.DATA_MANAGER)
+  async batchRemoveRlsRules(
+    @Query() query: BatchRemoveRlsRulesRequest,
+    @User() user: RequestUser,
+  ) {
+    const { customer_id, customer_name, user_id, username } = user;
+    const metadata = PackTheMetadata({
+      customer_id,
+      customer_name,
+      user_id,
+      username,
+    });
+    await this.catalogService.batchRemoveRlsRule(query, metadata);
+
+    return { message: 'OK' };
+  }
+
+  @Get('nimbus-dashboards')
+  @RequireAllPermissions(PERMISSIONS_GROUPS.CATALOG.permissions.DATA_MANAGER)
+  async getNimbusDashboards(
+    @User() user: RequestUser,
+    @Body() body: GetNimbusDashboardsRequest,
+  ) {
+    const { customer_id, customer_name, user_id, username } = user;
+    const metadata = PackTheMetadata({
+      customer_id,
+      customer_name,
+      user_id,
+      username,
+    });
+    const dashboards = await this.catalogService.getNimbusDashboards(
+      body,
+      metadata,
+    );
+
+    return JSON.parse(dashboards);
+  }
 }

src/modules/catalog/catalog.service.ts

@@ -6,6 +6,7 @@ import {
   Messages,
 } from '@dadosfera/protospack-v2/dist/lib/Catalog';
 import {
+  BadRequestException,
   HttpException,
   HttpStatus,
   Inject,
@@ -18,8 +19,18 @@ import { CatalogClientConfiguration } from './catalog-client';
 import { UsersService } from '../users/users.service';
 import { RolesService } from '../roles/roles.service';
 import { Metadata } from '@grpc/grpc-js';
-import { IUpdateDataRequest, TriggerCatalogReq } from './dtos';
-import { AddRlsUserToDashboardRequest } from '@dadosfera/protospack-v2/dist/lib/Catalog/interfaces/messages';
+import {
+  BatchRemoveRlsRulesRequest,
+  IUpdateDataRequest,
+  TriggerCatalogReq,
+} from './dtos';
+import {
+  AddRlsRuleRequest,
+  GetNimbusDashboardsRequest,
+  GetOneRlsRuleRequest,
+  GetRlsRulesRequest,
+} from '@dadosfera/protospack-v2/dist/lib/Catalog/interfaces/messages';
+import { parse } from 'path';
 
 class CatalogService implements OnModuleInit {
   catalogReadService: ReadService.CatalogReadServices;
@@ -334,6 +345,7 @@ class CatalogService implements OnModuleInit {
       } as typeof data_asset;
     });
   }
+
   async triggerCatalog(data: TriggerCatalogReq, metadata: Metadata) {
     const { session } = await lastValueFrom(
       this.catalogWriteService.TriggerDatasetCataloging(data, metadata),
@@ -346,16 +358,70 @@ class CatalogService implements OnModuleInit {
     );
     return res;
   }
-  async addRlsRule(data: AddRlsUserToDashboardRequest, metadata: Metadata) {
-    const newData: AddRlsUserToDashboardRequest = {
-      ...data,
-      rls_option: `rls_email`,
-    };
+
+  async addRlsRule(data: AddRlsRuleRequest, metadata: Metadata) {
     const res = await lastValueFrom(
-      this.catalogWriteService.AddRlsUserToDashboard(newData, metadata),
+      this.catalogWriteService.AddRlsRule(data, metadata),
     );
     return res;
   }
+
+  async removeRlsRule(id: number, metadata: Metadata) {
+    const res = await lastValueFrom(
+      this.catalogWriteService.RemoveRlsRule({ id }, metadata),
+    );
+    return res;
+  }
+
+  async batchRemoveRlsRule(
+    query: BatchRemoveRlsRulesRequest,
+    metadata: Metadata,
+  ) {
+    const { id_rls, nimbus_dashboard_id } = query;
+
+    if (id_rls && nimbus_dashboard_id) {
+      throw new BadRequestException(
+        "You can't delete using both parameters. Choose either 'id_rls' or 'nimbus_dashboard_id'",
+      );
+    }
+    if (id_rls) {
+      await lastValueFrom(
+        this.catalogWriteService.RemoveRlsRulesByRlsId({ id_rls }, metadata),
+      );
+    } else if (nimbus_dashboard_id) {
+      await lastValueFrom(
+        this.catalogWriteService.RemoveRlsRulesByDashboardId(
+          { nimbus_dashboard_id: parseInt(nimbus_dashboard_id) },
+          metadata,
+        ),
+      );
+    }
+    return 'OK';
+  }
+
+  async getRlsRules(data: GetRlsRulesRequest, metadata: Metadata) {
+    const res = await lastValueFrom(
+      this.catalogReadService.GetRlsRules(data, metadata),
+    );
+    return res.rls_rules;
+  }
+
+  async getOneRlsRule(id: number, metadata: Metadata) {
+    const res = await lastValueFrom(
+      this.catalogReadService.GetOneRlsRule({ id }, metadata),
+    );
+    return res.rls_rule;
+  }
+
+  async getNimbusDashboards(
+    data: GetNimbusDashboardsRequest,
+    metadata: Metadata,
+  ) {
+    const res = await lastValueFrom(
+      this.catalogReadService.GetNimbusDashboards(data, metadata),
+    );
+    return res.dashboards;
+  }
 }
 
 export { CatalogService };

src/modules/catalog/dtos/index.ts

@@ -312,3 +312,11 @@ export class GetDatasetCatalogTaskRes {
   @ApiProperty()
   updated_by: string;
 }
+
+export class BatchRemoveRlsRulesRequest {
+  @ApiPropertyOptional()
+  nimbus_dashboard_id?: string;
+
+  @ApiPropertyOptional()
+  id_rls?: string;
+}