Migration to add no_map_logo to users and organizations

db/migrate/20170228132222_add_no_map_logo_to_users.rb

@@ -0,0 +1,22 @@
+require 'carto/db/migration_helper'
+
+include Carto::Db::MigrationHelper
+
+NO_REMOVE_LOGO_PLANS_REGEXP = '^(FREE|MAGELLAN|JOHN SNOW|ACADEMY|ACADEMIC|ON HOLD|SITE LICENSE|CARTO FOR)'.freeze
+
+migration(
+  Proc.new do
+    add_column :users, :no_map_logo, :boolean
+    run "UPDATE users SET no_map_logo = (account_type !~* '#{NO_REMOVE_LOGO_PLANS_REGEXP}');"
+    run 'ALTER TABLE users ALTER no_map_logo SET DEFAULT false, ALTER no_map_logo SET NOT NULL'
+
+    add_column :organizations, :no_map_logo, :boolean
+    run 'UPDATE organizations SET no_map_logo =
+           COALESCE((SELECT no_map_logo FROM users WHERE id = organizations.owner_id), false)'
+    run 'ALTER TABLE organizations ALTER no_map_logo SET DEFAULT false, ALTER no_map_logo SET NOT NULL'
+  end,
+  Proc.new do
+    drop_column :users, :no_map_logo
+    drop_column :organizations, :no_map_logo
+  end
+)