Index: lib/src/message.dart |
diff --git a/lib/src/message.dart b/lib/src/message.dart |
index ba5c5fa5441d22853f3a21ffc03a616141d81fba..f013c9f5aaa5d573918760b87cbe27989842f589 100644 |
--- a/lib/src/message.dart |
+++ b/lib/src/message.dart |
@@ -11,6 +11,7 @@ import 'package:http_parser/http_parser.dart'; |
import 'package:stack_trace/stack_trace.dart'; |
import 'shelf_unmodifiable_map.dart'; |
+import 'util.dart'; |
/// Represents logic shared between [Request] and [Response]. |
abstract class Message { |
@@ -46,10 +47,11 @@ abstract class Message { |
/// Creates a new [Message]. |
/// |
/// If [headers] is `null`, it is treated as empty. |
- Message(this._body, |
- {Map<String, String> headers, Map<String, Object> context}) |
- : this.headers = new ShelfUnmodifiableMap<String>(headers, |
- ignoreKeyCase: true), |
+ Message(body, {Encoding encoding, Map<String, String> headers, |
nweiz
2015/01/21 01:51:00
Document what types [body] can be.
kevmoo
2015/01/21 02:15:25
Done.
|
+ Map<String, Object> context}) |
+ : this._body = _bodyToStream(body, encoding), |
+ this.headers = new ShelfUnmodifiableMap<String>( |
+ _adjustHeaders(headers, encoding), ignoreKeyCase: true), |
this.context = new ShelfUnmodifiableMap<Object>(context, |
ignoreKeyCase: false); |
@@ -130,3 +132,34 @@ abstract class Message { |
/// changes. |
Message change({Map<String, String> headers, Map<String, Object> context}); |
} |
+ |
+/// Converts [body] to a byte stream. |
+/// |
+/// [body] may be either a [String], a [Stream<List<int>>], or `null`. If it's a |
+/// [String], [encoding] will be used to convert it to a [Stream<List<int>>]. |
+Stream<List<int>> _bodyToStream(body, Encoding encoding) { |
+ if (encoding == null) encoding = UTF8; |
+ if (body == null) return new Stream.fromIterable([]); |
+ if (body is String) return new Stream.fromIterable([encoding.encode(body)]); |
+ if (body is Stream) return body; |
+ |
+ throw new ArgumentError('Response body "$body" must be a String or a ' |
+ 'Stream.'); |
+} |
+ |
+/// Adds information about [encoding] to [headers]. |
+/// |
+/// Returns a new map without modifying [headers]. |
+Map<String, String> _adjustHeaders( |
+ Map<String, String> headers, Encoding encoding) { |
+ if (headers == null) headers = const {}; |
+ if (encoding == null) return headers; |
+ if (headers['content-type'] == null) { |
+ return addHeader(headers, 'content-type', |
+ 'application/octet-stream; charset=${encoding.name}'); |
+ } |
+ |
+ var contentType = new MediaType.parse(headers['content-type']).change( |
+ parameters: {'charset': encoding.name}); |
+ return addHeader(headers, 'content-type', contentType.toString()); |
+} |