1 package org.make.api.migrations.db
2 
3 import org.make.api.migrations.db.SqlTools.ImprovedResultSet
4 
5 import java.sql.Connection
6 
7 @SuppressWarnings(Array("org.wartremover.warts.TraversableOps"))
8 class V147__Retro_compatibility_old_question_prefix extends Migration {
9 
10   val prefixes: Map[String, String] = Map(
11     "fr" -> "Il faut",
12     "de" -> "Man sollte",
13     "en" -> "We should",
14     "bg" -> "Трябва",
15     "hr" -> "Potrebno je",
16     "da" -> "Vi skal",
17     "es" -> "Es necesario",
18     "et" -> "On vaja",
19     "fi" -> "On toimittava näin:",
20     "el" -> "Πρέπει",
21     "hu" -> "Muszáj",
22     "it" -> "È necessario",
23     "lv" -> "Nepieciešams",
24     "lt" -> "Būtina",
25     "mt" -> "Il faut",
26     "nl" -> "Er moet",
27     "pl" -> "Należy",
28     "pt" -> "É necessário",
29     "ro" -> "Trebuie",
30     "sk" -> "Je potrebné",
31     "sl" -> "Treba je",
32     "sv" -> "Vi måste",
33     "cs" -> "Je třeba",
34     "uk" -> "Необхідно"
35   )
36 
37   private def getQuestionLanguages(implicit connection: Connection) =
38     connection
39       .prepareStatement(s"SELECT question_id, languages FROM question")
40       .executeQuery
41       .toMaps("question_id", "languages")
42 
43   private def updateConsultation(proposalPrefixes: String, questionId: String)(implicit connection: Connection) =
44     connection
45       .prepareStatement(s"""
46            |UPDATE operation_of_question
47            |SET proposal_prefixes = $$$${$proposalPrefixes}$$$$
48            |WHERE question_id = '$questionId'
49           """.stripMargin)
50       .execute()
51 
52   override def migrate(connection: Connection): Unit = {
53     implicit val C: Connection = connection
54     val questionLanguages = getQuestionLanguages
55     questionLanguages.foreach(questionLanguage => {
56       val questionId = questionLanguage("question_id")
57       val languagesString = questionLanguage("languages")
58       val proposalPrefixes =
59         languagesString
60           .slice(1, languagesString.length - 1)
61           .split(',')
62           .map(l => l -> prefixes(l))
63           .toMap
64           .toSeq
65           .map {
66             case (k, v) => s"\"$k\": \"$v\""
67           }
68           .mkString(", ")
69       updateConsultation(proposalPrefixes, questionId)
70     })
71   }
72 }