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.segment
21 
22 import org.make.api.feature.{ActiveFeatureServiceComponent, FeatureServiceComponent}
23 import org.make.core.RequestContext
24 import org.make.core.feature.FeatureSlug
25 import org.make.core.feature.FeatureSlug.{ConsultationDepartmentCompulsory, SequenceCustomDataSegment}
26 
27 import scala.concurrent.ExecutionContext.Implicits.global
28 import scala.concurrent.Future
29 
30 trait DefaultSegmentServiceComponent extends SegmentServiceComponent {
31   self: ActiveFeatureServiceComponent with FeatureServiceComponent =>
32 
33   override lazy val segmentService: DefaultSegmentService = new DefaultSegmentService
34 
35   class DefaultSegmentService extends SegmentService {
36     type SegmentResolver = RequestContext => Future[Option[String]]
37 
38     val defaultSegmentResolver: SegmentResolver = { _ =>
39       Future.successful(None)
40     }
41 
42     val segmentResolverFromDepartment: SegmentResolver = { requestContext =>
43       val customData = requestContext.customData
44       Future.successful(
45         customData
46           .get("declared_department")
47           .orElse(customData.get("detected_department"))
48       )
49     }
50 
51     val segmentResolverFromCustomData: SegmentResolver = { requestContext =>
52       Future.successful(requestContext.customData.get("segment").map(_.trim).filter(_.nonEmpty))
53     }
54 
55     val segmentResolvers: Map[FeatureSlug, SegmentResolver] = Map(
56       ConsultationDepartmentCompulsory -> segmentResolverFromDepartment,
57       SequenceCustomDataSegment -> segmentResolverFromCustomData
58     )
59 
60     override def resolveSegment(requestContext: RequestContext): Future[Option[String]] = {
61       requestContext.questionContext.questionId.map { _ =>
62         activeFeatureService
63           .find(maybeQuestionId = requestContext.questionContext.questionId.map(Seq(_)))
64           .flatMap { activeFeatures =>
65             featureService.findByFeatureIds(activeFeatures.map(_.featureId))
66           }
67           .flatMap { features =>
68             features
69               .flatMap(feature => segmentResolvers.get(feature.slug))
70               .headOption
71               .getOrElse(defaultSegmentResolver)(requestContext)
72           }
73       }.getOrElse(Future.successful(None))
74     }
75   }
76 
77 }