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.migrations.db
21 
22 import java.sql.Connection
23 import java.util.UUID
24 import scala.collection.mutable.HashSet
25 
26 class V126_1__Add_abusive_proposers_tpl_kinds extends Migration {
27 
28   private def nextId(): String = UUID.randomUUID().toString
29 
30   private val isProd = System.getenv("ENV_NAME") == "prod"
31 
32   private val templateAbusiveWarnIds: Map[String, (String, String)] =
33     Map("fr" -> ("3520961", "3520884"), "en" -> ("3536356", "3536356"), "de" -> ("3536371", "3536371"))
34 
35   private val templateAbusiveBlockIds: Map[String, (String, String)] =
36     Map("fr" -> ("3520981", "3520933"), "en" -> ("3536302", "3536302"), "de" -> ("3536641", "3536641"))
37 
38   private def makeAbusiveWarnTemplateRow(language: String): String = {
39     val (preprod, prod) = templateAbusiveWarnIds.getOrElse(language, templateAbusiveWarnIds("fr"))
40     s"('${nextId()}', 'MessageToAbusiveWarn', '$language', '${if (isProd) prod else preprod}')"
41   }
42 
43   private def makeAbusiveBlockTemplateRow(language: String): String = {
44     val (preprod, prod) = templateAbusiveBlockIds.getOrElse(language, templateAbusiveBlockIds("fr"))
45     s"('${nextId()}', 'MessageToAbusiveBlock', '$language', '${if (isProd) prod else preprod}')"
46   }
47 
48   @SuppressWarnings(Array("org.wartremover.warts.While", "org.wartremover.warts.MutableDataStructures"))
49   override def migrate(connection: Connection): Unit = {
50 
51     def insertTplRow(row: String): Unit =
52       connection
53         .prepareStatement(s"INSERT INTO crm_language_template (id, kind, language, template) VALUES $row")
54         .execute()
55 
56     val resultSet =
57       connection
58         .prepareStatement("SELECT language FROM crm_language_template")
59         .executeQuery()
60 
61     val languages = HashSet.empty[String]
62 
63     while (resultSet.next()) {
64       languages += resultSet.getString("language")
65     }
66 
67     languages.foreach(lang => {
68       insertTplRow(makeAbusiveWarnTemplateRow(lang))
69       insertTplRow(makeAbusiveBlockTemplateRow(lang))
70     })
71   }
72 }