summaryrefslogtreecommitdiff
path: root/db/migrate/20191125114345_add_admin_mode_protected_path.rb
blob: 7e9b0d5a2852c63b65ccd5a19192088bc9a0761d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
# frozen_string_literal: true

class AddAdminModeProtectedPath < ActiveRecord::Migration[5.2]
  include Gitlab::Database::MigrationHelpers

  DOWNTIME = false

  ADMIN_MODE_ENDPOINT = '/admin/session'

  OLD_DEFAULT_PROTECTED_PATHS = [
      '/users/password',
      '/users/sign_in',
      '/api/v3/session.json',
      '/api/v3/session',
      '/api/v4/session.json',
      '/api/v4/session',
      '/users',
      '/users/confirmation',
      '/unsubscribes/',
      '/import/github/personal_access_token'
  ]

  NEW_DEFAULT_PROTECTED_PATHS = OLD_DEFAULT_PROTECTED_PATHS.dup << ADMIN_MODE_ENDPOINT

  class ApplicationSetting < ActiveRecord::Base
    self.table_name = 'application_settings'
  end

  def up
    change_column_default :application_settings, :protected_paths, NEW_DEFAULT_PROTECTED_PATHS

    # schema allows nulls for protected_paths
    ApplicationSetting.where.not(protected_paths: nil).each do |application_setting|
      unless application_setting.protected_paths.include?(ADMIN_MODE_ENDPOINT)
        updated_protected_paths = application_setting.protected_paths << ADMIN_MODE_ENDPOINT

        application_setting.update(protected_paths: updated_protected_paths)
      end
    end
  end

  def down
    change_column_default :application_settings, :protected_paths, OLD_DEFAULT_PROTECTED_PATHS

    # schema allows nulls for protected_paths
    ApplicationSetting.where.not(protected_paths: nil).each do |application_setting|
      if application_setting.protected_paths.include?(ADMIN_MODE_ENDPOINT)
        updated_protected_paths = application_setting.protected_paths - [ADMIN_MODE_ENDPOINT]

        application_setting.update(protected_paths: updated_protected_paths)
      end
    end
  end
end