1 files changed, 7 insertions, 1 deletions
diff --git a/lib/gitlab/quick_actions/command_definition.rb b/lib/gitlab/quick_actions/command_definition.rb
index e7bfcb16582..93030fd454e 100644
--- a/lib/gitlab/quick_actions/command_definition.rb
+++ b/lib/gitlab/quick_actions/command_definition.rb
@@ -4,7 +4,7 @@ module Gitlab
   module QuickActions
     class CommandDefinition
       attr_accessor :name, :aliases, :description, :explanation, :params,
-        :condition_block, :parse_params_block, :action_block, :warning
+        :condition_block, :parse_params_block, :action_block, :warning, :types
 
       def initialize(name, attributes = {})
         @name = name
@@ -17,6 +17,7 @@ module Gitlab
         @condition_block = attributes[:condition_block]
         @parse_params_block = attributes[:parse_params_block]
         @action_block = attributes[:action_block]
+        @types = attributes[:types] || []
       end
 
       def all_names
@@ -28,6 +29,7 @@ module Gitlab
       end
 
       def available?(context)
+        return false unless valid_type?(context)
         return true unless condition_block
 
         context.instance_exec(&condition_block)
@@ -96,6 +98,10 @@ module Gitlab
 
         context.instance_exec(arg, &parse_params_block)
       end
+
+      def valid_type?(context)
+        types.blank? || types.any? { |type| context.quick_action_target.is_a?(type) }
+      end
     end
   end
 end