gitlab-dangerfiles 4.8.1 → 4.12.0

data/lib/danger/plugins/commit_messages.rb

@@ -0,0 +1,140 @@
+# frozen_string_literal: true
+
+require_relative "../../gitlab/dangerfiles/commit_linter"
+require_relative "../../gitlab/dangerfiles/merge_request_linter"
+
+module Danger
+  # Common helper functions for our danger scripts. See Danger::Helper
+  # for more details
+  class CommitMessages < Danger::Plugin
+    COMMIT_MSG_DOC = "https://docs.gitlab.com/ee/development/contributing/merge_request_workflow.html#commit-messages-guidelines"
+    MORE_INFO = "For more information, take a look at our [Commit message guidelines](#{COMMIT_MSG_DOC}).".freeze
+    THE_DANGER_JOB_TEXT = "the `%<job_name>s` job"
+    MAX_COMMITS_COUNT_EXCEEDED_MESSAGE = <<~MSG
+      This merge request includes more than %<max_commits_count>d commits. Each commit should meet the following criteria:
+
+      1. Have a well-written commit message.
+      1. Has all tests passing when used on its own (e.g. when using git checkout SHA).
+      1. Can be reverted on its own without also requiring the revert of commit that came before it.
+      1. Is small enough that it can be reviewed in isolation in under 30 minutes or so.
+
+      If this merge request contains commits that do not meet this criteria and/or contains intermediate work, please rebase these commits into a smaller number of commits or split this merge request into multiple smaller merge requests.
+    MSG
+
+    def build_message(commit, message, more_info: true)
+      [message].tap do |full_message|
+        full_message << ". #{MORE_INFO}" if more_info
+        full_message.unshift("#{commit.sha}: ") if commit.sha
+      end.join
+    end
+
+    def fail_commit(commit, message, more_info: true)
+      self.fail(build_message(commit, message, more_info: more_info))
+    end
+
+    def warn_commit(commit, message, more_info: true)
+      self.warn(build_message(commit, message, more_info: more_info))
+    end
+
+    def danger_job_link
+      if helper.ci?
+        "[#{format(THE_DANGER_JOB_TEXT, job_name: ENV.fetch('CI_JOB_NAME', nil))}](#{ENV.fetch('CI_JOB_URL', nil)})"
+      else
+        THE_DANGER_JOB_TEXT
+      end
+    end
+
+    # Perform various checks against commits. We're not using
+    # https://github.com/jonallured/danger-commit_lint because its output is not
+    # very helpful, and it doesn't offer the means of ignoring merge commits.
+    def lint_commit(commit)
+      linter = Gitlab::Dangerfiles::CommitLinter.new(commit)
+
+      # For now we'll ignore merge commits, as getting rid of those is a problem
+      # separate from enforcing good commit messages.
+      return linter if linter.merge?
+
+      # We ignore revert commits as they are well structured by Git already
+      return linter if linter.revert?
+
+      # If MR is set to squash, we ignore fixup commits
+      return linter if linter.fixup? && helper.squash_mr?
+
+      if linter.fixup?
+        msg = "Squash or fixup commits must be squashed before merge, or **edit** the merge request, " \
+              "enable **Squash commits when merge request is accepted** and re-run #{danger_job_link}."
+
+        if helper.draft_mr? || helper.squash_mr?
+          warn_commit(commit, msg, more_info: false)
+        else
+          fail_commit(commit, msg, more_info: false)
+        end
+
+        # Makes no sense to process other rules for fixup commits, they trigger just more noise
+        return linter
+      end
+
+      # Fail if a suggestion commit is used and squash is not enabled
+      if linter.suggestion?
+        unless helper.squash_mr?
+          msg = "If you are applying suggestions, **edit** the merge request, enable **Squash commits when " \
+                "merge request is accepted** and re-run #{danger_job_link}."
+          fail_commit(commit, msg, more_info: false)
+        end
+
+        return linter
+      end
+
+      linter.lint
+    end
+
+    def lint_mr_title(mr_title)
+      commit = Struct.new(:message, :sha).new(mr_title)
+
+      Gitlab::Dangerfiles::MergeRequestLinter.new(commit).lint
+    end
+
+    def count_non_fixup_commits(commit_linters)
+      commit_linters.count { |commit_linter| !commit_linter.fixup? }
+    end
+
+    def lint_commits(commits)
+      commit_linters = commits.map { |commit| lint_commit(commit) }
+
+      if helper.squash_mr?
+        multi_line_commit_linter = commit_linters.detect do |commit_linter|
+          !commit_linter.merge? && commit_linter.multi_line?
+        end
+
+        if multi_line_commit_linter&.failed?
+          warn_or_fail_commits(multi_line_commit_linter)
+          commit_linters.delete(multi_line_commit_linter) # Don't show an error (here) and a warning (below)
+        end
+      elsif count_non_fixup_commits(commit_linters) > max_commits_count
+        self.warn(format(MAX_COMMITS_COUNT_EXCEEDED_MESSAGE, max_commits_count: max_commits_count))
+      end
+
+      failed_commit_linters = commit_linters.select(&:failed?)
+      warn_or_fail_commits(failed_commit_linters, default_to_fail: !helper.squash_mr?)
+    end
+
+    def max_commits_count
+      helper.config.max_commits_count
+    end
+
+    def warn_or_fail_commits(failed_linters, default_to_fail: true)
+      level = default_to_fail ? :fail : :warn
+
+      Array(failed_linters).each do |linter|
+        linter.problems.each do |problem_key, problem_desc|
+          case problem_key
+          when :subject_too_short, :details_too_many_changes, :details_line_too_long
+            warn_commit(linter.commit, problem_desc)
+          else
+            __send__(:"#{level}_commit", linter.commit, problem_desc)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/danger/plugins/duo_code.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module Danger
+  # Contains method to check if Duo code suggestion added to MR as a reviewer.
+  class DuoCode < Danger::Plugin
+    def suggestion_added?
+      return false unless helper.ci?
+
+      helper.mr_reviewers.include?("GitLabDuo")
+    end
+  end
+end

