X-Git-Url: http://git.code-monkey.de/?p=embrace.git;a=blobdiff_plain;f=lib%2Fembrace%2Fimap.rb;fp=lib%2Fembrace%2Fimap.rb;h=0c545e32a3ada166341bfe36b0ae52561613bc9c;hp=cb129a08dbf071ad3f351fcc2d7756b0b29021e9;hb=276ab5edad1b440db6a7791a832794342ef94fda;hpb=c0b425d6a1cd0e286d58eca4c1ed415315bd48d2 diff --git a/lib/embrace/imap.rb b/lib/embrace/imap.rb index cb129a0..0c545e3 100644 --- a/lib/embrace/imap.rb +++ b/lib/embrace/imap.rb @@ -138,15 +138,23 @@ end end def handle_tagged_response(resp, req) - return unless req.is_a?(LoginRequest) - - case resp.status - when :ok - @state = :logged_in - - @mboxes.each { |mb| query_status(mb) } - else - raise(LoginError, "cannot login - #{resp.data}") + case req + when LoginRequest + case resp.status + when :ok + @state = :logged_in + + @mboxes.each { |mb| query_status(mb) } + else + raise(LoginError, "cannot login - #{resp.data}") + end + when StatusRequest + @mboxes.delete(req.mailbox) + + if @mboxes.empty? + @status = :done + logout + end end end @@ -161,12 +169,6 @@ end name, count = md.captures.first, md.captures.last.to_i MailboxStatusEvent.raise(name, count) - @mboxes.delete(name) - - if @mboxes.empty? - @status = :done - logout - end end end @@ -208,6 +210,8 @@ end end class StatusRequest < Request + attr_reader :mailbox + def initialize(session, mailbox) super(session)