require 'vmail/version'
require 'vmail/options'
require 'vmail/query'
require 'vmail/message_formatter'
require 'versionomy'

module Vmail
  extend self

  BUFFER_FILE = 'vmailbuffer'

  def start
    puts "Starting vmail #{ Vmail::VERSION }"
    check_ruby_version

    set_vmail_browser
    check_html_reader
    change_directory_to_vmail_home

    logfile = (vim == 'mvim' || vim == 'gvim') ? STDERR : 'vmail.log'
    config.merge! 'logfile' => logfile

    puts "Starting vmail imap client for #{ config['username'] }"

    set_inbox_poller

    puts "Working directory: #{ Dir.pwd }"

    server = start_imap_daemon
    mailbox, query_string = select_mailbox(server)

    start_vim(mailbox, query_string)

    if vim == 'mvim' || vim == 'gvim'
      DRb.thread.join
    end

    close_connection
  end

  private

  def options
    @options ||= Vmail::Options.new(ARGV)
  end

  def config
    @config ||= options.config
  end

  def has_clientserver?
    @has_clientserver ||= system("#{ vim } --version | grep +clientserver")
  end

  def change_directory_to_vmail_home
    working_dir = ENV['VMAIL_HOME'] || "#{ ENV['HOME'] }/.vmail/default"
    `mkdir -p #{ working_dir }`
    puts "Changing working directory to #{ working_dir }"
    Dir.chdir(working_dir)
  end

  def check_html_reader
    return if ENV['VMAIL_HTML_PART_READER']
    html_reader = %w( w3m elinks lynx ).detect {|x| `which #{ x }` != ''}
    if html_reader
      cmd = ['w3m -dump -T text/html -I utf-8 -O utf-8', 'lynx -stdin -dump', 'elinks -dump'].detect {|s| s.index(html_reader)}
      STDERR.puts "Setting VMAIL_HTML_PART_READER to '#{ cmd }'"
      ENV['VMAIL_HTML_PART_READER'] = cmd
    else
      abort "You need to install w3m, elinks, or lynx on your system in order to see html-only messages"
    end
  end

  def parse_query
    if ARGV[0] =~ /^\d+/
      ARGV.shift
    end
    mailbox = ARGV.shift || 'INBOX'
    query = Vmail::Query.parse(ARGV)
    [mailbox, query]
  end

  def select_mailbox(server)
    mailbox, query = parse_query
    query_string = Vmail::Query.args2string query
    server.select_mailbox mailbox

    STDERR.puts "Mailbox: #{ mailbox }"
    STDERR.puts "Query: #{ query.inspect }"
    STDERR.puts "Query String: #{ String.shellescape(query_string) }"

    [mailbox, query_string]
  end

  def check_ruby_version
    required_version = Versionomy::create(:major => 1, :minor => 9, :tiny => 0)
    ruby_version = Versionomy::parse(RUBY_VERSION)

    if required_version > ruby_version
      puts "This version of vmail requires Ruby version 1.9.0 or higher (1.9.2 is recommended)"
      exit
    end
  end

  def set_vmail_browser
    ENV['VMAIL_BROWSER'] ||= if RUBY_PLATFORM.downcase.include?('linux')
                               tools = ['gnome-open', 'kfmclient-exec', 'xdg-open', 'konqueror']
                               tool = tools.detect { |tool|
                                 `which #{ tool }`.size > 0
                               }
                               if tool.nil?
                                 puts "Can't find a VMAIL_BROWSER tool on your system. Please report this issue."
                               else
                                 tool
                               end
                             else
                               'open'
                             end

    puts "Setting VMAIL_BROWSER to '#{ ENV['VMAIL_BROWSER'] }'"
  end

  def set_inbox_poller
    if config['polling'] == true
      require 'vmail/inbox_poller'
      inbox_poller = Vmail::InboxPoller.start config
      Thread.new do
        inbox_poller.start_polling
      end
    else
      puts "INBOX polling disabled."
    end
  end

  def start_imap_daemon
    # require after the working dir is set
    require 'vmail/imap_client'

    $drb_uri = begin
                Vmail::ImapClient.daemon config
              rescue
                puts "Failure:", $!
                exit(1)
              end

    DRbObject.new_with_uri $drb_uri
  end

  def start_vim(mailbox, query_string)
    vimscript = File.expand_path("../vmail.vim", __FILE__)
    vimopts = config['vim_opts']
    contacts_file = options.contacts_file

    vim_options = {
      'DRB_URI' => $drb_uri,
      'VMAIL_CONTACTS_FILE' => contacts_file,
      'VMAIL_MAILBOX' => String.shellescape(mailbox),
      'VMAIL_QUERY' => %("#{ query_string }")
    }

    vim_command = "#{ vim } #{ server_opts } -S #{ vimscript } -c '#{ vimopts }' #{ BUFFER_FILE }"

    STDERR.puts vim_options
    STDERR.puts vim_command
    STDERR.puts "Using buffer file: #{ BUFFER_FILE }"
    File.open(BUFFER_FILE, "w") do |file|
      file.puts "\n\nVmail #{ Vmail::VERSION }\n\n"
      file.puts "Please wait while I fetch your messages.\n\n\n"
    end

    system(vim_options, vim_command)
  end

  def close_connection
    File.delete(BUFFER_FILE)

    STDERR.puts "Closing imap connection"
    begin
      Timeout::timeout(10) do
        $gmail.close
      end
    rescue Timeout::Error
      puts "Close connection attempt timed out"
    end
    puts "Bye"
    exit
  end

  def vim
    ENV['VMAIL_VIM'] || 'vim'
  end

  def server_opts
    return unless has_clientserver?
    server_name = "VMAIL:#{ config['username'] }"
    "--servername #{ server_name }"
  end
end

if __FILE__ == $0
  Vmail.start
end