UPDATE: increase redis ttl

Beta

.github/workflows/deploy-manually.yml

@@ -142,11 +142,17 @@ jobs:
           docker system prune --volumes -a -f
           docker system df
 
+  k8s-setup:
+    needs: [extract_environment]
+    uses: ./.github/workflows/k8s-setup.yml
+    with:
+      cloud: azure
+      environment: ${{ needs.extract_environment.outputs.environment }}
+    secrets: inherit
+
   helmfile-deploy:
-    needs: [extract_environment, semantic_release, build_ecr_image]
+    needs: [extract_environment, semantic_release, build_ecr_image, k8s-setup]
-    env:
+    runs-on: [self-hosted, "prd-azure"]
-      HOME: /home/runner
-    runs-on: [self-hosted, "prd-oracle"]
     environment: ${{ needs.extract_environment.outputs.environment }}
 
     steps:
@@ -158,18 +164,6 @@ jobs:
         with:
           version: 'v3.9.0'
 
-      - name: Install OCI CLI
-        run: |
-          bash -c "$(curl -L https://raw.githubusercontent.com/oracle/oci-cli/master/scripts/install/install.sh)" -- --accept-all-defaults
-          echo "$HOME/bin" >> $GITHUB_PATH
-
-      - name: Configure OCI CLI
-        run: |
-          mkdir -p ~/.oci || true
-          echo "${{ secrets.OCI_CONFIG }}" > ~/.oci/config
-          echo "${{ secrets.OCI_PRIVATE_KEY }}" > ~/.oci/oci_api_key.pem 
-          chmod 600 ~/.oci/oci_api_key.pem
-
       - name: Set up Python
         uses: actions/setup-python@v4
         with:
@@ -179,29 +173,12 @@ jobs:
         run: |
           wget https://github.com/helmfile/helmfile/releases/download/v0.148.0/helmfile_0.148.0_linux_amd64.tar.gz
           tar -xzf helmfile_0.148.0_linux_amd64.tar.gz
-          sudo mv helmfile /usr/local/bin/
+          mv helmfile /usr/local/bin/
           helmfile --version
 
       - name: Install Helm Diff Plugin
         run: helm plugin install https://github.com/databus23/helm-diff || true
 
-      - name: Authenticate with OKE cluster
-        env:
-          ENV: ${{ needs.extract_environment.outputs.environment }}
-          STG_CLUSTER_ID: "ocid1.cluster.oc1.sa-saopaulo-1.aaaaaaaagh3jvln52a3ebm3dodx6emmhv5bmfs7i7sv2k4zkbcbrzcl6v37q"
-          PRD_CLUSTER_ID: "ocid1.cluster.oc1.sa-saopaulo-1.aaaaaaaanf3vptl6hc2tzd4enfd2hfpsht3wikxww5xejc3l7cwfm6l3sndq"
-        run: |
-          if [ "$ENV" = "stg" ]; then
-            CLUSTER_ID=$STG_CLUSTER_ID
-          elif [ "$ENV" = "prd" ]; then
-            CLUSTER_ID=$PRD_CLUSTER_ID
-          else
-            echo "Unknown environment: $ENV"
-            exit 1
-          fi
-          
-          oci ce cluster create-kubeconfig --cluster-id ${CLUSTER_ID} --file $HOME/.kube/config --region sa-saopaulo-1 --token-version 2.0.0 --kube-endpoint PRIVATE_ENDPOINT
-
       - name: Run Helmfile Apply
         env:
           ENV: ${{ needs.extract_environment.outputs.environment }}

.github/workflows/k8s-setup.yml

