foreman_azure_rm 1.3.1 → 2.0.0.pre1

data/app/models/foreman_azure_rm/azure_rm_compute.rb

@@ -0,0 +1,95 @@
+module ForemanAzureRM
+  class AzureRMCompute
+    attr_accessor :sdk
+    attr_accessor :azure_vm
+
+    delegate :name, to: :azure_vm, allow_nil: true
+
+    def initialize(azure_vm: ComputeModels::VirtualMachine.new, sdk: sdk)
+      @azure_vm = azure_vm
+      @sdk = sdk
+    end
+
+    def id
+      @azure_vm.id
+    end
+
+    def persisted?
+      !!identity && !!id
+    end
+
+    def vm_size
+      @azure_vm.hardware_profile.vm_size
+    end
+
+    def wait_for(_timeout = 0, _interval = 0, &block)
+      instance_eval(&block)
+      return true
+    end
+
+    def ready?
+      vm_status == 'running'
+    end
+
+    def state
+      vm_status
+    end
+
+    def vm_status
+      sdk.check_vm_status(@azure_vm.resource_group, name)
+    end
+
+    def network_interface_card_ids
+      nics = @azure_vm.network_profile.network_interfaces
+      nics.map(&:id)
+    end
+
+    def provisioning_ip_address
+      interfaces.each do |nic|
+        nic.ip_configurations.each do |configuration|
+          next unless configuration.primary
+          if configuration.public_ipaddress.present?
+            ip_id     = configuration.public_ipaddress.id
+            ip_rg     = ip_id.split('/')[4]
+            ip_name   = ip_id.split('/')[-1]
+            public_ip = sdk.public_ip(ip_rg, ip_name)
+            return public_ip.ip_address
+          else
+            return configuration.private_ipaddress
+          end
+        end
+      end    
+    end
+
+    def interfaces
+      interfaces = []
+      unless network_interface_card_ids.nil?
+        network_interface_card_ids.each do |nic_id|
+          nic_rg   = nic_id.split('/')[4]
+          nic_name = nic_id.split('/')[-1]
+          interfaces << sdk.vm_nic(nic_rg, nic_name)
+        end
+      end
+      interfaces
+    end
+
+    def interfaces=(setifaces)
+      @azure_vm.network_profile.network_interfaces = setifaces
+    end
+
+    def ip_addresses
+      []
+    end
+
+    def identity
+      @azure_vm.name
+    end
+
+    def identity=(setuuid)
+      @azure_vm.name = setuuid
+    end
+
+    def image_id
+    end
+  end
+end

data/app/views/compute_resources/form/_azurerm.html.erb

@@ -1,9 +1,8 @@
 <%= text_f f, :url, :label => _('Client ID'), :required => true %>
-<%= password_f f, :password, :label => _('Client Secret'), :required => true %>
+<%= password_f f, :password, :label => _('Client Secret'), :keep_value => true, :required => true %>
 <%= text_f f, :user, :label => _('Subscription ID'), :required => true %>
-<%= text_f f, :uuid, :label => _('Tenant ID'), :required => true,
-    :help_inline => link_to_function(
-    _("Test Connection"), "testConnection(this)",
-    :class => "btn btn-default",
-    :'data-url' => test_connection_compute_resources_path) +
-    hidden_spinner('', :id => 'test_connection_indicator').html_safe  %>
+<%= text_f f, :uuid, :label => _('Tenant ID'), :required => true %>
+
+<div class="col-md-offset-2">
+    <%= test_connection_button_f(f, true) %>
+</div>

data/app/views/compute_resources_vms/form/azurerm/_base.html.erb

@@ -5,13 +5,13 @@
 %>
 
 <script>
