@terascope/job-components/dist/src/register-apis.js

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.registerApis = exports.getClient = exports.getAssetPath = exports.getOpConfig = void 0;
const fs_1 = __importDefault(require("fs"));
const path_1 = __importDefault(require("path"));
const utils_1 = require("@terascope/utils");
const execution_context_1 = require("./execution-context");
/** Get the first opConfig from an operation name */
function getOpConfig(job, name) {
    return job.operations.find((op) => op._op === name);
}
exports.getOpConfig = getOpConfig;
/* Get the asset path from a asset name or ID */
async function getAssetPath(assetDir, assets, name) {
    if (!assetDir) {
        throw new Error('No asset_directroy has been configured, cannot get asset path');
    }
    const assetIds = assets || [];
    if (!name) {
        throw new Error('Invalid asset name');
    }
    if (name.length === 40) {
        const assetPath = path_1.default.join(assetDir, name);
        if (fs_1.default.existsSync(assetDir))
            return assetPath;
    }
    const [assetName] = name.split(':').map((s) => s.trim());
    for (const id of assetIds) {
        const rawAssetJSON = fs_1.default.readFileSync(path_1.default.join(assetDir, id, 'asset.json'));
        const assetJSON = utils_1.parseJSON(rawAssetJSON);
        if (assetJSON.name === assetName) {
            return path_1.default.join(assetDir, id);
        }
    }
    throw new Error(`Unable to find asset "${name}"`);
}
exports.getAssetPath = getAssetPath;
/*
 * This will request a connection based on the 'connection' attribute of
 * an opConfig. Intended as a context API endpoint.
 */
function getClient(context, config, type) {
    const clientConfig = {
        type,
        cached: true,
        endpoint: 'default',
    };
    if (config && config.connection) {
        clientConfig.endpoint = config.connection || 'default';
        const isCached = config.connection_cache != null;
        clientConfig.cached = isCached ? config.connection_cache : true;
    }
    else {
        clientConfig.endpoint = 'default';
        clientConfig.cached = true;
    }
    return context.foundation.getConnection(clientConfig).client;
}
exports.getClient = getClient;
function registerApis(context, job, assetIds) {
    const cleanupApis = ['op_runner', 'executionContext', 'job_runner', 'assets'];
    for (const api of cleanupApis) {
        if (context.apis[api] != null) {
            delete context.apis[api];
        }
    }
    context.apis.registerAPI('executionContext', new execution_context_1.ExecutionContextAPI(context, job));
    context.apis.registerAPI('op_runner', {
        getClient(config, type) {
            return getClient(context, config, type);
        },
    });
    context.apis.registerAPI('job_runner', {
        getOpConfig(name) {
            return getOpConfig(job, name);
        },
    });
    const assetDir = context.sysconfig.teraslice.assets_directory;
    context.apis.registerAPI('assets', {
        getPath(name) {
            return getAssetPath(assetDir || '', assetIds || job.assets, name);
        },
    });
}
exports.registerApis = registerApis;
//# sourceMappingURL=register-apis.js.map