87 lines
2.5 KiB
Dart
87 lines
2.5 KiB
Dart
/// Easy helper hooks.
|
|
library angel_framework.hooks;
|
|
|
|
import 'dart:async';
|
|
import 'package:json_god/json_god.dart' as god;
|
|
import 'angel_framework.dart';
|
|
|
|
/// Transforms `e.data` into JSON-friendly data, i.e. a Map.
|
|
HookedServiceEventListener toJson() {
|
|
return (HookedServiceEvent e) {
|
|
if (e.data != null && e.data is! Map) e.data = god.serializeObject(e.data);
|
|
};
|
|
}
|
|
|
|
/// Transforms `e.data` into an instance of the given [type],
|
|
/// if it is not already.
|
|
HookedServiceEventListener toType(Type type) {
|
|
return (HookedServiceEvent e) {
|
|
if (e.data != null && e.data.runtimeType != type)
|
|
e.data = god.deserializeDatum(e.data, outputType: type);
|
|
};
|
|
}
|
|
|
|
/// Removes one or more [key]s from service results.
|
|
/// Works on single results, and iterable results.
|
|
HookedServiceEventListener remove(key, remover(key, obj)) {
|
|
return (HookedServiceEvent e) async {
|
|
if (!e.isAfter) throw new StateError("'remove' only works on after hooks.");
|
|
|
|
_remover(key, obj) {
|
|
if (remover != null)
|
|
return remover(key, obj);
|
|
else if (obj is List)
|
|
return obj..remove(key);
|
|
else if (obj is Iterable)
|
|
return obj.where((k) => !key);
|
|
else if (obj is Map)
|
|
return obj..remove(key);
|
|
else if (obj is Extensible)
|
|
return obj..properties.remove(key);
|
|
else
|
|
throw new ArgumentError("Cannot remove key 'key' from $obj.");
|
|
}
|
|
|
|
var keys = key is Iterable ? key : [key];
|
|
|
|
_removeAll(obj) async {
|
|
var r = obj;
|
|
|
|
for (var key in keys) {
|
|
r = await _remover(key, r);
|
|
}
|
|
|
|
return r;
|
|
}
|
|
|
|
if (e.result is Iterable) {
|
|
var r = await Future.wait(e.result.map(_removeAll));
|
|
e.result = e.result is List ? r.toList() : r;
|
|
} else
|
|
e.result = await _removeAll(e.result);
|
|
};
|
|
}
|
|
|
|
/// Disables a service method for access from a provider.
|
|
///
|
|
/// [provider] can be either a String, [Providers], an Iterable of String, or a
|
|
/// function that takes a [HookedServiceEvent] and returns a bool.
|
|
/// Futures are allowed.
|
|
HookedServiceEventListener disable([provider]) {
|
|
return (HookedServiceEvent e) async {
|
|
if (provider is Function) {
|
|
var r = await provider(e);
|
|
|
|
if (r != true) throw new AngelHttpException.methodNotAllowed();
|
|
} else {
|
|
_provide(p) => p is Providers ? p : new Providers(p.toString());
|
|
|
|
var providers =
|
|
provider is Iterable ? provider.map(_provide) : [_provide(provider)];
|
|
|
|
if (providers.any((Providers p) => p == e.params['provider'])) {
|
|
throw new AngelHttpException.methodNotAllowed();
|
|
}
|
|
}
|
|
};
|
|
}
|