1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2018 Make.org
4  *
5  * This program is free software: you can redistribute it and/or modify
6  *  it under the terms of the GNU Affero General Public License as
7  *  published by the Free Software Foundation, either version 3 of the
8  *  License, or (at your option) any later version.
9  *
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU Affero General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Affero General Public License
16  *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
17  *
18  */
19 
20 package org.make.api.proposal
21 
22 import cats.implicits._
23 import akka.actor.typed.{ActorSystem, Behavior}
24 import akka.util.Timeout
25 import grizzled.slf4j.Logging
26 import org.make.api.proposal.PublishedProposalEvent._
27 import org.make.api.technical.KafkaConsumerBehavior
28 import org.make.api.operation.OperationOfQuestionService
29 
30 import scala.concurrent.ExecutionContext.Implicits.global
31 import scala.concurrent.Future
32 import scala.concurrent.duration.DurationInt
33 
34 class ProposalConsumerBehavior(
35   proposalIndexerService: ProposalIndexerService,
36   operationOfQuestionService: OperationOfQuestionService
37 ) extends KafkaConsumerBehavior[ProposalEventWrapper]
38     with Logging {
39 
40   override protected val topicKey: String = ProposalKafkaProducerBehavior.topicKey
41   override val groupId = "proposal-consumer"
42 
43   implicit val timeout: Timeout = Timeout(5.seconds)
44 
45   override def handleMessage(message: ProposalEventWrapper)(system: ActorSystem[_]): Future[_] = {
46     message.event match {
47       case event: ProposalViewed               => doNothing(event)
48       case event: ReindexProposal              => onCreateOrUpdate(event)
49       case event: ProposalUpdated              => onCreateOrUpdate(event)
50       case event: ProposalReportedNotice       => doNothing(event)
51       case event: ProposalVotesVerifiedUpdated => onCreateOrUpdate(event)
52       case event: ProposalVotesUpdated         => onCreateOrUpdate(event)
53       case event: ProposalProposed             => onCreate(event)
54       case event: ProposalAccepted             => onCreateOrUpdate(event)
55       case event: ProposalRefused              => onCreateOrUpdate(event)
56       case event: ProposalPostponed            => onCreateOrUpdate(event)
57       case event: ProposalVoted                => onCreateOrUpdate(event)
58       case event: ProposalUnvoted              => onCreateOrUpdate(event)
59       case event: ProposalQualified            => onCreateOrUpdate(event)
60       case event: ProposalUnqualified          => onCreateOrUpdate(event)
61       case event: ProposalPatched              => onCreateOrUpdate(event)
62       case event: ProposalAddedToOperation     => onCreateOrUpdate(event)
63       case event: ProposalRemovedFromOperation => onCreateOrUpdate(event)
64       case event: ProposalLocked               => doNothing(event)
65       case event: ProposalAnonymized           => onCreateOrUpdate(event)
66       case event: SimilarProposalsAdded        => doNothing(event)
67       case event: ProposalKeywordsSet          => onCreateOrUpdate(event)
68     }
69 
70   }
71 
72   def onCreate(event: ProposalProposed): Future[Unit] =
73     event.question.fold(Future.unit)(operationOfQuestionService.incrementProposalsCount(_).void) >>
74       onCreateOrUpdate(event)
75 
76   def onCreateOrUpdate(event: ProposalEvent): Future[Unit] = {
77     proposalIndexerService.offer(event.id).recover {
78       case ex =>
79         error(s"Error presenting proposal to indexation queue: ${ex.getMessage}")
80     }
81   }
82 
83 }
84 
85 object ProposalConsumerBehavior {
86 
87   def apply(
88     proposalIndexerService: ProposalIndexerService,
89     operationOfQuestionService: OperationOfQuestionService
90   ): Behavior[KafkaConsumerBehavior.Protocol] =
91     new ProposalConsumerBehavior(proposalIndexerService, operationOfQuestionService).createBehavior(name)
92 
93   val name: String = "proposal-consumer"
94 }