platform/lib/src/parse_body.dart

125 lines
3.9 KiB
Dart
Raw Normal View History

import 'dart:async';
import 'dart:convert';
import 'dart:io';
import 'package:http_server/http_server.dart';
import 'package:mime/mime.dart';
import 'body_parse_result.dart';
import 'file_upload_info.dart';
import 'map_from_uri.dart';
/// Grabs data from an incoming request.
///
/// Supports URL-encoded and JSON, as well as multipart/* forms.
/// On a file upload request, only fields with the name **'file'** are processed
/// as files. Anything else is put in the body. You can change the upload file name
/// via the *fileUploadName* parameter. :)
2017-01-14 13:50:02 +00:00
///
/// Use [storeOriginalBuffer] to add the original request bytes to the result.
Future<BodyParseResult> parseBody(HttpRequest request,
{bool storeOriginalBuffer: false}) async {
var result = new _BodyParseResultImpl();
2017-10-12 02:32:42 +00:00
Future<List<int>> getBytes() {
return request
.fold<BytesBuilder>(new BytesBuilder(copy: false), (a, b) => a..add(b))
.then((b) => b.takeBytes());
2017-01-14 13:50:02 +00:00
}
2017-10-12 02:32:42 +00:00
Future<String> getBody() {
2017-01-14 13:50:02 +00:00
if (storeOriginalBuffer) {
2017-10-12 02:32:42 +00:00
return getBytes().then((bytes) {
result.originalBuffer = bytes;
return UTF8.decode(bytes);
});
2017-01-14 13:50:02 +00:00
} else
2017-10-12 02:32:42 +00:00
return request.transform(UTF8.decoder).join();
2017-01-14 13:50:02 +00:00
}
2016-12-05 02:31:25 +00:00
try {
if (request.headers.contentType != null) {
if (request.headers.contentType.primaryType == 'multipart' &&
request.headers.contentType.parameters.containsKey('boundary')) {
2017-01-14 13:50:02 +00:00
Stream<List<int>> stream;
if (storeOriginalBuffer) {
var bytes = result.originalBuffer = await getBytes();
var ctrl = new StreamController<List<int>>()
..add(bytes)
..close();
stream = ctrl.stream;
} else {
stream = request;
}
var parts = stream
2016-12-05 02:31:25 +00:00
.transform(new MimeMultipartTransformer(
request.headers.contentType.parameters['boundary']))
.map((part) =>
HttpMultipartFormData.parse(part, defaultEncoding: UTF8));
2016-12-05 02:31:25 +00:00
await for (HttpMultipartFormData part in parts) {
if (part.isBinary ||
part.contentDisposition.parameters.containsKey("filename")) {
2017-10-12 02:32:42 +00:00
BytesBuilder builder = await part.fold(
new BytesBuilder(copy: false),
2016-12-05 02:31:25 +00:00
(BytesBuilder b, d) => b..add(d is! String ? d : d.codeUnits));
var upload = new FileUploadInfo(
mimeType: part.contentType.mimeType,
name: part.contentDisposition.parameters['name'],
filename:
part.contentDisposition.parameters['filename'] ?? 'file',
data: builder.takeBytes());
result.files.add(upload);
} else if (part.isText) {
var text = await part.join();
buildMapFromUri(result.body,
'${part.contentDisposition.parameters["name"]}=$text');
}
}
2016-12-05 02:31:25 +00:00
} else if (request.headers.contentType.mimeType ==
ContentType.JSON.mimeType) {
2017-01-14 13:50:02 +00:00
result.body.addAll(JSON.decode(await getBody()));
2016-12-05 02:31:25 +00:00
} else if (request.headers.contentType.mimeType ==
'application/x-www-form-urlencoded') {
2017-01-14 13:50:02 +00:00
String body = await getBody();
2016-12-05 02:31:25 +00:00
buildMapFromUri(result.body, body);
2017-07-19 21:20:57 +00:00
} else if (storeOriginalBuffer == true) {
result.originalBuffer = await getBytes();
}
} else {
if (request.uri.hasQuery) {
buildMapFromUri(result.query, request.uri.query);
}
if (storeOriginalBuffer == true) {
result.originalBuffer = await getBytes();
}
}
2017-07-19 21:20:57 +00:00
} catch (e, st) {
result.error = e;
result.stack = st;
}
return result;
}
2017-01-14 13:50:02 +00:00
class _BodyParseResultImpl implements BodyParseResult {
@override
Map<String, dynamic> body = {};
@override
List<FileUploadInfo> files = [];
@override
List<int> originalBuffer;
@override
Map<String, dynamic> query = {};
2017-07-19 21:20:57 +00:00
@override
var error = null;
@override
StackTrace stack = null;
2017-01-14 13:50:02 +00:00
}