mirror of
https://github.com/titanscouting/tra-analysis.git
synced 2025-01-04 21:11:23 +00:00
393 lines
11 KiB
JavaScript
393 lines
11 KiB
JavaScript
/*!
|
|
* Copyright 2015 Google Inc. All Rights Reserved.
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
/*!
|
|
* @module common/service-object
|
|
*/
|
|
|
|
'use strict';
|
|
|
|
const arrify = require('arrify');
|
|
const exec = require('methmeth');
|
|
const extend = require('extend');
|
|
const is = require('is');
|
|
|
|
/**
|
|
* @type {module:common/util}
|
|
* @private
|
|
*/
|
|
const util = require('./util.js');
|
|
|
|
/**
|
|
* ServiceObject is a base class, meant to be inherited from by a "service
|
|
* object," like a BigQuery dataset or Storage bucket.
|
|
*
|
|
* Most of the time, these objects share common functionality; they can be
|
|
* created or deleted, and you can get or set their metadata.
|
|
*
|
|
* By inheriting from this class, a service object will be extended with these
|
|
* shared behaviors. Note that any method can be overridden when the service
|
|
* object requires specific behavior.
|
|
*
|
|
* @constructor
|
|
* @alias module:common/service-object
|
|
*
|
|
* @private
|
|
*
|
|
* @param {object} config - Configuration object.
|
|
* @param {string} config.baseUrl - The base URL to make API requests to.
|
|
* @param {string} config.createMethod - The method which creates this object.
|
|
* @param {string=} config.id - The identifier of the object. For example, the
|
|
* name of a Storage bucket or Pub/Sub topic.
|
|
* @param {object=} config.methods - A map of each method name that should be
|
|
* inherited.
|
|
* @param {object} config.methods[].reqOpts - Default request options for this
|
|
* particular method. A common use case is when `setMetadata` requires a
|
|
* `PUT` method to override the default `PATCH`.
|
|
* @param {object} config.parent - The parent service instance. For example, an
|
|
* instance of Storage if the object is Bucket.
|
|
*/
|
|
function ServiceObject(config) {
|
|
const self = this;
|
|
|
|
util.privatize(this, 'metadata', {});
|
|
|
|
util.privatize(this, 'baseUrl', config.baseUrl);
|
|
util.privatize(this, 'parent', config.parent); // Parent class.
|
|
util.privatize(this, 'id', config.id); // Name or ID (e.g. dataset ID, bucket name, etc.)
|
|
util.privatize(this, 'createMethod', config.createMethod);
|
|
util.privatize(this, 'methods', config.methods || {});
|
|
util.privatize(this, 'interceptors', []);
|
|
util.privatize(this, 'Promise', this.parent.Promise);
|
|
|
|
if (config.methods) {
|
|
const allMethodNames = Object.keys(ServiceObject.prototype);
|
|
allMethodNames
|
|
.filter(function(methodName) {
|
|
return (
|
|
// All ServiceObjects need `request`.
|
|
!/^request/.test(methodName) &&
|
|
// The ServiceObject didn't redefine the method.
|
|
self[methodName] === ServiceObject.prototype[methodName] &&
|
|
// This method isn't wanted.
|
|
!config.methods[methodName]
|
|
);
|
|
})
|
|
.forEach(function(methodName) {
|
|
self[methodName] = undefined;
|
|
});
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Create the object.
|
|
*
|
|
* @param {object=} options - Configuration object.
|
|
* @param {function} callback - The callback function.
|
|
* @param {?error} callback.err - An error returned while making this request.
|
|
* @param {object} callback.instance - The instance.
|
|
* @param {object} callback.apiResponse - The full API response.
|
|
*/
|
|
ServiceObject.prototype.create = function(options, callback) {
|
|
const self = this;
|
|
const args = [this.id];
|
|
|
|
if (is.fn(options)) {
|
|
callback = options;
|
|
}
|
|
|
|
if (is.object(options)) {
|
|
args.push(options);
|
|
}
|
|
|
|
// Wrap the callback to return *this* instance of the object, not the newly-
|
|
// created one.
|
|
function onCreate(err, instance) {
|
|
const args = [].slice.call(arguments);
|
|
|
|
if (!err) {
|
|
self.metadata = instance.metadata;
|
|
args[1] = self; // replace the created `instance` with this one.
|
|
}
|
|
|
|
callback.apply(null, args);
|
|
}
|
|
|
|
args.push(onCreate);
|
|
|
|
this.createMethod.apply(null, args);
|
|
};
|
|
|
|
/**
|
|
* Delete the object.
|
|
*
|
|
* @param {function=} callback - The callback function.
|
|
* @param {?error} callback.err - An error returned while making this request.
|
|
* @param {object} callback.apiResponse - The full API response.
|
|
*/
|
|
ServiceObject.prototype.delete = function(callback) {
|
|
const methodConfig = this.methods.delete || {};
|
|
callback = callback || util.noop;
|
|
|
|
const reqOpts = extend(
|
|
{
|
|
method: 'DELETE',
|
|
uri: '',
|
|
},
|
|
methodConfig.reqOpts
|
|
);
|
|
|
|
// The `request` method may have been overridden to hold any special behavior.
|
|
// Ensure we call the original `request` method.
|
|
ServiceObject.prototype.request.call(this, reqOpts, function(err, resp) {
|
|
callback(err, resp);
|
|
});
|
|
};
|
|
|
|
/**
|
|
* Check if the object exists.
|
|
*
|
|
* @param {function} callback - The callback function.
|
|
* @param {?error} callback.err - An error returned while making this request.
|
|
* @param {boolean} callback.exists - Whether the object exists or not.
|
|
*/
|
|
ServiceObject.prototype.exists = function(callback) {
|
|
this.get(function(err) {
|
|
if (err) {
|
|
if (err.code === 404) {
|
|
callback(null, false);
|
|
} else {
|
|
callback(err);
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
callback(null, true);
|
|
});
|
|
};
|
|
|
|
/**
|
|
* Get the object if it exists. Optionally have the object created if an options
|
|
* object is provided with `autoCreate: true`.
|
|
*
|
|
* @param {object=} config - The configuration object that will be used to
|
|
* create the object if necessary.
|
|
* @param {boolean} config.autoCreate - Create the object if it doesn't already
|
|
* exist.
|
|
* @param {function} callback - The callback function.
|
|
* @param {?error} callback.err - An error returned while making this request.
|
|
* @param {object} callback.instance - The instance.
|
|
* @param {object} callback.apiResponse - The full API response.
|
|
*/
|
|
ServiceObject.prototype.get = function(config, callback) {
|
|
const self = this;
|
|
|
|
if (is.fn(config)) {
|
|
callback = config;
|
|
config = {};
|
|
}
|
|
|
|
config = config || {};
|
|
|
|
const autoCreate = config.autoCreate && is.fn(this.create);
|
|
delete config.autoCreate;
|
|
|
|
function onCreate(err, instance, apiResponse) {
|
|
if (err) {
|
|
if (err.code === 409) {
|
|
self.get(config, callback);
|
|
return;
|
|
}
|
|
|
|
callback(err, null, apiResponse);
|
|
return;
|
|
}
|
|
|
|
callback(null, instance, apiResponse);
|
|
}
|
|
|
|
this.getMetadata(function(err, metadata) {
|
|
if (err) {
|
|
if (err.code === 404 && autoCreate) {
|
|
const args = [];
|
|
|
|
if (!is.empty(config)) {
|
|
args.push(config);
|
|
}
|
|
|
|
args.push(onCreate);
|
|
|
|
self.create.apply(self, args);
|
|
return;
|
|
}
|
|
|
|
callback(err, null, metadata);
|
|
return;
|
|
}
|
|
|
|
callback(null, self, metadata);
|
|
});
|
|
};
|
|
|
|
/**
|
|
* Get the metadata of this object.
|
|
*
|
|
* @param {function} callback - The callback function.
|
|
* @param {?error} callback.err - An error returned while making this request.
|
|
* @param {object} callback.metadata - The metadata for this object.
|
|
* @param {object} callback.apiResponse - The full API response.
|
|
*/
|
|
ServiceObject.prototype.getMetadata = function(callback) {
|
|
const self = this;
|
|
|
|
const methodConfig = this.methods.getMetadata || {};
|
|
|
|
const reqOpts = extend(
|
|
{
|
|
uri: '',
|
|
},
|
|
methodConfig.reqOpts
|
|
);
|
|
|
|
// The `request` method may have been overridden to hold any special behavior.
|
|
// Ensure we call the original `request` method.
|
|
ServiceObject.prototype.request.call(this, reqOpts, function(err, resp) {
|
|
if (err) {
|
|
callback(err, null, resp);
|
|
return;
|
|
}
|
|
|
|
self.metadata = resp;
|
|
|
|
callback(null, self.metadata, resp);
|
|
});
|
|
};
|
|
|
|
/**
|
|
* Set the metadata for this object.
|
|
*
|
|
* @param {object} metadata - The metadata to set on this object.
|
|
* @param {function=} callback - The callback function.
|
|
* @param {?error} callback.err - An error returned while making this request.
|
|
* @param {object} callback.instance - The instance.
|
|
* @param {object} callback.apiResponse - The full API response.
|
|
*/
|
|
ServiceObject.prototype.setMetadata = function(metadata, callback) {
|
|
const self = this;
|
|
|
|
callback = callback || util.noop;
|
|
|
|
const methodConfig = this.methods.setMetadata || {};
|
|
|
|
const reqOpts = extend(
|
|
true,
|
|
{
|
|
method: 'PATCH',
|
|
uri: '',
|
|
json: metadata,
|
|
},
|
|
methodConfig.reqOpts
|
|
);
|
|
|
|
// The `request` method may have been overridden to hold any special behavior.
|
|
// Ensure we call the original `request` method.
|
|
ServiceObject.prototype.request.call(this, reqOpts, function(err, resp) {
|
|
if (err) {
|
|
callback(err, resp);
|
|
return;
|
|
}
|
|
|
|
self.metadata = resp;
|
|
|
|
callback(null, resp);
|
|
});
|
|
};
|
|
|
|
/**
|
|
* Make an authenticated API request.
|
|
*
|
|
* @private
|
|
*
|
|
* @param {object} reqOpts - Request options that are passed to `request`.
|
|
* @param {string} reqOpts.uri - A URI relative to the baseUrl.
|
|
* @param {function} callback - The callback function passed to `request`.
|
|
*/
|
|
ServiceObject.prototype.request_ = function(reqOpts, callback) {
|
|
reqOpts = extend(true, {}, reqOpts);
|
|
|
|
const isAbsoluteUrl = reqOpts.uri.indexOf('http') === 0;
|
|
|
|
const uriComponents = [this.baseUrl, this.id || '', reqOpts.uri];
|
|
|
|
if (isAbsoluteUrl) {
|
|
uriComponents.splice(0, uriComponents.indexOf(reqOpts.uri));
|
|
}
|
|
|
|
reqOpts.uri = uriComponents
|
|
.filter(exec('trim')) // Limit to non-empty strings.
|
|
.map(function(uriComponent) {
|
|
const trimSlashesRegex = /^\/*|\/*$/g;
|
|
return uriComponent.replace(trimSlashesRegex, '');
|
|
})
|
|
.join('/');
|
|
|
|
const childInterceptors = arrify(reqOpts.interceptors_);
|
|
const localInterceptors = [].slice.call(this.interceptors);
|
|
|
|
reqOpts.interceptors_ = childInterceptors.concat(localInterceptors);
|
|
|
|
if (!callback) {
|
|
return this.parent.requestStream(reqOpts);
|
|
}
|
|
|
|
this.parent.request(reqOpts, callback);
|
|
};
|
|
|
|
/**
|
|
* Make an authenticated API request.
|
|
*
|
|
* @private
|
|
*
|
|
* @param {object} reqOpts - Request options that are passed to `request`.
|
|
* @param {string} reqOpts.uri - A URI relative to the baseUrl.
|
|
* @param {function} callback - The callback function passed to `request`.
|
|
*/
|
|
ServiceObject.prototype.request = function(reqOpts, callback) {
|
|
ServiceObject.prototype.request_.call(this, reqOpts, callback);
|
|
};
|
|
|
|
/**
|
|
* Make an authenticated API request.
|
|
*
|
|
* @private
|
|
*
|
|
* @param {object} reqOpts - Request options that are passed to `request`.
|
|
* @param {string} reqOpts.uri - A URI relative to the baseUrl.
|
|
*/
|
|
ServiceObject.prototype.requestStream = function(reqOpts) {
|
|
return ServiceObject.prototype.request_.call(this, reqOpts);
|
|
};
|
|
|
|
/*! Developer Documentation
|
|
*
|
|
* All async methods (except for streams) will return a Promise in the event
|
|
* that a callback is omitted.
|
|
*/
|
|
util.promisifyAll(ServiceObject);
|
|
|
|
module.exports = ServiceObject;
|