require "thor"

module Saber
  class CLI < Thor
    include Thor::Actions

    class_option "no-color",               banner: "Disable colorization in output", type: :boolean
    class_option "verbose", aliases: "-V", banner: "Enable verbose output mode", type: :boolean
    class_option "log",                    banner: "Log file", type: :string
    class_option "force",   aliases: "-f", banner: "Fore writing even if file exists", type: :boolean
    class_option "tracker", aliases: "-t", banner: "tracker name", type: :string

    attr_reader :o

    def initialize(*)
      super
      o = @o = options.dup

      Saber.ui = if o["log"] then
        require "logger"
        UI::Logger.new(::Logger.new(o["log"]))
      else
        the_shell = (options["no-color"] ? Thor::Shell::Basic.new : shell)
        UI::Shell.new(the_shell)
      end

      Saber.ui.debug! if o["verbose"]

      # Initialize environment in first time
      unless Rc.p.home.exists?
        Pa.mkdir Rc.p.home
      end
    end

    desc "clean", "clean up files doesn't in rtorrent client"
    def clean
      require "saber/task/clean"

      Task["clean"].invoke(:clean, [], o)
    end

    desc "server", "start saber-server daemon"
    def server
      AutoFetcher::Server.start
    end

    desc "client", "start saber-client daemon"
    def client
      AutoFetcher::Client.start
    end

    desc "drb_add [options] <id,...> <label>", "add a file to saber-server daemon via drb."
    def drb_add(ids_str, label="")
      return if label != "saber"

      require "drb"
      require "retort"
      Retort::Service.configure { |c| c.url = Rc.scgi_server }

      names = ids_str.split(",").map{|v| Retort::Torrent.action("name", v)}

      AutoFetcher::DRbClient.new.add(*names)
    end

    desc "upload [options] <torrent_file/file ...>", "[make a torrent file and] upoad a torrent file to the site"
    def upload(*files)
      require "saber/task/upload"

      Task["upload"].invoke(:upload, [options["tracker"] || ENV["SABER_TRACKER"], *files], o)
    end

    desc "generate [options] <type> <filename> [arg ...]", %~generate a meta data file (alias: "g")~
    def generate(type, filename, *args)
      require "saber/task/generate"

      Task["generate"].invoke(:generate, [options["tracker"] || ENV["SABER_TRACKER"], type, filename, *args], o)
    end
    map "g" => "generate"

    desc "fetch [options] <file ..>", "fetch files from seedbox."
    def fetch(*names)
      Fetcher.new.add_names(*names)
    end

    desc "send1 <src ..> <dest>", "send files to seedbox"
    def send1(*names)
      require "saber/task/send"

      Task["send"].invoke(:send1, names, o)
    end

    desc "make [options] <file ..>", "make a torent file and send it to local and/or remote watch directory"
    def make(*files)
      require "saber/task/make"

      Task["make"].invoke(:make, [options["tracker"] || ENV["SABER_TRACKER"], *files], o)
    end
  end
end