class KPM::DiagnosticFile

def config=(config_file = nil)
  @config = nil
  return if config_file.nil?
  @config = YAML.load_file(config_file) unless Dir[config_file][0].nil?
end

def export_data(account_id = nil, log_dir = nil)
  self.config = @config_file
  tenant_export_file = retrieve_tenant_config
  system_export_file = retrieve_system_config
  account_export_file = retrieve_account_data(account_id) unless account_id.nil?
  log_files = retrieve_log_files(log_dir)
  raise Interrupt, 'Account id or configuration file not found' unless File.exist?(system_export_file) && File.exist?(tenant_export_file)
  zip_file_name = TMP_DIR + File::Separator + ZIP_FILE
  Zip::File.open(zip_file_name, Zip::File::CREATE) do |zip_file|
    zip_file.add(TENANT_FILE, tenant_export_file)
    zip_file.add(SYSTEM_FILE, system_export_file)
    zip_file.add(ACCOUNT_FILE, account_export_file) unless account_id.nil?
    zip_file.add(ZIP_LOG_FILE, log_files) unless log_files.nil?
  end
  @logger.info "\e[32mDiagnostic data exported under #{zip_file_name} \e[0m"
  zip_file_name
end

def initialize(config_file = nil, killbill_api_credentials = nil, killbill_credentials = nil, killbill_url = nil,
               database_name = nil, database_credentials = nil, database_host = nil, database_port = nil, kaui_web_path = nil,
               killbill_web_path = nil, bundles_dir = nil, logger = nil)
  @killbill_api_credentials = killbill_api_credentials
  @killbill_credentials = killbill_credentials
  @killbill_url = killbill_url
  @database_name = database_name
  @database_credentials = database_credentials
  @database_host = database_host
  @database_port = database_port
  @config_file = config_file
  @kaui_web_path = kaui_web_path
  @killbill_web_path = killbill_web_path
  @logger = logger
  @original_logger_level = logger.level
  @catalina_base = nil
  @bundles_dir = bundles_dir
end

def retrieve_account_data(account_id)
  @logger.info 'Retrieving account data for id: ' + account_id
  # this suppress the message of where it put the account file, this is to avoid confusion
  @logger.level = Logger::WARN
  account = KPM::Account.new(@config_file, @killbill_api_credentials, @killbill_credentials,
                             @killbill_url, @database_name,
                             @database_credentials, @database_host, @database_port, nil, @logger)
  export_file = account.export_data(account_id)
  final = TMP_DIR + File::Separator + ACCOUNT_FILE
  FileUtils.move(export_file, final)
  @logger.level = @original_logger_level
  final
end

def retrieve_config(parent, child)
  item = nil
  unless @config.nil?
    config_parent = @config[parent]
    item = config_parent[child] unless config_parent.nil?
  end
  item
end

def retrieve_log_files(log_dir)
  if @catalina_base.nil? && log_dir.nil?
    @logger.warn "\e[91;1mUnable to find Tomcat process, logs won't be collected: make sure to run kpm using the same user as the Tomcat process or pass the option --log-dir\e[0m"
    return nil
  end
  @logger.info 'Collecting log files'
  log_base = log_dir || (@catalina_base + File::Separator + 'logs')
  log_items = Dir.glob(log_base + File::Separator + '*')
  zip_file_name = TMP_DIR + File::Separator + ZIP_LOG_FILE
  Zip::File.open(zip_file_name, Zip::File::CREATE) do |zip_file|
    log_items.each do |file|
      name = file.split('/').last
      zip_file.add(name, file)
    end
  end
  zip_file_name
end

def retrieve_system_config
  @logger.info 'Retrieving system configuration'
  system = KPM::System.new(@logger)
  export_data = system.information(@bundles_dir, true, @config_file, @kaui_web_path, @killbill_web_path)
  system_catalina_base(export_data)
  export_file = TMP_DIR + File::SEPARATOR + SYSTEM_FILE
  File.open(export_file, 'w') { |io| io.puts export_data }
  export_file
end

def retrieve_tenant_config
  @logger.info 'Retrieving tenant configuration'
  # this suppress the message of where it put the account file, this is to avoid confusion
  @logger.level = Logger::WARN
  @killbill_api_credentials ||= [retrieve_config('killbill', 'api_key'), retrieve_config('killbill', 'api_secret')] unless @config_file.nil?
  @killbill_credentials ||= [retrieve_config('killbill', 'user'), retrieve_config('killbill', 'password')] unless @config_file.nil?
  @killbill_url ||= 'http://' + retrieve_config('killbill', 'host').to_s + ':' + retrieve_config('killbill', 'port').to_s unless @config_file.nil?
  tenant_config = KPM::TenantConfig.new(@killbill_api_credentials,
                                        @killbill_credentials,
                                        @killbill_url,
                                        @logger)
  export_file = tenant_config.export
  final = TMP_DIR + File::Separator + TENANT_FILE
  FileUtils.move(export_file, final)
  @logger.level = @original_logger_level
  final
end

def system_catalina_base(export_data)
  @catalina_base = nil
  system_json = JSON.parse(export_data)
  return if system_json['java_system_information']['catalina.base'].nil?
  @catalina_base = system_json['java_system_information']['catalina.base']['value']
end