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.personality
21 
22 import org.make.api.operation.OperationOfQuestionServiceComponent
23 import org.make.api.proposal.ProposalSearchEngineComponent
24 import org.make.api.question._
25 import org.make.api.technical.IdGeneratorComponent
26 import org.make.core.technical.Pagination
27 import org.make.core.personality.{Personality, PersonalityId, PersonalityRoleId}
28 import org.make.core.question.QuestionId
29 import org.make.core.user.UserId
30 import org.make.core.Order
31 
32 import scala.concurrent.ExecutionContext.Implicits.global
33 import scala.concurrent.Future
34 
35 trait DefaultQuestionPersonalityServiceComponent extends QuestionPersonalityServiceComponent {
36   this: PersistentQuestionPersonalityServiceComponent
37     with IdGeneratorComponent
38     with QuestionServiceComponent
39     with OperationOfQuestionServiceComponent
40     with ProposalSearchEngineComponent =>
41 
42   override lazy val questionPersonalityService: DefaultQuestionPersonalityService =
43     new DefaultQuestionPersonalityService
44 
45   class DefaultQuestionPersonalityService extends QuestionPersonalityService {
46 
47     override def getPersonality(personalityId: PersonalityId): Future[Option[Personality]] = {
48       persistentQuestionPersonalityService.getById(personalityId)
49     }
50 
51     override def createPersonality(request: CreateQuestionPersonalityRequest): Future[Personality] = {
52       val personality: Personality = Personality(
53         personalityId = idGenerator.nextPersonalityId(),
54         userId = request.userId,
55         questionId = request.questionId,
56         personalityRoleId = request.personalityRoleId
57       )
58       persistentQuestionPersonalityService.persist(personality)
59     }
60 
61     override def updatePersonality(
62       personalityId: PersonalityId,
63       request: UpdateQuestionPersonalityRequest
64     ): Future[Option[Personality]] = {
65       persistentQuestionPersonalityService.getById(personalityId).flatMap {
66         case Some(personality) =>
67           persistentQuestionPersonalityService
68             .modify(personality.copy(userId = request.userId, personalityRoleId = request.personalityRoleId))
69             .map(Some.apply)
70         case None => Future.successful(None)
71       }
72     }
73 
74     override def find(
75       offset: Pagination.Offset,
76       end: Option[Pagination.End],
77       sort: Option[String],
78       order: Option[Order],
79       userId: Option[UserId],
80       questionId: Option[QuestionId],
81       personalityRoleId: Option[PersonalityRoleId]
82     ): Future[Seq[Personality]] = {
83       persistentQuestionPersonalityService.find(offset, end, sort, order, userId, questionId, personalityRoleId)
84     }
85 
86     override def count(
87       userId: Option[UserId],
88       questionId: Option[QuestionId],
89       personalityRoleId: Option[PersonalityRoleId]
90     ): Future[Int] = {
91       persistentQuestionPersonalityService.count(userId, questionId, personalityRoleId)
92     }
93 
94     override def deletePersonality(personalityId: PersonalityId): Future[Unit] = {
95       persistentQuestionPersonalityService.delete(personalityId)
96     }
97 
98   }
99 }