-
Notifications
You must be signed in to change notification settings - Fork 153
/
HeaderDictionary.cs
140 lines (115 loc) · 4.42 KB
/
HeaderDictionary.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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
using System;
using System.Collections.Generic;
using System.Linq;
using System.Net.Mail;
using System.Text;
using System.Text.RegularExpressions;
namespace AE.Net.Mail {
public class HeaderDictionary : SafeDictionary<string, HeaderValue> {
public HeaderDictionary() : base(StringComparer.OrdinalIgnoreCase) { }
public virtual string GetBoundary() {
return this["Content-Type"]["boundary"];
}
private static Regex[] rxDates = new[]{
@"\d{1,2}\s+[a-z]{3}\s+\d{2,4}\s+\d{1,2}\:\d{2}\:\d{1,2}\s+[\+\-\d\:]*",
@"\d{4}\-\d{1,2}-\d{1,2}\s+\d{1,2}\:\d{2}(?:\:\d{2})?(?:\s+[\+\-\d:]+)?",
}.Select(x => new Regex(x, RegexOptions.Compiled | RegexOptions.IgnoreCase)).ToArray();
public virtual DateTime GetDate() {
var value = this["Date"].RawValue.ToNullDate();
if (value == null) {
foreach (var rx in rxDates) {
var match = rx.Matches(this["Received"].RawValue ?? string.Empty)
.Cast<Match>().LastOrDefault();
if (match != null) {
value = match.Value.ToNullDate();
if (value != null) {
break;
}
}
}
}
//written this way so a break can be set on the null condition
if (value == null) return DateTime.MinValue;
return value.Value;
}
public virtual T GetEnum<T>(string name) where T : struct, IConvertible {
var value = this[name].RawValue;
if (string.IsNullOrEmpty(value)) return default(T);
var values = System.Enum.GetValues(typeof(T)).Cast<T>().ToArray();
return values.FirstOrDefault(x => x.ToString().Equals(value, StringComparison.OrdinalIgnoreCase));
}
public virtual void Add(string name, string value) {
this[name] = new HeaderValue(value);
}
public virtual void Add(string name, DateTime value) {
this[name] = new HeaderValue(value.GetRFC2060Date());
}
public virtual MailAddress[] GetMailAddresses(string header) {
const int notFound = -1;
var mailAddresses = new List<MailAddress>();
var headerValue = this[header].RawValue.Trim();
var mailAddressStartIndex = 0;
var mailAddressEndIndex = 0;
while (mailAddressEndIndex < headerValue.Length) {
// Start searching for the next comma by skipping the previous mailAddressEndIndex
mailAddressEndIndex = IndexOfExcludingQuoted(headerValue, ',', mailAddressEndIndex);
if (mailAddressEndIndex == notFound) {
mailAddressEndIndex = headerValue.Length;
}
var possibleMailAddress = headerValue.Substring(mailAddressStartIndex, mailAddressEndIndex - mailAddressStartIndex);
var mailAddress = possibleMailAddress.Trim().ToEmailAddress();
if (mailAddress != null) {
mailAddresses.Add(mailAddress);
mailAddressStartIndex = mailAddressEndIndex + 1;
mailAddressEndIndex = mailAddressStartIndex;
} else {
// Inscrease the end index by one so the search for the next comma skips beyond the previous found comma
mailAddressEndIndex++;
}
}
return mailAddresses.ToArray();
}
public static HeaderDictionary Parse(string headers, System.Text.Encoding encoding) {
headers = Utilities.DecodeWords(headers, encoding);
var temp = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
var lines = headers.Split(new char[] { '\r', '\n' }, StringSplitOptions.RemoveEmptyEntries);
int i;
string key = null, value;
foreach (var line in lines) {
if (key != null && (line[0] == '\t' || line[0] == ' ')) {
temp[key] += line.TrimStartOnce();
} else {
if (key != null)
temp[key] = temp[key].TrimEndOnce(); // It trims the last line of the previous key
i = line.IndexOf(':');
if (i > -1) {
key = line.Substring(0, i).TrimStartOnce();
value = line.Substring(i + 1).TrimStartOnce();
temp.Set(key, value);
}
}
}
var result = new HeaderDictionary();
foreach (var item in temp) {
result.Add(item.Key, new HeaderValue(item.Value));
}
return result;
}
private static int IndexOfExcludingQuoted(string str, char searchChar, int startIndex)
{
bool withinQuotes = false;
for (int i = startIndex; i < str.Length; i++) {
if (!withinQuotes) {
if (str[i] == searchChar)
return i;
if (str[i] == '"')
withinQuotes = true;
}
else if (str[i] == '"') {
withinQuotes = false;
}
}
return -1;
}
}
}