summaryrefslogtreecommitdiff
path: root/db/migrate/20200728182311_add_o_auth_paths_to_protected_paths.rb
blob: 7a5af0135fae05123be3d445675d9ce036a3f5ac (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
55
56
57
58
59
60
61
62
# frozen_string_literal: true

class AddOAuthPathsToProtectedPaths < ActiveRecord::Migration[6.0]
  include Gitlab::Database::MigrationHelpers

  DOWNTIME = false

  ADD_PROTECTED_PATHS = [
    '/oauth/authorize',
    '/oauth/token'
  ].freeze

  EXISTING_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',
    '/admin/session'
  ].freeze

  NEW_DEFAULT_PROTECTED_PATHS = (EXISTING_DEFAULT_PROTECTED_PATHS + ADD_PROTECTED_PATHS).freeze

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

  def up
    change_column_default :application_settings, :protected_paths, NEW_DEFAULT_PROTECTED_PATHS

    ApplicationSetting.reset_column_information

    ApplicationSetting.where.not(protected_paths: nil).each do |application_setting|
      missing_paths = ADD_PROTECTED_PATHS - application_setting.protected_paths

      next if missing_paths.empty?

      updated_protected_paths = application_setting.protected_paths + missing_paths
      application_setting.update!(protected_paths: updated_protected_paths)
    end
  end

  def down
    change_column_default :application_settings, :protected_paths, EXISTING_DEFAULT_PROTECTED_PATHS

    ApplicationSetting.reset_column_information

    ApplicationSetting.where.not(protected_paths: nil).each do |application_setting|
      paths_to_remove = application_setting.protected_paths - EXISTING_DEFAULT_PROTECTED_PATHS

      next if paths_to_remove.empty?

      updated_protected_paths = application_setting.protected_paths - paths_to_remove
      application_setting.update!(protected_paths: updated_protected_paths)
    end
  end
end