require 'hammer_cli'
require 'foreman_api'
require 'hammer_cli_foreman/commands'
require 'hammer_cli_foreman/associating_commands'

module HammerCLIForeman

  class Template < HammerCLI::Apipie::Command

    resource ForemanApi::Resources::ConfigTemplate

    class ListCommand < HammerCLIForeman::ListCommand

      output do
        from "config_template" do
          field :id, "Id"
          field :name, "Name"
          field :type, "Type"
        end
      end

      def retrieve_data
        templates = super
        templates.each do |tpl|
          if tpl["config_template"]["snippet"]
            tpl["config_template"]["type"] = "snippet"
          else
            tpl["config_template"]["type"] = tpl["config_template"]["template_kind"]["name"] if tpl["config_template"]["template_kind"]
          end
        end
      end

      apipie_options
    end


    class InfoCommand < HammerCLIForeman::InfoCommand

      #FIXME: Add name identifier when the find by name issue is fixed in the api
      #       Api currently does not accept names containing a dot
      identifiers :id

      output ListCommand.output_definition do
        from "config_template" do
          field :operatingsystem_ids, "OS ids", Fields::List
        end
      end

      def retrieve_data
        template = super
        if template["config_template"]["snippet"]
          template["config_template"]["type"] = "snippet"
        else
          template["config_template"]["type"] = template["config_template"]["template_kind"]["name"]
        end
        template
      end

    end


    class ListKindsCommand < HammerCLIForeman::ListCommand

      command_name "kinds"
      desc "List available config template kinds."

      output do
        from "template_kind" do
          field :name, "Name"
        end
      end

      def retrieve_data
        snippet = { "template_kind" => { "name" => "snippet" }}
        super << snippet
      end

      resource ForemanApi::Resources::TemplateKind, "index"
    end


    class DumpCommand < HammerCLIForeman::InfoCommand

      command_name "dump"
      desc "View config template content."

      identifiers :id

      def print_data(template)
        puts template["config_template"]["template"]
      end

    end


    class CreateCommand < HammerCLIForeman::CreateCommand

      option "--file", "TEMPLATE", "Path to a file that contains the template", :attribute_name => :template, :required => true,
        :format => HammerCLI::Options::Normalizers::File.new
      option "--type", "TYPE", "Template type. Eg. snippet, script, provision", :required => true

      success_message "Config template created"
      failure_message "Could not create the config template"

      def snippet
        type == "snippet"
      end

      def template_kind_id
        kinds = ForemanApi::Resources::TemplateKind.new(resource_config).index()[0]
        table = kinds.inject({}){ |result, k| result.update(k["template_kind"]["name"] => k["template_kind"]["id"]) }
        table[type]
      end

      apipie_options :without => [:template_combinations_attributes, :template, :snippet, :template_kind_id]
    end


    class UpdateCommand < HammerCLIForeman::UpdateCommand

      option "--file", "TEMPLATE", "Path to a file that contains the template", :attribute_name => :template,
        :format => HammerCLI::Options::Normalizers::File.new
      option "--type", "TYPE", "Template type. Eg. snippet, script, provision"

      identifiers :id

      success_message "Config template updated"
      failure_message "Could not update the config template"

      def snippet
        type == "snippet"
      end

      def template_kind_id
        kinds = ForemanApi::Resources::TemplateKind.new(resource_config).index()[0]
        table = kinds.inject({}){ |result, k| result.update(k["template_kind"]["name"] => k["template_kind"]["id"]) }
        table[type]
      end

      apipie_options :without => [:template_combinations_attributes, :template, :snippet, :template_kind_id]
    end


    class DeleteCommand < HammerCLIForeman::DeleteCommand

      identifiers :id

      success_message "Config template deleted"
      failure_message "Could not delete the config template"
    end


    include HammerCLIForeman::AssociatingCommands::OperatingSystem


    autoload_subcommands
  end

end

HammerCLI::MainCommand.subcommand 'template', "Manipulate Foreman's config templates.", HammerCLIForeman::Template