[TRELLO-2799] Remove Albert classification for RGPD deletion (#1806)

* [TRELLO-2799] Fix Albert

* [TRELLO-2799] Remove Albert classification for RGPD deletion

app/loader/SignalConsoApplicationLoader.scala

@@ -253,6 +253,8 @@ class SignalConsoComponents(
 
   val ipBlackListRepository = new IpBlackListRepository(dbConfig)
 
+  val albertClassificationRepository = new AlbertClassificationRepository(dbConfig)
+
   val crypter = new JcaCrypter(applicationConfiguration.crypter)
   val signer  = new JcaSigner(applicationConfiguration.signer)
 
@@ -509,7 +511,8 @@ class SignalConsoComponents(
     engagementOrchestrator,
     reportRepository,
     reportFileOrchestrator,
-    eventRepository
+    eventRepository,
+    albertClassificationRepository
   )
 
   val reportAdminActionOrchestrator = new ReportAdminActionOrchestrator(
@@ -720,7 +723,7 @@ class SignalConsoComponents(
     companyRepository,
     emailNotificationOrchestrator,
     ipBlackListRepository,
-    new AlbertClassificationRepository(dbConfig),
+    albertClassificationRepository,
     albertService,
     frontRoute,
     cookieAuthenticator,

app/models/report/Report.scala

@@ -81,6 +81,12 @@ case class Report(
     // /!\ Some reports don't have description
     // in some subcategories, there is no description field (or it's called differently)
     details.find(_.label == "Description :").map(_.value.trim).filterNot(_.isEmpty)
+
+  def getReponseConsoDescription: Option[String] =
+    details
+      .find(x => x.label == "Your question :" || x.label == "Votre question :")
+      .map(_.value.trim)
+      .filterNot(_.isEmpty)
 }
 
 object Report {

app/orchestrators/RgpdOrchestrator.scala

@@ -5,6 +5,7 @@ import models.report.ReportStatus.SuppressionRGPD
 import models.report._
 import play.api.Logger
 import play.api.libs.json.Json
+import repositories.albert.AlbertClassificationRepositoryInterface
 import repositories.event.EventFilter
 import repositories.event.EventRepositoryInterface
 import repositories.report.ReportRepositoryInterface
@@ -20,7 +21,8 @@ class RgpdOrchestrator(
     engagementOrchestrator: EngagementOrchestrator,
     reportRepository: ReportRepositoryInterface,
     reportFileOrchestrator: ReportFileOrchestrator,
-    eventRepository: EventRepositoryInterface
+    eventRepository: EventRepositoryInterface,
+    albertClassificationRepository: AlbertClassificationRepositoryInterface
 )(implicit val executionContext: ExecutionContext) {
   val logger = Logger(this.getClass)
 
@@ -59,8 +61,11 @@ class RgpdOrchestrator(
       _ <- reportConsumerReviewOrchestrator.deleteDetails(emptiedReport.id)
       _ <- engagementOrchestrator.deleteDetails(emptiedReport.id)
       _ <- reportFileOrchestrator.removeFromReportId(emptiedReport.id)
-      _ = logger.info(s"Report ${report.id} was emptied")
-    } yield emptiedReport
+      _ <- albertClassificationRepository.removeByReportId(emptiedReport.id)
+    } yield {
+      logger.info(s"Report ${report.id} was emptied")
+      emptiedReport
+    }
   }
 
 }

app/repositories/albert/AlbertClassificationRepository.scala

@@ -11,6 +11,7 @@ import scala.concurrent.Future
 trait AlbertClassificationRepositoryInterface {
   def getByReportId(reportId: UUID): Future[Option[AlbertClassification]]
   def createOrUpdate(element: AlbertClassification): Future[AlbertClassification]
+  def removeByReportId(reportId: UUID): Future[Int]
 }
 
 class AlbertClassificationRepository(dbConfig: DatabaseConfig[JdbcProfile])(implicit ec: ExecutionContext)
@@ -20,13 +21,17 @@ class AlbertClassificationRepository(dbConfig: DatabaseConfig[JdbcProfile])(impl
 
   import dbConfig._
 
-  def getByReportId(reportId: UUID): Future[Option[AlbertClassification]] = db.run(
+  override def getByReportId(reportId: UUID): Future[Option[AlbertClassification]] = db.run(
     table.filter(_.reportId === reportId).result.headOption
   )
 
-  def createOrUpdate(element: AlbertClassification): Future[AlbertClassification] = db
+  override def createOrUpdate(element: AlbertClassification): Future[AlbertClassification] = db
     .run(
       table.insertOrUpdate(element)
     )
     .map(_ => element)
+
+  override def removeByReportId(reportId: UUID): Future[Int] = db.run(
+    table.filter(_.reportId === reportId).delete
+  )
 }

app/services/AlbertService.scala

@@ -167,12 +167,25 @@ class AlbertService(albertConfiguration: AlbertConfiguration)(implicit ec: Execu
        |""".stripMargin
 
   def classifyReport(report: Report): Future[Option[String]] =
-    report.getDescription match {
-      case Some(description) =>
+    (report.getDescription, report.getReponseConsoDescription) match {
+      case (Some(description), Some(question)) =>
+        val text =
+          s"""$description
+             |
+             |Ma question :
+             |$question""".stripMargin
+        chatCompletion(chatPrompt(text))
+          .map(Some(_))
+          .recover(_ => None)
+      case (Some(description), None) =>
         chatCompletion(chatPrompt(description))
           .map(Some(_))
           .recover(_ => None)
-      case None => Future.successful(None)
+      case (None, Some(question)) =>
+        chatCompletion(chatPrompt(question))
+          .map(Some(_))
+          .recover(_ => None)
+      case (None, None) => Future.successful(None)
     }
 
   def qualifyReportBasedOnCodeConso(report: Report): Future[Option[String]] =

conf/db/migration/default/V47__albert_delete_cascade.sql

@@ -0,0 +1,5 @@
+ALTER TABLE albert_classification
+DROP CONSTRAINT fk_reportId;
+
+ALTER TABLE albert_classification
+ADD CONSTRAINT fk_reportId FOREIGN KEY (report_id) REFERENCES reports(id) ON DELETE CASCADE;