openproject/db/migrate/20131031170857_fix_watcher_work_package_associations.rb

class FixWatcherWorkPackageAssociations < ActiveRecord::Migration
  def up
    rename_watchable_type('Issue', 'WorkPackage')
    adapt_planning_element_ids
    rename_watchable_type('Timelines::PlanningElement', 'WorkPackage')
  end

  def down
    revert_planning_element_ids_and_types
    rename_watchable_type('WorkPackage', 'Issue')
  end

  private

  def rename_watchable_type(source_type, target_type)
    ActiveRecord::Base.connection.execute "UPDATE #{watchers_table}
                                           SET watchable_type=#{quote_value(target_type)}
                                           WHERE watchable_type=#{quote_value(source_type)}"
  end

  def adapt_planning_element_ids
    if postgres?
      ActiveRecord::Base.connection.execute <<-SQL
        UPDATE #{watchers_table}
         SET watchable_id = tmp.new_id
         FROM (
          SELECT first.watchable_id, second.new_id
          FROM #{watchers_table} AS first
          LEFT JOIN #{legacy_planning_elements_table} AS second
          ON first.watchable_id = second.id
          WHERE first.watchable_type = #{quote_value('Timelines::PlanningElement')}
          ) AS tmp
          WHERE watchers.watchable_id = tmp.watchable_id
          AND watchers.watchable_type = #{quote_value('Timelines::PlanningElement')};
      SQL
    elsif mysql?
      ActiveRecord::Base.connection.execute <<-SQL
        UPDATE #{watchers_table} AS first
        LEFT JOIN #{legacy_planning_elements_table} AS second ON first.watchable_id = second.id
        SET watchable_id = second.new_id
        WHERE watchable_type = #{quote_value('Timelines::PlanningElement')};
      SQL
    end
  end

  def revert_planning_element_ids_and_types
    if postgres?
      ActiveRecord::Base.connection.execute <<-SQL
        UPDATE #{watchers_table}
         SET watchable_id = tmp.id,
             watchable_type = #{quote_value('Timelines::PlanningElement')}
         FROM (
          SELECT first.watchable_id, second.id
          FROM #{watchers_table} AS first
          INNER JOIN #{legacy_planning_elements_table} AS second
          ON first.watchable_id = second.new_id
          WHERE first.watchable_type = #{quote_value('WorkPackage')}
          ) AS tmp
          WHERE watchers.watchable_id = tmp.watchable_id
          AND watchers.watchable_type = #{quote_value('WorkPackage')};
      SQL
    elsif mysql?
      ActiveRecord::Base.connection.execute <<-SQL
        UPDATE #{watchers_table} AS first
        INNER JOIN #{legacy_planning_elements_table} AS second ON first.watchable_id = second.new_id
        SET watchable_id = second.id,
            watchable_type = #{quote_value('Timelines::PlanningElement')}
        WHERE watchable_type = #{quote_value('WorkPackage')};
      SQL
    end
  end

  def watchers_table
    ActiveRecord::Base.connection.quote_table_name 'watchers'
  end

  def legacy_planning_elements_table
    ActiveRecord::Base.connection.quote_table_name 'legacy_planning_elements'
  end

  def quote_value(value)
    ActiveRecord::Base.connection.quote value
  end

  def postgres?
    ActiveRecord::Base.connection.instance_values["config"][:adapter] == "postgresql"
  end

  def mysql?
    ActiveRecord::Base.connection.instance_values["config"][:adapter] == "mysql2"
  end

end