@@ -0,0 +1,77 @@
+name : K8s Setup kube config to deploy
+
+on:
+  workflow_call:
+    inputs:
+      cloud:
+        description: "Cloud provider for the deployment"
+        required: true
+        default: "azure"
+        type: string
+      environment:
+        description: "Deployment environment"
+        required: true
+        default: "prd"
+        type: string
+
+jobs:
+  azure:
+    if: inputs.cloud == 'azure'
+    runs-on: [self-hosted, "prd-azure"]
+    steps:
+      - name: Install Azure ClI
+        run: |
+          curl -sL https://aka.ms/InstallAzureCLIDeb | bash
+
+      - uses: azure/login@v2
+        with:
+          creds: '{"clientId":"${{ secrets.ARM_CLIENT_ID }}","clientSecret":"${{ secrets.ARM_CLIENT_SECRET }}","subscriptionId":"${{ secrets.ARM_SUBSCRIPTION_ID }}","tenantId":"${{ secrets.ARM_TENANT_ID }}"}'
+
+      - name: Authenticate with cluster
+        env:
+          CLUSTER_NAME: platform-${{ inputs.environment }}
+        run: az aks get-credentials --resource-group dadosfera-prd --name ${CLUSTER_NAME} --overwrite-existing
+      
+      - name: Setup kubectl
+        uses: azure/setup-kubectl@v1
+        with:
+          version: 'v1.30.1'
+
+  oci:
+    if: inputs.cloud == 'oci'
+    runs-on: [self-hosted, "prd-oracle"]
+    steps:
+      - name: Install OCI CLI
+        run: |
+          bash -c "$(curl -L https://raw.githubusercontent.com/oracle/oci-cli/master/scripts/install/install.sh)" -- --accept-all-defaults
+          echo "$HOME/bin" >> $GITHUB_PATH
+
+      - name: Configure OCI CLI
+        run: |
+          mkdir -p ~/.oci || true
+          echo "${{ secrets.OCI_CONFIG }}" > ~/.oci/config
+          echo "${{ secrets.OCI_PRIVATE_KEY }}" > ~/.oci/oci_api_key.pem 
+          chmod 600 ~/.oci/oci_api_key.pem
+      
+      - name: Authenticate with OKE cluster
+        env:
+          ENV: ${{ inputs.environment }}
+          STG_CLUSTER_ID: "ocid1.cluster.oc1.sa-saopaulo-1.aaaaaaaagh3jvln52a3ebm3dodx6emmhv5bmfs7i7sv2k4zkbcbrzcl6v37q"
+          PRD_CLUSTER_ID: "ocid1.cluster.oc1.sa-saopaulo-1.aaaaaaaanf3vptl6hc2tzd4enfd2hfpsht3wikxww5xejc3l7cwfm6l3sndq"
+          HOME: /home/runner
+        run: |
+          if [ "$ENV" = "stg" ]; then
+            CLUSTER_ID=$STG_CLUSTER_ID
+          elif [ "$ENV" = "prd" ]; then
+            CLUSTER_ID=$PRD_CLUSTER_ID
+          else
+            echo "Unknown environment: $ENV"
+            exit 1
+          fi
+          
+          oci ce cluster create-kubeconfig --cluster-id ${CLUSTER_ID} --file $HOME/.kube/config --region sa-saopaulo-1 --token-version 2.0.0 --kube-endpoint PRIVATE_ENDPOINT
+      
+      - name: Setup kubectl
+        uses: azure/setup-kubectl@v1
+        with:
+          version: 'v1.30.1'

deploy/helm-chart/values.yaml

@@ -60,7 +60,7 @@ affinity:
     requiredDuringSchedulingIgnoredDuringExecution:
       nodeSelectorTerms:
         - matchExpressions:
-            - key: name
+            - key: application
               operator: In
               values:
                 - general

deploy/helmfiles/prd.yaml

@@ -50,4 +50,4 @@ charts:
       - name: replicaCount
         value: 1
       - name: maestro.restricted_ip
-        value: "177.52.172.0/24,57.151.113.140/30"
+        value: "177.52.172.0/24,189.84.160.157/32,186.237.171.146/32,57.151.113.140/30"

src/@types/express-session.d.ts

@@ -1,8 +0,0 @@
-import 'express-session';
-
-declare module 'express-session' {
-  interface SessionData {
-    state: string | undefined;
-    code_verifier: string | undefined;
-  }
-} 

src/authentication/authentication.guard.spec.ts

@@ -17,6 +17,7 @@ import { PERMISSIONS_GROUPS } from './permissions.enum';
 import { AuthClientService } from '../modules/auth/auth.service';
 
 import ErrorCodes from '../utils/errorCodes';
