Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(31)

Unified Diff: lib/src/message.dart

Issue 851423003: Request now has the same body model as Response (Closed) Base URL: https://github.com/dart-lang/shelf.git@master
Patch Set: rebase Created 5 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « CHANGELOG.md ('k') | lib/src/request.dart » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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());
+}
« no previous file with comments | « CHANGELOG.md ('k') | lib/src/request.dart » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698