platform/lib/src/http/response_context.dart

179 lines
5 KiB
Dart
Raw Normal View History

library angel_framework.http.response_context;
import 'dart:async';
import 'dart:convert';
import 'dart:io';
2016-10-22 20:41:36 +00:00
import 'package:angel_route/angel_route.dart';
import 'package:json_god/json_god.dart' as god;
import 'package:mime/mime.dart';
import '../extensible.dart';
import 'angel_base.dart';
import 'controller.dart';
2016-04-18 03:27:23 +00:00
/// A convenience wrapper around an outgoing HTTP request.
class ResponseContext extends Extensible {
2016-10-22 20:41:36 +00:00
bool _isOpen = true;
2016-04-18 03:27:23 +00:00
/// The [Angel] instance that is sending a response.
AngelBase app;
2016-04-18 03:27:23 +00:00
/// Can we still write to this response?
2016-10-22 20:41:36 +00:00
bool get isOpen => _isOpen;
2016-04-18 03:27:23 +00:00
/// A set of UTF-8 encoded bytes that will be written to the response.
2016-10-22 20:41:36 +00:00
final BytesBuilder buffer = new BytesBuilder();
2016-04-18 03:27:23 +00:00
/// Sets the status code to be sent with this response.
2016-10-22 20:41:36 +00:00
void status(int code) {
2016-04-22 01:27:28 +00:00
underlyingResponse.statusCode = code;
2016-04-18 03:27:23 +00:00
}
/// The underlying [HttpResponse] under this instance.
2016-10-22 20:41:36 +00:00
final HttpResponse underlyingResponse;
2016-04-18 03:27:23 +00:00
2016-10-22 20:41:36 +00:00
ResponseContext(this.underlyingResponse, this.app);
2016-04-18 03:27:23 +00:00
/// Any and all cookies to be sent to the user.
List<Cookie> get cookies => underlyingResponse.cookies;
/// Set this to true if you will manually close the response.
bool willCloseItself = false;
/// Sends a download as a response.
2016-10-22 20:41:36 +00:00
download(File file, {String filename}) async {
header("Content-Disposition",
'attachment; filename="${filename ?? file.path}"');
2016-04-18 03:27:23 +00:00
header(HttpHeaders.CONTENT_TYPE, lookupMimeType(file.path));
header(HttpHeaders.CONTENT_LENGTH, file.lengthSync().toString());
2016-10-22 20:41:36 +00:00
buffer.add(await file.readAsBytes());
end();
2016-04-18 03:27:23 +00:00
}
/// Prevents more data from being written to the response.
2016-10-22 20:41:36 +00:00
void end() {
_isOpen = false;
}
2016-04-18 03:27:23 +00:00
/// Sets a response header to the given value, or retrieves its value.
header(String key, [String value]) {
2016-10-22 20:41:36 +00:00
if (value == null)
return underlyingResponse.headers[key];
else
underlyingResponse.headers.set(key, value);
2016-04-18 03:27:23 +00:00
}
/// Serializes JSON to the response.
2016-10-22 20:41:36 +00:00
void json(value) {
2016-04-18 03:27:23 +00:00
write(god.serialize(value));
header(HttpHeaders.CONTENT_TYPE, ContentType.JSON.toString());
end();
}
/// Returns a JSONP response.
2016-10-22 20:41:36 +00:00
void jsonp(value, {String callbackName: "callback"}) {
2016-04-18 03:27:23 +00:00
write("$callbackName(${god.serialize(value)})");
header(HttpHeaders.CONTENT_TYPE, "application/javascript");
end();
}
/// Renders a view to the response stream, and closes the response.
2016-04-22 02:40:37 +00:00
Future render(String view, [Map data]) async {
write(await app.viewGenerator(view, data));
2016-04-18 03:27:23 +00:00
header(HttpHeaders.CONTENT_TYPE, ContentType.HTML.toString());
end();
}
/// Redirects to user to the given URL.
2016-10-22 20:41:36 +00:00
void redirect(String url, {int code: 301}) {
2016-04-18 03:27:23 +00:00
header(HttpHeaders.LOCATION, url);
status(code ?? 301);
2016-04-18 03:27:23 +00:00
write('''
<!DOCTYPE html>
<html>
<head>
<title>Redirecting...</title>
<meta http-equiv="refresh" content="0; url=$url">
</head>
<body>
<h1>Currently redirecting you...</h1>
<br />
Click <a href="$url">here</a> if you are not automatically redirected...
2016-04-18 03:27:23 +00:00
<script>
window.location = "$url";
</script>
</body>
</html>
''');
end();
}
/// Redirects to the given named [Route].
2016-10-22 20:41:36 +00:00
void redirectTo(String name, [Map params, int code]) {
2016-11-23 09:10:47 +00:00
_findRoute(Route route) {
for (Route child in route.children) {
final resolved = _findRoute(child);
if (resolved != null) return resolved;
}
return route.children
.firstWhere((r) => r.name == name, orElse: () => null);
}
Route matched = _findRoute(app.root);
if (matched != null) {
2016-10-22 20:41:36 +00:00
redirect(matched.makeUri(params), code: code);
return;
}
throw new ArgumentError.notNull('Route to redirect to ($name)');
}
2016-06-27 00:20:42 +00:00
/// Redirects to the given [Controller] action.
2016-10-22 20:41:36 +00:00
void redirectToAction(String action, [Map params, int code]) {
2016-06-27 00:20:42 +00:00
// UserController@show
List<String> split = action.split("@");
2016-10-22 20:41:36 +00:00
// Todo: AngelResponseException
2016-06-27 00:20:42 +00:00
if (split.length < 2)
2016-10-22 20:41:36 +00:00
throw new Exception(
"Controller redirects must take the form of 'Controller@action'. You gave: $action");
2016-06-27 00:20:42 +00:00
Controller controller = app.controller(split[0]);
if (controller == null)
throw new Exception("Could not find a controller named '${split[0]}'");
Route matched = controller.routeMappings[split[1]];
2016-06-27 00:20:42 +00:00
if (matched == null)
2016-10-22 20:41:36 +00:00
throw new Exception(
"Controller '${split[0]}' does not contain any action named '${split[1]}'");
2016-06-27 00:20:42 +00:00
2016-10-22 20:41:36 +00:00
redirect(matched.makeUri(params), code: code);
2016-06-27 00:20:42 +00:00
}
2016-04-18 03:27:23 +00:00
/// Streams a file to this response as chunked data.
///
/// Useful for video sites.
2016-10-22 20:41:36 +00:00
Future streamFile(File file,
2016-04-18 03:27:23 +00:00
{int chunkSize, int sleepMs: 0, bool resumable: true}) async {
if (!isOpen) return;
header(HttpHeaders.CONTENT_TYPE, lookupMimeType(file.path));
willCloseItself = true;
await file.openRead().pipe(underlyingResponse);
}
/// Writes data to the response.
2016-10-22 20:41:36 +00:00
void write(value, {Encoding encoding: UTF8}) {
if (isOpen) {
if (value is List<int>)
buffer.add(value);
2016-11-23 09:10:47 +00:00
else
buffer.add(encoding.encode(value.toString()));
2016-10-22 20:41:36 +00:00
}
2016-04-18 03:27:23 +00:00
}
2016-10-22 20:41:36 +00:00
}