Merge pull request #35 from klayrHQ/update-validator

Update validator

src/validator/dto/get-validator-res.dto.ts

@@ -18,11 +18,26 @@ export class GetValidatorResponseDto {
   consecutiveMissedBlocks?: number;
   reportMisbehaviorHeights?: string;
   sharingCoefficients?: string;
+  status?: string;
 }
 
 export const getValidatorResponse: ApiResponseOptions = {
   status: 200,
   description: 'The validator have been successfully fetched.',
   type: GetValidatorResponseDto,
-  isArray: false,
+  isArray: true,
+};
+
+export class GetValidatorCountsResponseDto {
+  active: number;
+  ineligible: number;
+  standby: number;
+  punished: number;
+  banned: number;
+}
+
+export const getValidatorCountsResponse: ApiResponseOptions = {
+  status: 200,
+  description: 'The validator counts have been successfully fetched.',
+  type: GetValidatorCountsResponseDto,
 };

src/validator/validator.controller.ts

@@ -1,13 +1,19 @@
 import { Controller, Get, Query, UsePipes, ValidationPipe } from '@nestjs/common';
 import { ApiTags, ApiResponse } from '@nestjs/swagger';
 import { ValidatorRepoService } from './validator.repo-service';
-import { GetValidatorResponseDto, getValidatorResponse } from './dto/get-validator-res.dto';
+import {
+  GetValidatorCountsResponseDto,
+  GetValidatorResponseDto,
+  getValidatorCountsResponse,
+  getValidatorResponse,
+} from './dto/get-validator-res.dto';
 import { GatewayResponse, ValidatorSortTypes } from 'src/utils/controller-helpers';
 import { ValidatorQueryDto } from './dto/get-validator.dto';
 import { MAX_VALIDATORS_TO_FETCH } from 'src/utils/constants';
 import { Prisma } from '@prisma/client';
 import { NodeApiService } from 'src/node-api/node-api.service';
 import { Generator } from 'src/node-api/types';
+import { ValidatorStatus } from './types';
 
 @ApiTags('Validators')
 @Controller('pos')
@@ -62,6 +68,23 @@ export class ValidatorController {
     });
   }
 
+  @Get('validators/status-count')
+  @ApiResponse(getValidatorCountsResponse)
+  async countValidatorsByStatus(): Promise<GatewayResponse<{ [key: string]: number }>> {
+    const statuses = Object.values(ValidatorStatus);
+    const counts = await Promise.all(
+      statuses.map(async (status) => ({
+        [status]: await this.validatorRepoService.countValidators({ where: { status } }),
+      })),
+    );
+
+    const formattedCounts = counts.reduce((acc, countObj) => ({ ...acc, ...countObj }));
+    return {
+      data: formattedCounts,
+      meta: {},
+    };
+  }
+
   private getValidatorResponse(
     validator: Prisma.ValidatorGetPayload<{ include: { account: true } }>,
     list: Generator[],