@datastax/astra-db-ts
Version:
Data API TypeScript client
171 lines (170 loc) • 9.59 kB
JavaScript
;
// Copyright Datastax, Inc
// SPDX-License-Identifier: Apache-2.0
// noinspection ExceptionCaughtLocallyJS
var _AstraAdmin_defaultOpts, _AstraAdmin_httpClient, _AstraAdmin_environment;
Object.defineProperty(exports, "__esModule", { value: true });
exports.AstraAdmin = void 0;
const tslib_1 = require("tslib");
const astra_db_admin_js_1 = require("../administration/astra-db-admin.js");
const db_js_1 = require("../db/db.js");
const utils_js_1 = require("../administration/utils.js");
const constants_js_1 = require("../lib/api/constants.js");
const devops_api_http_client_js_1 = require("../lib/api/clients/devops-api-http-client.js");
const index_js_1 = require("../lib/index.js");
const utils_js_2 = require("../lib/utils.js");
const constants_js_2 = require("../lib/constants.js");
const timeouts_js_1 = require("../lib/api/timeouts/timeouts.js");
const admin_opts_handler_js_1 = require("../client/opts-handlers/admin-opts-handler.js");
const db_opts_handler_js_1 = require("../client/opts-handlers/db-opts-handler.js");
class AstraAdmin extends index_js_1.HierarchicalLogger {
constructor(rootOpts, adminOpts) {
const defaultOpts = {
...rootOpts,
adminOptions: admin_opts_handler_js_1.AdminOptsHandler.concat([rootOpts.adminOptions, adminOpts]),
dbOptions: {
...rootOpts.dbOptions,
token: index_js_1.TokenProvider.opts.concat([rootOpts.adminOptions.adminToken, adminOpts?.adminToken, rootOpts.dbOptions.token]),
},
};
super(rootOpts.client, defaultOpts.adminOptions.logging);
_AstraAdmin_defaultOpts.set(this, void 0);
_AstraAdmin_httpClient.set(this, void 0);
_AstraAdmin_environment.set(this, void 0);
tslib_1.__classPrivateFieldSet(this, _AstraAdmin_defaultOpts, defaultOpts, "f");
tslib_1.__classPrivateFieldSet(this, _AstraAdmin_environment, tslib_1.__classPrivateFieldGet(this, _AstraAdmin_defaultOpts, "f").adminOptions.astraEnv ?? 'prod', "f");
tslib_1.__classPrivateFieldSet(this, _AstraAdmin_httpClient, new devops_api_http_client_js_1.DevOpsAPIHttpClient({
baseUrl: tslib_1.__classPrivateFieldGet(this, _AstraAdmin_defaultOpts, "f").adminOptions.endpointUrl ?? constants_js_1.DEFAULT_DEVOPS_API_ENDPOINTS[tslib_1.__classPrivateFieldGet(this, _AstraAdmin_environment, "f")],
logger: this,
fetchCtx: rootOpts.fetchCtx,
caller: rootOpts.caller,
tokenProvider: tslib_1.__classPrivateFieldGet(this, _AstraAdmin_defaultOpts, "f").adminOptions.adminToken,
additionalHeaders: tslib_1.__classPrivateFieldGet(this, _AstraAdmin_defaultOpts, "f").additionalHeaders,
timeoutDefaults: timeouts_js_1.Timeouts.cfg.concat([rootOpts.adminOptions.timeoutDefaults, tslib_1.__classPrivateFieldGet(this, _AstraAdmin_defaultOpts, "f").adminOptions.timeoutDefaults]),
}), "f");
Object.defineProperty(this, constants_js_2.$CustomInspect, {
value: () => 'AstraAdmin()',
});
}
db(endpointOrId, regionOrOptions, maybeOptions) {
const [endpoint, dbOpts] = resolveEndpointOrIdOverload(endpointOrId, regionOrOptions, maybeOptions);
return new db_js_1.Db(tslib_1.__classPrivateFieldGet(this, _AstraAdmin_defaultOpts, "f"), endpoint, db_opts_handler_js_1.DbOptsHandler.parse(dbOpts));
}
dbAdmin(endpointOrId, regionOrOptions, maybeOptions) {
const [endpoint, dbOpts] = resolveEndpointOrIdOverload(endpointOrId, regionOrOptions, maybeOptions);
return new astra_db_admin_js_1.AstraDbAdmin(this.db(endpoint, dbOpts), tslib_1.__classPrivateFieldGet(this, _AstraAdmin_defaultOpts, "f"), admin_opts_handler_js_1.AdminOptsHandler.empty, tslib_1.__classPrivateFieldGet(this, _AstraAdmin_defaultOpts, "f").dbOptions.token, endpoint);
}
async dbInfo(id, options) {
const tm = tslib_1.__classPrivateFieldGet(this, _AstraAdmin_httpClient, "f").tm.single('databaseAdminTimeoutMs', options);
const resp = await tslib_1.__classPrivateFieldGet(this, _AstraAdmin_httpClient, "f").request({
method: constants_js_1.HttpMethods.Get,
path: `/databases/${id}`,
methodName: 'admin.dbInfo',
}, tm);
return (0, utils_js_1.buildAstraDatabaseAdminInfo)(resp.data, tslib_1.__classPrivateFieldGet(this, _AstraAdmin_environment, "f"));
}
async listDatabases(options) {
const params = {};
if (typeof options?.include === 'string') {
params.include = options.include;
}
if (typeof options?.provider === 'string') {
params.provider = options.provider;
}
if (typeof options?.limit === 'number') {
params.limit = String(options.limit);
}
if (typeof options?.startingAfter === 'string') {
params.starting_after = options.startingAfter;
}
const tm = tslib_1.__classPrivateFieldGet(this, _AstraAdmin_httpClient, "f").tm.single('databaseAdminTimeoutMs', options);
const resp = await tslib_1.__classPrivateFieldGet(this, _AstraAdmin_httpClient, "f").request({
method: constants_js_1.HttpMethods.Get,
path: `/databases`,
params: params,
methodName: 'admin.listDatabases',
}, tm);
return resp.data.map((d) => (0, utils_js_1.buildAstraDatabaseAdminInfo)(d, tslib_1.__classPrivateFieldGet(this, _AstraAdmin_environment, "f")));
}
async createDatabase(config, options) {
const definition = {
capacityUnits: 1,
tier: 'serverless',
dbType: 'vector',
keyspace: config.keyspace || constants_js_1.DEFAULT_KEYSPACE,
...config,
};
const tm = tslib_1.__classPrivateFieldGet(this, _AstraAdmin_httpClient, "f").tm.multipart('databaseAdminTimeoutMs', options);
const resp = await tslib_1.__classPrivateFieldGet(this, _AstraAdmin_httpClient, "f").requestLongRunning({
method: constants_js_1.HttpMethods.Post,
path: '/databases',
data: definition,
methodName: 'admin.createDatabase',
}, {
id: (resp) => resp.headers.location,
target: 'ACTIVE',
legalStates: ['INITIALIZING', 'PENDING'],
defaultPollInterval: 10000,
timeoutManager: tm,
options,
});
const endpoint = (0, utils_js_2.buildAstraEndpoint)(resp.headers.location, definition.region);
const db = this.db(endpoint, { ...options?.dbOptions, keyspace: definition.keyspace });
return new astra_db_admin_js_1.AstraDbAdmin(db, tslib_1.__classPrivateFieldGet(this, _AstraAdmin_defaultOpts, "f"), admin_opts_handler_js_1.AdminOptsHandler.empty, tslib_1.__classPrivateFieldGet(this, _AstraAdmin_defaultOpts, "f").adminOptions.adminToken, endpoint);
}
async dropDatabase(db, options) {
const id = typeof db === 'string' ? db : db.id;
const tm = tslib_1.__classPrivateFieldGet(this, _AstraAdmin_httpClient, "f").tm.multipart('databaseAdminTimeoutMs', options);
await tslib_1.__classPrivateFieldGet(this, _AstraAdmin_httpClient, "f").requestLongRunning({
method: constants_js_1.HttpMethods.Post,
path: `/databases/${id}/terminate`,
methodName: 'admin.dropDatabase',
}, {
id: id,
target: 'TERMINATED',
legalStates: ['TERMINATING'],
defaultPollInterval: 10000,
timeoutManager: tm,
options,
});
}
async findAvailableRegions(options) {
const tm = tslib_1.__classPrivateFieldGet(this, _AstraAdmin_httpClient, "f").tm.single('databaseAdminTimeoutMs', options);
const filterByOrg = options?.onlyOrgEnabledRegions !== false ? 'enabled' : 'disabled';
const resp = await tslib_1.__classPrivateFieldGet(this, _AstraAdmin_httpClient, "f").request({
method: constants_js_1.HttpMethods.Get,
path: '/regions/serverless',
params: {
'filter-by-org': filterByOrg,
'region-type': 'vector',
},
methodName: 'admin.findAvailableRegions',
}, tm);
return resp.data.map((region) => ({
classification: region.classification,
cloudProvider: region.cloudProvider,
displayName: region.displayName,
enabled: region.enabled,
name: region.name,
reservedForQualifiedUsers: region.reservedForQualifiedUsers,
zone: region.zone,
}));
}
get _httpClient() {
return tslib_1.__classPrivateFieldGet(this, _AstraAdmin_httpClient, "f");
}
}
exports.AstraAdmin = AstraAdmin;
_AstraAdmin_defaultOpts = new WeakMap(), _AstraAdmin_httpClient = new WeakMap(), _AstraAdmin_environment = new WeakMap();
const resolveEndpointOrIdOverload = (endpointOrId, regionOrOptions, maybeOptions) => {
const dbOpts = (typeof regionOrOptions === 'string')
? maybeOptions
: regionOrOptions;
if (typeof regionOrOptions === 'string' && (endpointOrId.startsWith('http'))) {
throw new Error('Unexpected db() argument: database id can\'t start with "http(s)://". Did you mean to call `.db(endpoint, { keyspace })`?');
}
const endpoint = (typeof regionOrOptions === 'string')
? 'https://' + endpointOrId + '-' + regionOrOptions + '.apps.astra.datastax.com'
: endpointOrId;
return [endpoint, dbOpts];
};