data/lib/danger/plugins/internal/helper.rb

@@ -25,12 +25,14 @@ module Danger
       pipeline: '~"maintenance::pipelines" for CI',
       ci_template: '~"ci::templates"',
       analytics_instrumentation: '~"analytics instrumentation"',
-      import_integrate_be: '~"group::import and integrate" (backend)',
-      import_integrate_fe: '~"group::import and integrate" (frontend)',
       Authentication: '~"group::authentication"',
       Authorization: '~"group::authorization"',
       Compliance: '~"group::compliance"',
-      Verify: '~"devops::verify"'
+      Verify: '~"devops::verify"',
+      "AST:Composition Analysis": '~"group::composition analysis"',
+      "AST:Dynamic Analysis": '~"group::dynamic analysis"',
+      "AST:Secret Detection": '~"group::secret detection"',
+      "AST:Static Analysis": '~"group::static analysis"'
     }.freeze
     # rubocop:enable Style/HashSyntax
 
@@ -181,7 +183,9 @@ module Danger
     # @return [{Symbol => Array<String>}] a hash of the type +{ category1: ["file1", "file2"], category2: ["file3", "file4"] }+
     #   using filename regex (+filename_regex+) and specific change regex (+changes_regex+) from the given +categories+ hash.
     def changes_by_category(categories = [])
-      all_changed_files.each_with_object(Hash.new { |h, k| h[k] = [] }) do |file, hash|
+      changed_and_deleted_files = all_changed_files + changes.deleted.files
+
+      changed_and_deleted_files.each_with_object(Hash.new { |h, k| h[k] = [] }) do |file, hash|
         categories_for_file(file, categories).each { |category| hash[category] << file }
       end
     end
@@ -244,7 +248,7 @@ module Danger
     # @return [String] the GFM for a category label, making its best guess if it's not
     #   a category we know about.
     def label_for_category(category)
-      CATEGORY_LABELS[category] ||
+      helper.config.custom_labels_for_categories[category] || CATEGORY_LABELS[category] ||
 
         if category.start_with?("`")
           category.to_s
