openproject/db/migrate/20211209092519_query_views.rb

class QueryViews < ActiveRecord::Migration[6.1]
  def up
    add_column :queries,
               :starred,
               :boolean,
               default: false

    execute <<~SQL.squish
      INSERT INTO
        views (
          type,
          query_id,
          created_at,
          updated_at
        )
      SELECT
        'work_packages_table',
        id,
        created_at,
        updated_at
      FROM queries
      WHERE
        hidden = false
    SQL

    execute <<~SQL.squish
      UPDATE
        queries
      SET
        starred = true
      WHERE
        id IN (SELECT navigatable_id FROM menu_items WHERE type = 'MenuItems::QueryMenuItem')
    SQL

    execute <<~SQL.squish
      DELETE FROM
        menu_items
      WHERE
        type = 'MenuItems::QueryMenuItem'
    SQL

    remove_column :queries,
                  :hidden

    rename_column :queries,
                  :is_public,
                  :public
  end

  def down
    rename_column :queries,
                  :public,
                  :is_public

    add_column :queries,
               :hidden,
               :boolean,
               default: false

    # Consciously avoiding the use of a PostgreSQL 13.0 feature (gen_random_uuid())
    Query.where(starred: true).find_each do |query|
      ::MenuItem.create(
        type: 'MenuItems::QueryMenuItem',
        navigatable_id: query.id,
        name: SecureRandom.uuid,
        title: query.name
      )
    end

    execute <<~SQL.squish
      UPDATE
        queries
      SET
        hidden = true
      WHERE
        id NOT IN (SELECT query_id FROM views WHERE type = 'work_packages_table')
    SQL

    execute <<~SQL.squish
      DELETE FROM
        views
      WHERE type = 'work_packages_table'
    SQL

    remove_column :queries,
                  :starred
  end
end