-    function azure_rm_location_callback() {
-        azure_rm_get_size_from_location();
-        azure_rm_subnets_from_location();
+    function azure_rm_region_callback() {
+        azure_rm_get_size_from_region();
+        azure_rm_subnets_from_region();
     }
 
-    function azure_rm_get_size_from_location() {
-        var location = $('#azure_rm_location').val();
+    function azure_rm_get_size_from_region() {
+        var region = $('#azure_rm_region').val();
         var size_spinner = $('#azure_rm_size_spinner');
         var sizes = $('#azure_rm_size');
         var imageId = $('#azure_rm_image_id').val();
@@ -22,7 +22,7 @@
         }
         size_spinner.removeClass('hide');
         $.ajax({
-            data: {"location_string": location, "image_id": imageId},
+            data: {"region_string": region, "image_id": imageId},
             type: 'get',
             url: '/azure_rm/sizes',
             complete: function () {
@@ -47,19 +47,19 @@
         });
     }
 
-    function azure_rm_subnets_from_location() {
+    function azure_rm_subnets_from_region() {
         var imageId = $('#azure_rm_image_id').val();
         var subnets = $('#azure_rm_subnet');
-        var location = $('#azure_rm_location').val();
+        var region = $('#azure_rm_region').val();
         if (typeof tfm == 'undefined') {  // earlier than 1.13
             foreman.tools.showSpinner();
         } else {
             tfm.tools.showSpinner();
         }
         $.ajax({
-            data: {"image_id": imageId, "location": location},
-            type: "get",
-            url: "/azure_rm/subnets",
+            data: {"image_id": imageId, "region": region},
+            type: 'get',
+            url: '/azure_rm/subnets',
             complete: function () {
                 reloadOnAjaxComplete('#azure_rm_subnet');
                 if (typeof tfm == 'undefined') {  // earlier than 1.13
@@ -77,33 +77,39 @@
                 subnets.empty();
                 $.each(request_subnets, function () {
                     //noinspection JSAnnotator
-                    subnets.append($("<option />").val(this.id).text(`${this.virtual_network_name} - ${this.name} (${this.address_prefix})`));
+                    subnets.append($("<option />").val(this.id).text(`${this.id.split('/')[8]} - ${this.name} (${this.address_prefix})`));
                 });
             }
         });
     }
 </script>
 
-<%= selectable_f f, :location, compute_resource.locations,
-                 { :include_blank => _('Please select an Azure region') },
-                 {
-                     :label       => _('Azure Region'),
-                     :required    => true,
-                     :id          => 'azure_rm_location',
-                     :label_size  => "col-md-2",
-                     :onchange    => 'azure_rm_location_callback();',
-                     :help_inline => spinner_button_f(f, _('Reload Images, Sizes, vNets'),
-                                                      'azure_rm_location_callback();',
-                                                      {
-                                                          :id            => 'load_subnets_btn',
-                                                          :spinner_id    => 'load_subnets_indicator',
-                                                          :class         => 'btn-success',
-                                                          :spinner_class => 'spinner-inverse'
-                                                      })
-                 }
+<% #This view has been modified and wraps the properties of azure_vm
+   # using fields_for rails method.
 %>
 
-<%= selectable_f f, :resource_group, resource_groups,
+<%= f.fields_for :azure_vm do |vm_f| %>
+
+  <%= selectable_f vm_f, :location, compute_resource.regions,
+               { :include_blank => _('Please select an Azure region') },
+               {
+                   :label       => _('Azure Region'),
+                   :required    => true,
+                   :id          => 'azure_rm_region',
+                   :label_size  => "col-md-2",
+                   :onchange    => 'azure_rm_region_callback();',
+                   :help_inline => spinner_button_f(f, _('Reload Images, Sizes, vNets'),
+                                                    'azure_rm_region_callback();',
+                                                    {
+                                                        :id            => 'load_subnets_btn',
+                                                        :spinner_id    => 'load_subnets_indicator',
+                                                        :class         => 'btn-success',
+                                                        :spinner_class => 'spinner-inverse'
+                                                    })
+               }
+  %>
+
+  <%= selectable_f vm_f, :resource_group, resource_groups,
                  { :include_blank => true },
                  {
                      :disabled => resource_groups.empty?,
@@ -111,87 +117,76 @@
                      :required => true,
                      :id       => 'azure_rm_rg'
                  }
-%>
+  %>
 
-<%= selectable_f f, :vm_size, [],
+  <%= selectable_f vm_f, :vm_size, [],
                  { :include_blank => _('Please first select an Azure region') },
                  {
                      :label    => _('VM Size'),
                      :required => true,
                      :id       => 'azure_rm_size'
                  }
-%>
+  %>
 
-<%= selectable_f f, :platform, %w(Linux Windows),
+  <%= selectable_f vm_f, :platform, %w(Linux Windows),
                  {},
                  {
                      :label    => _('Platform'),
                      :required => true
                  }
-%>
+  %>
 
-<%= text_f f, :username,
+  <%= text_f vm_f, :username,
            {
                :label    => _('Username'),
                :required => true
            }
-%>
-<%= text_f f, :password,
+  %>
+
+  <%= password_f vm_f, :password,
            {
                :label    => _('Password'),
-               :required => true
+               :required => true,
+               :placeholder => "********"
            }
-%>
+  %>
 
-<%= textarea_f f, :ssh_key_data,
+  <%= textarea_f vm_f, :ssh_key_data,
                {
                    :label => _('SSH Key')
                }
-%>
-
-<%= number_f f, :os_disk_size,
-             {
-                 :class    => "col-md-2",
-                 :label    => _('OS Disk Size (GB)'), :label_size => "col-md-2",
-                 :required => true
-             }
-%>
+  %>
 
-<%= checkbox_f f, :premium_os_disk,
+  <%= checkbox_f vm_f, :premium_os_disk,
                {
                    :label => _('Premium OS Disk'), :label_size => "col-md-2"
                },
                'true',
                'false'
-%>
+  %>
 
-<%= selectable_f f, :os_disk_caching, %w(None ReadOnly ReadWrite),
+  <%= selectable_f vm_f, :os_disk_caching, %w(None ReadOnly ReadWrite),
                  {},
                  {
                      :label    => _('OS Disk Caching'),
                      :required => true,
                      :class    => "col-md-2"
                  }
-%>
-
-<%= text_f f, :puppet_master,
-           {
-               :label => _('Puppet Master'),
-               :help_inline => _('For the Azure Puppet extension for Windows')
-           }
-%>
+  %>
 
-<%= text_f f, :script_command,
+  <%= text_f vm_f, :script_command,
            {
-               :label => _('Custom Script Command')
+               :label => _('Custom Script Command'),
+               :help_inline => _("To perform commands as root, prefix it with 'sudo'.")
            }
-%>
+  %>
 
-<%= text_f f, :script_uris,
+  <%= text_f vm_f, :script_uris,
            {
                :label => _('Comma seperated file URIs')
            }
-%>
+  %>
+<% end %>
 
 <div id="image_selection">
   <%= select_f f, :image_id, images, :uuid, :name,
@@ -204,4 +199,4 @@
                    :id         => 'azure_rm_image_id'
                }
   %>
-</div>
+</div>

data/app/views/compute_resources_vms/index/_azurerm.html.erb

@@ -13,8 +13,8 @@
     <tr>
       <td><%= link_to_if_authorized vm.name, hash_for_compute_resource_vm_path(:compute_resource_id => @compute_resource, :id => vm.identity).merge(:auth_object => @compute_resource, :authorizer => :authorizer) %></td>
       <td><%= vm.vm_size %></td>
-      <td><%= vm.resource_group %></td>
-      <td><%= vm.location %></td>
+      <td><%= vm.azure_vm.resource_group %></td>
+      <td><%= vm.azure_vm.location %></td>
       <td> <span <%= vm_power_class(vm.ready?) %>> <%= vm_state(vm) %> </span> </td>
       <td>
         <%= action_buttons(
@@ -26,4 +26,4 @@
       </td>
     </tr>
   <% end %>
-</table>
+</table>

data/app/views/compute_resources_vms/show/_azurerm.html.erb

@@ -1,14 +1,26 @@
-<% title @vm.name %>
+<% title @vm.azure_vm.name %>
 <div class='col-md-12'>
   <table class="<%= table_css_classes %>">
     <thead>
     <tr><th colspan="2"><%=_('Properties') %></th></tr>
     </thead>
     <tbody>
-      <%= prop :id %>
-      <%= prop :location %>
-      <%= prop :vm_size %>
-      <%= prop :platform %>
+      <tr>
+        <td>Id</td>
+        <td><%=@vm.id%></td>
+      </tr>
+      <tr>
+        <td>Location</td>
+        <td><%=@vm.azure_vm.location%></td>
+      </tr>
+      <tr>
+        <td>VM Size</td>
+        <td><%=@vm.vm_size%></td>
+      </tr>
+      <tr>
+        <td>Platform</td>
+        <td><%=@vm.azure_vm.storage_profile.os_disk.os_type%></td>
+      </tr>
     </tbody>
   </table>
 </div>

data/lib/foreman_azure_rm.rake

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+# Tests
+namespace :test do
+  desc 'Test foreman_azure_rm'
+  Rake::TestTask.new(:foreman_azure_rm) do |t|
+    test_dir = File.join(File.dirname(__FILE__), '..', 'test')
+    t.libs << ['test', test_dir]
+    t.pattern = "#{test_dir}/**/*_test.rb"
+    t.verbose = true
+    t.warning = false
+  end
+end

data/lib/foreman_azure_rm/azure_sdk_adapter.rb

@@ -0,0 +1,141 @@
+module ForemanAzureRM
+  Storage = Azure::Storage::Profiles::Latest::Mgmt
+  Network = Azure::Network::Profiles::Latest::Mgmt
+  Compute = Azure::Compute::Profiles::Latest::Mgmt
+  Resources = Azure::Resources::Profiles::Latest::Mgmt
+
+  StorageModels = Storage::Models
+  NetworkModels = Network::Models
+  ComputeModels = Compute::Models
+  ResourceModels = Resources::Models
+
+  class AzureSDKAdapter
+    def initialize(tenant, app_ident, secret_key, sub_id)
+      @tenant           = tenant
+      @app_ident        = app_ident
+      @secret_key       = secret_key
+      @sub_id           = sub_id
+    end
+
+    def resource_client
+      @resource_client ||= Resources::Client.new(azure_credentials)
+    end
+
+    def compute_client
+      @compute_client ||= Compute::Client.new(azure_credentials)
+    end
+
+    def network_client
+      @network_client ||= Network::Client.new(azure_credentials)
+    end
+
+    def storage_client
+      @storage_client ||= Storage::Client.new(azure_credentials)
+    end
+
+    def azure_credentials
+      provider = MsRestAzure::ApplicationTokenProvider.new(
+      @tenant,
+      @app_ident,
+      @secret_key)
+      
+      credentials = MsRest::TokenCredentials.new(provider)
+
+      {
+        credentials: credentials,
+        subscription_id: @sub_id
+      }
+    end
+
+    def rgs
+      rgs      = resource_client.resource_groups.list
+      rgs.map(&:name)
+    end
+
+    def vnets
+      network_client.virtual_networks.list_all
+    end
+
+    def subnets(resource_group, vnet_name)
+      network_client.subnets.list(resource_group, vnet_name)
+    end
+
+    def public_ip(rg_name, pip_name)
+      network_client.public_ipaddresses.get(rg_name, pip_name)
+    end
+
+    def vm_nic(rg_name,nic_name)
+      network_client.network_interfaces.get(rg_name, nic_name)
+    end
+
+    def list_vm_sizes(region)
+      stripped_region = region.gsub(/\s+/, '').downcase
+      compute_client.virtual_machine_sizes.list(stripped_region).value()
+    end
+
+    def list_vms(rg_name)
+      # List all VMs in a resource group
+      virtual_machines = compute_client.virtual_machines.list(rg_name)
+    end
+
+    def get_vm(rg_name, vm_name)
+      compute_client.virtual_machines.get(rg_name, vm_name)
+    end
+
+    def get_storage_accts
+      result = storage_client.storage_accounts.list
+      result.value
+    end
+
+    def create_or_update_vm(rg_name, vm_name, parameters)
+      compute_client.virtual_machines.create_or_update(rg_name, vm_name, parameters)
+    end
+
+    def create_or_update_vm_extensions(rg_name, vm_name, vm_extension_name, extension_params)
+      compute_client.virtual_machine_extensions.create_or_update(rg_name,
+                                                          vm_name,
+                                                          vm_extension_name,
+                                                          extension_params) 
+    end
+
+    def create_or_update_pip(rg_name, pip_name, parameters)
+      network_client.public_ipaddresses.create_or_update(rg_name, pip_name, parameters)
+    end
+
+    def create_or_update_nic(rg_name, nic_name, parameters)
+      network_client.network_interfaces.create_or_update(rg_name, nic_name, parameters)
+    end
+
+    def delete_pip(rg_name, pip_name)
+      network_client.public_ipaddresses.delete(rg_name, pip_name)
+    end
+
+    def delete_nic(rg_name, nic_name)
+      network_client.network_interfaces.delete(rg_name, nic_name)
+    end
+
+    def delete_vm(rg_name, vm_name)
+      compute_client.virtual_machines.delete(rg_name, vm_name)
+    end
+
+    def delete_disk(rg_name, osdisk_name)
+      compute_client.disks.delete(rg_name, osdisk_name)
+    end
+
+    def check_vm_status(rg_name, vm_name)
+      virtual_machine = compute_client.virtual_machines.get(rg_name, vm_name, expand:'instanceView')
+      get_status(virtual_machine)
+    end
+
+    def get_status(virtual_machine)
+      vm_statuses = virtual_machine.instance_view.statuses
+      vm_status = nil
+      vm_statuses.each do |status|
+        if status.code.include? 'PowerState'
+          vm_status = status.code.split('/')[1]
+        end
+      end
+      vm_status
+    end
+  end
+end