@@ -288,6 +292,13 @@ module Danger
       gitlab_helper.mr_json["assignees"]
     end
 
+    # @return [Array<Hash>] +[]+ when not in the CI context, and the MR reviewers otherwise.
+    def mr_reviewers
+      return [] unless ci?
+
+      gitlab_helper.mr_json["reviewers"]
+    end
+
     # @return [String] +""+ when not in the CI context, and the MR title otherwise.
     def mr_title
       return "" unless ci?

data/lib/danger/plugins/roulette.rb

@@ -9,6 +9,7 @@ module Danger
   # for more details
   class Roulette < Danger::Plugin
     HOURS_WHEN_PERSON_CAN_BE_PICKED = (6..14).freeze
+    BOT_REVIEWERS = %w[GitLabDuo].freeze
     HTTPError = Class.new(StandardError)
 
     def prepare_categories(changes_keys)
@@ -45,7 +46,7 @@ module Danger
     # @param categories [Array<Symbol>] An array of categories symbols.
     #
     # @return [Array<Spin>]
-    def spin(project = nil, categories = [:none], ux_fallback_wider_community_reviewer: teammate_pedroms)
+    def spin(project = nil, categories = [:none], ux_fallback_wider_community_reviewer: ux_fallback_reviewer)
       # TODO: Deprecate the project argument. It prevents us from
       # memorizing Spinner and can cause unexpected results if it's
       # passing a different project than the merge request project.
@@ -87,10 +88,40 @@ module Danger
       Gitlab::Dangerfiles::Teammate.warnings
     end
 
+    # Automatically assigns reviewers from roulette spins if configured to do so
+    #
+    # @param spins [Array<Spin>] The roulette spins to potentially assign from
+    def assign_reviewers_from_roulette(spins)
+      return if human_reviewers?
+
+      reviewers_to_assign = find_reviewers_to_assign(spins)
+
+      if reviewers_to_assign.any?
+        post_assignment_message(reviewers_to_assign)
+      else
+        warn("No reviewers available for assignment")
+      end
+    end
+
+    # Determines if auto-assignment should happen based on configuration
+    #
+    # @return [Boolean]
+    def auto_assign_reviewers?
+      return false if helper.config.auto_assign_for_roulette_roles.empty?
+
+      configured_labels = helper.config.auto_assign_for_roulette_labels
+
+      return true if configured_labels.empty?
+
+      mr_labels = helper.mr_labels
+      configured_labels.any? { |label| mr_labels.include?(label) }
+    end
+
     private
 
-    def teammate_pedroms
-      @teammate_pedroms ||= Gitlab::Dangerfiles::Teammate.find_member("pedroms")
+    def ux_fallback_reviewer
+      teammates = %w[pedroms annabeldunstone seggenberger jmiocene clavimoniere]
+      Gitlab::Dangerfiles::Teammate.find_member(teammates.sample)
     end
 
     def spin_for_approval_rule?(rule)
@@ -157,5 +188,48 @@ module Danger
     def labels
       @labels ||= helper.mr_labels
     end
+
+    # Checks if any non-bot reviewers are already assigned to the merge request
+    #
+    # @return [Boolean]
+    def human_reviewers?
+      helper.mr_reviewers.any? { |reviewer| !BOT_REVIEWERS.include?(reviewer["username"]) }
+    end
+
+    # Find reviewers to assign based on configured roles
+    #
+    # @param spins [Array<Spin>] The roulette spins
+    # @return [Array<String>] Array of usernames to assign
+    def find_reviewers_to_assign(spins)
+      roles_to_assign = helper.config.auto_assign_for_roulette_roles
+      reviewers_to_assign = []
+
+      spins.each do |spin|
+        if roles_to_assign.include?(:reviewer) && spin.reviewer&.username
+          reviewers_to_assign << spin.reviewer.username
+        end
+
+        if roles_to_assign.include?(:maintainer) && spin.maintainer&.username
+          reviewers_to_assign << spin.maintainer.username
+        end
+
+        if reviewers_to_assign.any?
+          break
+        end
+      end
+
+      reviewers_to_assign
+    end
+
+    # Posts the assignment message with the selected reviewers
+    #
+    # @param reviewers_to_assign [Array<String>] Array of usernames to assign
+    def post_assignment_message(reviewers_to_assign)
+      role_text = helper.config.auto_assign_for_roulette_roles.map(&:to_s).join(' and ')
+      message = "🎲 Assigned #{role_text}s based on reviewer roulette.\n/assign_reviewer #{reviewers_to_assign.map { |u| "@#{u}" }.join(' ')}"
+      markdown(message)
+    rescue StandardError => e
+      warn("Failed to assign reviewers: #{e.message}")
+    end
   end
 end

