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.technical
21 
22 import org.make.core.Order
23 import scalikejdbc.PagingSQLBuilder
24 import org.make.core.technical.Pagination
25 
26 object PersistentServiceUtils {
27 
28   def sortOrderQuery[Persistent, Model](
29     offset: Pagination.Offset,
30     end: Option[Pagination.End],
31     sort: Option[String],
32     order: Option[Order],
33     query: PagingSQLBuilder[Persistent]
34   )(implicit companion: PersistentCompanion[Persistent, Model]): PagingSQLBuilder[Persistent] = {
35     val queryOrdered = (sort, order) match {
36       case (Some(field), Some(Order.desc)) if companion.columnNames.contains(field) =>
37         query
38           .orderBy(companion.alias.field(field))
39           .desc
40           .offset(offset.extractInt)
41       case (Some(field), _) if companion.columnNames.contains(field) =>
42         query
43           .orderBy(companion.alias.field(field))
44           .asc
45           .offset(offset.extractInt)
46       case (_, _) =>
47         query
48           .orderBy(companion.defaultSortColumns.toList: _*)
49           .asc
50           .offset(offset.extractInt)
51     }
52 
53     end.fold(queryOrdered)(end => queryOrdered.limit(end.extractInt - offset.extractInt))
54   }
55 }