#!/usr/bin/env ruby
require "optparse"

$LOAD_PATH.unshift(File.expand_path("../lib/", __dir__))
require "imap/backup"

KNOWN_COMMANDS = [
  {name: "setup",   help: "Create/edit the configuration file"},
  {name: "backup",  help: "Do the backup (default)"},
  {name: "folders", help: "List folders for all (or selected) accounts"},
  {name: "status",  help: "List count of non backed-up emails per folder"},
  {name: "help",    help: "Show usage"}
].freeze

options = {command: "backup"}
parser = OptionParser.new do |opts|
  opts.banner = "Usage: #{$PROGRAM_NAME} [options] COMMAND"

  opts.separator ""
  opts.separator "Commands:"
  KNOWN_COMMANDS.each do |command|
    opts.separator format("\t%- 20<name>s %<help>s", command)
  end
  opts.separator ""
  opts.separator "Common options:"

  opts.on(
    "-a",
    "--accounts ACCOUNT1[,ACCOUNT2,...]",
    Array,
    "only these accounts"
  ) do |account|
    options[:accounts] = account
  end

  opts.on_tail("-h", "--help", "Show usage") do
    puts opts
    exit
  end

  opts.on_tail("--version", "Show version") do
    puts Imap::Backup::VERSION
    exit
  end
end
parser.parse!

if ARGV.size > 0
  options[:command] = ARGV.shift
end

if KNOWN_COMMANDS.find { |c| c[:name] == options[:command] }.nil?
  raise "Unknown command '#{options[:command]}'"
end

if options[:command] == "help"
  puts opts
  exit
end

begin
  configuration = Imap::Backup::Configuration::List.new(options[:accounts])
rescue Imap::Backup::ConfigurationNotFound
  Imap::Backup::Configuration::Setup.new.run
  exit
end

configuration.setup_logging

case options[:command]
when "setup"
  Imap::Backup::Configuration::Setup.new.run
when "backup"
  configuration.each_connection do |connection|
    connection.run_backup
  end
when "folders"
  configuration.each_connection do |connection|
    puts connection.username
    folders = connection.folders
    if folders.nil?
      warn "Unable to list account folders"
      exit 1
    end
    folders.each { |f| puts "\t" + f.name }
  end
when "status"
  configuration.each_connection do |connection|
    puts connection.username
    folders = connection.status
    folders.each do |f|
      missing_locally = f[:remote] - f[:local]
      puts "#{f[:name]}: #{missing_locally.size}"
    end
  end
end