ManageIQ/manageiq-providers-azure

class ManageIQ::Providers::Azure::CloudManager::OrchestrationStack < ManageIQ::Providers::CloudManager::OrchestrationStack
  require_dependency 'manageiq/providers/azure/cloud_manager/orchestration_stack/status'
  include ManageIQ::Providers::Azure::CloudManager::Deployment

  def self.raw_create_stack(orchestration_manager, stack_name, template, options = {})
    resource_group, create_options = make_create_options(template, options)

    # no error is raised if the resource_group already exists
    orchestration_manager.with_provider_connection do |configure|
      Azure::Armrest::ResourceGroupService.new(configure).create(resource_group, orchestration_manager.provider_region)
    end

    orchestration_manager.with_provider_connection do |configure|
      Azure::Armrest::TemplateDeploymentService.new(configure).create(stack_name, resource_group, create_options).id
    end
  rescue StandardError => err
    _log.error("stack=[#{stack_name}], error: #{err}")
    raise MiqException::MiqOrchestrationProvisionError, err.to_s, err.backtrace
  end

  def self.make_create_options(template, options)
    resource_group = options[:resource_group]
    create_options = {:template => JSON.parse(template.content)}
    create_options.merge!(options).except!(:resource_group)

    if create_options[:parameters]
      create_options[:parameters] = create_options[:parameters].map { |k, v| [k, {'value' => v}] }.to_h
    end

    return resource_group, 'properties' => create_options
  end

  def raw_update_stack(template, options)
    resource_group, create_options = self.class.make_create_options(template, options)

    # use the same API for stack update and creation
    ext_management_system.with_provider_connection do |configure|
      Azure::Armrest::TemplateDeploymentService.new(configure).create(name, resource_group, create_options)
    end
  rescue StandardError => err
    _log.error("stack=[#{name}], error: #{err}")
    raise MiqException::MiqOrchestrationUpdateError, err.to_s, err.backtrace
  end

  def raw_delete_stack
    ext_management_system.with_provider_connection do |configure|
      Azure::Armrest::TemplateDeploymentService.new(configure).delete_associated_resources(name, resource_group)
    end
  rescue StandardError => err
    _log.error("stack=[#{name}], error: #{err}")
    raise MiqException::MiqOrchestrationDeleteError, err.to_s, err.backtrace
  end

  def raw_status
    ext_management_system.with_provider_connection do |configure|
      tds = ::Azure::Armrest::TemplateDeploymentService.new(configure)
      deployment = tds.get(name, resource_group)
      if deployment_failed?(deployment)
        fail_reason = deployment_failure_reason(tds.list_deployment_operations(name, resource_group))
      end
      Status.new(deployment.properties.provisioning_state, fail_reason)
    end
  rescue StandardError => err
    if err.to_s =~ /[D|d]eployment.+ could not be found/
      raise MiqException::MiqOrchestrationStackNotExistError,
            _("%{name} does not exist on %{management}") % {:name => name, :management => ext_management_system.name}
    end

    _log.error("stack=[#{name}], error: #{err}")
    raise MiqException::MiqOrchestrationStatusError, err.to_s, err.backtrace
  end

  def self.display_name(number = 1)
    n_('Orchestration Stack (Microsoft Azure)', 'Orchestration Stacks (Microsoft Azure)', number)
  end
end