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 package post
22 package indexed
23 
24 import com.sksamuel.elastic4s.{ElasticApi, ElasticDsl}
25 import com.sksamuel.elastic4s.requests.searches.queries.Query
26 import com.sksamuel.elastic4s.requests.searches.sort.{FieldSort, SortOrder}
27 import org.make.core.technical.Pagination
28 import org.make.core.reference.Country
29 
30 final case class PostSearchQuery(
31   filters: Option[PostSearchFilters] = None,
32   limit: Option[Pagination.Limit] = None,
33   offset: Option[Pagination.Offset] = None,
34   sort: Option[String] = None,
35   order: Option[Order] = None
36 )
37 
38 final case class PostSearchFilters(
39   postIds: Option[PostIdsSearchFilter] = None,
40   displayHome: Option[DisplayHomeSearchFilter] = None,
41   countries: Option[PostCountriesFilter] = None
42 )
43 
44 object PostSearchFilters extends ElasticDsl {
45 
46   def parse(
47     postIds: Option[PostIdsSearchFilter] = None,
48     displayHome: Option[DisplayHomeSearchFilter] = None,
49     postCountries: Option[PostCountriesFilter] = None
50   ): Option[PostSearchFilters] = {
51     (postIds, displayHome, postCountries) match {
52       case (None, None, None) => None
53       case _                  => Some(PostSearchFilters(postIds, displayHome, postCountries))
54     }
55   }
56 
57   def getPostSearchFilters(postSearchQuery: PostSearchQuery): Seq[Query] = {
58     Seq(
59       buildPostIdsSearchFilter(postSearchQuery.filters),
60       buildDisplayHomeSearchFilter(postSearchQuery.filters),
61       buildPostCountriesSearchFilter(postSearchQuery.filters)
62     ).flatten
63   }
64 
65   def getOffsetSearch(postSearchQuery: PostSearchQuery): Int =
66     postSearchQuery.offset.fold(0)(_.extractInt)
67 
68   def getLimitSearch(postSearchQuery: PostSearchQuery): Int =
69     postSearchQuery.limit.fold(10)(_.extractInt)
70 
71   def getSort(postSearchQuery: PostSearchQuery): Option[FieldSort] = {
72     val order: SortOrder = postSearchQuery.order.map(_.sortOrder).getOrElse(SortOrder.Asc)
73 
74     postSearchQuery.sort.map { sort =>
75       FieldSort(field = sort, order = order)
76     }
77   }
78 
79   def buildPostIdsSearchFilter(maybeFilters: Option[PostSearchFilters]): Option[Query] = {
80     for {
81       filters                      <- maybeFilters
82       PostIdsSearchFilter(postIds) <- filters.postIds
83     } yield ElasticApi.termsQuery(PostElasticsearchFieldNames.postId, postIds.map(_.value))
84   }
85 
86   def buildDisplayHomeSearchFilter(maybeFilters: Option[PostSearchFilters]): Option[Query] = {
87     for {
88       filters                              <- maybeFilters
89       DisplayHomeSearchFilter(displayHome) <- filters.displayHome
90     } yield ElasticApi.termQuery(PostElasticsearchFieldNames.displayHome, displayHome)
91   }
92 
93   def buildPostCountriesSearchFilter(maybeFilters: Option[PostSearchFilters]): Option[Query] = {
94     for {
95       filters                        <- maybeFilters
96       PostCountriesFilter(countries) <- filters.countries
97     } yield ElasticApi.termsQuery(PostElasticsearchFieldNames.country, countries.map(_.value))
98   }
99 
100 }
101 
102 final case class PostIdsSearchFilter(postIds: Seq[PostId])
103 final case class DisplayHomeSearchFilter(displayHome: Boolean)
104 final case class PostCountriesFilter(countries: Seq[Country])