require 'hammer_cli_foreman/image'

module HammerCLIForeman

  class ComputeResource < HammerCLIForeman::Command
    resource :compute_resources

    module ProviderNameLegacy
      def extend_data(data)
        # api used to return provider_friendly_name as a provider
        data['provider_friendly_name'] ||= data['provider']
        data
      end
    end

    class ListCommand < HammerCLIForeman::ListCommand
      include ProviderNameLegacy

      output do
        field :id, _("Id")
        field :name, _("Name")
        field :provider_friendly_name, _("Provider")
      end

      build_options
    end


    class InfoCommand < HammerCLIForeman::InfoCommand
      include ProviderNameLegacy

      PROVIDER_SPECIFIC_FIELDS = {
        'ovirt' => [
          Fields::Field.new(:label => _('Datacenter'), :path => [:datacenter])
        ],
        'ec2' => [
          Fields::Field.new(:label => _('Region'), :path => [:region])
        ],
        'vmware' => [
          Fields::Field.new(:label => _('Datacenter'), :path => [:datacenter]),
          Fields::Field.new(:label => _('Server'), :path => [:server])
        ],
        'openstack' => [
          Fields::Field.new(:label => _('Tenant'), :path => [:tenant])
        ],
        'rackspace' => [
          Fields::Field.new(:label => _('Region'), :path => [:region])
        ],
        'libvirt' => [
        ]
      }

      output ListCommand.output_definition do
        field :url, _("Url")
        field :description, _("Description")
        field :user, _("User")
        HammerCLIForeman::References.taxonomies(self)
        HammerCLIForeman::References.timestamps(self)
      end

      def print_data(data)
        provider = data["provider"].downcase
        output_definition.fields.concat(PROVIDER_SPECIFIC_FIELDS[provider] || [])
        super(data)
      end

      build_options
    end


    class CreateCommand < HammerCLIForeman::CreateCommand

      VALIDATION_PER_PROVIDER = {
        'libvirt'   => [:option_url],
        'ovirt'     => [:option_url, :option_user, :option_password, :option_datacenter],
        'openstack' => [:option_url, :option_user, :option_password],
        'racksapce' => [:option_url],
        'ec2'       => [:option_region, :option_user, :option_password],
        'vmware'    => [:option_user, :option_password, :option_datacenter, :option_server]
      }

      success_message _("Compute resource created.")
      failure_message _("Could not create the compute resource")

      build_options

      validate_options do
        required_options = (VALIDATION_PER_PROVIDER[
          option(:option_provider).value.to_s.downcase
        ] || []).unshift(:option_name, :option_provider).uniq

        all(*required_options).required
      end
    end

    class UpdateCommand < HammerCLIForeman::UpdateCommand
      success_message _("Compute resource updated.")
      failure_message _("Could not update the compute resource")

      build_options :without => :name
    end


    class DeleteCommand < HammerCLIForeman::DeleteCommand
      success_message _("Compute resource deleted.")
      failure_message _("Could not delete the compute resource")

      build_options
    end

    class AvailableClustersCommand < HammerCLIForeman::ListCommand
      action :available_clusters
      command_name 'clusters'

      output do
        field :id, _('Id')
        field :name, _('Name')
      end

      build_options
    end

    class AvailableNetworksCommand < HammerCLIForeman::ListCommand
      action :available_networks
      command_name 'networks'

      output do
        field :id, _('Id')
        field :name, _('Name')
      end

      build_options
    end

    class AvailableImagesCommand < HammerCLIForeman::ListCommand
      action :available_images
      command_name 'images'

      output do
        field :id, _('Id')
        field :name, _('Name')
      end

      build_options
    end

    class AvailableFlavorsCommand < HammerCLIForeman::ListCommand
      action :available_flavors
      command_name 'flavors'

      output do
        field :id, _('Id')
        field :name, _('Name')
      end

      build_options
    end

    class AvailableFoldersCommand < HammerCLIForeman::ListCommand
      action :available_folders
      command_name 'folders'

      output do
        field :id, _('Id')
        field :name, _('Name')
      end

      build_options
    end

    class AvailableZonesCommand < HammerCLIForeman::ListCommand
      action :available_zones
      command_name 'zones'

      output do
        field :id, _('Id')
        field :name, _('Name')
      end

      build_options
    end

    class AvailableResourcePoolsCommand < HammerCLIForeman::ListCommand
      action :available_resource_pools
      command_name 'resource-pools'

      output do
        field :id, _('Id')
        field :name, _('Name')
      end

      build_options
    end

    class AvailableStorageDomainsCommand < HammerCLIForeman::ListCommand
      action :available_storage_domains
      command_name 'storage-domains'

      output do
        field :id, _('Id')
        field :name, _('Name')
      end

      build_options
    end

    class AvailableStoragePodsCommand < HammerCLIForeman::ListCommand
      action :available_storage_pods
      command_name 'storage-pods'

      output do
        field :id, _('Id')
        field :name, _('Name')
      end

      build_options
    end

    class AvailableSecurityGroupsCommand < HammerCLIForeman::ListCommand
      action :available_security_groups
      command_name 'security-groups'

      output do
        field :id, _('Id')
        field :name, _('Name')
      end

      build_options
    end

    autoload_subcommands
    subcommand 'image', HammerCLIForeman::Image.desc, HammerCLIForeman::Image
  end

end