module HammerCLIKatello::ContentExportHelper

# File lib/hammer_cli_katello/content_export_helper.rb, line 114
def self.included(base)
  if base.command_name.first.to_sym == :version
    setup_version(base)
  elsif base.command_name.first.to_sym == :library
    setup_library(base)
  end
end

# File lib/hammer_cli_katello/content_export_helper.rb, line 122
def self.setup_library(base)
  base.action(:library)
  base.success_message _("Library environment is being exported in task %{id}.")
  base.failure_message _("Could not export the library")
  base.option "--fail-on-missing-content", :flag,
                _("Fails if any of the repositories belonging"\
                  " to this organization are unexportable.")

  base.build_options do |o|
    o.expand(:all).including(:organizations)
  end
end

# File lib/hammer_cli_katello/content_export_helper.rb, line 135
def self.setup_version(base)
  base.action(:version)
  setup_version_options(base)
  base.success_message _("Content view version is being exported in task %{id}.")
  base.failure_message _("Could not export the content view version")

  base.extend_with(
    HammerCLIKatello::CommandExtensions::LifecycleEnvironments.new(only: :option_sources)
  )
  base.include(LifecycleEnvironmentNameMapping)

  base.class_eval do
    def request_params
      super.tap do |opts|
        opts["id"] = resolver.content_view_version_id(options)
      end
    end
  end
end

# File lib/hammer_cli_katello/content_export_helper.rb, line 155
def self.setup_version_options(base)
  base.option "--fail-on-missing-content", :flag,
                _("Fails if any of the repositories belonging"\
                  " to this version are unexportable.")

  base.option "--version", "VERSION", _("Filter versions by version number."),
             :attribute_name => :option_version,
             :required => false

  base.build_options do |o|
    o.expand(:all).including(:content_views, :organizations, :environments)
    o.without(:environment_ids, :environment_id)
  end

  base.validate_options do
    unless option(:option_id).exist?
      any(:option_id, :option_content_view_name, :option_content_view_id).required
      any(:option_version, :option_environment_id, :option_environment_name).required
      unless option(:option_content_view_id).exist?
        any(:option_organization_id, :option_organization_name, \
                            :option_organization_label).required
      end
    end
  end
end

# File lib/hammer_cli_katello/content_export_helper.rb, line 7
def execute
  warn_unexportable_repositories
  response = super
  if option_async?
    output.print_message _("Once the task completes the export metadata must be generated "\
      + "with the command:")
    output.print_message(" hammer content-export generate-metadata --task-id #{@task['id']}")
    HammerCLI::EX_OK
  elsif response != HammerCLI::EX_OK
    response
  else
    export_history = fetch_export_history_from_task(reload_task(@task))
    if export_history
      generate_metadata_json(export_history)
      HammerCLI::EX_OK
    else
      output.print_error _("Could not fetch the export history")
      HammerCLI::EX_CANTCREAT
    end
  end
end

# File lib/hammer_cli_katello/content_export_helper.rb, line 42
def fetch_export_history(export_history_id)
  index(:content_exports, :id => export_history_id).first if export_history_id
end

# File lib/hammer_cli_katello/content_export_helper.rb, line 46
def fetch_export_history_from_task(task)
  # checking this here implies the task object was loaded recently
  if %w(error warning).include?(task['result'])
    raise _("Can not fetch export history from an unfinished task")
  end

  export_history_id = task.dig('output', 'export_history_id')
  fetch_export_history(export_history_id)
end

# File lib/hammer_cli_katello/content_export_helper.rb, line 75
def fetch_repositories
  repo_options = {
    library: true,
    content_type: 'yum',
    search: 'download_policy != immediate'
  }
  if version_command?
    repo_options[:content_view_version_id] = resolver.content_view_version_id(options)
  else
    repo_options[:organization_id] = options["option_organization_id"]
  end
  index(:repositories, repo_options)
end

# File lib/hammer_cli_katello/content_export_helper.rb, line 56
def generate_metadata_json(export_history)
  metadata_json = export_history["metadata"].to_json
  begin
    metadata_path = "#{export_history['path']}/metadata.json"
    File.write(metadata_path, metadata_json)
    output.print_message _("Generated #{metadata_path}")
  rescue SystemCallError
    filename = "metadata-#{export_history['id']}.json"
    File.write(filename, metadata_json)
    output.print_message _("Unable to access/write to '#{export_history['path']}'. "\
                           "Generated '#{Dir.pwd}/#{filename}' instead. "\
                           "This file is necessary to perform an import.")
  end
end

# File lib/hammer_cli_katello/content_export_helper.rb, line 33
def reload_task(task)
  task_id = if task.is_a? Hash
              task['id']
            else
              task
            end
  show(:foreman_tasks, id: task_id)
end

# File lib/hammer_cli_katello/content_export_helper.rb, line 147
def request_params
  super.tap do |opts|
    opts["id"] = resolver.content_view_version_id(options)
  end
end

# File lib/hammer_cli_katello/content_export_helper.rb, line 29
def send_request
  @task = super
end

# File lib/hammer_cli_katello/content_export_helper.rb, line 71
def version_command?
  self.class.command_name.first.to_sym == :version
end

# File lib/hammer_cli_katello/content_export_helper.rb, line 89
def warn_unexportable_repositories
  repos = fetch_repositories
  unless repos.empty?
    if version_command?
      output.print_message _("NOTE: Unable to fully export this version because"\
        " it contains repositories without the 'immediate' download policy."\
        " Update the download policy and sync affected repositories."\
        " Once synced republish the content view"\
        " and export the generated version.")
    else
      output.print_message _("NOTE: Unable to fully export this organization's library because"\
        " it contains repositories without the 'immediate' download policy."\
        " Update the download policy and sync affected"\
        " repositories to include them in the export.")
    end
    output.print_message _("Use the following command to update the "\
      "download policy of these repositories.")
    output.print_message "hammer repository update --id=<REPOSITORY_ID> "\
      "--download-policy='immediate'"
    output.print_message ""
    print_record(::HammerCLIKatello::Repository::ListCommand.output_definition, repos)
    exit(HammerCLI::EX_SOFTWARE) if option_fail_on_missing_content?
  end
end