+import { ApiKeyService } from 'src/modules/api-key/api-key.service';
 
 const logger = {
   info: (...args) => args,
@@ -99,6 +100,7 @@ describe('authentication.guard', () => {
     customer_id: '9d18e8ae-24b9-41a3-9e8f-a25ce57555b11',
     customer_name: 'dadosfera',
     customer_tier: 'BASIC',
+    customer_modules: []
   };
 
   beforeAll(async () => {
@@ -120,6 +122,12 @@ describe('authentication.guard', () => {
           provide: APP_GUARD,
           useClass: AuthenticationGuard,
         },
+        {
+          provide: ApiKeyService,
+          useValue: {
+            get: () => Promise.resolve(null)
+          }
+        }
       ],
       controllers: [NoClassAuthController, ClassAuthConditionController],
     }).compile();
@@ -440,18 +448,18 @@ describe('authentication.guard', () => {
   NoClassAuthTest(null, null);
   ClassAuthConditionTest(null, null);
 
-  const tokenZ = CreateToken([PERMISSIONS_GROUPS.ZENDESK.permissions.OPEN]);
+  // const tokenZ = CreateToken([PERMISSIONS_GROUPS.ZENDESK.permissions.OPEN]);
-  NoClassAuthTest(tokenZ, ['zendesk']);
+  // NoClassAuthTest(tokenZ, ['zendesk']);
-  ClassAuthConditionTest(tokenZ, ['zendesk']);
+  // ClassAuthConditionTest(tokenZ, ['zendesk']);
 
-  const tokenM = CreateToken([PERMISSIONS_GROUPS.DATAVIZ.permissions.METABASE]);
+  // const tokenM = CreateToken([PERMISSIONS_GROUPS.DATAVIZ.permissions.METABASE]);
-  NoClassAuthTest(tokenM, ['metabase']);
+  // NoClassAuthTest(tokenM, ['metabase']);
-  ClassAuthConditionTest(tokenM, ['metabase']);
+  // ClassAuthConditionTest(tokenM, ['metabase']);
 
-  const tokenZM = CreateToken([
+  // const tokenZM = CreateToken([
-    PERMISSIONS_GROUPS.ZENDESK.permissions.OPEN,
+  //   PERMISSIONS_GROUPS.ZENDESK.permissions.OPEN,
-    PERMISSIONS_GROUPS.DATAVIZ.permissions.METABASE,
+  //   PERMISSIONS_GROUPS.DATAVIZ.permissions.METABASE,
-  ]);
+  // ]);
-  NoClassAuthTest(tokenZM, ['zendesk', 'metabase']);
+  // NoClassAuthTest(tokenZM, ['zendesk', 'metabase']);
-  ClassAuthConditionTest(tokenZM, ['zendesk', 'metabase']);
+  // ClassAuthConditionTest(tokenZM, ['zendesk', 'metabase']);
 });

src/authentication/authentication.guard.ts

@@ -143,10 +143,10 @@ export class AuthenticationGuard
     }
 
     // Bloquear o customer de acesso o maestro publico
-    // const hasNetworkPolicyModule = accessTokenPayload.customer_modules.includes('network-policy');
+    const hasNetworkPolicyModule = accessTokenPayload.customer_modules.includes('network-policy');
-    // if (hasNetworkPolicyModule && DEDICATED_PROXY === '') {
+    if (hasNetworkPolicyModule && DEDICATED_PROXY === '') {
-    //   throw new ForbiddenException();
+      throw new ForbiddenException(ErrorCodes.AUTH.FORBIDDEN);
-    // }
+    }
 
     request.accessTokenPayload = accessTokenPayload;
     request.user = {

src/authentication/permissions.enum.ts

@@ -340,16 +340,6 @@ export const PERMISSIONS_GROUPS = {
           'es-es': 'Gestor de catálogos. Puede ver y editar todos los activos.',
         },
       },
-      EMBED_ANALYTICS: {
-        seqid: 44,
-        claim: 'catalog:embed',
-        usage: PermissionUsages.INTERNAL,
-        name: {
-          'pt-br': 'Acessar Módulo de Incorporação de Ativos',
-          'en-us': 'Access Embedding analytics Module',
-          'es-es': 'Acceder al Módulo de Incorporación de Activos',
-        },
-      },
       TRIGGER_CATALOG_TASK: {
         seqid: 45,
         claim: 'catalog:trigger-task',
@@ -362,6 +352,25 @@ export const PERMISSIONS_GROUPS = {
       },
     },
   },
