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.core
21 
22 import java.time.{LocalDate, ZoneId, ZonedDateTime}
23 import akka.http.scaladsl.unmarshalling.Unmarshaller
24 import akka.stream.Materializer
25 import enumeratum.values.{StringEnum, StringEnumEntry}
26 import enumeratum.{Enum, EnumEntry}
27 import org.make.core.demographics.DemographicsCardId
28 import org.make.core.idea.IdeaId
29 import org.make.core.operation.OperationId
30 import org.make.core.personality.PersonalityRoleId
31 import org.make.core.proposal.{ProposalId, ProposalKeywordKey}
32 import org.make.core.question.QuestionId
33 import org.make.core.reference.{Country, LabelId, Language}
34 import org.make.core.tag.{TagId, TagTypeId}
35 import org.make.core.user.UserId
36 
37 import scala.concurrent.{ExecutionContext, Future}
38 import org.make.core.technical.Pagination._
39 import org.make.core.widget.SourceId
40 
41 @SuppressWarnings(Array("org.wartremover.warts.Throw"))
42 trait ParameterExtractors {
43 
44   implicit def eitherNoneOrT[T](
45     implicit unmarshaller: Unmarshaller[String, T]
46   ): Unmarshaller[String, Either[None.type, T]] = {
47 
48     Unmarshaller.identityUnmarshaller.transform {
49       implicit executionContext: ExecutionContext => implicit materializer: Materializer =>
50         _.flatMap {
51           case "None" => Future.successful(Left(None))
52           case other  => unmarshaller(other).map(Right(_))
53         }
54     }
55   }
56 
57   implicit val proposalKeywordKeyFromStringUnmarshaller: Unmarshaller[String, ProposalKeywordKey] =
58     Unmarshaller.strict[String, ProposalKeywordKey](ProposalKeywordKey.apply)
59 
60   implicit val localDateFromStringUnmarshaller: Unmarshaller[String, LocalDate] =
61     Unmarshaller.strict[String, LocalDate] { string =>
62       LocalDate.parse(string)
63     }
64 
65   implicit val zonedDateTimeFromStringUnmarshaller: Unmarshaller[String, ZonedDateTime] =
66     Unmarshaller.strict[String, ZonedDateTime] {
67       case value if value.contains('T') => ZonedDateTime.parse(value)
68       case value                        => LocalDate.parse(value).atStartOfDay(ZoneId.systemDefault())
69     }
70 
71   implicit val languageFromStringUnmarshaller: Unmarshaller[String, Language] =
72     Unmarshaller.strict[String, Language] { string =>
73       Language(string.toLowerCase())
74     }
75 
76   implicit val countryFromStringUnmarshaller: Unmarshaller[String, Country] =
77     Unmarshaller.strict[String, Country] { string =>
78       Country(string.toUpperCase())
79     }
80 
81   implicit val proposalIdFromStringUnmarshaller: Unmarshaller[String, ProposalId] =
82     Unmarshaller.strict[String, ProposalId] { string =>
83       ProposalId(string)
84     }
85 
86   implicit val userIdFromStringUnmarshaller: Unmarshaller[String, UserId] =
87     Unmarshaller.strict[String, UserId] { string =>
88       UserId(string)
89     }
90 
91   implicit val labelIdFromStringUnmarshaller: Unmarshaller[String, LabelId] =
92     Unmarshaller.strict[String, LabelId] { string =>
93       LabelId(string)
94     }
95 
96   implicit val tagIdFromStringUnmarshaller: Unmarshaller[String, TagId] =
97     Unmarshaller.strict[String, TagId] { string =>
98       TagId(string)
99     }
100 
101   implicit val tagTypeIdFromStringUnmarshaller: Unmarshaller[String, TagTypeId] =
102     Unmarshaller.strict[String, TagTypeId] { string =>
103       TagTypeId(string)
104     }
105 
106   implicit val operationIdFromStringUnmarshaller: Unmarshaller[String, OperationId] =
107     Unmarshaller.strict[String, OperationId] { string =>
108       OperationId(string)
109     }
110 
111   implicit val questionIdFromStringUnmarshaller: Unmarshaller[String, QuestionId] =
112     Unmarshaller.strict[String, QuestionId] { string =>
113       QuestionId(string)
114     }
115 
116   implicit val sourceIdFromStringUnmarshaller: Unmarshaller[String, SourceId] =
117     Unmarshaller.strict[String, SourceId] { string =>
118       SourceId(string)
119     }
120 
121   implicit val ideaIdFromStringUnmarshaller: Unmarshaller[String, IdeaId] =
122     Unmarshaller.strict[String, IdeaId] { string =>
123       IdeaId(string)
124     }
125 
126   implicit val personalityRoleIdFromStringUnmarshaller: Unmarshaller[String, PersonalityRoleId] =
127     Unmarshaller.strict[String, PersonalityRoleId] { role =>
128       PersonalityRoleId(role)
129     }
130 
131   implicit val demographicsCardIdFromStringUnmarshaller: Unmarshaller[String, DemographicsCardId] =
132     Unmarshaller.strict[String, DemographicsCardId] { role =>
133       DemographicsCardId(role)
134     }
135 
136   implicit def enumeratumEnumUnmarshaller[A <: EnumEntry](implicit basicEnum: Enum[A]): Unmarshaller[String, A] =
137     Unmarshaller.strict(basicEnum.withNameInsensitiveEither(_).fold(e => throw new Exception(e), identity))
138 
139   implicit def enumeratumStringEnumUnmarshaller[A <: StringEnumEntry](
140     implicit stringEnum: StringEnum[A]
141   ): Unmarshaller[String, A] =
142     Unmarshaller.strict(s => stringEnum.withValueEither(s).fold(e => throw new Exception(e), identity))
143 
144   implicit val startFromIntUnmarshaller: Unmarshaller[String, Offset] =
145     Unmarshaller.intFromStringUnmarshaller.map(Offset.apply)
146   implicit val endFromIntUnmarshaller: Unmarshaller[String, End] = Unmarshaller.intFromStringUnmarshaller.map(End.apply)
147   implicit val limitFromIntUnmarshaller: Unmarshaller[String, Limit] =
148     Unmarshaller.intFromStringUnmarshaller.map(Limit.apply)
149 
150 }