# frozen_string_literal: true
require 'optparse'
require 'amazing_print'
module DatadogBackup
class Cli
include ::DatadogBackup::Options
def all_diff_futures
LOGGER.info("Starting diffs on #{::DatadogBackup::ThreadPool::TPOOL.max_length} threads")
any_resource_instance
.all_file_ids_for_selected_resources
.map do |id|
Concurrent::Promises.future_on(::DatadogBackup::ThreadPool::TPOOL, id) do |id|
[id, getdiff(id)]
end
end
end
def any_resource_instance
resource_instances.first
end
def backup
resource_instances.each(&:purge)
resource_instances.each(&:backup)
any_resource_instance.all_files
end
def initialize_client
@options[:client] ||= Dogapi::Client.new(
datadog_api_key,
datadog_app_key
)
end
def definitive_resource_instance(id)
matching_resource_instance(any_resource_instance.class_from_id(id))
end
def diffs
futures = all_diff_futures
::DatadogBackup::ThreadPool.watcher.join
format_diff_output(
Concurrent::Promises
.zip(*futures)
.value!
.compact
)
end
def getdiff(id)
result = definitive_resource_instance(id).diff(id)
case result
when ''
nil
when "\n"
nil
when ''
nil
else
result
end
end
def format_diff_output(diff_output)
case diff_format
when nil, :color
diff_output.map do |id, diff|
" ---\n id: #{id}\n#{diff}"
end.join("\n")
when :html
'' +
diff_output.map do |id, diff|
"
---
id: #{id}
" + diff
end.join('
') +
''
else
raise 'Unexpected diff_format.'
end
end
def initialize(options)
@options = options
initialize_client
end
def matching_resource_instance(klass)
resource_instances.select { |resource_instance| resource_instance.instance_of?(klass) }.first
end
def resource_instances
@resource_instances ||= resources.map do |resource|
resource.new(@options)
end
end
def restore
futures = all_diff_futures
watcher = ::DatadogBackup::ThreadPool.watcher
futures.each do |future|
id, diff = *future.value!
next unless diff
if @options[:force_restore]
definitive_resource_instance(id).restore(id)
else
puts '--------------------------------------------------------------------------------'
puts format_diff_output([id, diff])
puts '(r)estore to Datadog, overwrite local changes and (d)ownload, (s)kip, or (q)uit?'
response = $stdin.gets.chomp
case response
when 'q'
exit
when 'r'
puts "Restoring #{id} to Datadog."
definitive_resource_instance(id).restore(id)
when 'd'
puts "Downloading #{id} from Datadog."
definitive_resource_instance(id).get_and_write_file(id)
when 's'
next
else
puts 'Invalid response, please try again.'
response = $stdin.gets.chomp
end
end
end
watcher.join if watcher.status
end
def run!
puts(send(action.to_sym))
rescue SystemExit, Interrupt
::DatadogBackup::ThreadPool.shutdown
end
end
end