1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2020 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.widget
21 
22 import org.apache.commons.text.StringEscapeUtils
23 import org.make.api.ConfigComponent
24 import org.make.api.technical.IdGeneratorComponent
25 import org.make.api.technical.security.{SecurityConfigurationComponent, SecurityHelper}
26 import org.make.core.question.Question
27 import org.make.core.reference.{Country, Language}
28 import org.make.core.sequence.SequenceKind
29 import org.make.core.technical.Pagination
30 import org.make.core.user.UserId
31 import org.make.core.widget.{Source, SourceId, Widget, WidgetId}
32 import org.make.core.question.QuestionId
33 import org.make.core.{DateHelperComponent, Order}
34 
35 import scala.concurrent.Future
36 
37 trait DefaultWidgetServiceComponent extends WidgetServiceComponent {
38   self: ConfigComponent
39     with DateHelperComponent
40     with IdGeneratorComponent
41     with PersistentWidgetServiceComponent
42     with SecurityConfigurationComponent =>
43 
44   override val widgetService: WidgetService = new WidgetService {
45 
46     override def get(id: WidgetId): Future[Option[Widget]] = persistentWidgetService.get(id)
47 
48     override def count(sourceId: SourceId): Future[Int] = persistentWidgetService.count(sourceId)
49 
50     override def count(questionId: QuestionId): Future[Int] = persistentWidgetService.count(questionId)
51 
52     override def list(
53       sourceId: SourceId,
54       offset: Option[Pagination.Offset],
55       end: Option[Pagination.End],
56       sort: Option[String],
57       order: Option[Order]
58     ): Future[Seq[Widget]] = persistentWidgetService.list(sourceId, offset, end, sort, order)
59 
60     override def create(
61       source: Source,
62       question: Question,
63       country: Country,
64       language: Language,
65       sequenceKind: SequenceKind,
66       author: UserId
67     ): Future[Widget] = {
68       val id = idGenerator.nextWidgetId()
69       val sequenceKindUrl =
70         if (sequenceKind == SequenceKind.Standard) "" else s"&sequenceKind=${sequenceKind.entryName}"
71       val url =
72         s"?questionSlug=${question.slug}&source=${source.source}&country=${country.value}&language=${language.value}&widgetId=${id.value}$sequenceKindUrl"
73       val widgetHeight: Int = sequenceKind match {
74         case SequenceKind.Standard => 550
75         case _                     => 572
76       }
77       val hash = SecurityHelper.createSecureHash(url, securityConfiguration.secureHashSalt)
78       val style = s"display: block; max-width: 635px; margin: 0 auto; min-height: ${widgetHeight}px"
79       val script =
80         s"""<iframe frameborder="0" scrolling="no" referrerpolicy="no-referrer-when-downgrade" width="100%" height="$widgetHeight" style="$style" src="${config
81           .getString("make-api.urls.widget")}/${StringEscapeUtils.escapeHtml4(s"$url&hash=")}$hash"></iframe>"""
82       persistentWidgetService.persist(
83         Widget(
84           id = id,
85           sourceId = source.id,
86           questionId = question.questionId,
87           country = country,
88           language = language,
89           author = author,
90           version = Widget.Version.V6,
91           script = script,
92           createdAt = dateHelper.now(),
93           sequenceKind = sequenceKind
94         )
95       )
96     }
97 
98   }
99 }