data/lib/danger/rules/commit_messages/Dangerfile

@@ -1,127 +1,5 @@
 # frozen_string_literal: true
 
-require_relative "../../../gitlab/dangerfiles/commit_linter"
-require_relative "../../../gitlab/dangerfiles/merge_request_linter"
-
-COMMIT_MESSAGE_GUIDELINES = "https://docs.gitlab.com/ee/development/contributing/merge_request_workflow.html#commit-messages-guidelines"
-MORE_INFO = "For more information, take a look at our [Commit message guidelines](#{COMMIT_MESSAGE_GUIDELINES})."
-THE_DANGER_JOB_TEXT = "the `%<job_name>s` job"
-MAX_COMMITS_COUNT = helper.config.max_commits_count
-MAX_COMMITS_COUNT_EXCEEDED_MESSAGE = <<~MSG
-This merge request includes more than %<max_commits_count>d commits. Each commit should meet the following criteria:
-
-1. Have a well-written commit message.
-1. Has all tests passing when used on its own (e.g. when using git checkout SHA).
-1. Can be reverted on its own without also requiring the revert of commit that came before it.
-1. Is small enough that it can be reviewed in isolation in under 30 minutes or so.
-
-If this merge request contains commits that do not meet this criteria and/or contains intermediate work, please rebase these commits into a smaller number of commits or split this merge request into multiple smaller merge requests.
-MSG
-
-def fail_commit(commit, message, more_info: true)
-  self.fail(build_message(commit, message, more_info: more_info))
-end
-
-def warn_commit(commit, message, more_info: true)
-  self.warn(build_message(commit, message, more_info: more_info))
-end
-
-def build_message(commit, message, more_info: true)
-  [message].tap do |full_message|
-    full_message << ". #{MORE_INFO}" if more_info
-    full_message.unshift("#{commit.sha}: ") if commit.sha
-  end.join
-end
-
-def danger_job_link
-  helper.ci? ? "[#{format(THE_DANGER_JOB_TEXT, job_name: ENV["CI_JOB_NAME"])}](#{ENV['CI_JOB_URL']})" : THE_DANGER_JOB_TEXT
-end
-
-# Perform various checks against commits. We're not using
-# https://github.com/jonallured/danger-commit_lint because its output is not
-# very helpful, and it doesn't offer the means of ignoring merge commits.
-def lint_commit(commit)
-  linter = Gitlab::Dangerfiles::CommitLinter.new(commit)
-
-  # For now we'll ignore merge commits, as getting rid of those is a problem
-  # separate from enforcing good commit messages.
-  return linter if linter.merge?
-
-  # We ignore revert commits as they are well structured by Git already
-  return linter if linter.revert?
-
-  # If MR is set to squash, we ignore fixup commits
-  return linter if linter.fixup? && helper.squash_mr?
-
-  if linter.fixup?
-    msg = "Squash or fixup commits must be squashed before merge, or **edit** the merge request, enable **Squash commits when merge request is accepted** and re-run #{danger_job_link}."
-    if helper.draft_mr? || helper.squash_mr?
-      warn_commit(commit, msg, more_info: false)
-    else
-      fail_commit(commit, msg, more_info: false)
-    end
-
-    # Makes no sense to process other rules for fixup commits, they trigger just more noise
-    return linter
-  end
-
-  # Fail if a suggestion commit is used and squash is not enabled
-  if linter.suggestion?
-    unless helper.squash_mr?
-      fail_commit(commit, "If you are applying suggestions, **edit** the merge request, enable **Squash commits when merge request is accepted** and re-run #{danger_job_link}.", more_info: false)
-    end
-
-    return linter
-  end
-
-  linter.lint
-end
-
-def lint_mr_title(mr_title)
-  commit = Struct.new(:message, :sha).new(mr_title)
-
-  Gitlab::Dangerfiles::MergeRequestLinter.new(commit).lint
-end
-
-def count_non_fixup_commits(commit_linters)
-  commit_linters.count { |commit_linter| !commit_linter.fixup? }
-end
-
-def lint_commits(commits)
-  commit_linters = commits.map { |commit| lint_commit(commit) }
-
-  if helper.squash_mr?
-    multi_line_commit_linter = commit_linters.detect { |commit_linter| !commit_linter.merge? && commit_linter.multi_line? }
-
-    if multi_line_commit_linter && multi_line_commit_linter.failed?
-      warn_or_fail_commits(multi_line_commit_linter)
-      commit_linters.delete(multi_line_commit_linter) # Don't show an error (here) and a warning (below)
-    end
-  else
-    if count_non_fixup_commits(commit_linters) > MAX_COMMITS_COUNT
-      self.warn(format(MAX_COMMITS_COUNT_EXCEEDED_MESSAGE, max_commits_count: MAX_COMMITS_COUNT))
-    end
-  end
-
-  failed_commit_linters = commit_linters.select { |commit_linter| commit_linter.failed? }
-  warn_or_fail_commits(failed_commit_linters, default_to_fail: !helper.squash_mr?)
-end
-
-def warn_or_fail_commits(failed_linters, default_to_fail: true)
-  level = default_to_fail ? :fail : :warn
-
-  Array(failed_linters).each do |linter|
-    linter.problems.each do |problem_key, problem_desc|
-      case problem_key
-      when :subject_too_short, :subject_above_warning, :details_too_many_changes, :details_line_too_long
-        warn_commit(linter.commit, problem_desc)
-      else
-        self.__send__(:"#{level}_commit", linter.commit, problem_desc)
-      end
-    end
-  end
-end
-
 # As part of https://gitlab.com/groups/gitlab-org/-/epics/4826 we are
 # vendoring workhorse commits from the stand-alone gitlab-workhorse
 # repo. There is no point in linting commits that we want to vendor as
