summaryrefslogtreecommitdiffstats
path: root/samples/OAuthServiceProvider/Code/CustomOAuthTypeProvider.cs
blob: 67da17cbe634d3fe1d8233c710be7c715bb55225 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
namespace OAuthServiceProvider.Code {
	using System;
	using System.Collections.Generic;
	using System.Linq;
	using System.Web;
	using DotNetOpenAuth.Messaging;
	using DotNetOpenAuth.OAuth.ChannelElements;
	using DotNetOpenAuth.OAuth.Messages;

	/// <summary>
	/// A custom class that will cause the OAuth library to use our custom message types
	/// where we have them.
	/// </summary>
	public class CustomOAuthMessageFactory : OAuthServiceProviderMessageFactory {
		/// <summary>
		/// Initializes a new instance of the <see cref="CustomOAuthMessageFactory"/> class.
		/// </summary>
		/// <param name="tokenManager">The token manager instance to use.</param>
		public CustomOAuthMessageFactory(IServiceProviderTokenManager tokenManager)
			: base(tokenManager) {
		}

		public override IDirectedProtocolMessage GetNewRequestMessage(MessageReceivingEndpoint recipient, IDictionary<string, string> fields) {
			var message = base.GetNewRequestMessage(recipient, fields);

			// inject our own type here to replace the standard one
			if (message is UnauthorizedTokenRequest) {
				message = new RequestScopedTokenMessage(recipient, message.Version);
			}

			return message;
		}
	}
}