summaryrefslogtreecommitdiff
path: root/qa/qa/tools/delete_subgroups.rb
blob: 11b45365d4c2b6aebf722183908fbe614999a754 (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
63
64
65
66
67
# frozen_string_literal: true

# This script deletes all subgroups of a group specified by ENV['TOP_LEVEL_GROUP_NAME']
# Required environment variables: GITLAB_QA_ACCESS_TOKEN and GITLAB_ADDRESS
# Optional environment variable: TOP_LEVEL_GROUP_NAME (defaults to 'gitlab-qa-sandbox-group')
# Run `rake delete_subgroups`

module QA
  module Tools
    class DeleteSubgroups
      include Support::API

      def initialize
        raise ArgumentError, "Please provide GITLAB_ADDRESS" unless ENV['GITLAB_ADDRESS']
        raise ArgumentError, "Please provide GITLAB_QA_ACCESS_TOKEN" unless ENV['GITLAB_QA_ACCESS_TOKEN']

        @api_client = Runtime::API::Client.new(ENV['GITLAB_ADDRESS'], personal_access_token: ENV['GITLAB_QA_ACCESS_TOKEN'])
      end

      def run
        $stdout.puts 'Fetching subgroups for deletion...'

        sub_group_ids = fetch_subgroup_ids
        $stdout.puts "\nNumber of Sub Groups not already marked for deletion: #{sub_group_ids.length}"

        delete_subgroups(sub_group_ids) unless sub_group_ids.empty?
        $stdout.puts "\nDone"
      end

      private

      def delete_subgroups(sub_group_ids)
        $stdout.puts "Deleting #{sub_group_ids.length} subgroups..."
        sub_group_ids.each do |subgroup_id|
          delete_response = delete Runtime::API::Request.new(@api_client, "/groups/#{subgroup_id}").url
          dot_or_f = delete_response.code == 202 ? "\e[32m.\e[0m" : "\e[31mF\e[0m"
          print dot_or_f
        end
      end

      def fetch_group_id
        group_name = ENV['TOP_LEVEL_GROUP_NAME'] || 'gitlab-qa-sandbox-group'
        group_search_response = get Runtime::API::Request.new(@api_client, "/groups/#{group_name}" ).url
        JSON.parse(group_search_response.body)["id"]
      end

      def fetch_subgroup_ids
        group_id = fetch_group_id
        sub_groups_ids = []
        page_no = '1'

        # When we reach the last page, the x-next-page header is a blank string
        while page_no.present?
          $stdout.print '.'

          sub_groups_response = get Runtime::API::Request.new(@api_client, "/groups/#{group_id}/subgroups", page: page_no, per_page: '100').url
          sub_groups_ids.concat(JSON.parse(sub_groups_response.body)
            .reject { |subgroup| !subgroup["marked_for_deletion_on"].nil? }.map { |subgroup| subgroup['id'] })

          page_no = sub_groups_response.headers[:x_next_page].to_s
        end

        sub_groups_ids.uniq
      end
    end
  end
end