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 scala.collection.mutable.ArrayBuffer
24 import SqlTools.ImprovedResultSet
25 
26 @SuppressWarnings(Array("org.wartremover.warts.IterableOps"))
27 class V135_1__Add_language_to_crm_tpl extends Migration {
28 
29   private def getQuestionIds(implicit connection: Connection): ArrayBuffer[String] =
30     connection
31       .prepareStatement(s"SELECT question_id FROM crm_question_template")
32       .executeQuery
33       .toListOfRows(List("question_id"))
34       .map(_.head)
35 
36   private def getQuestionsLang(questionId: String)(implicit connection: Connection): String =
37     connection
38       .prepareStatement(s"SELECT default_language FROM question WHERE question_id = '$questionId'")
39       .executeQuery
40       .toListOfRows(List("default_language"))
41       .head
42       .head
43 
44   private def genUpdateLangQuery(questionId: String, lang: String)(implicit connection: Connection): Unit =
45     connection
46       .prepareStatement(s"UPDATE crm_question_template SET language = '$lang' WHERE question_id = '$questionId'")
47       .execute
48 
49   override def migrate(connection: Connection): Unit = {
50     implicit val C: Connection = connection
51     val questionIds = getQuestionIds
52     val addLanguageToTemplate: (String, String) => Unit = genUpdateLangQuery
53     questionIds
54       .zip(questionIds.map(getQuestionsLang))
55       .foreach(addLanguageToTemplate.tupled)
56   }
57 }