feat: adjust keypair generation and error handling in KeygenService

- Added OnApplicationBootstrap lifecycle method to check for existing keypair and generate a new one if not present.
- Enhanced generateKey method to handle keypair creation errors more effectively.

Author: kastov

libs/contract/constants/errors/errors.ts

@@ -443,4 +443,9 @@ export const ERRORS = {
         message: 'LIMITED and EXPIRED statuses are not allowed to be set manually.',
         httpCode: 400,
     },
+    KEYPAIR_CREATION_ERROR: {
+        code: 'A090',
+        message: 'Keypair creation error',
+        httpCode: 500,
+    },
 } as const;

src/modules/keygen/commands/get-node-jwt/get-node-jwt.handler.ts

@@ -42,6 +42,9 @@ export class GetNodeJwtHandler
             expiresIn: '9999d',
         });
 
+        // TODO: remove this after testing
+        this.logger.log(`Node JWT: ${token}, signed with private key: \n ${privKey}`);
+
         try {
             return {
                 isOk: true,

src/modules/keygen/keygen.service.ts

@@ -1,7 +1,7 @@
 import { generateKeyPair } from 'crypto';
 import { promisify } from 'util';
 
-import { Injectable, Logger } from '@nestjs/common';
+import { Injectable, Logger, OnApplicationBootstrap } from '@nestjs/common';
 
 import { ICommandResponse } from '@common/types/command-response.type';
 import { ERRORS } from '@libs/contracts/constants/errors';
@@ -12,11 +12,31 @@ import { KeygenEntity } from './entities/keygen.entity';
 const generateKeyPairAsync = promisify(generateKeyPair);
 
 @Injectable()
-export class KeygenService {
+export class KeygenService implements OnApplicationBootstrap {
     private readonly logger = new Logger(KeygenService.name);
 
     constructor(private readonly keygenRepository: KeygenRepository) {}
 
+    async onApplicationBootstrap() {
+        const pubKey = await this.keygenRepository.findFirstByCriteria({});
+
+        if (pubKey) {
+            this.logger.log('Keypair already exists, skipping...');
+            return;
+        }
+
+        const result = await this.generateKey();
+
+        if (!result.isOk) {
+            this.logger.error('Failed to generate keypair, please restart application.');
+            return;
+        }
+
+        this.logger.log('Keypair generated successfully.');
+
+        return;
+    }
+
     public async generateKey(): Promise<ICommandResponse<KeygenEntity>> {
         try {
             const pubKey = await this.keygenRepository.findFirstByCriteria({});
@@ -28,6 +48,30 @@ export class KeygenService {
                 };
             }
 
+            const newEntity = await this.createKeypair();
+
+            if (!newEntity) {
+                return {
+                    isOk: false,
+                    ...ERRORS.KEYPAIR_CREATION_ERROR,
+                };
+            }
+
+            return {
+                isOk: true,
+                response: newEntity,
+            };
+        } catch (error) {
+            this.logger.error(error);
+            return {
+                isOk: false,
+                ...ERRORS.GET_PUBLIC_KEY_ERROR,
+            };
+        }
+    }
+
+    private async createKeypair(): Promise<KeygenEntity | null> {
+        try {
             const { privateKey, publicKey } = await generateKeyPairAsync('rsa', {
                 modulusLength: 2048,
                 publicKeyEncoding: {
@@ -47,16 +91,10 @@ export class KeygenService {
 
             const newEntity = await this.keygenRepository.create(keygenEntity);
 
-            return {
-                isOk: true,
-                response: newEntity,
-            };
+            return newEntity;
         } catch (error) {
             this.logger.error(error);
-            return {
-                isOk: false,
-                ...ERRORS.GET_PUBLIC_KEY_ERROR,
-            };
+            return null;
         }
     }
 }