first import

git-svn-id: svn+ssh://svn.gna.org/svn/feed2imap/trunk/feed2imap@5 f70e237a-67f3-0310-a06c-d2b8a7116972

1 files changed, 118 insertions, 0 deletions

diff --git a/lib/feed2imap/imap.rb b/lib/feed2imap/imap.rb
new file mode 100644
index 0000000..591f561
--- /dev/null
+++ b/lib/feed2imap/imap.rb
@@ -0,0 +1,118 @@
+=begin
+Feed2Imap - RSS/Atom Aggregator uploading to an IMAP Server
+Copyright (c) 2005 Lucas Nussbaum <lucas@lucas-nussbaum.net>
+
+This program is free software; you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation; either version 2 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program; if not, write to the Free Software
+Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+=end
+
+# Imap connection handling
+require 'net/imap'
+begin
+  require 'openssl'
+rescue
+end
+require 'uri'
+
+# This class is a container of IMAP accounts.
+# Thanks to it, accounts are re-used : several feeds
+# using the same IMAP account will create only one
+# IMAP connection.
+class ImapAccounts < Hash
+  def add_account(uri)
+    u = URI::Generic::build({ :scheme => uri.scheme,
+                              :userinfo => uri.userinfo,
+                              :host => uri.host,
+                              :port => uri.port })
+    if not include?(u)
+      ac = ImapAccount::new(u)
+      self[u] = ac
+    end
+    return self[u]
+  end
+end
+
+# This class is an IMAP account, with the given fd
+# once the connection has been established
+class ImapAccount
+  attr_reader :uri
+
+  def initialize(uri)
+    @uri = uri
+  end
+
+  # connects to the IMAP server
+  # raises an exception if it fails
+  def connect
+    port = 143
+    usessl = false
+    if uri.scheme == 'imap'
+      port = 143
+      usessl = false
+    elsif uri.scheme == 'imaps'
+      port = 993
+      usessl = true
+    else
+      raise "Unknown scheme: #{uri.scheme}"
+    end
+    # use given port if port given
+    port = uri.port if uri.port 
+    @connection = Net::IMAP::new(uri.host, port, usessl)
+    user, password = uri.userinfo.split(':',2)
+    @connection.login(user, password)
+  end
+
+  # disconnect from the IMAP server
+  def disconnect
+    @connection.disconnect if @connection
+  end
+
+  # Returns true if the folder exist
+  def folder_exist?(folder)
+    return !@connection.list('', folder).nil?
+  end
+
+  # Creates the given folder
+  def create_folder(folder)
+    @connection.create(folder)
+    @connection.subscribe(folder)
+    self
+  end
+
+  # Put the mail in the given folder
+  # You should check whether the folder exist first.
+  def putmail(folder, mail)
+    # TODO check response
+    @connection.append(folder, mail)
+  end
+
+  def updatemail(folder, mail, idx)
+    # TODO check response
+    # TODO keep flags of deleted mail
+    @connection.select(folder)
+    searchres = @connection.search(['HEADER', 'X-CacheIndex', "-#{idx}-"])
+    if searchres.length == 1
+      @connection.store(searchres[0], "+FLAGS", [:Deleted])
+      @connection.expunge
+    elsif searchres.length != 0
+      raise "Search returned multiple results !!"
+    end
+    putmail(folder, mail)
+  end
+
+  def to_s
+    uri.to_s
+  end
+end
+