Simplify label helper and correct version

1 files changed, 54 insertions, 66 deletions

diff --git a/lib/api/helpers/label_helpers.rb b/lib/api/helpers/label_helpers.rb
index f444358ebe0..c11e7d614ab 100644
--- a/lib/api/helpers/label_helpers.rb
+++ b/lib/api/helpers/label_helpers.rb
@@ -3,91 +3,79 @@
 module API
   module Helpers
     module LabelHelpers
-      extend ActiveSupport::Concern
-
-      included do
-        helpers do
-          params :label_create_params do
-            requires :name, type: String, desc: 'The name of the label to be created'
-            requires :color, type: String, desc: "The color of the label given in 6-digit hex notation with leading '#' sign (e.g. #FFAABB) or one of the allowed CSS color names"
-            optional :description, type: String, desc: 'The description of label to be created'
-          end
-
-          params :label_update_params do
-            requires :name,  type: String, desc: 'The name of the label to be updated'
-            optional :new_name, type: String, desc: 'The new name of the label'
-            optional :color, type: String, desc: "The new color of the label given in 6-digit hex notation with leading '#' sign (e.g. #FFAABB) or one of the allowed CSS color names"
-            optional :description, type: String, desc: 'The new description of label'
-            at_least_one_of :new_name, :color, :description
-          end
+      extend Grape::API::Helpers
 
-          def find_label(parent, id, include_ancestor_groups: true)
-            labels = available_labels_for(parent, include_ancestor_groups: include_ancestor_groups)
-            label = labels.find_by_id(id) || labels.find_by_title(id)
+      params :label_create_params do
+        requires :name, type: String, desc: 'The name of the label to be created'
+        requires :color, type: String, desc: "The color of the label given in 6-digit hex notation with leading '#' sign (e.g. #FFAABB) or one of the allowed CSS color names"
+        optional :description, type: String, desc: 'The description of label to be created'
+      end
 
-            label || not_found!('Label')
-          end
+      def find_label(parent, id, include_ancestor_groups: true)
+        labels = available_labels_for(parent, include_ancestor_groups: include_ancestor_groups)
+        label = labels.find_by_id(id) || labels.find_by_title(id)
 
-          def get_labels(parent, entity)
-            present paginate(available_labels_for(parent)), with: entity, current_user: current_user, parent: parent
-          end
+        label || not_found!('Label')
+      end
 
-          def create_label(parent, entity)
-            authorize! :admin_label, parent
+      def get_labels(parent, entity)
+        present paginate(available_labels_for(parent)), with: entity, current_user: current_user, parent: parent
+      end
 
-            label = available_labels_for(parent).find_by_title(params[:name])
-            conflict!('Label already exists') if label
+      def create_label(parent, entity)
+        authorize! :admin_label, parent
 
-            priority = params.delete(:priority)
-            label_params = declared_params(include_missing: false)
+        label = available_labels_for(parent).find_by_title(params[:name])
+        conflict!('Label already exists') if label
 
-            label =
-              if parent.is_a?(Project)
-                ::Labels::CreateService.new(label_params).execute(project: parent)
-              else
-                ::Labels::CreateService.new(label_params).execute(group: parent)
-              end
+        priority = params.delete(:priority)
+        label_params = declared_params(include_missing: false)
 
-            if label.persisted?
-              if parent.is_a?(Project)
-                label.prioritize!(parent, priority) if priority
-              end
+        label =
+          if parent.is_a?(Project)
+            ::Labels::CreateService.new(label_params).execute(project: parent)
+          else
+            ::Labels::CreateService.new(label_params).execute(group: parent)
+          end
 
-              present label, with: entity, current_user: current_user, parent: parent
-            else
-              render_validation_error!(label)
-            end
+        if label.persisted?
+          if parent.is_a?(Project)
+            label.prioritize!(parent, priority) if priority
           end
 
-          def update_label(parent, entity)
-            authorize! :admin_label, parent
+          present label, with: entity, current_user: current_user, parent: parent
+        else
+          render_validation_error!(label)
+        end
+      end
 
-            label = find_label(parent, params[:name], include_ancestor_groups: false)
-            update_priority = params.key?(:priority)
-            priority = params.delete(:priority)
+      def update_label(parent, entity)
+        authorize! :admin_label, parent
 
-            label = ::Labels::UpdateService.new(declared_params(include_missing: false)).execute(label)
-            render_validation_error!(label) unless label.valid?
+        label = find_label(parent, params[:name], include_ancestor_groups: false)
+        update_priority = params.key?(:priority)
+        priority = params.delete(:priority)
 
-            if parent.is_a?(Project) && update_priority
-              if priority.nil?
-                label.unprioritize!(parent)
-              else
-                label.prioritize!(parent, priority)
-              end
-            end
+        label = ::Labels::UpdateService.new(declared_params(include_missing: false)).execute(label)
+        render_validation_error!(label) unless label.valid?
 
-            present label, with: entity, current_user: current_user, parent: parent
+        if parent.is_a?(Project) && update_priority
+          if priority.nil?
+            label.unprioritize!(parent)
+          else
+            label.prioritize!(parent, priority)
           end
+        end
 
-          def delete_label(parent)
-            authorize! :admin_label, parent
+        present label, with: entity, current_user: current_user, parent: parent
+      end
 
-            label = find_label(parent, params[:name], include_ancestor_groups: false)
+      def delete_label(parent)
+        authorize! :admin_label, parent
 
-            destroy_conditionally!(label)
-          end
-        end
+        label = find_label(parent, params[:name], include_ancestor_groups: false)
+
+        destroy_conditionally!(label)
       end
     end
   end