fix: change SurveyResult to return isCurrentAccountAnswer property

Author: DanielAraldi

src/data/protocols/db/survey-result/load-survey-result-repository.ts

@@ -1,5 +1,8 @@
 import { SurveyResultModel } from '@/domain/models/survey-result';
 
 export interface LoadSurveyResultRepository {
-  loadBySurveyId(surveyId: string): Promise<SurveyResultModel | null>;
+  loadBySurveyId(
+    surveyId: string,
+    accountId: string
+  ): Promise<SurveyResultModel | null>;
 }

src/data/protocols/db/survey-result/save-survey-result-repository.ts

@@ -1,4 +1,3 @@
-import { SurveyResultModel } from '@/domain/models/survey-result';
 import { SaveSurveyResultParams } from '@/domain/usecases/survey-result/save-survey-result';
 
 export interface SaveSurveyResultRepository {

src/data/usecases/survey-result/load-survey-result/db-load-survey-result.ts

@@ -11,9 +11,13 @@ export class DbLoadSurveyResult implements LoadSurveyResult {
     private readonly loadSurveyByIdRepository: LoadSurveyByIdRepository
   ) {}
 
-  async load(surveyId: string): Promise<SurveyResultModel | null> {
+  async load(
+    surveyId: string,
+    accountId: string
+  ): Promise<SurveyResultModel | null> {
     let surveyResult = await this.loadSurveyResultRepository.loadBySurveyId(
-      surveyId
+      surveyId,
+      accountId
     );
 
     if (!surveyResult) {
@@ -28,6 +32,7 @@ export class DbLoadSurveyResult implements LoadSurveyResult {
           Object.assign({}, answer, {
             count: 0,
             percent: 0,
+            isCurrentAccountAnswer: false,
           })
         ),
       };

src/data/usecases/survey-result/save-survey-result/db-save-survey-result.ts

@@ -14,6 +14,9 @@ export class DbSaveSurveyResult implements SaveSurveyResult {
 
   async save(data: SaveSurveyResultParams): Promise<SurveyResultModel | null> {
     await this.saveSurveyResultRepository.save(data);
-    return this.loadSurveyResultRepository.loadBySurveyId(data.surveyId);
+    return this.loadSurveyResultRepository.loadBySurveyId(
+      data.surveyId,
+      data.accountId
+    );
   }
 }

src/domain/models/survey-result.ts

@@ -11,4 +11,5 @@ type SurveyResultAnswerModel = {
   count: number;
   percent: number;
   image?: string;
+  isCurrentAccountAnswer: boolean;
 };

src/domain/usecases/survey-result/load-survey-result.ts

@@ -1,5 +1,5 @@
 import { SurveyResultModel } from '@/domain/models/survey-result';
 
 export interface LoadSurveyResult {
-  load(surveyId: string): Promise<SurveyResultModel | null>;
+  load(surveyId: string, accountId: string): Promise<SurveyResultModel | null>;
 }

src/infra/db/mongodb/survey-result/survey-result-mongo-repository.ts

@@ -30,7 +30,10 @@ export class SurveyResultMongoRepository
     );
   }
 
-  async loadBySurveyId(surveyId: string): Promise<SurveyResultModel | null> {
+  async loadBySurveyId(
+    surveyId: string,
+    accountId: string
+  ): Promise<SurveyResultModel | null> {
     const surveyResultCollection = await MongoHelper.getCollection(
       'surveyResults'
     );
@@ -71,6 +74,15 @@ export class SurveyResultMongoRepository
         count: {
           $sum: 1,
         },
+        currentAccountAnswer: {
+          $push: {
+            $cond: [
+              { $eq: ['$data.accountId', MongoHelper.objectId(accountId)] },
+              '$data.answerId',
+              null,
+            ],
+          },
+        },
       })
       .builder('$project', {
         _id: 0,
@@ -110,6 +122,12 @@ export class SurveyResultMongoRepository
                       else: 0,
                     },
                   },
+                  isCurrentAccountAnswer: {
+                    $eq: [
+                      '$$item.answerId',
+                      { $arrayElemAt: ['$currentAccountAnswer', 0] },
+                    ],
+                  },
                 },
               ],
             },
@@ -152,6 +170,7 @@ export class SurveyResultMongoRepository
           answerId: '$answers.answerId',
           answer: '$answers.answer',
           image: '$answers.image',
+          isCurrentAccountAnswer: '$answers.isCurrentAccountAnswer',
         },
         count: {
           $sum: '$answers.count',
@@ -169,8 +188,13 @@ export class SurveyResultMongoRepository
           answerId: '$_id.answerId',
           answer: '$_id.answer',
           image: '$_id.image',
-          count: '$count',
-          percent: '$percent',
+          count: {
+            $round: ['$count'],
+          },
+          percent: {
+            $round: ['$percent'],
+          },
+          isCurrentAccountAnswer: '$_id.isCurrentAccountAnswer',
         },
       })
       .builder('$sort', {

src/presentation/controllers/survey-result/load-survey-result/load-survey-result-controller.ts

@@ -23,7 +23,10 @@ export class LoadSurveyResultController implements Controller {
       const survey = await this.loadSurveyById.loadById(surveyId);
       if (!survey) return forbidden(new InvalidParamError('surveyId'));
 
-      const surveyResult = await this.loadSurveyResult.load(surveyId);
+      const surveyResult = await this.loadSurveyResult.load(
+        surveyId,
+        httpRequest.accountId
+      );
       if (!surveyResult) return forbidden(new InvalidParamError('surveyId'));
 
       return ok(surveyResult);