1 files changed, 106 insertions, 0 deletions
diff --git a/src/DotNetOpenAuth.OpenId/Yadis/DiscoveryResult.cs b/src/DotNetOpenAuth.OpenId/Yadis/DiscoveryResult.cs
new file mode 100644
index 0000000..06c6fc7
--- /dev/null
+++ b/src/DotNetOpenAuth.OpenId/Yadis/DiscoveryResult.cs
@@ -0,0 +1,106 @@
+//-----------------------------------------------------------------------
+// <copyright file="DiscoveryResult.cs" company="Scott Hanselman, Andrew Arnott">
+//     Copyright (c) Scott Hanselman, Andrew Arnott. All rights reserved.
+// </copyright>
+//-----------------------------------------------------------------------
+
+namespace DotNetOpenAuth.Yadis {
+	using System;
+	using System.IO;
+	using System.Net.Mime;
+	using System.Web.UI.HtmlControls;
+	using System.Xml;
+	using DotNetOpenAuth.Messaging;
+
+	/// <summary>
+	/// Contains the result of YADIS discovery.
+	/// </summary>
+	internal class DiscoveryResult {
+		/// <summary>
+		/// The original web response, backed up here if the final web response is the preferred response to use
+		/// in case it turns out to not work out.
+		/// </summary>
+		private CachedDirectWebResponse htmlFallback;
+
+		/// <summary>
+		/// Initializes a new instance of the <see cref="DiscoveryResult"/> class.
+		/// </summary>
+		/// <param name="requestUri">The user-supplied identifier.</param>
+		/// <param name="initialResponse">The initial response.</param>
+		/// <param name="finalResponse">The final response.</param>
+		public DiscoveryResult(Uri requestUri, CachedDirectWebResponse initialResponse, CachedDirectWebResponse finalResponse) {
+			this.RequestUri = requestUri;
+			this.NormalizedUri = initialResponse.FinalUri;
+			if (finalResponse == null || finalResponse.Status != System.Net.HttpStatusCode.OK) {
+				this.ApplyHtmlResponse(initialResponse);
+			} else {
+				this.ContentType = finalResponse.ContentType;
+				this.ResponseText = finalResponse.GetResponseString();
+				this.IsXrds = true;
+				if (initialResponse != finalResponse) {
+					this.YadisLocation = finalResponse.RequestUri;
+				}
+
+				// Back up the initial HTML response in case the XRDS is not useful.
+				this.htmlFallback = initialResponse;
+			}
+		}
+
+		/// <summary>
+		/// Gets the URI of the original YADIS discovery request.  
+		/// This is the user supplied Identifier as given in the original
+		/// YADIS discovery request.
+		/// </summary>
+		public Uri RequestUri { get; private set; }
+
+		/// <summary>
+		/// Gets the fully resolved (after redirects) URL of the user supplied Identifier.
+		/// This becomes the ClaimedIdentifier.
+		/// </summary>
+		public Uri NormalizedUri { get; private set; }
+
+		/// <summary>
+		/// Gets the location the XRDS document was downloaded from, if different
+		/// from the user supplied Identifier.
+		/// </summary>
+		public Uri YadisLocation { get; private set; }
+
+		/// <summary>
+		/// Gets the Content-Type associated with the <see cref="ResponseText"/>.
+		/// </summary>
+		public ContentType ContentType { get; private set; }
+
+		/// <summary>
+		/// Gets the text in the final response.
+		/// This may be an XRDS document or it may be an HTML document, 
+		/// as determined by the <see cref="IsXrds"/> property.
+		/// </summary>
+		public string ResponseText { get; private set; }
+
+		/// <summary>
+		/// Gets a value indicating whether the <see cref="ResponseText"/> 
+		/// represents an XRDS document. False if the response is an HTML document.
+		/// </summary>
+		public bool IsXrds { get; private set; }
+
+		/// <summary>
+		/// Reverts to the HTML response after the XRDS response didn't work out.
+		/// </summary>
+		internal void TryRevertToHtmlResponse() {
+			if (this.htmlFallback != null) {
+				this.ApplyHtmlResponse(this.htmlFallback);
+				this.htmlFallback = null;
+			}
+		}
+
+		/// <summary>
+		/// Applies the HTML response to the object.
+		/// </summary>
+		/// <param name="initialResponse">The initial response.</param>
+		private void ApplyHtmlResponse(CachedDirectWebResponse initialResponse) {
+			this.ContentType = initialResponse.ContentType;
+			this.ResponseText = initialResponse.GetResponseString();
+			this.IsXrds = this.ContentType != null && this.ContentType.MediaType == ContentTypes.Xrds;
+		}
+	}
+}