murgatroid99 | c92df4c | 2017-05-23 10:28:47 -0700 | [diff] [blame] | 1 | /** |
| 2 | * @license |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 3 | * Copyright 2015 gRPC authors. |
murgatroid99 | cca5ffa | 2015-01-15 14:06:56 -0800 | [diff] [blame] | 4 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 5 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 6 | * you may not use this file except in compliance with the License. |
| 7 | * You may obtain a copy of the License at |
murgatroid99 | cca5ffa | 2015-01-15 14:06:56 -0800 | [diff] [blame] | 8 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 9 | * http://www.apache.org/licenses/LICENSE-2.0 |
murgatroid99 | cca5ffa | 2015-01-15 14:06:56 -0800 | [diff] [blame] | 10 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 11 | * Unless required by applicable law or agreed to in writing, software |
| 12 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 13 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 14 | * See the License for the specific language governing permissions and |
| 15 | * limitations under the License. |
murgatroid99 | cca5ffa | 2015-01-15 14:06:56 -0800 | [diff] [blame] | 16 | * |
| 17 | */ |
| 18 | |
murgatroid99 | dca966d | 2015-02-19 14:37:18 -0800 | [diff] [blame] | 19 | 'use strict'; |
| 20 | |
murgatroid99 | 6d6009f | 2015-10-15 09:57:31 -0700 | [diff] [blame] | 21 | var path = require('path'); |
murgatroid99 | 6f60766 | 2016-04-27 16:38:33 -0700 | [diff] [blame] | 22 | var fs = require('fs'); |
murgatroid99 | 6d6009f | 2015-10-15 09:57:31 -0700 | [diff] [blame] | 23 | |
| 24 | var SSL_ROOTS_PATH = path.resolve(__dirname, '..', '..', 'etc', 'roots.pem'); |
| 25 | |
murgatroid99 | 55739d5 | 2015-06-03 10:58:21 -0700 | [diff] [blame] | 26 | var _ = require('lodash'); |
murgatroid99 | cca5ffa | 2015-01-15 14:06:56 -0800 | [diff] [blame] | 27 | |
| 28 | var ProtoBuf = require('protobufjs'); |
| 29 | |
murgatroid99 | e787955 | 2015-02-12 12:21:15 -0800 | [diff] [blame] | 30 | var client = require('./src/client.js'); |
murgatroid99 | cca5ffa | 2015-01-15 14:06:56 -0800 | [diff] [blame] | 31 | |
murgatroid99 | e787955 | 2015-02-12 12:21:15 -0800 | [diff] [blame] | 32 | var server = require('./src/server.js'); |
murgatroid99 | cca5ffa | 2015-01-15 14:06:56 -0800 | [diff] [blame] | 33 | |
murgatroid99 | 9b9708a | 2016-06-01 11:42:20 -0700 | [diff] [blame] | 34 | var common = require('./src/common.js'); |
| 35 | |
murgatroid99 | 84e3cde | 2015-08-20 11:27:05 -0700 | [diff] [blame] | 36 | var Metadata = require('./src/metadata.js'); |
| 37 | |
murgatroid99 | e190f35 | 2016-01-20 13:52:08 -0800 | [diff] [blame] | 38 | var grpc = require('./src/grpc_extension'); |
murgatroid99 | cca5ffa | 2015-01-15 14:06:56 -0800 | [diff] [blame] | 39 | |
murgatroid99 | b5b5f02 | 2017-03-16 16:42:10 -0700 | [diff] [blame] | 40 | var protobuf_js_5_common = require('./src/protobuf_js_5_common'); |
| 41 | var protobuf_js_6_common = require('./src/protobuf_js_6_common'); |
| 42 | |
murgatroid99 | 7229f88 | 2017-05-10 16:24:05 -0700 | [diff] [blame] | 43 | var constants = require('./src/constants.js'); |
| 44 | |
murgatroid99 | 6f60766 | 2016-04-27 16:38:33 -0700 | [diff] [blame] | 45 | grpc.setDefaultRootsPem(fs.readFileSync(SSL_ROOTS_PATH, 'ascii')); |
| 46 | |
murgatroid99 | b306854 | 2017-03-21 11:25:01 -0700 | [diff] [blame] | 47 | /** |
murgatroid99 | c92df4c | 2017-05-23 10:28:47 -0700 | [diff] [blame] | 48 | * @namespace grpc |
| 49 | */ |
| 50 | |
| 51 | /** |
| 52 | * Load a ProtoBuf.js object as a gRPC object. |
| 53 | * @memberof grpc |
| 54 | * @alias grpc.loadObject |
murgatroid99 | b306854 | 2017-03-21 11:25:01 -0700 | [diff] [blame] | 55 | * @param {Object} value The ProtoBuf.js reflection object to load |
| 56 | * @param {Object=} options Options to apply to the loaded file |
murgatroid99 | c92df4c | 2017-05-23 10:28:47 -0700 | [diff] [blame] | 57 | * @param {bool=} [options.binaryAsBase64=false] deserialize bytes values as |
| 58 | * base64 strings instead of Buffers |
| 59 | * @param {bool=} [options.longsAsStrings=true] deserialize long values as |
| 60 | * strings instead of objects |
| 61 | * @param {bool=} [options.enumsAsStrings=true] deserialize enum values as |
| 62 | * strings instead of numbers. Only works with Protobuf.js 6 values. |
| 63 | * @param {bool=} [options.deprecatedArgumentOrder=false] use the beta method |
| 64 | * argument order for client methods, with optional arguments after the |
| 65 | * callback. This option is only a temporary stopgap measure to smooth an |
| 66 | * API breakage. It is deprecated, and new code should not use it. |
| 67 | * @param {(number|string)=} [options.protobufjsVersion='detect'] 5 and 6 |
| 68 | * respectively indicate that an object from the corresponding version of |
| 69 | * Protobuf.js is provided in the value argument. If the option is 'detect', |
| 70 | * gRPC wll guess what the version is based on the structure of the value. |
| 71 | * @return {Object<string, *>} The resulting gRPC object. |
murgatroid99 | b306854 | 2017-03-21 11:25:01 -0700 | [diff] [blame] | 72 | */ |
murgatroid99 | c02910b | 2016-02-17 12:59:26 -0800 | [diff] [blame] | 73 | exports.loadObject = function loadObject(value, options) { |
murgatroid99 | b1a0231 | 2017-03-17 13:45:15 -0700 | [diff] [blame] | 74 | options = _.defaults(options, common.defaultGrpcOptions); |
murgatroid99 | b306854 | 2017-03-21 11:25:01 -0700 | [diff] [blame] | 75 | options = _.defaults(options, {'protobufjsVersion': 'detect'}); |
| 76 | var protobufjsVersion; |
| 77 | if (options.protobufjsVersion === 'detect') { |
| 78 | if (protobuf_js_6_common.isProbablyProtobufJs6(value)) { |
| 79 | protobufjsVersion = 6; |
| 80 | } else if (protobuf_js_5_common.isProbablyProtobufJs5(value)) { |
| 81 | protobufjsVersion = 5; |
| 82 | } else { |
| 83 | var error_message = 'Could not detect ProtoBuf.js version. Please ' + |
| 84 | 'specify the version number with the "protobufjs_version" option'; |
| 85 | throw new Error(error_message); |
| 86 | } |
murgatroid99 | b1a0231 | 2017-03-17 13:45:15 -0700 | [diff] [blame] | 87 | } else { |
murgatroid99 | b306854 | 2017-03-21 11:25:01 -0700 | [diff] [blame] | 88 | protobufjsVersion = options.protobufjsVersion; |
| 89 | } |
| 90 | switch (protobufjsVersion) { |
| 91 | case 6: return protobuf_js_6_common.loadObject(value, options); |
| 92 | case 5: |
murgatroid99 | b1a0231 | 2017-03-17 13:45:15 -0700 | [diff] [blame] | 93 | return protobuf_js_5_common.loadObject(value, options); |
murgatroid99 | b306854 | 2017-03-21 11:25:01 -0700 | [diff] [blame] | 94 | default: |
| 95 | throw new Error('Unrecognized protobufjsVersion', protobufjsVersion); |
murgatroid99 | cca5ffa | 2015-01-15 14:06:56 -0800 | [diff] [blame] | 96 | } |
murgatroid99 | 9cd90a6 | 2015-07-27 11:23:13 -0700 | [diff] [blame] | 97 | }; |
| 98 | |
| 99 | var loadObject = exports.loadObject; |
murgatroid99 | cca5ffa | 2015-01-15 14:06:56 -0800 | [diff] [blame] | 100 | |
| 101 | /** |
murgatroid99 | c92df4c | 2017-05-23 10:28:47 -0700 | [diff] [blame] | 102 | * Load a gRPC object from a .proto file. |
| 103 | * @memberof grpc |
| 104 | * @alias grpc.load |
murgatroid99 | c02910b | 2016-02-17 12:59:26 -0800 | [diff] [blame] | 105 | * @param {string|{root: string, file: string}} filename The file to load |
murgatroid99 | 71dbb86 | 2015-04-20 11:22:51 -0700 | [diff] [blame] | 106 | * @param {string=} format The file format to expect. Must be either 'proto' or |
| 107 | * 'json'. Defaults to 'proto' |
murgatroid99 | c02910b | 2016-02-17 12:59:26 -0800 | [diff] [blame] | 108 | * @param {Object=} options Options to apply to the loaded file |
murgatroid99 | c92df4c | 2017-05-23 10:28:47 -0700 | [diff] [blame] | 109 | * @param {bool=} [options.convertFieldsToCamelCase=false] Load this file with |
| 110 | * field names in camel case instead of their original case |
| 111 | * @param {bool=} [options.binaryAsBase64=false] deserialize bytes values as |
| 112 | * base64 strings instead of Buffers |
| 113 | * @param {bool=} [options.longsAsStrings=true] deserialize long values as |
| 114 | * strings instead of objects |
| 115 | * @param {bool=} [options.deprecatedArgumentOrder=false] use the beta method |
| 116 | * argument order for client methods, with optional arguments after the |
| 117 | * callback. This option is only a temporary stopgap measure to smooth an |
| 118 | * API breakage. It is deprecated, and new code should not use it. |
murgatroid99 | cca5ffa | 2015-01-15 14:06:56 -0800 | [diff] [blame] | 119 | * @return {Object<string, *>} The resulting gRPC object |
| 120 | */ |
murgatroid99 | c02910b | 2016-02-17 12:59:26 -0800 | [diff] [blame] | 121 | exports.load = function load(filename, format, options) { |
murgatroid99 | b5b5f02 | 2017-03-16 16:42:10 -0700 | [diff] [blame] | 122 | options = _.defaults(options, common.defaultGrpcOptions); |
murgatroid99 | b1c69e4 | 2017-05-02 14:01:35 -0700 | [diff] [blame] | 123 | options.protobufjsVersion = 5; |
| 124 | if (!format) { |
| 125 | format = 'proto'; |
| 126 | } |
| 127 | var convertFieldsToCamelCaseOriginal = ProtoBuf.convertFieldsToCamelCase; |
| 128 | if(options && options.hasOwnProperty('convertFieldsToCamelCase')) { |
| 129 | ProtoBuf.convertFieldsToCamelCase = options.convertFieldsToCamelCase; |
| 130 | } |
| 131 | var builder; |
| 132 | try { |
| 133 | switch(format) { |
| 134 | case 'proto': |
| 135 | builder = ProtoBuf.loadProtoFile(filename); |
| 136 | break; |
| 137 | case 'json': |
| 138 | builder = ProtoBuf.loadJsonFile(filename); |
| 139 | break; |
| 140 | default: |
| 141 | throw new Error('Unrecognized format "' + format + '"'); |
| 142 | } |
| 143 | } finally { |
| 144 | ProtoBuf.convertFieldsToCamelCase = convertFieldsToCamelCaseOriginal; |
| 145 | } |
| 146 | return loadObject(builder.ns, options); |
murgatroid99 | 9cd90a6 | 2015-07-27 11:23:13 -0700 | [diff] [blame] | 147 | }; |
murgatroid99 | cca5ffa | 2015-01-15 14:06:56 -0800 | [diff] [blame] | 148 | |
murgatroid99 | 1d2f289 | 2016-06-02 14:33:22 -0700 | [diff] [blame] | 149 | var log_template = _.template( |
| 150 | '{severity} {timestamp}\t{file}:{line}]\t{message}', |
| 151 | {interpolate: /{([\s\S]+?)}/g}); |
| 152 | |
murgatroid99 | cca5ffa | 2015-01-15 14:06:56 -0800 | [diff] [blame] | 153 | /** |
murgatroid99 | 9b9708a | 2016-06-01 11:42:20 -0700 | [diff] [blame] | 154 | * Sets the logger function for the gRPC module. For debugging purposes, the C |
| 155 | * core will log synchronously directly to stdout unless this function is |
| 156 | * called. Note: the output format here is intended to be informational, and |
| 157 | * is not guaranteed to stay the same in the future. |
| 158 | * Logs will be directed to logger.error. |
murgatroid99 | c92df4c | 2017-05-23 10:28:47 -0700 | [diff] [blame] | 159 | * @memberof grpc |
| 160 | * @alias grpc.setLogger |
murgatroid99 | 9b9708a | 2016-06-01 11:42:20 -0700 | [diff] [blame] | 161 | * @param {Console} logger A Console-like object. |
| 162 | */ |
| 163 | exports.setLogger = function setLogger(logger) { |
| 164 | common.logger = logger; |
murgatroid99 | 1d2f289 | 2016-06-02 14:33:22 -0700 | [diff] [blame] | 165 | grpc.setDefaultLoggerCallback(function(file, line, severity, |
| 166 | message, timestamp) { |
| 167 | logger.error(log_template({ |
| 168 | file: path.basename(file), |
| 169 | line: line, |
| 170 | severity: severity, |
| 171 | message: message, |
| 172 | timestamp: timestamp.toISOString() |
| 173 | })); |
murgatroid99 | 9b9708a | 2016-06-01 11:42:20 -0700 | [diff] [blame] | 174 | }); |
| 175 | }; |
| 176 | |
| 177 | /** |
| 178 | * Sets the logger verbosity for gRPC module logging. The options are members |
| 179 | * of the grpc.logVerbosity map. |
murgatroid99 | c92df4c | 2017-05-23 10:28:47 -0700 | [diff] [blame] | 180 | * @memberof grpc |
| 181 | * @alias grpc.setLogVerbosity |
murgatroid99 | 9b9708a | 2016-06-01 11:42:20 -0700 | [diff] [blame] | 182 | * @param {Number} verbosity The minimum severity to log |
| 183 | */ |
| 184 | exports.setLogVerbosity = function setLogVerbosity(verbosity) { |
| 185 | common.logVerbosity = verbosity; |
| 186 | grpc.setLogVerbosity(verbosity); |
| 187 | }; |
| 188 | |
murgatroid99 | 366e64d | 2015-07-15 17:01:49 -0700 | [diff] [blame] | 189 | exports.Server = server.Server; |
murgatroid99 | cca5ffa | 2015-01-15 14:06:56 -0800 | [diff] [blame] | 190 | |
murgatroid99 | 84e3cde | 2015-08-20 11:27:05 -0700 | [diff] [blame] | 191 | exports.Metadata = Metadata; |
| 192 | |
murgatroid99 | 7229f88 | 2017-05-10 16:24:05 -0700 | [diff] [blame] | 193 | exports.status = constants.status; |
murgatroid99 | 9cd90a6 | 2015-07-27 11:23:13 -0700 | [diff] [blame] | 194 | |
murgatroid99 | 7229f88 | 2017-05-10 16:24:05 -0700 | [diff] [blame] | 195 | exports.propagate = constants.propagate; |
murgatroid99 | 0b09457 | 2015-08-14 10:48:45 -0700 | [diff] [blame] | 196 | |
murgatroid99 | 7229f88 | 2017-05-10 16:24:05 -0700 | [diff] [blame] | 197 | exports.callError = constants.callError; |
murgatroid99 | b6ab1b4 | 2015-01-21 10:30:36 -0800 | [diff] [blame] | 198 | |
murgatroid99 | 7229f88 | 2017-05-10 16:24:05 -0700 | [diff] [blame] | 199 | exports.writeFlags = constants.writeFlags; |
murgatroid99 | 4a1474f | 2015-08-17 14:00:31 -0700 | [diff] [blame] | 200 | |
murgatroid99 | 7229f88 | 2017-05-10 16:24:05 -0700 | [diff] [blame] | 201 | exports.logVerbosity = constants.logVerbosity; |
murgatroid99 | 9b9708a | 2016-06-01 11:42:20 -0700 | [diff] [blame] | 202 | |
murgatroid99 | 153b09d | 2015-09-25 16:04:03 -0700 | [diff] [blame] | 203 | exports.credentials = require('./src/credentials.js'); |
murgatroid99 | b6ab1b4 | 2015-01-21 10:30:36 -0800 | [diff] [blame] | 204 | |
| 205 | /** |
| 206 | * ServerCredentials factories |
murgatroid99 | c92df4c | 2017-05-23 10:28:47 -0700 | [diff] [blame] | 207 | * @constructor ServerCredentials |
| 208 | * @memberof grpc |
murgatroid99 | b6ab1b4 | 2015-01-21 10:30:36 -0800 | [diff] [blame] | 209 | */ |
| 210 | exports.ServerCredentials = grpc.ServerCredentials; |
murgatroid99 | 8c3ed00 | 2015-02-18 15:00:56 -0800 | [diff] [blame] | 211 | |
murgatroid99 | 9cd90a6 | 2015-07-27 11:23:13 -0700 | [diff] [blame] | 212 | /** |
murgatroid99 | c92df4c | 2017-05-23 10:28:47 -0700 | [diff] [blame] | 213 | * Create insecure server credentials |
| 214 | * @name grpc.ServerCredentials.createInsecure |
| 215 | * @kind function |
| 216 | * @return grpc.ServerCredentials |
murgatroid99 | 9cd90a6 | 2015-07-27 11:23:13 -0700 | [diff] [blame] | 217 | */ |
murgatroid99 | c92df4c | 2017-05-23 10:28:47 -0700 | [diff] [blame] | 218 | |
| 219 | /** |
| 220 | * A private key and certificate pair |
| 221 | * @typedef {Object} grpc.ServerCredentials~keyCertPair |
| 222 | * @property {Buffer} privateKey The server's private key |
| 223 | * @property {Buffer} certChain The server's certificate chain |
| 224 | */ |
| 225 | |
| 226 | /** |
| 227 | * Create SSL server credentials |
| 228 | * @name grpc.ServerCredentials.createInsecure |
| 229 | * @kind function |
| 230 | * @param {?Buffer} rootCerts Root CA certificates for validating client |
| 231 | * certificates |
| 232 | * @param {Array<grpc.ServerCredentials~keyCertPair>} keyCertPairs A list of |
| 233 | * private key and certificate chain pairs to be used for authenticating |
| 234 | * the server |
| 235 | * @param {boolean} [checkClientCertificate=false] Indicates that the server |
| 236 | * should request and verify the client's certificates |
| 237 | * @return grpc.ServerCredentials |
| 238 | */ |
| 239 | |
murgatroid99 | e023e98 | 2015-03-18 17:17:33 -0700 | [diff] [blame] | 240 | exports.makeGenericClientConstructor = client.makeClientConstructor; |
murgatroid99 | 76ba1ff | 2015-08-28 14:57:04 -0700 | [diff] [blame] | 241 | |
murgatroid99 | 76ba1ff | 2015-08-28 14:57:04 -0700 | [diff] [blame] | 242 | exports.getClientChannel = client.getClientChannel; |
| 243 | |
murgatroid99 | 76ba1ff | 2015-08-28 14:57:04 -0700 | [diff] [blame] | 244 | exports.waitForClientReady = client.waitForClientReady; |
murgatroid99 | 9030c81 | 2016-09-16 13:25:08 -0700 | [diff] [blame] | 245 | |
murgatroid99 | c92df4c | 2017-05-23 10:28:47 -0700 | [diff] [blame] | 246 | /** |
| 247 | * @memberof grpc |
| 248 | * @alias grpc.closeClient |
| 249 | * @param {grpc.Client} client_obj The client to close |
| 250 | */ |
murgatroid99 | 9030c81 | 2016-09-16 13:25:08 -0700 | [diff] [blame] | 251 | exports.closeClient = function closeClient(client_obj) { |
murgatroid99 | ffac55d | 2017-05-05 13:54:03 -0700 | [diff] [blame] | 252 | client.Client.prototype.close.apply(client_obj); |
murgatroid99 | 9030c81 | 2016-09-16 13:25:08 -0700 | [diff] [blame] | 253 | }; |
murgatroid99 | ffac55d | 2017-05-05 13:54:03 -0700 | [diff] [blame] | 254 | |
murgatroid99 | ffac55d | 2017-05-05 13:54:03 -0700 | [diff] [blame] | 255 | exports.Client = client.Client; |