This repository has been archived by the owner on Nov 20, 2018. It is now read-only.
/
RequestCookiesFeature.cs
96 lines (84 loc) · 3.27 KB
/
RequestCookiesFeature.cs
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
using System;
using System.Collections.Generic;
using Microsoft.AspNetCore.Http.Internal;
using Microsoft.Extensions.Primitives;
using Microsoft.Net.Http.Headers;
namespace Microsoft.AspNetCore.Http.Features
{
public class RequestCookiesFeature : IRequestCookiesFeature
{
// Lambda hoisted to static readonly field to improve inlining https://github.com/dotnet/roslyn/issues/13624
private readonly static Func<IFeatureCollection, IHttpRequestFeature> _nullRequestFeature = f => null;
private FeatureReferences<IHttpRequestFeature> _features;
private StringValues _original;
private IRequestCookieCollection _parsedValues;
public RequestCookiesFeature(IRequestCookieCollection cookies)
{
if (cookies == null)
{
throw new ArgumentNullException(nameof(cookies));
}
_parsedValues = cookies;
}
public RequestCookiesFeature(IFeatureCollection features)
{
if (features == null)
{
throw new ArgumentNullException(nameof(features));
}
_features = new FeatureReferences<IHttpRequestFeature>(features);
}
private IHttpRequestFeature HttpRequestFeature =>
_features.Fetch(ref _features.Cache, _nullRequestFeature);
public IRequestCookieCollection Cookies
{
get
{
if (_features.Collection == null)
{
if (_parsedValues == null)
{
_parsedValues = RequestCookieCollection.Empty;
}
return _parsedValues;
}
var headers = HttpRequestFeature.Headers;
StringValues current;
if (!headers.TryGetValue(HeaderNames.Cookie, out current))
{
current = string.Empty;
}
if (_parsedValues == null || _original != current)
{
_original = current;
_parsedValues = RequestCookieCollection.Parse(current.ToArray());
}
return _parsedValues;
}
set
{
_parsedValues = value;
_original = StringValues.Empty;
if (_features.Collection != null)
{
if (_parsedValues == null || _parsedValues.Count == 0)
{
HttpRequestFeature.Headers.Remove(HeaderNames.Cookie);
}
else
{
var headers = new List<string>();
foreach (var pair in _parsedValues)
{
headers.Add(new CookieHeaderValue(pair.Key, pair.Value).ToString());
}
_original = headers.ToArray();
HttpRequestFeature.Headers[HeaderNames.Cookie] = _original;
}
}
}
}
}
}