#!/usr/bin/env ruby

require 'rmega'
require 'rmega/cli'

include Rmega::CLI::Helpers

if ARGV.empty?
  ARGV << '--help'
else
  cli_options[:url] = ARGV[0]
end

OptionParser.new do |opts|
  opts.banner = "Usage:\n"
  opts.banner << "\t#{File.basename(__FILE__)} url [options]\n"
  opts.banner << "\t#{File.basename(__FILE__)} path [options]\n"
  opts.banner << "Examples:\n"
  opts.banner << "\t#{File.basename(__FILE__)} 'https://mega.nz/#!aBkHBKLX!n4kqzbJooqcPZjN1tEJzQ4QQwskHf7YqKa'\n"
  opts.banner << "\t#{File.basename(__FILE__)} https://mega.nz/file/aBkHBKLX#n4kqzbJooqcPZjN1tEJzQ4QQwskHf7YqKa\n"
  opts.banner << "\t#{File.basename(__FILE__)} https://mega.nz/folder/aBkHBKLX#n4kqzbJooqcPZjN1tEJzQ4QQwskHf7YqKa\n"
  opts.banner << "\t#{File.basename(__FILE__)} https://www.reddit.com/r/megalinks3\n"
  opts.banner << "\t#{File.basename(__FILE__)} mylinks.txt\n"
  opts.banner << "\t#{File.basename(__FILE__)} /remote/docs/myfile.txt -u email@localhost\n"
  opts.banner << "Options:"

  opts.on("-o PATH", "--output", "Local destination path") { |path|
    cli_options[:output] = path
  }

  apply_opt_parser_options(opts)
end.parse!

cli_rescue do
  if cli_options[:user]
    session = Rmega::Session.new.login(cli_options[:user], cli_options[:pass] || cli_prompt_password)
    root = session.storage.root
    node = traverse_storage(root, cli_options[:url].dup)
    raise("Node not found - #{cli_options[:url]}") unless node
    node.download(cli_options[:output] || Dir.pwd)
  else
    urls = []

    if mega_url?(cli_options[:url])
      # a valid MEGA urls
      urls = [cli_options[:url]]
    else
      # A text file with a list of MEGA urls (considering only files < 1 Mb)
      if File.exist?(cli_options[:url])
        if File.size(cli_options[:url]) < 1_000_000
          File.open(cli_options[:url], "rb") do |file|
            file.each_line do |line|
              line.strip!
              urls << line if mega_url?(line)
            end
          end
        end
      else
        # A link to a web page with some MEGA urls in its body
        html = Rmega::Session.new.http_get_content(cli_options[:url])

        urls = html.scan(Rmega::Nodes::Factory::URL_REGEXP)
        urls << html.scan(Rmega::Nodes::Factory::URL_REGEXP_NEW)
        urls = urls.flatten.select { |u| Rmega::Nodes::Factory.url?(u) }.uniq

        if cli_options[:debug] and urls.any?
          Rmega.logger.debug("#{urls.size} URL(s) founded:\n" + urls.join("\n"))
        end
      end
    end

    raise("Nothing to download") if urls.empty?

    urls.each_with_index do |url, index|
      node = Rmega::Nodes::Factory.build_from_url(url)
      puts "[#{index+1}/#{urls.size}] #{url}" if urls.size > 1
      node.download(cli_options[:output] || Dir.pwd)
    end
  end
end