platform/packages/mock_request/lib/src/headers.dart
2021-03-19 07:10:19 +08:00

151 lines
3.6 KiB
Dart

import 'dart:io';
class MockHttpHeaders extends HttpHeaders {
final Map<String, List<String>> _data = {};
final List<String> _noFolding = [];
Uri _host;
List<String> get doNotFold => List<String>.unmodifiable(_noFolding);
@override
ContentType get contentType {
if (_data.containsKey(HttpHeaders.contentTypeHeader)) {
return ContentType.parse(_data[HttpHeaders.contentTypeHeader].join(','));
} else {
return null;
}
}
@override
set contentType(ContentType value) =>
set(HttpHeaders.contentTypeHeader, value.value);
@override
DateTime get date => _data.containsKey(HttpHeaders.dateHeader)
? HttpDate.parse(_data[HttpHeaders.dateHeader].join(','))
: null;
@override
set date(DateTime value) =>
set(HttpHeaders.dateHeader, HttpDate.format(value));
@override
DateTime get expires => _data.containsKey(HttpHeaders.expiresHeader)
? HttpDate.parse(_data[HttpHeaders.expiresHeader].join(','))
: null;
@override
set expires(DateTime value) =>
set(HttpHeaders.expiresHeader, HttpDate.format(value));
@override
DateTime get ifModifiedSince =>
_data.containsKey(HttpHeaders.ifModifiedSinceHeader)
? HttpDate.parse(_data[HttpHeaders.ifModifiedSinceHeader].join(','))
: null;
@override
set ifModifiedSince(DateTime value) =>
set(HttpHeaders.ifModifiedSinceHeader, HttpDate.format(value));
@override
String get host {
if (_host != null) {
return _host.host;
} else if (_data.containsKey(HttpHeaders.hostHeader)) {
_host = Uri.parse(_data[HttpHeaders.hostHeader].join(','));
return _host.host;
} else {
return null;
}
}
@override
int get port {
host; // Parse it
return _host?.port;
}
@override
List<String> operator [](String name) => _data[name.toLowerCase()];
@override
void add(String name, Object value, {bool preserveHeaderCase = false}) {
var lower = preserveHeaderCase ? name : name.toLowerCase();
if (_data.containsKey(lower)) {
if (value is Iterable) {
_data[lower].addAll(value.map((x) => x.toString()).toList());
} else {
_data[lower].add(value.toString());
}
} else {
if (value is Iterable) {
_data[lower] = value.map((x) => x.toString()).toList();
} else {
_data[lower] = [value.toString()];
}
}
}
@override
void clear() {
_data.clear();
}
@override
void forEach(void Function(String name, List<String> values) f) {
_data.forEach(f);
}
@override
void noFolding(String name) {
_noFolding.add(name.toLowerCase());
}
@override
void remove(String name, Object value) {
var lower = name.toLowerCase();
if (_data.containsKey(lower)) {
if (value is Iterable) {
for (var x in value) {
_data[lower].remove(x.toString());
}
} else {
_data[lower].remove(value.toString());
}
}
}
@override
void removeAll(String name) {
_data.remove(name.toLowerCase());
}
@override
void set(String name, Object value, {bool preserveHeaderCase = false}) {
var lower = preserveHeaderCase ? name : name.toLowerCase();
_data.remove(lower);
if (value is Iterable) {
_data[lower] = value.map((x) => x.toString()).toList();
} else {
_data[lower] = [value.toString()];
}
}
@override
String value(String name) => _data[name.toLowerCase()]?.join(',');
@override
String toString() {
var b = StringBuffer();
_data.forEach((k, v) {
b.write('$k: ');
b.write(v.join(','));
b.writeln();
});
return b.toString();
}
}