platform/lib/src/parse_body.dart
thosakwe 1acfeeee24 +5
2017-01-14 08:50:02 -05:00

104 lines
3.4 KiB
Dart

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. :)
///
/// Use [storeOriginalBuffer] to add the original request bytes to the result.
Future<BodyParseResult> parseBody(HttpRequest request,
{bool storeOriginalBuffer: false}) async {
var result = new _BodyParseResultImpl();
Future<List<int>> getBytes() async {
return await request.fold(<int>[], (a, b) => a..addAll(b));
}
Future<String> getBody() async {
if (storeOriginalBuffer) {
List<int> bytes = await getBytes();
return UTF8.decode(result.originalBuffer = bytes);
} else
return await request.transform(UTF8.decoder).join();
}
try {
if (request.headers.contentType != null) {
if (request.headers.contentType.primaryType == 'multipart' &&
request.headers.contentType.parameters.containsKey('boundary')) {
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
.transform(new MimeMultipartTransformer(
request.headers.contentType.parameters['boundary']))
.map((part) =>
HttpMultipartFormData.parse(part, defaultEncoding: UTF8));
await for (HttpMultipartFormData part in parts) {
if (part.isBinary ||
part.contentDisposition.parameters.containsKey("filename")) {
BytesBuilder builder = await part.fold(new BytesBuilder(),
(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');
}
}
} else if (request.headers.contentType.mimeType ==
ContentType.JSON.mimeType) {
result.body.addAll(JSON.decode(await getBody()));
} else if (request.headers.contentType.mimeType ==
'application/x-www-form-urlencoded') {
String body = await getBody();
buildMapFromUri(result.body, body);
}
} else if (request.uri.hasQuery) {
buildMapFromUri(result.query, request.uri.query);
}
} catch (e) {
//
}
return result;
}
class _BodyParseResultImpl implements BodyParseResult {
@override
Map<String, dynamic> body = {};
@override
List<FileUploadInfo> files = [];
@override
List<int> originalBuffer;
@override
Map<String, dynamic> query = {};
}