using System.Web.Mvc;
using NHibernate;
using NHibernate.Context;

namespace Agendas.NHibernate
{
	public class NHibernateSessionPerActionAttribute : ActionFilterAttribute
	{
		private readonly ISessionFactory _sessionFactory;

		public NHibernateSessionPerActionAttribute(ISessionFactory sessionFactory)
		{
			_sessionFactory = sessionFactory;
		}

		public override void OnActionExecuting(ActionExecutingContext filterContext)
		{
			var session = _sessionFactory.OpenSession();
			CurrentSessionContext.Bind(session);

			base.OnActionExecuting(filterContext);
		}

		public override void OnResultExecuted(ResultExecutedContext filterContext)
		{
			base.OnResultExecuted(filterContext);

			var session = _sessionFactory.GetCurrentSession();

			session.Flush();
			session.Close();

		}
	}
}