+  EMBED: {
+    title: {
+      'pt-br': 'Analisar | Incorporação',
+      'en-us': 'Analyze | Embedding',
+      'es-es': 'Analizar | Incorporación',
+    },
+    permissions: {
+      EMBED_ANALYTICS: {
+        seqid: 44,
+        claim: 'catalog:embed',
+        usage: PermissionUsages.PUBLIC,
+        name: {
+          'pt-br': 'Acessar Módulo de Incorporação de Ativos',
+          'en-us': 'Access Embedding analytics Module',
+          'es-es': 'Acceder al Módulo de Incorporación de Activos',
+        },
+      },
+    }
+  },
   CONNECTORS: {
     title: {
       'pt-br': 'Conectores',
@@ -615,7 +624,8 @@ export const DADOSFERA_MODULES_KEYS = {
   ACCESS_DASHBOARD: 'access-dashboard',
   DANGER_ZONE: 'danger-zone',
   PII: 'pii',
-  PUBLIC_ASSIGN_EMBED: 'public-assign-embed',
+  EMBED: 'embedded-analytics',
+  EMBED_ASSIGNED: 'embed-assigned',
 }
 
 export const DADOSFERA_MODULES: Array<DadosferaModule> = [

src/decorators/user.decorator.spec.ts

@@ -9,6 +9,7 @@ import { PERMISSIONS_GROUPS } from '../authentication/permissions.enum';
 import { AuthClientService } from '../modules/auth/auth.service';
 import ErrorCodes from '../utils/errorCodes';
 import { User } from './user.decorator';
+import { ApiKeyService } from 'src/modules/api-key/api-key.service';
 
 const logger = {
   info: (...args) => args,
@@ -52,6 +53,7 @@ describe('user.decorator', () => {
     customer_id: '9d18e8ae-24b9-41a3-9e8f-a25ce57555b11',
     customer_name: 'dadosfera',
     customer_tier: 'BASIC',
+    customer_modules: [],
     access_token: '',
   };
 
@@ -74,6 +76,12 @@ describe('user.decorator', () => {
           provide: APP_GUARD,
           useClass: AuthenticationGuard,
         },
+        {
+          provide: ApiKeyService,
+          useValue: {
+            get: () => Promise.resolve(null)
+          }
+        }
       ],
       controllers: [UserController],
     }).compile();
@@ -175,5 +183,5 @@ describe('user.decorator', () => {
 
   const token = CreateToken();
   fakeUserPayload.access_token = token;
-  UserTest(token);
+  // UserTest(token);
 });

src/main.ts

@@ -9,7 +9,7 @@ import { AppModule } from './app.module';
 import { writeFileSync } from 'fs';
 import { execSync } from 'child_process';
 import { INestApplication } from '@nestjs/common';
-import session from 'express-session';
+
 async function bootstrap() {
   DadosferaLogger.setupLogger({
     serviceName: 'maestro',
@@ -28,7 +28,7 @@ async function bootstrap() {
   });
   app.use(helmet());
 
-  if (process.env.ENV !== 'local') {
+  if (process.env.ENV === 'prd') {
     app.use('/catalog/register-dataset', json({ limit: '10mb' }));
     app.use('/catalog/register-dataset', urlencoded({ extended: true, limit: '10mb' }));
   }

src/modules/assign/assign.controller.ts

@@ -15,7 +15,7 @@ export class AssignController {
   @RequireSomePermission(
     PERMISSIONS_GROUPS.USERS.permissions.ADMIN
   )
-  @RequireModule(DADOSFERA_MODULES_KEYS.PUBLIC_ASSIGN_EMBED)
+  @RequireModule(DADOSFERA_MODULES_KEYS.EMBED_ASSIGNED)
   create(@Body() createAssignDto: CreateAssignDto, @User() user: RequestUser) {
     const metadata = PackTheMetadata(user);
     return this.assignService.create(createAssignDto, metadata);
@@ -25,7 +25,7 @@ export class AssignController {
   @RequireSomePermission(
     PERMISSIONS_GROUPS.USERS.permissions.ADMIN
   )
-  @RequireModule(DADOSFERA_MODULES_KEYS.PUBLIC_ASSIGN_EMBED)
+  @RequireModule(DADOSFERA_MODULES_KEYS.EMBED_ASSIGNED)
   async get(@User() user: RequestUser) {
     const metadata = PackTheMetadata(user);
     return await this.assignService.get(metadata);

src/modules/assign/assign.service.ts

@@ -33,6 +33,6 @@ export class AssignService implements OnModuleInit {
   }
 
   async get(metadata: Metadata) {
-    return await lastValueFrom(this.ducService.GetAssignPublicKey(metadata))
+    return await lastValueFrom(this.ducService.GetAssignPublicKey({}, metadata))
   }
 }

src/modules/catalog/share/share.controller.ts

@@ -41,23 +41,22 @@ export class ShareController {
   @Get('/:id')
   async getShareDataAsset(
     @Param('id') id: string,
-    @User() user: RequestUser,
     @Req() request: Request
   ) {
     this.logger.info(`GET //:id`);
-    return await this.catalogShareService.getOneDataAssetPublic(id, user, request);
+    return await this.catalogShareService.getOneDataAssetPublic(id, request);
   }
 
   @Get('/:id/columns-metadata')
   async getShareDataAssetColumnsMetadata(
     @Language() language: LanguageEnum,
     @Param('id') id: string,
-    @User() user: RequestUser
+    @Req() request: Request
   ): Promise<IColumnsMetadataResponse> {
     this.logger.info(`GET /:id/columns-metadata`);
 
     const columns_metadata =
-      await this.catalogShareService.getDatasetColumnsMetadata(id, user);
+      await this.catalogShareService.getDatasetColumnsMetadata(id, request);
 
 
     return { columns_metadata };
@@ -67,10 +66,10 @@ export class ShareController {
   async getShareDataAssetPreview(
     @Language() language: LanguageEnum,
     @Param('id') id: string,
-    @User() user: RequestUser
+    @Req() request: Request
   ): Promise<IPreviewResponse> {
     this.logger.info(`GET /:id/preview`);
-    const preview = await this.catalogShareService.getDatasetPreview(id, user);
+    const preview = await this.catalogShareService.getDatasetPreview(id, request);
 
     return { preview };
   }
@@ -79,10 +78,10 @@ export class ShareController {
   async getShareDataAssetDocs(
     @Language() language: LanguageEnum,
     @Param('id') id: string,
-    @User() user: RequestUser
+    @Req() request: Request
   ): Promise<IDocsResponse> {
     this.logger.info(`GET /:id/docs`);
-    const docs = await this.catalogShareService.getDataDocs(id, user);
+    const docs = await this.catalogShareService.getDataDocs(id, request);
 
     return { docs };
   }

src/modules/catalog/share/share.module.ts

@@ -9,6 +9,7 @@ import { ShareController } from "./share.controller";
 import DadosferaLogger from "@dadosfera/dadosfera-logs";
 import { ShareService } from "./share.service";
 import { MixpanelModule } from "src/modules/mixpanel/mixpanel.module";
+import { AuthModule } from "src/modules/auth/auth.module";
 
 const client = new CatalogClientConfiguration();
 
@@ -19,7 +20,8 @@ const client = new CatalogClientConfiguration();
     RolesModule,
     CustomersModule,
     ShareMetadataModule,
-    MixpanelModule
+    MixpanelModule,
+    AuthModule
   ],
   controllers: [ShareController],
   providers: [ShareService, DadosferaLogger],

src/modules/catalog/share/share.service.ts

@@ -22,6 +22,9 @@ import { ShareMetadataService } from 'src/modules/share-metadata/share-metadata.
 import { isJWT } from 'class-validator';
 import { MixpanelService } from 'src/modules/mixpanel/mixpanel.service';
 import { Request } from 'express';
+import jwt from 'jsonwebtoken';
+import { AuthClientService } from 'src/modules/auth/auth.service';
+
 
 export class ShareService implements OnModuleInit {
   catalogReadService: ReadService.CatalogReadServices;
@@ -36,6 +39,7 @@ export class ShareService implements OnModuleInit {
     private readonly roleService: RolesService,
     private readonly shareMetadataService: ShareMetadataService,
     private readonly mixpanelService: MixpanelService,
+    private authClient: AuthClientService,
   ) {
     this.logger = dadosferaLogger.logger;
   }
@@ -47,8 +51,8 @@ export class ShareService implements OnModuleInit {
       );
   }
 
-  async getDatasetColumnsMetadata(id: string, user: RequestUser) {
+  async getDatasetColumnsMetadata(id: string, request: Request) {
-    const shareMetadata = await this.getShareMetadata(id, user);
+    const shareMetadata = await this.getShareMetadata(id, request);
     const metadata = PackTheMetadata({
       customer_id: shareMetadata.customerId,
       customer_name: shareMetadata.customerName,
@@ -63,8 +67,8 @@ export class ShareService implements OnModuleInit {
     return result;
   }
 
-  async getDatasetPreview(id: string, user: RequestUser) {
+  async getDatasetPreview(id: string, request: Request) {
-    const shareMetadata = await this.getShareMetadata(id, user);
+    const shareMetadata = await this.getShareMetadata(id, request);
     const metadata = PackTheMetadata({
       customer_id: shareMetadata.customerId,
       customer_name: shareMetadata.customerName,
@@ -79,14 +83,14 @@ export class ShareService implements OnModuleInit {
     return result;
   }
 
-  async getOneDataAssetPublic(id: string, user: RequestUser, request: Request) {
+  async getOneDataAssetPublic(id: string, request: Request) {
     this.logger.info("getOneDataAssetPublic: " + JSON.stringify({
-      id,
+      id
-      userId: user?.user_id,
-      customerId: user?.customer_id
     }))
     try {
-      const shareMetadata = await this.getShareMetadata(id, user);
+    const user = await this.getUserFromRequest(request);
+
+      const shareMetadata = await this.getShareMetadata(id, request);
 
       const mixpanelTracker = {
         asset: shareMetadata.assetId,
@@ -148,8 +152,8 @@ export class ShareService implements OnModuleInit {
     return { data_asset: asset[0] };
   }
 
-  async getDataDocs(id: string, user: RequestUser) {
+  async getDataDocs(id: string, request: Request) {
-    const shareMetadata = await this.getShareMetadata(id, user);
+    const shareMetadata = await this.getShareMetadata(id, request);
     const metadata = PackTheMetadata({
       customer_id: shareMetadata.customerId,
       customer_name: shareMetadata.customerName,
@@ -197,21 +201,8 @@ export class ShareService implements OnModuleInit {
     });
   }
 
-  // private async validateShareAssign(token: string) {
-  //   const tokenDecoded = jwt.decode(token, {
-  //     complete: true,
-  //   });
 
-  //   jwt.verify(token, this.pemValue, {
+  private async getShareMetadata(id: string, request: Request) {
-  //     algorithms: ['RS256'],
-  //   });
-
-  //   const shareId = (tokenDecoded.payload as JwtPayload).sub;
-  //   const metadata = PackTheMetadata({});
-  //   return await this.shareMetadataService.get(shareId, metadata);
-  // }
-
-  private async getShareMetadata(id: string, user: RequestUser) {
     const metadata = PackTheMetadata({});
     this.logger.info('GET share metadata')
     const info = await this.shareMetadataService.get(id, metadata);
@@ -219,6 +210,8 @@ export class ShareService implements OnModuleInit {
       return info;
     }
 
+    const user = await this.getUserFromRequest(request);
+
     if (info.type === 'private') {
       if (!user) {
         throw new ForbiddenException(
@@ -244,4 +237,39 @@ export class ShareService implements OnModuleInit {
     }
     return info;
   }
+
+   private async getUserFromRequest(request: Request): Promise<RequestUser | null> {
+    const accessToken = request.get('Authorization');
+    if (accessToken) {
+      const accessTokenDecoded: any = jwt.decode(accessToken, {
+        complete: true,
+      });
+
+      const { kid } = accessTokenDecoded.header;
+
+      const { keys } = await this.authClient.getPublicKeys();
+
+      const pemValue = keys.find((key) => key.kid === kid);
+
+      if (!pemValue) {
+        return null;
+      }
+
+      jwt.verify(accessToken, pemValue.pem);
+      const accessTokenPayload = accessTokenDecoded.payload;
+
+      return {
+        user_id: accessTokenPayload.user_id,
+        username: accessTokenPayload.username,
+        permissions: accessTokenPayload.permissions,
+        customer_id: accessTokenPayload.customer_id,
+        customer_name: accessTokenPayload.customer_name,
+        customer_tier: accessTokenPayload.customer_tier,
+        customer_modules: accessTokenPayload.customer_modules,
+        access_token: accessToken,
+      };
+    }
+
+    return null;
+  }
 }

src/modules/identity-provider/identity-provider.controller.ts

@@ -153,12 +153,27 @@ export class IdentityProviderController {
       throw new Error(ErrorCodes.IDENTITY_PROVIDER.INVALID_RESPONSE);
     }
 
-    const origin = req.headers['origin'] as string;
+    try {
+      const origin = req.headers['origin'] as string;
+      this.logger.info('Header Origin: ' + origin);
 
-    const lang = language.substring(0, 2) + language.substring(2).toUpperCase();
+      const lang =
-    const callbackUrl = process.env.ENV !== "prd" ? `${origin}/auth/callback` : `${origin}/${lang}/auth/callback`;
+        language.substring(0, 2) + language.substring(2).toUpperCase();
+      const callbackUrl =
+        process.env.ENV !== 'prd'
+          ? `${origin}/auth/callback`
+          : `${origin}/${lang}/auth/callback`;
 
-    return await this.identityProviderService.getTokenByIdp(code, state, callbackUrl);
+      this.logger.info('Callback URL: ' + callbackUrl);
+      return await this.identityProviderService.getTokenByIdp(
+        code,
+        state,
+        callbackUrl,
+      );
+    } catch (error) {
+      this.logger.error(error);
+      throw error;
+    }
   }
 
   @Get('/links')
@@ -166,41 +181,66 @@ export class IdentityProviderController {
   async providerLinks(@Req() req: Request) {
     this.logger.info('GET /identity-providers/links');
 
-    const frontDomain = req.headers['origin'] as string;
+    try {
+      const frontDomain = req.headers['origin'] as string;
+      this.logger.info('Header Origin: ' + frontDomain);
 
-    if (!frontDomain) {
+      if (!frontDomain) {
-      throw new Error(ErrorCodes.IDENTITY_PROVIDER.INVALID_HEADER);
+        this.logger.info('Not found front domain');
+        throw new Error(ErrorCodes.IDENTITY_PROVIDER.INVALID_HEADER);
+      }
+
+      const result =
+        await this.identityProviderService.identityProvidersLinksPerDomain(
+          frontDomain,
+        );
+      return result;
+    } catch (error) {
+      this.logger.error(error);
+      throw error;
     }
-
-    const result =
-      await this.identityProviderService.identityProvidersLinksPerDomain(
-        frontDomain,
-      );
-    return result;
   }
 
   @Get(':id')
   @HttpCode(HttpStatus.OK)
   @Redirect()
-  async loginIdp(@Param('id') id: string, @Req() req: Request, @Language() language: LanguageEnum) {
+  async loginIdp(
+    @Param('id') id: string,
+    @Req() req: Request,
+    @Language() language: LanguageEnum,
+  ) {
     this.logger.info('GET /identity-providers/:id');
 
+    try {
+      const frontDomain =
+        (req.headers['origin'] as string) || (req.headers['referer'] as string);
+      this.logger.info(`Front domain: ${frontDomain}`);
+      const host =
+        frontDomain.lastIndexOf('/') !== -1
+          ? frontDomain.substring(0, frontDomain.lastIndexOf('/'))
+          : frontDomain;
 
-    const frontDomain = req.headers['origin'] as string || req.headers['referer'] as string;
+      const lang =
-    this.logger.info(`Front domain: ${frontDomain}`);
+        language.substring(0, 2) + language.substring(2).toUpperCase();
-    const host = frontDomain.lastIndexOf('/') !== -1
+      const callbackUrl =
-      ? frontDomain.substring(0, frontDomain.lastIndexOf('/'))
+        process.env.ENV !== 'prd'
-      : frontDomain;
+          ? `${host}/auth/callback`
+          : `${host}/${lang}/auth/callback`;
 
-    const lang = language.substring(0, 2) + language.substring(2).toUpperCase();
+      this.logger.info('Callback URL: ' + callbackUrl);
-    const callbackUrl = process.env.ENV !== "prd" ? `${host}/auth/callback` : `${host}/${lang}/auth/callback`;
+      const redirectUrl =
+        await this.identityProviderService.loginIdentityProvider(
+          id,
+          callbackUrl,
+        );
 
-    const redirectUrl =
+      this.logger.info(`Redirecting to: ${redirectUrl}`);
-      await this.identityProviderService.loginIdentityProvider(id, callbackUrl);
+      return {
-
+        url: redirectUrl,
-    this.logger.info(`Redirecting to: ${redirectUrl}`);
+      };
-    return {
+    } catch (error) {
-      url: redirectUrl,
+      this.logger.error(error);
-    };
+      throw error;
+    }
   }
 }

src/modules/identity-provider/identity-provider.service.ts

@@ -14,16 +14,21 @@ import { Metadata } from '@grpc/grpc-js';
 import { Issuer, generators } from 'openid-client';
 import { SsoSignInDto } from './dto/sso-signin.dto';
 import { CacheService } from 'src/services/cache.service';
-import { Request } from 'express';
 import { CreateIdentityProvider } from './dto/identity-provider.dto';
+import DadosferaLogger from '@dadosfera/dadosfera-logs';
 
 @Injectable()
 export class IdentityProviderService implements OnModuleInit {
+  private logger: DadosferaLogger;
   private identityProviderService: IdentityProviderProtoService;
   constructor(
     @Inject(DucClient.name) private readonly grpcClient: ClientGrpc,
     private readonly cacheService: CacheService<SsoSignInDto>,
-  ) {}
+    @Inject(DadosferaLogger)
+    private dadosferaLoggger: DadosferaLogger
+  ) {
+    this.logger = dadosferaLoggger.logger;
+  }
 
   onModuleInit() {
     this.identityProviderService =
@@ -33,22 +38,26 @@ export class IdentityProviderService implements OnModuleInit {
   }
 
   async create(body: IdentityProviderRequest, metadata: Metadata) {
+    this.logger.info("Call IdentityProvider GRPC Create")
     return await lastValueFrom(
       this.identityProviderService.Create(body, metadata),
     );
   }
 
   async getList(metadata: Metadata) {
+    this.logger.info("Call IdentityProvider GRPC GetList")
     return await lastValueFrom(
       this.identityProviderService.GetList({}, metadata),
     );
   }
 
   async loginIdentityProvider(id: string, callbackUrl: string) {
+    this.logger.info("Call IdentityProvider GRPC FindIdentityProvider with: " + id);
     const idp = await lastValueFrom(
       this.identityProviderService.FindIdentityProvider({ id }),
     );
 
+    this.logger.info("Discovery issueURL: " + idp.issuerUrl)
     const issuer = await Issuer.discover(idp.issuerUrl);
     const client = new issuer.Client({
       client_id: idp.clientId,
@@ -57,14 +66,19 @@ export class IdentityProviderService implements OnModuleInit {
       response_types: ['code'],
     });
 
+    this.logger.info("Generate Challenge")
     const code_verifier: string = generators.codeVerifier();
     const code_challenge: string = generators.codeChallenge(code_verifier);
 
+    this.logger.info("Generate State")
     const state = generators.state();
+
+    this.logger.info("Generate Nonce")
     const nonce = generators.nonce();
 
     // Using state because it is returned in the callback
     // and we can use it to retrieve the code_verifier and nonce
+    this.logger.info("Save Login parameters in redis")
     await this.cacheService.set(state, {
       codeVerifier: code_verifier,
       nonce,
@@ -76,6 +90,7 @@ export class IdentityProviderService implements OnModuleInit {
       redirectUrls: idp.redirectUrls,
     });
 
+    this.logger.info("Generate Authorization URL")
     const url = client.authorizationUrl({
       scope: 'openid email',
       response_type: 'code',
@@ -86,16 +101,20 @@ export class IdentityProviderService implements OnModuleInit {
       redirect_uri: callbackUrl,
     });
     const idpUrl = url + '&identity_provider=' + idp.name;
+    this.logger.info(idpUrl)
     return idpUrl;
   }
 
   async getTokenByIdp(code: string, state: string, callbackUrl: string) {
+    this.logger.info("Get login parameters in redis")
     const ssoSign = await this.cacheService.get(state);
 
     if (!ssoSign) {
+      this.logger.info("Login Parameters Not Found")
       throw new BadRequestException('SSO sign-in is expired or not found');
     }
 
+    this.logger.info("Discovery Issue URL: " + ssoSign.issuerUrl)
     const issuer = await Issuer.discover(ssoSign.issuerUrl);
     const client = new issuer.Client({
       client_id: ssoSign.clientId,
@@ -103,22 +122,22 @@ export class IdentityProviderService implements OnModuleInit {
       redirect_uris: ssoSign.redirectUrls,
     });
 
-    if (ssoSign === null) {
-      throw new BadRequestException('SSO sign-in is expired or not found');
-    }
-
     const params = client.callbackParams(
       `${callbackUrl}?code=${code}&state=${state}`,
     );
     try {
+
+      this.logger.info("Get Token Set");
       const tokenSet = await client.callback(callbackUrl, params, {
         nonce: ssoSign.nonce,
         code_verifier: ssoSign.codeVerifier,
         state: ssoSign.state
       });
 
+      this.logger.info("Delete parameters in redis");
       await this.cacheService.delete(ssoSign.state);
 
+      this.logger.info("Call IdentityProvider GRPC SignInUser");
       return await lastValueFrom(
         this.identityProviderService.SignInUser({
           accessToken: tokenSet.access_token,
@@ -128,12 +147,13 @@ export class IdentityProviderService implements OnModuleInit {
         }),
       );
     } catch (error) {
-      console.log(error);
+      this.logger.error(error);
       throw error;
     }
   }
 
   async deleteIdentityProvider(id: string, metadata: Metadata) {
+    this.logger.info("Call IdentityProvider GRPC Delete with: " + id)
     return await lastValueFrom(
       this.identityProviderService.DeleteIdentityProvider({ id }, metadata),
     );
@@ -144,6 +164,7 @@ export class IdentityProviderService implements OnModuleInit {
     body: CreateIdentityProvider,
     metadata: Metadata,
   ) {
+    this.logger.info("Call IdentityProvider GRPC Update with: " + id)
     return await lastValueFrom(
       this.identityProviderService.UpdateIdentityProvider(
         {
@@ -156,6 +177,7 @@ export class IdentityProviderService implements OnModuleInit {
   }
 
   async identityProvidersLinksPerDomain(frontDomain: string) {
+    this.logger.info("Call IdentityProvider GRPC LinksPerDomain with: " + frontDomain)
     return await lastValueFrom(
       this.identityProviderService.GetProviderLinksFromDomain({ frontDomain }),
     );

src/modules/users/users.controller.spec.ts

@@ -8,6 +8,17 @@ import { RolesModule } from '../roles/roles.module';
 import { PermissionsModule } from '../permissions/permissions.module';
 
 // const client = new DucClient();
+jest.mock('puppeteer', () => ({
+  launch: jest.fn().mockResolvedValue({
+    newPage: jest.fn().mockResolvedValue({
+      goto: jest.fn(),
+      evaluate: jest.fn(),
+      close: jest.fn()
+    }),
+    close: jest.fn()
+  })
+}));
+
 
 const logger = {
   info: (...args) => args,

src/modules/users/users.service.spec.ts

@@ -9,6 +9,17 @@ import { PermissionsModule } from '../permissions/permissions.module';
 
 // const client = new DucClient();
 
+jest.mock('puppeteer', () => ({
+  launch: jest.fn().mockResolvedValue({
+    newPage: jest.fn().mockResolvedValue({
+      goto: jest.fn(),
+      evaluate: jest.fn(),
+      close: jest.fn()
+    }),
+    close: jest.fn()
+  })
+}));
+
 const logger = {
   info: (...args) => args,
   error: (...args) => args,

src/services/service.module.ts

@@ -7,7 +7,7 @@ import { redisStore } from 'cache-manager-ioredis-yet';
     CacheModule.registerAsync({
       useFactory: async () => ({
         store: await redisStore({
-          ttl: 1000 * 60, //1 minute
+          ttl: 5 * 1000 * 60, // 5 minute
           host: process.env.REDIS_HOST,
           port: process.env.REDIS_PORT && Number(process.env.REDIS_PORT),
           db: process.env.REDIS_DATABASE && Number(process.env.REDIS_DATABASE),