OLD | NEW |
1 // Copyright (c) 2015, the Dart project authors. Please see the AUTHORS file | 1 // Copyright (c) 2015, the Dart project authors. Please see the AUTHORS file |
2 // for details. All rights reserved. Use of this source code is governed by a | 2 // for details. All rights reserved. Use of this source code is governed by a |
3 // BSD-style license that can be found in the LICENSE file. | 3 // BSD-style license that can be found in the LICENSE file. |
4 | 4 |
5 library test.one_off_handler; | 5 library test.util.one_off_handler; |
6 | 6 |
7 import 'package:path/path.dart' as p; | 7 import 'package:path/path.dart' as p; |
8 import 'package:shelf/shelf.dart' as shelf; | 8 import 'package:shelf/shelf.dart' as shelf; |
9 | 9 |
| 10 import '../utils.dart'; |
| 11 |
10 /// A Shelf handler that provides support for one-time handlers. | 12 /// A Shelf handler that provides support for one-time handlers. |
11 /// | 13 /// |
12 /// This is useful for handlers that only expect to be hit once before becoming | 14 /// This is useful for handlers that only expect to be hit once before becoming |
13 /// invalid and don't need to have a persistent URL. | 15 /// invalid and don't need to have a persistent URL. |
14 class OneOffHandler { | 16 class OneOffHandler { |
15 /// A map from URL paths to handlers. | 17 /// A map from URL paths to handlers. |
16 final _handlers = new Map<String, shelf.Handler>(); | 18 final _handlers = new Map<String, shelf.Handler>(); |
17 | 19 |
18 /// The counter of handlers that have been activated. | 20 /// The counter of handlers that have been activated. |
19 var _counter = 0; | 21 var _counter = 0; |
20 | 22 |
21 /// The actual [shelf.Handler] that dispatches requests. | 23 /// The actual [shelf.Handler] that dispatches requests. |
22 shelf.Handler get handler => _onRequest; | 24 shelf.Handler get handler => _onRequest; |
23 | 25 |
24 /// Creates a new one-off handler that forwards to [handler]. | 26 /// Creates a new one-off handler that forwards to [handler]. |
25 /// | 27 /// |
26 /// Returns a string that's the URL path for hitting this handler, relative to | 28 /// Returns a string that's the URL path for hitting this handler, relative to |
27 /// the URL for the one-off handler itself. | 29 /// the URL for the one-off handler itself. |
28 /// | 30 /// |
29 /// [handler] will be unmounted as soon as it receives a request. | 31 /// [handler] will be unmounted as soon as it receives a request. |
30 String create(shelf.Handler handler) { | 32 String create(shelf.Handler handler) { |
31 var path = _counter.toString(); | 33 var path = _counter.toString(); |
32 _handlers[path] = handler; | 34 _handlers[path] = handler; |
33 _counter++; | 35 _counter++; |
34 return path; | 36 return path; |
35 } | 37 } |
36 | 38 |
37 /// Dispatches [request] to the appropriate handler. | 39 /// Dispatches [request] to the appropriate handler. |
38 _onRequest(shelf.Request request) { | 40 _onRequest(shelf.Request request) { |
39 var components = p.url.split(request.url.path); | 41 var components = p.url.split(shelfUrl(request).path); |
40 | |
41 // For shelf < 0.6.0, the first component of the path is always "/". We can | |
42 // safely skip it. | |
43 if (components.isNotEmpty && components.first == "/") { | |
44 components.removeAt(0); | |
45 } | |
46 | |
47 if (components.isEmpty) return new shelf.Response.notFound(null); | 42 if (components.isEmpty) return new shelf.Response.notFound(null); |
48 | 43 |
49 var handler = _handlers.remove(components.removeAt(0)); | 44 var path = components.removeAt(0); |
| 45 var handler = _handlers.remove(path); |
50 if (handler == null) return new shelf.Response.notFound(null); | 46 if (handler == null) return new shelf.Response.notFound(null); |
51 return handler(request); | 47 return handler(shelfChange(request, path: path)); |
52 } | 48 } |
53 } | 49 } |
OLD | NEW |