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.util.chaining.scalaUtilChainingOps
24 import io.circe.syntax._
25 import io.circe.parser.decode
26 import io.circe.generic.auto._
27 import SqlTools.ImprovedResultSet
28 import org.make.core.technical.Multilingual
29 import org.make.core.reference.Language
30 import org.make.core.demographics.LabelsValue
31 
32 @SuppressWarnings(Array("org.wartremover.warts.TraversableOps"))
33 class V137_0__Multilingual_demographics extends Migration {
34 
35   private case class LegacyLabelValue(label: String, value: String)
36 
37   @SuppressWarnings(Array("org.wartremover.warts.Throw"))
38   private def migrateParameters(params: String, lang: String): String =
39     decode[Seq[LegacyLabelValue]](params) match {
40       case Right(list) =>
41         list
42           .map({
43             case LegacyLabelValue(label, value) =>
44               LabelsValue(Multilingual(Language(lang) -> label), value)
45           })
46           .asJson
47           .noSpaces
48       case Left(err) => throw err
49     }
50 
51   private val addNewColumns: Connection => Unit =
52     _.prepareStatement("""
53         |ALTER TABLE demographics_card
54         |ADD COLUMN languages STRING[], ADD COLUMN titles JSONB
55       """.stripMargin)
56       .execute()
57 
58   private val destroyOldCumns: Connection => Unit =
59     _.prepareStatement("ALTER TABLE demographics_card DROP COLUMN language, DROP COLUMN title")
60       .execute()
61 
62   private def genUpdateLangQuery(cardId: String, lang: String, title: String, params: String)(
63     connection: Connection
64   ): Unit = {
65     val parameters = migrateParameters(params, lang)
66     connection
67       .prepareStatement(s"""
68         |UPDATE demographics_card
69         |SET (languages, titles, parameters) = (ARRAY['$lang'], $$$${"$lang": "$title"}$$$$, $$$$${parameters}$$$$)
70         |WHERE id = '$cardId'
71       """.stripMargin)
72       .execute()
73   }
74 
75   override def migrate(connection: Connection): Unit =
76     connection
77       .prepareStatement(s"SELECT id, language, title, parameters FROM demographics_card")
78       .executeQuery
79       .toMaps("id", "language", "title", "parameters")
80       .tap(_ => connection.commit())
81       .tap(_ => addNewColumns(connection))
82       .tap(_ => connection.commit())
83       .foreach(
84         dataMap =>
85           genUpdateLangQuery(dataMap("id"), dataMap("language"), dataMap("title"), dataMap("parameters"))(connection)
86       )
87       .tap(_ => connection.commit())
88       .tap(_ => destroyOldCumns(connection))
89 }