88 lines
3.5 KiB
JavaScript
88 lines
3.5 KiB
JavaScript
"use strict";
|
|
/**
|
|
* Copyright 2021 Google LLC
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.generateServiceStub = void 0;
|
|
/* global window */
|
|
/* global AbortController */
|
|
const node_fetch_1 = require("node-fetch");
|
|
const abort_controller_1 = require("abort-controller");
|
|
const featureDetection_1 = require("./featureDetection");
|
|
function generateServiceStub(rpcs, protocol, servicePath, servicePort, authClient, requestEncoder, responseDecoder) {
|
|
const fetch = featureDetection_1.hasWindowFetch()
|
|
? window.fetch
|
|
: node_fetch_1.default;
|
|
const serviceStub = {};
|
|
for (const [rpcName, rpc] of Object.entries(rpcs)) {
|
|
serviceStub[rpcName] = (request, options, _metadata, callback) => {
|
|
// We cannot use async-await in this function because we need to return the canceller object as soon as possible.
|
|
// Using plain old promises instead.
|
|
const cancelController = featureDetection_1.hasAbortController()
|
|
? new AbortController()
|
|
: new abort_controller_1.AbortController();
|
|
const cancelSignal = cancelController.signal;
|
|
let cancelRequested = false;
|
|
const fetchParameters = requestEncoder(rpc, protocol, servicePath, servicePort, request);
|
|
const url = fetchParameters.url;
|
|
const headers = fetchParameters.headers;
|
|
for (const key of Object.keys(options)) {
|
|
headers[key] = options[key][0];
|
|
}
|
|
authClient
|
|
.getRequestHeaders()
|
|
.then(authHeader => {
|
|
const fetchRequest = {
|
|
headers: {
|
|
...authHeader,
|
|
...headers,
|
|
},
|
|
body: fetchParameters.body,
|
|
method: fetchParameters.method,
|
|
signal: cancelSignal,
|
|
};
|
|
if (fetchParameters.method === 'get' ||
|
|
fetchParameters.method === 'delete') {
|
|
delete fetchRequest['body'];
|
|
}
|
|
return fetch(url, fetchRequest);
|
|
})
|
|
.then((response) => {
|
|
return Promise.all([
|
|
Promise.resolve(response.ok),
|
|
response.arrayBuffer(),
|
|
]);
|
|
})
|
|
.then(([ok, buffer]) => {
|
|
const response = responseDecoder(rpc, ok, buffer);
|
|
callback(null, response);
|
|
})
|
|
.catch((err) => {
|
|
if (!cancelRequested || err.name !== 'AbortError') {
|
|
callback(err);
|
|
}
|
|
});
|
|
return {
|
|
cancel: () => {
|
|
cancelRequested = true;
|
|
cancelController.abort();
|
|
},
|
|
};
|
|
};
|
|
}
|
|
return serviceStub;
|
|
}
|
|
exports.generateServiceStub = generateServiceStub;
|
|
//# sourceMappingURL=fallbackServiceStub.js.map
|