OLD | NEW |
| (Empty) |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 define("mojo/public/js/bindings/router", [ | |
6 "mojo/public/js/bindings/codec", | |
7 "mojo/public/js/bindings/connector", | |
8 "mojo/public/js/bindings/validator", | |
9 ], function(codec, connector, validator) { | |
10 | |
11 function Router(handle, connectorFactory) { | |
12 if (connectorFactory === undefined) | |
13 connectorFactory = connector.Connector; | |
14 this.connector_ = new connectorFactory(handle); | |
15 this.incomingReceiver_ = null; | |
16 this.nextRequestID_ = 0; | |
17 this.completers_ = new Map(); | |
18 this.payloadValidators_ = []; | |
19 | |
20 this.connector_.setIncomingReceiver({ | |
21 accept: this.handleIncomingMessage_.bind(this), | |
22 }); | |
23 this.connector_.setErrorHandler({ | |
24 onError: this.handleConnectionError_.bind(this), | |
25 }); | |
26 } | |
27 | |
28 Router.prototype.close = function() { | |
29 this.completers_.clear(); // Drop any responders. | |
30 this.connector_.close(); | |
31 }; | |
32 | |
33 Router.prototype.accept = function(message) { | |
34 this.connector_.accept(message); | |
35 }; | |
36 | |
37 Router.prototype.reject = function(message) { | |
38 // TODO(mpcomplete): no way to trasmit errors over a Connection. | |
39 }; | |
40 | |
41 Router.prototype.acceptAndExpectResponse = function(message) { | |
42 // Reserve 0 in case we want it to convey special meaning in the future. | |
43 var requestID = this.nextRequestID_++; | |
44 if (requestID == 0) | |
45 requestID = this.nextRequestID_++; | |
46 | |
47 message.setRequestID(requestID); | |
48 var result = this.connector_.accept(message); | |
49 if (!result) | |
50 return Promise.reject(Error("Connection error")); | |
51 | |
52 var completer = {}; | |
53 this.completers_.set(requestID, completer); | |
54 return new Promise(function(resolve, reject) { | |
55 completer.resolve = resolve; | |
56 completer.reject = reject; | |
57 }); | |
58 }; | |
59 | |
60 Router.prototype.setIncomingReceiver = function(receiver) { | |
61 this.incomingReceiver_ = receiver; | |
62 }; | |
63 | |
64 Router.prototype.setPayloadValidators = function(payloadValidators) { | |
65 this.payloadValidators_ = payloadValidators; | |
66 }; | |
67 | |
68 Router.prototype.encounteredError = function() { | |
69 return this.connector_.encounteredError(); | |
70 }; | |
71 | |
72 Router.prototype.handleIncomingMessage_ = function(message) { | |
73 var noError = validator.validationError.NONE; | |
74 var messageValidator = new validator.Validator(message); | |
75 var err = messageValidator.validateMessageHeader(); | |
76 for (var i = 0; err === noError && i < this.payloadValidators_.length; ++i) | |
77 err = this.payloadValidators_[i](messageValidator); | |
78 | |
79 if (err == noError) | |
80 this.handleValidIncomingMessage_(message); | |
81 else | |
82 this.handleInvalidIncomingMessage_(message, err); | |
83 }; | |
84 | |
85 Router.prototype.handleValidIncomingMessage_ = function(message) { | |
86 if (message.expectsResponse()) { | |
87 if (this.incomingReceiver_) { | |
88 this.incomingReceiver_.acceptWithResponder(message, this); | |
89 } else { | |
90 // If we receive a request expecting a response when the client is not | |
91 // listening, then we have no choice but to tear down the pipe. | |
92 this.close(); | |
93 } | |
94 } else if (message.isResponse()) { | |
95 var reader = new codec.MessageReader(message); | |
96 var requestID = reader.requestID; | |
97 var completer = this.completers_.get(requestID); | |
98 this.completers_.delete(requestID); | |
99 completer.resolve(message); | |
100 } else { | |
101 if (this.incomingReceiver_) | |
102 this.incomingReceiver_.accept(message); | |
103 } | |
104 } | |
105 | |
106 Router.prototype.handleInvalidIncomingMessage_ = function(message, error) { | |
107 this.close(); | |
108 } | |
109 | |
110 Router.prototype.handleConnectionError_ = function(result) { | |
111 this.completers_.forEach(function(value) { | |
112 value.reject(result); | |
113 }); | |
114 this.close(); | |
115 }; | |
116 | |
117 // The TestRouter subclass is only intended to be used in unit tests. | |
118 // It defeats valid message handling and delgates invalid message handling. | |
119 | |
120 function TestRouter(handle, connectorFactory) { | |
121 Router.call(this, handle, connectorFactory); | |
122 } | |
123 | |
124 TestRouter.prototype = Object.create(Router.prototype); | |
125 | |
126 TestRouter.prototype.handleValidIncomingMessage_ = function() { | |
127 }; | |
128 | |
129 TestRouter.prototype.handleInvalidIncomingMessage_ = | |
130 function(message, error) { | |
131 this.validationErrorHandler(error); | |
132 }; | |
133 | |
134 var exports = {}; | |
135 exports.Router = Router; | |
136 exports.TestRouter = TestRouter; | |
137 return exports; | |
138 }); | |
OLD | NEW |