Moved all the project template logic that would be common between MVC and web forms web sites into its own library.

1 files changed, 150 insertions, 0 deletions

diff --git a/projecttemplates/RelyingPartyLogic/Database.cs b/projecttemplates/RelyingPartyLogic/Database.cs
new file mode 100644
index 0000000..a1e17a6
--- /dev/null
+++ b/projecttemplates/RelyingPartyLogic/Database.cs
@@ -0,0 +1,150 @@
+//-----------------------------------------------------------------------
+// <copyright file="Database.cs" company="Andrew Arnott">
+//     Copyright (c) Andrew Arnott. All rights reserved.
+// </copyright>
+//-----------------------------------------------------------------------
+
+namespace RelyingPartyLogic {
+	using System;
+	using System.Collections.Generic;
+	using System.Data;
+	using System.Data.SqlClient;
+	using System.Linq;
+	using System.ServiceModel;
+	using System.Text;
+	using System.Web;
+
+	public class Database : IHttpModule, IDisposable {
+		private const string DataContextKey = "DataContext";
+
+		private const string DataContextTransactionKey = "DataContextTransaction";
+
+		/// <summary>
+		/// Initializes a new instance of the <see cref="Database"/> class.
+		/// </summary>
+		public Database() {
+		}
+
+		public static User LoggedInUser {
+			get { return DataContext.AuthenticationToken.Where(token => token.ClaimedIdentifier == HttpContext.Current.User.Identity.Name).Select(token => token.User).FirstOrDefault(); }
+		}
+
+		/// <summary>
+		/// Gets the transaction-protected database connection for the current request.
+		/// </summary>
+		public static DatabaseEntities DataContext {
+			get {
+				DatabaseEntities dataContext = DataContextSimple;
+				if (dataContext == null) {
+					dataContext = new DatabaseEntities();
+					try {
+						dataContext.Connection.Open();
+					} catch (EntityException entityEx) {
+						var sqlEx = entityEx.InnerException as SqlException;
+						if (sqlEx != null) {
+							if (sqlEx.Class == 14 && sqlEx.Number == 15350) {
+								// Most likely the database schema hasn't been created yet.
+								HttpContext.Current.Response.Redirect("~/Setup.aspx");
+							}
+						}
+
+						throw;
+					}
+
+					DataContextTransactionSimple = dataContext.Connection.BeginTransaction();
+					DataContextSimple = dataContext;
+				}
+
+				return dataContext;
+			}
+		}
+
+		private static DatabaseEntities DataContextSimple {
+			get {
+				if (HttpContext.Current != null) {
+					return HttpContext.Current.Items[DataContextKey] as DatabaseEntities;
+				} else if (OperationContext.Current != null) {
+					object data;
+					if (OperationContext.Current.IncomingMessageProperties.TryGetValue(DataContextKey, out data)) {
+						return data as DatabaseEntities;
+					} else {
+						return null;
+					}
+				} else {
+					throw new InvalidOperationException();
+				}
+			}
+
+			set {
+				if (HttpContext.Current != null) {
+					HttpContext.Current.Items[DataContextKey] = value;
+				} else if (OperationContext.Current != null) {
+					OperationContext.Current.IncomingMessageProperties[DataContextKey] = value;
+				} else {
+					throw new InvalidOperationException();
+				}
+			}
+		}
+
+		private static IDbTransaction DataContextTransactionSimple {
+			get {
+				if (HttpContext.Current != null) {
+					return HttpContext.Current.Items[DataContextTransactionKey] as IDbTransaction;
+				} else if (OperationContext.Current != null) {
+					object data;
+					if (OperationContext.Current.IncomingMessageProperties.TryGetValue(DataContextTransactionKey, out data)) {
+						return data as IDbTransaction;
+					} else {
+						return null;
+					}
+				} else {
+					throw new InvalidOperationException();
+				}
+			}
+
+			set {
+				if (HttpContext.Current != null) {
+					HttpContext.Current.Items[DataContextTransactionKey] = value;
+				} else if (OperationContext.Current != null) {
+					OperationContext.Current.IncomingMessageProperties[DataContextTransactionKey] = value;
+				} else {
+					throw new InvalidOperationException();
+				}
+			}
+		}
+
+		public void Dispose() {
+		}
+
+		void IHttpModule.Init(HttpApplication context) {
+			context.EndRequest += this.Application_EndRequest;
+			context.Error += this.Application_Error;
+		}
+
+		protected void Application_EndRequest(object sender, EventArgs e) {
+			CommitAndCloseDatabaseIfNecessary();
+		}
+
+		protected void Application_Error(object sender, EventArgs e) {
+			if (DataContextTransactionSimple != null) {
+				DataContextTransactionSimple.Rollback();
+				DataContextTransactionSimple.Dispose();
+				DataContextTransactionSimple = null;
+			}
+		}
+
+		private static void CommitAndCloseDatabaseIfNecessary() {
+			var dataContext = DataContextSimple;
+			if (dataContext != null) {
+				dataContext.SaveChanges();
+				if (DataContextTransactionSimple != null) {
+					DataContextTransactionSimple.Commit();
+					DataContextTransactionSimple.Dispose();
+				}
+
+				dataContext.Dispose();
+				DataContextSimple = null;
+			}
+		}
+	}
+}