@@ -130,4 +8,4 @@ def workhorse_changes?
   git.diff.any? { |file| file.path.start_with?('workhorse/') }
 end
 
-lint_commits(git.commits) unless workhorse_changes?
+commit_messages.lint_commits(git.commits) unless workhorse_changes?

data/lib/danger/rules/duo_code_review/Dangerfile

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+return if duo_code.suggestion_added?
+
+case helper.config.duo_code_review
+when :optional
+  markdown("We advise getting a review from GitLab Duo Code. You can assign `@GitLabDuo` as a reviewer to this merge request.")
+when :mandatory
+  failure("A review from GitLab Duo Code is mandatory. Please assign `@GitLabDuo` as a reviewer to this merge request.")
+end

data/lib/danger/rules/simple_roulette/Dangerfile

@@ -8,9 +8,6 @@ MESSAGE = <<MARKDOWN
 Changes that require review have been detected! A merge request is normally
 reviewed by both a reviewer and a maintainer in its primary category and by a
 maintainer in all other categories.
-MARKDOWN
-
-TABLE_MARKDOWN = <<MARKDOWN
 
 To spread load more evenly across eligible reviewers, Danger has picked a candidate for each
 review slot. Feel free to
@@ -26,9 +23,11 @@ Please consider assigning a reviewer or maintainer who is a
 [domain expert](https://about.gitlab.com/handbook/engineering/projects/##{PROJECT_NAME})
 in the area of the merge request.
 
+MARKDOWN
+
+NO_AUTO_ASSIGN = <<MARKDOWN
 Once you've decided who will review this merge request, mention them as you
 normally would! Danger does not automatically notify them for you.
-
 MARKDOWN
 
 TABLE_HEADER_WITH_CATEGORIES = <<MARKDOWN
@@ -101,5 +100,11 @@ if changes.any?
 
   markdown(format(WARNING_MESSAGE, warnings: warnings)) if warnings
 
-  markdown(TABLE_MARKDOWN + table_header + rows.join("\n")) unless rows.empty?
+  markdown(table_header + rows.join("\n")) unless rows.empty?
+
+  if roulette.auto_assign_reviewers?
+    roulette.assign_reviewers_from_roulette(random_roulette_spins)
+  else
+    markdown(NO_AUTO_ASSIGN)
+  end
 end

data/lib/gitlab/dangerfiles/approval.rb

@@ -4,18 +4,18 @@ require_relative "spin"
 
 module Gitlab
   module Dangerfiles
-    Approval = Struct.new(:category, :spin) do
+    Approval = Struct.new(:category, :spin, :name) do
       def self.from_approval_rule(rule, maintainer)
         category =
           if rule["section"] == "codeowners"
             "`#{rule['name']}`"
           else
-            rule["section"]
+            rule["section"]&.downcase
           end.to_sym
 
         spin = Spin.new(category, nil, maintainer, :reviewer)
 
-        new(category, spin)
+        new(category, spin, rule["name"])
       end
     end
   end

data/lib/gitlab/dangerfiles/capability.rb

@@ -15,8 +15,6 @@ module Gitlab
           none: None,
           test: Test,
           tooling: Tooling,
-          import_integrate_be: ImportIntegrateBE,
-          import_integrate_fe: ImportIntegrateFE,
           ux: UX
         }.freeze
       end
