require 'thread'

module Redwood

class PollManager
  include Singleton

  DELAY = 300

  def initialize
    @mutex = Mutex.new
    @last_poll = nil
    
    self.class.i_am_the_instance self
  end

  def buffer
    BufferManager.spawn_unless_exists("<poll for new messages>", :hidden => true) { PollMode.new }
  end

  def poll
    BufferManager.flash "Polling for new messages..."
    num, numi = buffer.mode.poll
    if num > 0
      BufferManager.flash "Loaded #{num} new messages, #{numi} to inbox." 
    else
      BufferManager.flash "No new messages." 
    end
    [num, numi]
  end

  def start_thread
    Redwood::reporting_thread do
      while true
        sleep DELAY / 2
        poll if @last_poll.nil? || (Time.now - @last_poll) >= DELAY
      end
    end
  end

  def do_poll
    total_num = total_numi = 0
    @mutex.synchronize do
      Index.usual_sources.each do |source|
#        yield "source #{source} is done? #{source.done?} (cur_offset #{source.cur_offset} >= #{source.end_offset})"
        yield "Loading from #{source}... " unless source.done? || source.broken?
        num = 0
        numi = 0
        add_new_messages_from source do |m, offset, entry|
          ## always preserve the labels on disk.
          m.labels = entry[:label].split(/\s+/).map { |x| x.intern } if entry
          yield "Found message at #{offset} with labels {#{m.labels * ', '}}"
          unless entry
            num += 1
            numi += 1 if m.labels.include? :inbox
          end
          m
        end
        yield "Found #{num} messages, #{numi} to inbox" unless num == 0
        total_num += num
        total_numi += numi
      end

      yield "Done polling; loaded #{total_num} new messages total"
      @last_poll = Time.now
      @polling = false
    end
    [total_num, total_numi]
  end

  ## this is the main mechanism for adding new messages to the
  ## index. it's called both by sup-import and by PollMode.
  ##
  ## for each new message in the source, this yields the message, the
  ## source offset, and the index entry on disk (if any). it expects
  ## the yield to return the message (possibly altered in some way),
  ## and then adds it (if new) or updates it (if previously seen).
  ##
  ## the labels of the yielded message are the source labels. it is
  ## likely that callers will want to replace these with the index
  ## labels, if they exist, so that state is not lost when e.g. a new
  ## version of a message from a mailing list comes in.
  def add_new_messages_from source
    return if source.done? || source.broken?

    source.each do |offset, labels|
      if source.broken?
        Redwood::log "error loading messages from #{source}: #{source.broken_msg}"
        return
      end
      
      labels.each { |l| LabelManager << l }

      begin
        m = Message.new :source => source, :source_info => offset, :labels => labels
        if m.source_marked_read?
          m.remove_label :unread
          labels.delete :unread
        end

        docid, entry = Index.load_entry_for_id m.id
        m = yield m, offset, entry
        next unless m
        if entry
          Index.update_message m, docid, entry
        else
          Index.add_message m
          UpdateManager.relay self, :add, m
        end
      rescue MessageFormatError, SourceError => e
        Redwood::log "ignoring erroneous message at #{source}##{offset}: #{e.message}"
      end
    end
  end
end

end