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.partner
21 
22 import org.make.api.technical.IdGeneratorComponent
23 import org.make.core.partner.{Partner, PartnerId, PartnerKind}
24 import org.make.core.question.QuestionId
25 import org.make.core.user.UserId
26 import org.make.core.Order
27 
28 import scala.concurrent.ExecutionContext.Implicits.global
29 import scala.concurrent.Future
30 import org.make.core.technical.Pagination
31 
32 trait DefaultPartnerServiceComponent extends PartnerServiceComponent {
33   this: PersistentPartnerServiceComponent with IdGeneratorComponent =>
34 
35   override lazy val partnerService: DefaultPartnerService = new DefaultPartnerService
36 
37   class DefaultPartnerService extends PartnerService {
38 
39     override def getPartner(partnerId: PartnerId): Future[Option[Partner]] = {
40       persistentPartnerService.getById(partnerId)
41     }
42 
43     override def createPartner(request: CreatePartnerRequest): Future[Partner] = {
44       val partner: Partner = Partner(
45         partnerId = idGenerator.nextPartnerId(),
46         name = request.name,
47         logo = request.logo,
48         link = request.link,
49         organisationId = request.organisationId,
50         partnerKind = request.partnerKind,
51         questionId = request.questionId,
52         weight = request.weight
53       )
54       persistentPartnerService.persist(partner)
55     }
56 
57     override def updatePartner(partnerId: PartnerId, request: UpdatePartnerRequest): Future[Option[Partner]] = {
58       persistentPartnerService.getById(partnerId).flatMap {
59         case Some(partner) =>
60           persistentPartnerService
61             .modify(
62               partner.copy(
63                 name = request.name,
64                 logo = request.logo,
65                 link = request.link,
66                 organisationId = request.organisationId,
67                 partnerKind = request.partnerKind,
68                 weight = request.weight
69               )
70             )
71             .map(Some.apply)
72         case None => Future.successful(None)
73       }
74     }
75 
76     override def find(
77       offset: Pagination.Offset,
78       end: Option[Pagination.End],
79       sort: Option[String],
80       order: Option[Order],
81       questionId: Option[QuestionId],
82       organisationId: Option[UserId],
83       partnerKind: Option[PartnerKind]
84     ): Future[Seq[Partner]] = {
85       persistentPartnerService.find(offset, end, sort, order, questionId, organisationId, partnerKind)
86     }
87 
88     override def count(
89       questionId: Option[QuestionId],
90       organisationId: Option[UserId],
91       partnerKind: Option[PartnerKind]
92     ): Future[Int] = {
93       persistentPartnerService.count(questionId, organisationId, partnerKind)
94     }
95 
96     override def deletePartner(partnerId: PartnerId): Future[Unit] = {
97       persistentPartnerService.delete(partnerId)
98     }
99 
100   }
101 }