BigW Consortium Gitlab

20160301124843_add_visibility_level_to_groups.rb 1.1 KB
Newer Older
1
# rubocop:disable all
2
class AddVisibilityLevelToGroups < ActiveRecord::Migration
Douwe Maan committed
3 4 5 6 7 8 9 10 11 12 13 14 15
  def up
    add_column :namespaces, :visibility_level, :integer, null: false, default: Gitlab::VisibilityLevel::PUBLIC
    add_index :namespaces, :visibility_level

    # Unfortunately, this is needed on top of the `default`, since we don't want the configuration specific
    # `allowed_visibility_level` to end up in schema.rb
    if allowed_visibility_level < Gitlab::VisibilityLevel::PUBLIC
      execute("UPDATE namespaces SET visibility_level = #{allowed_visibility_level}")
    end
  end

  def down
    remove_column :namespaces, :visibility_level
16 17
  end

Douwe Maan committed
18 19
  private

20
  def allowed_visibility_level
Douwe Maan committed
21 22 23 24 25 26 27
    application_settings = select_one("SELECT restricted_visibility_levels FROM application_settings ORDER BY id DESC LIMIT 1")
    if application_settings
      restricted_visibility_levels = YAML.safe_load(application_settings["restricted_visibility_levels"]) rescue nil
    end
    restricted_visibility_levels ||= []

    allowed_levels = Gitlab::VisibilityLevel.values - restricted_visibility_levels
28
    allowed_levels.max
29 30
  end
end