diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2021-05-21 12:10:16 +0300 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2021-05-21 12:10:16 +0300 |
commit | 34ad6d995bcab9f88a236bfed15aebdad76df960 (patch) | |
tree | 4e47f30ed36fbdebd15933b10a04b80ae98f0475 /rubocop/cop/usage_data/instrumentation_superclass.rb | |
parent | 1cf90c700837637f7989bda8fe81c616f19e8150 (diff) |
Add latest changes from gitlab-org/gitlab@master
Diffstat (limited to 'rubocop/cop/usage_data/instrumentation_superclass.rb')
-rw-r--r-- | rubocop/cop/usage_data/instrumentation_superclass.rb | 63 |
1 files changed, 63 insertions, 0 deletions
diff --git a/rubocop/cop/usage_data/instrumentation_superclass.rb b/rubocop/cop/usage_data/instrumentation_superclass.rb new file mode 100644 index 00000000000..2ff2ed47a23 --- /dev/null +++ b/rubocop/cop/usage_data/instrumentation_superclass.rb @@ -0,0 +1,63 @@ +# frozen_string_literal: true + +module RuboCop + module Cop + module UsageData + # This cop checks that metric instrumentation classes subclass one of the allowed base classes. + # + # @example + # + # # good + # class CountIssues < DatabaseMetric + # # ... + # end + # + # # bad + # class CountIssues < BaseMetric + # # ... + # end + class InstrumentationSuperclass < RuboCop::Cop::Cop + MSG = "Instrumentation classes should subclass one of the following: %{allowed_classes}." + + BASE_PATTERN = "(const nil? !#allowed_class?)" + + def_node_matcher :class_definition, <<~PATTERN + (class (const _ !#allowed_class?) #{BASE_PATTERN} ...) + PATTERN + + def_node_matcher :class_new_definition, <<~PATTERN + [!^(casgn {nil? cbase} #allowed_class? ...) + !^^(casgn {nil? cbase} #allowed_class? (block ...)) + (send (const {nil? cbase} :Class) :new #{BASE_PATTERN})] + PATTERN + + def on_class(node) + class_definition(node) do + register_offense(node.children[1]) + end + end + + def on_send(node) + class_new_definition(node) do + register_offense(node.children.last) + end + end + + private + + def allowed_class?(class_name) + allowed_classes.include?(class_name) + end + + def allowed_classes + cop_config['AllowedClasses'] || [] + end + + def register_offense(offense_node) + message = format(MSG, allowed_classes: allowed_classes.join(', ')) + add_offense(offense_node, message: message) + end + end + end + end +end |