@@ -60,20 +58,6 @@ module Gitlab
         end
       end
 
-      class ImportIntegrateBE < Capability
-        def has_capability?(teammate)
-          kind == :reviewer &&
-            teammate.role.match?(/Backend Engineer.+Manage:Import and Integrate/)
-        end
-      end
-
-      class ImportIntegrateFE < Capability
-        def has_capability?(teammate)
-          kind == :reviewer &&
-            teammate.role.match?(/Frontend Engineer.+Manage:Import and Integrate/)
-        end
-      end
-
       class UX < Capability
         def has_capability?(teammate)
           super && teammate.member_of_the_group?(labels)

data/lib/gitlab/dangerfiles/commit_linter.rb

@@ -9,9 +9,9 @@ module Gitlab
       MAX_CHANGED_FILES_IN_COMMIT = 3
       MAX_CHANGED_LINES_IN_COMMIT = 30
       # Issue, MR, Epic
-      SHORT_REFERENCE_REGEX = %r{(\S*([\w\-\/]+)?(?<!`)(#|!|&)\d+(?<!`))}.freeze
+      SHORT_REFERENCE_REGEX = %r{(\S*([\w\-\/]+)?(?<!`)(#|!|&)(?>\d+)(?!`))}.freeze
       # Milestone
-      MS_SHORT_REFERENCE_REGEX = %r{(\S*([\w\-\/]+)?(?<!`)%"?\d{1,3}\.\d{1,3}"?(?<!`))}.freeze
+      MS_SHORT_REFERENCE_REGEX = %r{(\S*([\w\-\/]+)?(?<!`)%(?>"?\d{1,3}\.\d{1,3}"?)(?!`))}.freeze
       SUGGESTIONS_APPLIED_COMMIT_REGEX = /Apply \d+ suggestion\(s\) to \d+ file\(s\)/.freeze
 
       def self.problems_mapping
@@ -25,8 +25,8 @@ module Gitlab
             "to the commit message, and are displayed as plain text outside of GitLab",
             message_contains_unicode_emoji: "Avoid the use of Unicode Emoji. These add no value to the commit " \
             "message, and may not be displayed properly everywhere",
-            message_contains_short_reference: "Use full URLs instead of short references (`gitlab-org/gitlab#123` or " \
-            "`!123`), as short references are displayed as plain text outside of GitLab"
+            message_contains_short_reference: "Use full URLs instead of short references (`gitlab-org/gitlab#123`, " \
+            "`!123`, or `%%12.3`), as short references are displayed as plain text outside of GitLab"
           }
         )
       end

