From 78bd4508d13c3e3f2ae963dee83bdf68e017dc69 Mon Sep 17 00:00:00 2001
From: Sam <sam.saffron@gmail.com>
Date: Thu, 23 Jul 2015 14:39:38 +1000
Subject: [PATCH] Add multisite support to remap task

---
 script/discourse | 58 ++++++++++++++++++++++++++++++------------------
 1 file changed, 36 insertions(+), 22 deletions(-)

diff --git a/script/discourse b/script/discourse
index 8b7f979720b..dda1d493b59 100755
--- a/script/discourse
+++ b/script/discourse
@@ -6,40 +6,28 @@ class DiscourseCLI < Thor
   class_option :verbose, default: false, aliases: :v
 
   desc "remap", "Remap a string sequence accross all tables"
-  def remap(from, to)
+  def remap(from, to, global=nil)
     load_rails
 
+    global = global == "--global"
+
     puts "Rewriting all occurences of #{from} to #{to}"
     puts "THIS TASK WILL REWRITE DATA, ARE YOU SURE (type YES)"
+    puts "WILL RUN ON ALL #{RailsMultisite::ConnectionManagement.all_dbs.length} DBS" if global
     text = STDIN.gets
     if text.strip != "YES"
       puts "aborting."
       exit
     end
 
-    sql = "SELECT table_name, column_name
-FROM information_schema.columns
-WHERE table_schema='public' and (data_type like 'char%' or data_type like 'text%') and is_updatable = 'YES'"
-
-    cnn = ActiveRecord::Base.connection.raw_connection
-
-    results = cnn.async_exec(sql).to_a
-
-    results.each do |result|
-      table_name = result["table_name"]
-      column_name = result["column_name"]
-      puts "Remapping #{table_name} #{column_name}"
-      begin
-        result = cnn.async_exec("UPDATE #{table_name}
-                        SET #{column_name} = replace(#{column_name}, $1, $2)
-                        WHERE NOT #{column_name} IS NULL
-                          AND #{column_name} <> replace(#{column_name}, $1, $2)", [from, to])
-        puts "#{result.cmd_tuples} rows affected!"
-      rescue => ex
-        puts "Error: #{ex}"
+    if global
+      RailsMultisite::ConnectionManagement.each_connection do |db|
+        puts "","Remapping tables on #{db}...",""
+        do_remap(from, to)
       end
+    else
+      do_remap(from, to)
     end
-
   end
 
   desc "backup", "Backup a discourse forum"
@@ -152,6 +140,32 @@ WHERE table_schema='public' and (data_type like 'char%' or data_type like 'text%
     require File.expand_path(File.dirname(__FILE__) + "/../config/environment")
   end
 
+  def do_remap(from, to)
+    sql = "SELECT table_name, column_name
+FROM information_schema.columns
+WHERE table_schema='public' and (data_type like 'char%' or data_type like 'text%') and is_updatable = 'YES'"
+
+    cnn = ActiveRecord::Base.connection.raw_connection
+
+    results = cnn.async_exec(sql).to_a
+
+    results.each do |result|
+      table_name = result["table_name"]
+      column_name = result["column_name"]
+      puts "Remapping #{table_name} #{column_name}"
+      begin
+        result = cnn.async_exec("UPDATE #{table_name}
+                        SET #{column_name} = replace(#{column_name}, $1, $2)
+                        WHERE NOT #{column_name} IS NULL
+                          AND #{column_name} <> replace(#{column_name}, $1, $2)", [from, to])
+        puts "#{result.cmd_tuples} rows affected!"
+      rescue => ex
+        puts "Error: #{ex}"
+      end
+    end
+  end
+
+
 end
 
 DiscourseCLI.start(ARGV)