Subversion Repositories general

package ak.hostadmiral.core.model;

import java.util.*;

import net.sf.hibernate.*;

import net.sf.hibernate.type.Type;

import ak.hostadmiral.util.CollectionInfo;

import ak.hostadmiral.util.HibernateUtil;

import ak.hostadmiral.util.ModelException;

import ak.hostadmiral.util.ModelSecurityException;

public class MailboxManager

	implements

		UserBeforeDeleteListener,

		UserDeletingListener,

		SystemUserBeforeDeleteListener,

		SystemUserDeletingListener,

		InetDomainBeforeDeleteListener,

		InetDomainDeletingListener

{

	private static MailboxManager mailboxManager = null;

	private static boolean registered = false;

	public static MailboxManager getInstance()

	{

		return mailboxManager;

	}

	protected static void register()

	{

		synchronized(MailboxManager.class) {

			if(registered) return;

			registered = true;

			try {

				HibernateUtil.getConfiguration().addResource(

					"ak/hostadmiral/core/model/Mailbox.hbm.xml");

				mailboxManager = new MailboxManager();

			}

			catch(Exception ex) {

				ex.printStackTrace();

				throw new RuntimeException(ex.getMessage());

			}

		}

	}

	static {

		register();

	}

	private Collection createdListeners      = new ArrayList();

	private Collection modifiedListeners     = new ArrayList();

	private Collection beforeDeleteListeners = new ArrayList();

	private Collection deletingListeners     = new ArrayList();

	private Collection deletedListeners      = new ArrayList();

	private MailboxManager()

	{

		UserManager.getInstance().addBeforeDeleteListener(this);

		SystemUserManager.getInstance().addBeforeDeleteListener(this);

		InetDomainManager.getInstance().addBeforeDeleteListener(this);

	}

	public Mailbox create(User editor)

		throws ModelException

	{

		if(!allowedToCreate(editor)) throw new ModelSecurityException();

		Mailbox mailbox = new Mailbox();

        // FIXME: make this configurable

		mailbox.addPasswordStore(new PasswordStoreMd5());

		mailbox.addPasswordStore(new PasswordStoreCrypt());

		mailbox.addPasswordStore(new PasswordStorePlain());

		return mailbox;

	}

	public boolean allowedToCreate(User editor)

		throws ModelException

	{

		return Mailbox.allowedToCreate(this, editor);

	}

	public Mailbox get(User editor, Long id)

		throws ModelException

	{

		Mailbox mailbox;

		try {

			mailbox = (Mailbox)HibernateUtil.currentSession().load(Mailbox.class, id);

		}

		catch(HibernateException ex)

		{

			throw new ModelException(ex);

		}

		if(!mailbox.viewableBy(editor))

			throw new ModelSecurityException();

		return mailbox;

	}

	public boolean loginExists(User editor, Mailbox mailbox, String login)

		throws ModelException

	{

		if(mailbox.getDomain() == null)

			throw new ModelException("Cannot check unique login for mailbox without domain");

		try {

			if(mailbox.getId() == null)

				return ((Integer)HibernateUtil.currentSession().iterate(

					"select count(*) from Mailbox where login = ? and domain = ?",

					new Object[] { login, mailbox.getDomain() },

					new Type[] { Hibernate.STRING, Hibernate.entity(InetDomain.class) } )

					.next()).intValue() > 0;

			else

				return ((Integer)HibernateUtil.currentSession().iterate(

					"select count(*) from Mailbox b where login = ? and domain = ? and b != ?",

					new Object[] { login, mailbox.getDomain(), mailbox },

					new Type[] { Hibernate.STRING, Hibernate.entity(InetDomain.class),

						Hibernate.entity(Mailbox.class) } )

					.next()).intValue() > 0;

		}

		catch(HibernateException ex)

		{

			throw new ModelException(ex);

		}

	}

	protected Mailbox findForLogin(String login)

		throws ModelException

	{

		try {

			List list = HibernateUtil.currentSession().find(

				"select mb from Mailbox mb left join fetch mb.domain as d"

				+ " left join fetch mb.owner left join fetch mb.systemUser where mb.login=?",

				login, Hibernate.STRING);

			if(list.size() == 0)

				return null;

			else

				return (Mailbox)list.get(0);

		}

		catch(HibernateException ex)

		{

			throw new ModelException(ex);

		}

	}

	public void save(User editor, Mailbox mailbox)

		throws ModelException

	{

		if(!mailbox.editableBy(editor))

			throw new ModelSecurityException();

        boolean isNew = mailbox.isNew();

		//mailbox.setModUser(editor); // FIXME

		try {

			HibernateUtil.currentSession().saveOrUpdate(mailbox);

		}

		catch(HibernateException ex)

		{

			throw new ModelException(ex);

		}

        // inform listeners

        if(isNew) {

        	for(Iterator i = createdListeners.iterator(); i.hasNext(); ) {

        		MailboxCreatedListener listener = (MailboxCreatedListener)i.next();

    			listener.mailboxCreated(editor, mailbox);

        	}

        }

        else {

            Mailbox oldMailbox = mailbox.getOrigin();

            if(oldMailbox == null) oldMailbox = mailbox;

        	for(Iterator i = modifiedListeners.iterator(); i.hasNext(); ) {

        		MailboxModifiedListener listener = (MailboxModifiedListener)i.next();

    			listener.mailboxModified(editor, mailbox, oldMailbox);

        	}

        }

	}

    public void addCreatedListener(MailboxCreatedListener listener)

    {

    	createdListeners.add(listener);

    }

    public void removeCreatedListener(MailboxCreatedListener listener)

    {

    	createdListeners.remove(listener);

    }

    public void addModifiedListener(MailboxModifiedListener listener)

    {

    	modifiedListeners.add(listener);

    }

    public void removeModifiedListener(MailboxModifiedListener listener)

    {

    	modifiedListeners.remove(listener);

    }

    public void addBeforeDeleteListener(MailboxBeforeDeleteListener listener)

    {

    	beforeDeleteListeners.add(listener);

    }

    public void removeBeforeDeleteListener(MailboxBeforeDeleteListener listener)

    {

    	beforeDeleteListeners.remove(listener);

    }

    public void addDeletingListener(MailboxDeletingListener listener)

    {

    	deletingListeners.add(listener);

    }

    public void removeDeletingListener(MailboxDeletingListener listener)

    {

    	deletingListeners.remove(listener);

    }

    public void addDeletedListener(MailboxDeletedListener listener)

    {

    	deletedListeners.add(listener);

    }

    public void removeDeletedListener(MailboxDeletedListener listener)

    {

    	deletedListeners.remove(listener);

    }

    public Collection beforeDelete(User editor, Mailbox mailbox, Collection known)

		throws ModelException

    {

    	Collection cascade = new ArrayList();

    	for(Iterator i = beforeDeleteListeners.iterator(); i.hasNext(); ) {

    		MailboxBeforeDeleteListener listener = (MailboxBeforeDeleteListener)i.next();

			Collection subcascade = listener.mailboxBeforeDelete(editor, mailbox, known);

    		if(subcascade != null)

    			cascade.addAll(subcascade);

    	}

    	return cascade;

    }

	public void delete(User editor, Mailbox mailbox)

		throws ModelException

	{

	    // check rights

		if(!mailbox.deleteableBy(editor))

			throw new ModelSecurityException();

        // inform deleting listeners

    	for(Iterator i = deletingListeners.iterator(); i.hasNext(); ) {

    		MailboxDeletingListener listener = (MailboxDeletingListener)i.next();

			listener.mailboxDeleting(editor, mailbox);

    	}

        // backup copy

        Mailbox oldMailbox = new Mailbox(mailbox);

        // delete it

		try {

			HibernateUtil.currentSession().delete(mailbox);

		}

		catch(HibernateException ex)

		{

			throw new ModelException(ex);

		}

        // inform deleted listeners

    	for(Iterator i = deletedListeners.iterator(); i.hasNext(); ) {

    		MailboxDeletedListener listener = (MailboxDeletedListener)i.next();

			listener.mailboxDeleted(editor, oldMailbox);

    	}

	}

	public Collection listMailboxes(User editor)

		throws ModelException

	{

		return listMailboxes(null, 0, 0, null, editor);

	}

	public Collection listMailboxes(CollectionInfo info, int pageSize, int pageNumber,

			Integer[] sortingKeys, User editor)

		throws ModelException

	{

		try {

			if(editor.isSuperuser()) {

				if(info != null) {

					info.setSize(((Integer)HibernateUtil.currentSession().iterate(

						"select count(*) from Mailbox").next()).intValue());

				}

				return HibernateUtil.pageableList(pageSize, pageNumber,

					"select {mb.*}, {d.*}, {o.*}, {su.*}"

					+ " from      mailboxes   as mb"

					+ " left join domains     as d  on mb.domain     = d.id"

					+ " left join users       as o  on mb.owner      = o.id"

					+ " left join systemusers as su on mb.systemUser = su.id"

					+ HibernateUtil.formOrderClause(sortingKeys, sortKeys),

					Mailbox.class,

					new String[] { "mb", "d", "o", "su" },

					new Class[] { Mailbox.class, InetDomain.class, User.class, SystemUser.class },

					null,

					null);

			}

			else {

				if(info != null) {

					List countlist = HibernateUtil.sqlQuery(

						"select count(*) from ("

						+ " select mb.id from mailboxes mb"

						+ "   where mb.owner=?"

						+ " union"

						+ " select mb.id from mailboxes mb"

						+ "   left join domains as d on mb.domain = d.id"

						+ "   where d.owner=?"

						+ ") as count_table",

						new Object[] { editor.getId(), editor.getId() });

					info.setSize(((Long)countlist.get(0)).intValue());

				}

				return HibernateUtil.pageableList(pageSize, pageNumber,

					"select {mb.*}, {d.*}, {o.*}, {su.*}"

					+ "   from      mailboxes   as mb"

					+ "   left join domains     as d  on mb.domain     = d.id"

					+ "   left join users       as o  on mb.owner      = o.id"

					+ "   left join systemusers as su on mb.systemUser = su.id"

					+ " where mb.owner=?"

					+ " union"

					+ " select {mb.*}, {d.*}, {o.*}, {su.*}"

					+ "   from      mailboxes   as mb"

					+ "   left join domains     as d  on mb.domain     = d.id"

					+ "   left join users       as o  on mb.owner      = o.id"

					+ "   left join systemusers as su on mb.systemUser = su.id"

					+ " where d.owner=?"

					+ HibernateUtil.formOrderClause(sortingKeys, sortKeys),

					Mailbox.class,

					new String[] { "mb", "d", "o", "su" },

					new Class[] { Mailbox.class, InetDomain.class, User.class, SystemUser.class },

					new Object[] { editor, editor },

					new Type[] { Hibernate.entity(User.class), Hibernate.entity(User.class) });

			}

		}

		catch(HibernateException ex)

		{

			ex.printStackTrace();

			throw new ModelException(ex);

		}

	}

	public boolean areMailboxesAvailable(User editor)

		throws ModelException

	{

		try {

			if(editor.isSuperuser()

				|| InetDomainManager.getInstance().areInetDomainsAvailable(editor))

			{

				return true;

			}

			else {

				return ((Integer)HibernateUtil.currentSession().iterate(

					"select count(*) from Mailbox mb left join mb.domain as d"

					+ " where d.owner=? or mb.owner=?",

					new Object[] { editor, editor },

					new Type[] { Hibernate.entity(User.class), Hibernate.entity(User.class) })

					.next()).intValue() > 0;

			}

		}

		catch(HibernateException ex)

		{

			throw new ModelException(ex);

		}

	}

	public Collection userBeforeDelete(User editor, User user, Collection known)

		throws ModelException

	{

        Collection mailboxes;

		try {

			mailboxes = HibernateUtil.currentSession().find(

				"select mb from Mailbox mb left join fetch mb.domain"

				+ " left join fetch mb.systemUser where mb.owner = ?",

				user, Hibernate.entity(User.class) );

		}

		catch(HibernateException ex)

		{

			throw new ModelException(ex);

		}

		return iterateBeforeDelete(editor, mailboxes, known);

    }

	public void userDeleting(User editor, User user)

		throws ModelException

	{

        Collection mailboxes;

		try {

			mailboxes = HibernateUtil.currentSession().find(

				"select mb from Mailbox mb left join fetch mb.domain"

				+ " left join fetch mb.systemUser where mb.owner = ?",

				user, Hibernate.entity(User.class) );

		}

		catch(HibernateException ex)

		{

			throw new ModelException(ex);

		}

		for(Iterator i = mailboxes.iterator(); i.hasNext(); ) {

			delete(editor, (Mailbox)i.next());

		}

    }

	public Collection inetDomainBeforeDelete(User editor, InetDomain domain, Collection known)

		throws ModelException

	{

        Collection mailboxes;

		try {

			mailboxes = HibernateUtil.currentSession().find(

				"select mb from Mailbox mb left join fetch mb.owner"

				+ " left join fetch mb.systemUser where mb.domain = ?",

				domain, Hibernate.entity(InetDomain.class) );

		}

		catch(HibernateException ex)

		{

			throw new ModelException(ex);

		}

		return iterateBeforeDelete(editor, mailboxes, known);

    }

	public void inetDomainDeleting(User editor, InetDomain domain)

		throws ModelException

	{

        Collection mailboxes;

		try {

			mailboxes = HibernateUtil.currentSession().find(

				"select mb from Mailbox mb left join fetch mb.owner"

				+ " left join fetch mb.systemUser where mb.domain = ?",

				domain, Hibernate.entity(InetDomain.class) );

		}

		catch(HibernateException ex)

		{

			throw new ModelException(ex);

		}

		for(Iterator i = mailboxes.iterator(); i.hasNext(); ) {

			delete(editor, (Mailbox)i.next());

		}

    }

	public Collection systemUserBeforeDelete(User editor, SystemUser user, Collection known)

		throws ModelException

	{

        Collection mailboxes;

		try {

			mailboxes = HibernateUtil.currentSession().find(

				"select mb from Mailbox mb left join fetch mb.domain"

				+ " left join fetch mb.owner where mb.systemUser = ?",

				user, Hibernate.entity(SystemUser.class) );

		}

		catch(HibernateException ex)

		{

			throw new ModelException(ex);

		}

		return iterateBeforeDelete(editor, mailboxes, known);

    }

	public void systemUserDeleting(User editor, SystemUser user)

		throws ModelException

	{

        Collection mailboxes;

		try {

			mailboxes = HibernateUtil.currentSession().find(

				"select mb from Mailbox mb left join fetch mb.domain"

				+ " left join fetch mb.owner where mb.systemUser = ?",

				user, Hibernate.entity(SystemUser.class) );

		}

		catch(HibernateException ex)

		{

			throw new ModelException(ex);

		}

		for(Iterator i = mailboxes.iterator(); i.hasNext(); ) {

			delete(editor, (Mailbox)i.next());

		}

    }

	private Collection iterateBeforeDelete(User editor, Collection mailboxes, Collection known)

		throws ModelException

	{

    	Collection cascade = new ArrayList();

		for(Iterator i = mailboxes.iterator(); i.hasNext(); ) {

			Mailbox mailbox = (Mailbox)i.next();

            if(mailbox.viewableBy(editor)) {

				if(mailbox.deleteableBy(editor))

					cascade.add(new CascadeDeleteElement(mailbox, CascadeDeleteElement.DELETE,

						this.beforeDelete(editor, mailbox, known)));

				else

					cascade.add(new CascadeDeleteElement(mailbox, CascadeDeleteElement.FORBIDDEN,

						null));

			}

			else {

				cascade.add(new CascadeDeleteElement(Mailbox.createLimitedCopy(mailbox),

					CascadeDeleteElement.FORBIDDEN, null));

			}

		}

    	return cascade;

	}

	public static final Integer SORT_LOGIN  = new Integer(1);

	public static final Integer SORT_DOMAIN = new Integer(2);

	protected static Map sortKeys = new HashMap();

	static {

		sortKeys.put(SORT_LOGIN,  "login0_");

		sortKeys.put(SORT_DOMAIN, "name1_");

	}

	public static final Comparator LOGIN_COMPARATOR  = new LoginComparator();

	public static final Comparator DOMAIN_COMPARATOR = new DomainComparator();

	private static class LoginComparator

		implements Comparator

	{

		public int compare(Object o1, Object o2)

		{

			if(!(o1 instanceof Mailbox) || !(o2 instanceof Mailbox))

				throw new ClassCastException("not a Mailbox");

		    Mailbox a1 = (Mailbox)o1;

		    Mailbox a2 = (Mailbox)o2;

		    if(a1 == null && a2 == null)

		    	return 0;

		    else if(a1 == null && a2 != null)

		    	return -1;

		    else if(a1 != null && a2 == null)

		    	return 1;

		    else

		    	return a1.getLogin().compareToIgnoreCase(a2.getLogin());

		}

		public boolean equals(Object obj)

		{

			return (obj instanceof LoginComparator);

		}

	}

	private static class DomainComparator

		implements Comparator

	{

		public int compare(Object o1, Object o2)

		{

			if(!(o1 instanceof Mailbox) || !(o2 instanceof Mailbox))

				throw new ClassCastException("not a Mailbox");

		    Mailbox a1 = (Mailbox)o1;

		    Mailbox a2 = (Mailbox)o2;

		    if(a1 == null && a2 == null)

		    	return 0;

		    else if(a1 == null && a2 != null)

		    	return -1;

		    else if(a1 != null && a2 == null)

		    	return 1;

		    else

		    	return a1.getDomain().getName().compareToIgnoreCase(a2.getDomain().getName());

		}

		public boolean equals(Object obj)

		{

			return (obj instanceof DomainComparator);

		}

	}

}