from mod_python import psp
import imaplib
import os

import webmail

def index(req):
  req.content_type = "text/html"
  passwd = req.get_basic_auth_pw()
  user = req.user
  wm = webmail.Webmail(user, passwd, req)
  
  if req.form.has_key("folder"):
    wm.set_folder(req.form['folder'])
  
  vars = {'folders': wm.get_folders(),
          'folder': wm.current_folder,
          'mainpage': wm.get_messages(),
         }

  if req.form.has_key("msg"):
    vars['message'] = wm.get_message(req.form['msg'])
  else:
    vars['message'] = ""
  
  return psp.PSP(req, "templates/main.html", vars=vars)

def inc(req):
  file_path = os.path.dirname(req.canonical_filename) + "/_inc" + req.path_info
  file_type = file_path.split(".")[-1:][0]

  if file_type == "css":
    req.content_type = "text/css"
  elif file_type == "js":
    req.content_type = "text/javascript"
  req.flush()
  req.sendfile(file_path)