support label tag mapping for Azure

AlexanderZagaynov · Apr 20, 2018 · 3a1d2a0 · 3a1d2a0
1 parent 1987641
commit 3a1d2a0
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/app/controllers/ops_controller/settings/label_tag_mapping.rb b/app/controllers/ops_controller/settings/label_tag_mapping.rb
@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module OpsController::Settings::LabelTagMapping
   extend ActiveSupport::Concern
 
@@ -13,6 +15,7 @@ module OpsController::Settings::LabelTagMapping
   MAPPABLE_ENTITIES = {
     nil                   => nil, # map all entities
     "Vm"                  => "Amazon",
+    "VmAzure"             => "Azure",
     "Image"               => "Amazon",
     "ContainerProject"    => "Kubernetes",
     "ContainerRoute"      => "Kubernetes",
@@ -79,7 +82,7 @@ def entity_ui_name_or_all(entity)
     if entity
       provider = MAPPABLE_ENTITIES[entity]
       model = case entity
-              when 'Vm'
+              when 'Vm', 'VmAzure'
                 "ManageIQ::Providers::#{provider}::CloudManager::Vm"
               when 'Image'
                 "ManageIQ::Providers::#{provider}::CloudManager::Template"
@@ -172,8 +175,8 @@ def label_tag_mapping_add(entity, label_name, cat_description)
       provider = ALL_PREFIX
       entity_str = ''
     else
-      provider = MAPPABLE_ENTITIES[entity].downcase
-      entity_str = entity.underscore
+      provider = MAPPABLE_ENTITIES[entity].downcase.inquiry
+      entity_str = (provider.azure? && entity.sub(/Azure$/, '') || entity).underscore
     end
 
     cat_name = "#{provider}:#{entity_str}:" + Classification.sanitize_name(label_name.tr("/", ":"))
@@ -192,8 +195,9 @@ def label_tag_mapping_add(entity, label_name, cat_description)
                                                    :description  => cat_description,
                                                    :single_value => true,
                                                    :read_only    => true)
-        ContainerLabelTagMapping.create!(:labeled_resource_type => entity, :label_name => label_name,
-                                         :tag => category.tag)
+        ContainerLabelTagMapping.create!(:labeled_resource_type => entity,
+                                         :label_name            => label_name,
+                                         :tag                   => category.tag)
       end
     rescue StandardError => bang
       add_flash(_("Error during 'add': %{message}") % {:message => bang.message}, :error)