from imaplib import *
import sys

"""
this is a small python script to list your imap folders in mutt
usage:

	mailboxes `python ~/.mutt/list.py mail "ssh -q foo.com /usr/sbin/imapd"`

	or

	mailboxes `python ~/.mutt/list.py bar.com nick@bar.com secret "" [--subscr]`

using --subscr, only subscribed folders are listed.
"""

class ImapLister:
	def __init__(self, server = None, user = None, pwd = None, dir = None,
			tunnel = None, subscr = False):
		self.dirs = []

		if not tunnel:
			sock = IMAP4_SSL(server)
			sock.login(user, pwd)
		else:
			sock = IMAP4_stream(tunnel)
		if subscr:
			self.dirs.append("INBOX")
			mylist = sock.lsub
		else:
			mylist = sock.list
		for i in mylist(dir)[1]:
			if i.split(" ")[-1] == dir:
				i = "INBOX"
			self.dirs.append(i.split(" ")[-1])
		self.dirs.sort()

if __name__ == "__main__":
	if len(sys.argv) == 3:
		server = ImapLister(dir = sys.argv[1], tunnel = sys.argv[2])
	else:
		# we have two more parameter for real imap servers
		if len(sys.argv) > 5:
			subscr = True
		else:
			subscr = False
		server = ImapLister(server = sys.argv[1], user = sys.argv[2],
				pwd = sys.argv[3], dir = sys.argv[4], subscr=subscr)
	for i in server.dirs:
		sys.stdout.write("imaps://%s/%s " % (sys.argv[1], i))