data/lib/gitlab/dangerfiles/config.rb

@@ -21,31 +21,54 @@ module Gitlab
       #           match changed lines in files that match +filename_regex+. Used in `helper.changes_by_category`, `helper.changes`, and `helper.categories_for_file`.
       attr_accessor :files_to_category
 
+      # @!attribute custom_labels_for_categories
+      #   @return [{String => String}] A hash of the form +{ category_name => label }+.
+      #           Used in `helper.custom_labels_for_categories`.
+      attr_accessor :custom_labels_for_categories
+
+      # @!attribute disabled_roulette_categories
+      #   @return [Array] indicating which categories would be disabled for the simple roulette. Default to `[]` (all categories are enabled)
+      attr_accessor :disabled_roulette_categories
+
+      # Rule: changes_size
       # @!attribute code_size_thresholds
       #   @return [{ high: Integer, medium: Integer }] a hash of the form +{ high: 42, medium: 12 }+ where +:high+ is the lines changed threshold which triggers an error, and +:medium+ is the lines changed threshold which triggers a warning. Also, see +DEFAULT_CHANGES_SIZE_THRESHOLDS+ for the format of the hash.
       attr_accessor :code_size_thresholds
 
+      # Rule: commit_messages
       # @!attribute max_commits_count
       #   @return [Integer] the maximum number of allowed non-squashed/non-fixup commits for a given MR. A warning is triggered if the MR has more commits.
       attr_accessor :max_commits_count
 
-      # @!attribute disabled_roulette_categories
-      #   @return [Array] indicating which categories would be disabled for the simple roulette. Default to `[]` (all categories are enabled)
-      attr_accessor :disabled_roulette_categories
+      # Rule: duo_code_review
+      # @!attribute duo_code_review
+      #   @return [Symbol] whether a review from GitLab Duo Code is `:mandatory` or `:optional`. Default to `:optional`.
+      attr_accessor :duo_code_review
 
+      # Rule: simple_roulette
       # @!attribute included_optional_codeowners_sections_for_roulette
       #   @return [Array] indicating which optional codeowners sections should be included in roulette. Default to `[]`.
       attr_accessor :included_optional_codeowners_sections_for_roulette
 
+      # Rule: simple_roulette
       # @!attribute excluded_required_codeowners_sections_for_roulette
       #   @return [Array] indicating which required codeowners sections should be excluded from roulette. Default to `[]`.
       attr_accessor :excluded_required_codeowners_sections_for_roulette
 
+      # @!attribute auto_assign_for_roulette_roles
+      #   @return [Array<Symbol>] which roles to auto assign as reviewers, given roulette recommendations (:reviewer, :maintainer, or both). If empty, auto-assignment is disabled. Default to `[]`.
+      attr_accessor :auto_assign_for_roulette_roles
+
+      # @!attribute auto_assign_for_roulette_labels
+      #   @return [Array<String>] MR labels that allow auto reviewer assignment. If empty, applies to all MRs, provided :auto_assign_for_roulette_roles is not empty. Default to `[]`.
+      attr_accessor :auto_assign_for_roulette_labels
+
       DEFAULT_CHANGES_SIZE_THRESHOLDS = { high: 2_000, medium: 500 }.freeze
       DEFAULT_COMMIT_MESSAGES_MAX_COMMITS_COUNT = 10
 
       def initialize
         @files_to_category = {}
+        @custom_labels_for_categories = {}
         @project_root = nil
         @project_name = ENV["CI_PROJECT_NAME"]
         @ci_only_rules = []
@@ -54,6 +77,9 @@ module Gitlab
         @disabled_roulette_categories = []
         @included_optional_codeowners_sections_for_roulette = []
         @excluded_required_codeowners_sections_for_roulette = []
+        @auto_assign_for_roulette_roles = []
+        @auto_assign_for_roulette_labels = []
+        @duo_code_review = :optional
       end
     end
   end