@dbx-tools/appkit-mastra 0.6.8 → 0.6.9

package/lib/src/databricks-aitools.js

@@ -0,0 +1,150 @@
+/**
+ * Startup provisioning of Databricks AI Tools skills for a Mastra workspace.
+ *
+ * Databricks AI Tools (`databricks aitools`) are Databricks-owned Agent-Skill
+ * (`SKILL.md`) trees installed and kept up to date through the Databricks CLI.
+ * The CLI installs them globally under `~/.databricks/aitools/skills/<name>/`
+ * (with a `.state.json` manifest) and can also write a resolved, agent-agnostic
+ * copy to any directory with `databricks aitools install --path <dir>
+ * --skills-only`.
+ *
+ * {@link provisionDatabricksAITools} folds those skills into a Mastra workspace
+ * as extra LOCAL skill scan paths so an agent can use them without anyone
+ * hand-copying `SKILL.md` files. It never reimplements the CLI's sourcing: it
+ * either points Mastra at the already-installed global tree, or shells out to
+ * the CLI to materialize a curated subset into a temp dir.
+ *
+ * The option is `false | true | "auto" | DatabricksAIToolsOptions`:
+ *
+ * - `false` (default) - off.
+ * - `"auto"` - enable only when the global tree already exists OR the
+ *   `databricks` CLI is resolvable; otherwise silently no-op. The "on if the
+ *   CLI is around" default.
+ * - `true` - required: fail startup (subject to `failOnError`) when neither the
+ *   installed tree nor the CLI can supply skills.
+ * - an options bag for a specific `skills` subset, `experimental` skills, a
+ *   `refresh` that re-runs the CLI even when a tree exists, or an explicit
+ *   `path`.
+ *
+ * @module
+ */
+import { mkdtemp } from "node:fs/promises";
+import { existsSync } from "node:fs";
+import { homedir, tmpdir } from "node:os";
+import { join } from "node:path";
+import { spawn } from "@dbx-tools/core";
+import { error, log, string } from "@dbx-tools/shared-core";
+const logger = log.logger("mastra/databricks-aitools");
+/** Global skills tree the `databricks aitools` CLI installs into. */
+const GLOBAL_AITOOLS_SKILLS_PATH = join(homedir(), ".databricks", "aitools", "skills");
+/** The Databricks CLI binary name; resolved on `PATH`. */
+const DATABRICKS_CLI = "databricks";
+/* ------------------------------- helpers ------------------------------- */
+/** Normalize the `databricksAITools` option into a flat options bag, or `undefined` when off. */
+export function normalizeDatabricksAIToolsOption(option) {
+    if (option === undefined || option === false)
+        return undefined;
+    if (option === true)
+        return { mode: "require" };
+    if (option === "auto")
+        return { mode: "auto" };
+    return { mode: option.mode ?? "auto", ...option };
+}
+/**
+ * Materialize Databricks AI Tools skills at app startup and return them as
+ * extra local skill scan paths.
+ *
+ * Resolution: reuse the installed global tree when present (no CLI call) unless
+ * `refresh`/`skills`/`path` ask for a fresh CLI materialization; otherwise run
+ * `databricks aitools install --path <dir> --skills-only` when the CLI resolves.
+ * A `"require"` provision that yields nothing fails startup unless
+ * `failOnError: false`.
+ */
+export async function provisionDatabricksAITools(option) {
+    const options = normalizeDatabricksAIToolsOption(option);
+    const empty = { localSkillPaths: [] };
+    if (!options)
+        return empty;
+    const mode = options.mode ?? "auto";
+    const failOnError = options.failOnError ?? mode === "require";
+    const wantsCliFetch = options.refresh === true ||
+        options.path !== undefined ||
+        string.parseList(options.skills).length > 0 ||
+        options.experimental === true;
+    try {
+        // Fast path: reuse the already-installed global tree, no CLI call.
+        if (!wantsCliFetch && installedTreeExists()) {
+            logger.debug("using installed aitools tree", { path: GLOBAL_AITOOLS_SKILLS_PATH });
+            return { localSkillPaths: [GLOBAL_AITOOLS_SKILLS_PATH], source: "installed" };
+        }
+        const cliPath = await resolveCli(options.cli);
+        if (cliPath) {
+            const dir = await materializeViaCli(cliPath, options);
+            if (dir)
+                return { localSkillPaths: [dir], source: "cli" };
+        }
+        // Nothing fetched. Fall back to the installed tree if it's there.
+        if (installedTreeExists()) {
+            return { localSkillPaths: [GLOBAL_AITOOLS_SKILLS_PATH], source: "installed" };
+        }
+        if (failOnError) {
+            throw new Error(`Databricks AI Tools requested but no installed skills tree at "${GLOBAL_AITOOLS_SKILLS_PATH}" and the "${DATABRICKS_CLI}" CLI is not resolvable. Install with \`databricks aitools install\`, or set databricksAITools: "auto".`);
+        }
+        logger.debug("databricks aitools unavailable; skipping");
+        return empty;
+    }
+    catch (err) {
+        if (failOnError) {
+            throw new Error(`failed to provision Databricks AI Tools skills: ${error.errorMessage(err)}`, { cause: error.toError(err) });
+        }
+        logger.warn("skipped", { error: error.errorMessage(err) });
+        return empty;
+    }
+}
+/** Whether the CLI's global skills tree exists and holds at least one skill. */
+function installedTreeExists() {
+    return existsSync(GLOBAL_AITOOLS_SKILLS_PATH);
+}
+/** Resolve the `databricks` CLI, returning its invocation path or `undefined`. */
+async function resolveCli(cli) {
+    const command = string.trimToNull(cli) ?? DATABRICKS_CLI;
+    try {
+        const result = await spawn(command, ["aitools", "version"], {
+            stdout: "capture",
+            stderr: "capture",
+            check: false,
+        });
+        if (result.exitCode === 0)
+            return command;
+    }
+    catch {
+        // CLI absent or `aitools` unsupported - caller falls back / fails.
+    }
+    return undefined;
+}
+/**
+ * Run `databricks aitools install --path <dir> --skills-only` to materialize a
+ * resolved, agent-agnostic skill set into a directory (no agents, no state).
+ */
+async function materializeViaCli(cli, options) {
+    const dir = string.trimToNull(options.path) ?? (await mkdtemp(join(tmpdir(), "databricks-aitools-")));
+    const args = ["aitools", "install", "--path", dir, "--skills-only"];
+    const skills = string.parseList(options.skills);
+    if (skills.length > 0)
+        args.push("--skills", skills.join(","));
+    if (options.experimental)
+        args.push("--experimental");
+    const result = await spawn(cli, args, {
+        stdout: "capture",
+        stderr: "capture",
+        check: false,
+    });
+    if (result.exitCode !== 0) {
+        throw new Error(`databricks aitools install failed (exit ${result.exitCode}): ${result.stderr || result.stdout}`);
+    }
+    if (!existsSync(dir))
+        return undefined;
+    logger.debug("materialized aitools skills via CLI", { path: dir, skills });
+    return dir;
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiZGF0YWJyaWNrcy1haXRvb2xzLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vc3JjL2RhdGFicmlja3MtYWl0b29scy50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQTs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7R0E2Qkc7QUFFSCxPQUFPLEVBQUUsT0FBTyxFQUFFLE1BQU0sa0JBQWtCLENBQUM7QUFDM0MsT0FBTyxFQUFFLFVBQVUsRUFBRSxNQUFNLFNBQVMsQ0FBQztBQUNyQyxPQUFPLEVBQUUsT0FBTyxFQUFFLE1BQU0sRUFBRSxNQUFNLFNBQVMsQ0FBQztBQUMxQyxPQUFPLEVBQUUsSUFBSSxFQUFFLE1BQU0sV0FBVyxDQUFDO0FBQ2pDLE9BQU8sRUFBRSxLQUFLLEVBQUUsTUFBTSxpQkFBaUIsQ0FBQztBQUN4QyxPQUFPLEVBQUUsS0FBSyxFQUFFLEdBQUcsRUFBRSxNQUFNLEVBQUUsTUFBTSx3QkFBd0IsQ0FBQztBQUU1RCxNQUFNLE1BQU0sR0FBRyxHQUFHLENBQUMsTUFBTSxDQUFDLDJCQUEyQixDQUFDLENBQUM7QUFFdkQscUVBQXFFO0FBQ3JFLE1BQU0sMEJBQTBCLEdBQUcsSUFBSSxDQUFDLE9BQU8sRUFBRSxFQUFFLGFBQWEsRUFBRSxTQUFTLEVBQUUsUUFBUSxDQUFDLENBQUM7QUFFdkYsMERBQTBEO0FBQzFELE1BQU0sY0FBYyxHQUFHLFlBQVksQ0FBQztBQWdEcEMsNkVBQTZFO0FBRTdFLGlHQUFpRztBQUNqRyxNQUFNLFVBQVUsZ0NBQWdDLENBQzlDLE1BQTJDO0lBRTNDLElBQUksTUFBTSxLQUFLLFNBQVMsSUFBSSxNQUFNLEtBQUssS0FBSztRQUFFLE9BQU8sU0FBUyxDQUFDO0lBQy9ELElBQUksTUFBTSxLQUFLLElBQUk7UUFBRSxPQUFPLEVBQUUsSUFBSSxFQUFFLFNBQVMsRUFBRSxDQUFDO0lBQ2hELElBQUksTUFBTSxLQUFLLE1BQU07UUFBRSxPQUFPLEVBQUUsSUFBSSxFQUFFLE1BQU0sRUFBRSxDQUFDO0lBQy9DLE9BQU8sRUFBRSxJQUFJLEVBQUUsTUFBTSxDQUFDLElBQUksSUFBSSxNQUFNLEVBQUUsR0FBRyxNQUFNLEVBQUUsQ0FBQztBQUNwRCxDQUFDO0FBRUQ7Ozs7Ozs7OztHQVNHO0FBQ0gsTUFBTSxDQUFDLEtBQUssVUFBVSwwQkFBMEIsQ0FDOUMsTUFBMkM7SUFFM0MsTUFBTSxPQUFPLEdBQUcsZ0NBQWdDLENBQUMsTUFBTSxDQUFDLENBQUM7SUFDekQsTUFBTSxLQUFLLEdBQWlDLEVBQUUsZUFBZSxFQUFFLEVBQUUsRUFBRSxDQUFDO0lBQ3BFLElBQUksQ0FBQyxPQUFPO1FBQUUsT0FBTyxLQUFLLENBQUM7SUFFM0IsTUFBTSxJQUFJLEdBQUcsT0FBTyxDQUFDLElBQUksSUFBSSxNQUFNLENBQUM7SUFDcEMsTUFBTSxXQUFXLEdBQUcsT0FBTyxDQUFDLFdBQVcsSUFBSSxJQUFJLEtBQUssU0FBUyxDQUFDO0lBQzlELE1BQU0sYUFBYSxHQUNqQixPQUFPLENBQUMsT0FBTyxLQUFLLElBQUk7UUFDeEIsT0FBTyxDQUFDLElBQUksS0FBSyxTQUFTO1FBQzFCLE1BQU0sQ0FBQyxTQUFTLENBQUMsT0FBTyxDQUFDLE1BQU0sQ0FBQyxDQUFDLE1BQU0sR0FBRyxDQUFDO1FBQzNDLE9BQU8sQ0FBQyxZQUFZLEtBQUssSUFBSSxDQUFDO0lBRWhDLElBQUksQ0FBQztRQUNILG1FQUFtRTtRQUNuRSxJQUFJLENBQUMsYUFBYSxJQUFJLG1CQUFtQixFQUFFLEVBQUUsQ0FBQztZQUM1QyxNQUFNLENBQUMsS0FBSyxDQUFDLDhCQUE4QixFQUFFLEVBQUUsSUFBSSxFQUFFLDBCQUEwQixFQUFFLENBQUMsQ0FBQztZQUNuRixPQUFPLEVBQUUsZUFBZSxFQUFFLENBQUMsMEJBQTBCLENBQUMsRUFBRSxNQUFNLEVBQUUsV0FBVyxFQUFFLENBQUM7UUFDaEYsQ0FBQztRQUVELE1BQU0sT0FBTyxHQUFHLE1BQU0sVUFBVSxDQUFDLE9BQU8sQ0FBQyxHQUFHLENBQUMsQ0FBQztRQUM5QyxJQUFJLE9BQU8sRUFBRSxDQUFDO1lBQ1osTUFBTSxHQUFHLEdBQUcsTUFBTSxpQkFBaUIsQ0FBQyxPQUFPLEVBQUUsT0FBTyxDQUFDLENBQUM7WUFDdEQsSUFBSSxHQUFHO2dCQUFFLE9BQU8sRUFBRSxlQUFlLEVBQUUsQ0FBQyxHQUFHLENBQUMsRUFBRSxNQUFNLEVBQUUsS0FBSyxFQUFFLENBQUM7UUFDNUQsQ0FBQztRQUVELGtFQUFrRTtRQUNsRSxJQUFJLG1CQUFtQixFQUFFLEVBQUUsQ0FBQztZQUMxQixPQUFPLEVBQUUsZUFBZSxFQUFFLENBQUMsMEJBQTBCLENBQUMsRUFBRSxNQUFNLEVBQUUsV0FBVyxFQUFFLENBQUM7UUFDaEYsQ0FBQztRQUVELElBQUksV0FBVyxFQUFFLENBQUM7WUFDaEIsTUFBTSxJQUFJLEtBQUssQ0FDYixrRUFBa0UsMEJBQTBCLGNBQWMsY0FBYyx5R0FBeUcsQ0FDbE8sQ0FBQztRQUNKLENBQUM7UUFDRCxNQUFNLENBQUMsS0FBSyxDQUFDLDBDQUEwQyxDQUFDLENBQUM7UUFDekQsT0FBTyxLQUFLLENBQUM7SUFDZixDQUFDO0lBQUMsT0FBTyxHQUFHLEVBQUUsQ0FBQztRQUNiLElBQUksV0FBVyxFQUFFLENBQUM7WUFDaEIsTUFBTSxJQUFJLEtBQUssQ0FDYixtREFBbUQsS0FBSyxDQUFDLFlBQVksQ0FBQyxHQUFHLENBQUMsRUFBRSxFQUM1RSxFQUFFLEtBQUssRUFBRSxLQUFLLENBQUMsT0FBTyxDQUFDLEdBQUcsQ0FBQyxFQUFFLENBQzlCLENBQUM7UUFDSixDQUFDO1FBQ0QsTUFBTSxDQUFDLElBQUksQ0FBQyxTQUFTLEVBQUUsRUFBRSxLQUFLLEVBQUUsS0FBSyxDQUFDLFlBQVksQ0FBQyxHQUFHLENBQUMsRUFBRSxDQUFDLENBQUM7UUFDM0QsT0FBTyxLQUFLLENBQUM7SUFDZixDQUFDO0FBQ0gsQ0FBQztBQUVELGdGQUFnRjtBQUNoRixTQUFTLG1CQUFtQjtJQUMxQixPQUFPLFVBQVUsQ0FBQywwQkFBMEIsQ0FBQyxDQUFDO0FBQ2hELENBQUM7QUFFRCxrRkFBa0Y7QUFDbEYsS0FBSyxVQUFVLFVBQVUsQ0FBQyxHQUF1QjtJQUMvQyxNQUFNLE9BQU8sR0FBRyxNQUFNLENBQUMsVUFBVSxDQUFDLEdBQUcsQ0FBQyxJQUFJLGNBQWMsQ0FBQztJQUN6RCxJQUFJLENBQUM7UUFDSCxNQUFNLE1BQU0sR0FBRyxNQUFNLEtBQUssQ0FBQyxPQUFPLEVBQUUsQ0FBQyxTQUFTLEVBQUUsU0FBUyxDQUFDLEVBQUU7WUFDMUQsTUFBTSxFQUFFLFNBQVM7WUFDakIsTUFBTSxFQUFFLFNBQVM7WUFDakIsS0FBSyxFQUFFLEtBQUs7U0FDYixDQUFDLENBQUM7UUFDSCxJQUFJLE1BQU0sQ0FBQyxRQUFRLEtBQUssQ0FBQztZQUFFLE9BQU8sT0FBTyxDQUFDO0lBQzVDLENBQUM7SUFBQyxNQUFNLENBQUM7UUFDUCxtRUFBbUU7SUFDckUsQ0FBQztJQUNELE9BQU8sU0FBUyxDQUFDO0FBQ25CLENBQUM7QUFFRDs7O0dBR0c7QUFDSCxLQUFLLFVBQVUsaUJBQWlCLENBQzlCLEdBQVcsRUFDWCxPQUFpQztJQUVqQyxNQUFNLEdBQUcsR0FDUCxNQUFNLENBQUMsVUFBVSxDQUFDLE9BQU8sQ0FBQyxJQUFJLENBQUMsSUFBSSxDQUFDLE1BQU0sT0FBTyxDQUFDLElBQUksQ0FBQyxNQUFNLEVBQUUsRUFBRSxxQkFBcUIsQ0FBQyxDQUFDLENBQUMsQ0FBQztJQUM1RixNQUFNLElBQUksR0FBRyxDQUFDLFNBQVMsRUFBRSxTQUFTLEVBQUUsUUFBUSxFQUFFLEdBQUcsRUFBRSxlQUFlLENBQUMsQ0FBQztJQUNwRSxNQUFNLE1BQU0sR0FBRyxNQUFNLENBQUMsU0FBUyxDQUFDLE9BQU8sQ0FBQyxNQUFNLENBQUMsQ0FBQztJQUNoRCxJQUFJLE1BQU0sQ0FBQyxNQUFNLEdBQUcsQ0FBQztRQUFFLElBQUksQ0FBQyxJQUFJLENBQUMsVUFBVSxFQUFFLE1BQU0sQ0FBQyxJQUFJLENBQUMsR0FBRyxDQUFDLENBQUMsQ0FBQztJQUMvRCxJQUFJLE9BQU8sQ0FBQyxZQUFZO1FBQUUsSUFBSSxDQUFDLElBQUksQ0FBQyxnQkFBZ0IsQ0FBQyxDQUFDO0lBRXRELE1BQU0sTUFBTSxHQUFHLE1BQU0sS0FBSyxDQUFDLEdBQUcsRUFBRSxJQUFJLEVBQUU7UUFDcEMsTUFBTSxFQUFFLFNBQVM7UUFDakIsTUFBTSxFQUFFLFNBQVM7UUFDakIsS0FBSyxFQUFFLEtBQUs7S0FDYixDQUFDLENBQUM7SUFDSCxJQUFJLE1BQU0sQ0FBQyxRQUFRLEtBQUssQ0FBQyxFQUFFLENBQUM7UUFDMUIsTUFBTSxJQUFJLEtBQUssQ0FDYiwyQ0FBMkMsTUFBTSxDQUFDLFFBQVEsTUFBTSxNQUFNLENBQUMsTUFBTSxJQUFJLE1BQU0sQ0FBQyxNQUFNLEVBQUUsQ0FDakcsQ0FBQztJQUNKLENBQUM7SUFDRCxJQUFJLENBQUMsVUFBVSxDQUFDLEdBQUcsQ0FBQztRQUFFLE9BQU8sU0FBUyxDQUFDO0lBQ3ZDLE1BQU0sQ0FBQyxLQUFLLENBQUMscUNBQXFDLEVBQUUsRUFBRSxJQUFJLEVBQUUsR0FBRyxFQUFFLE1BQU0sRUFBRSxDQUFDLENBQUM7SUFDM0UsT0FBTyxHQUFHLENBQUM7QUFDYixDQUFDIiwic291cmNlc0NvbnRlbnQiOlsiLyoqXG4gKiBTdGFydHVwIHByb3Zpc2lvbmluZyBvZiBEYXRhYnJpY2tzIEFJIFRvb2xzIHNraWxscyBmb3IgYSBNYXN0cmEgd29ya3NwYWNlLlxuICpcbiAqIERhdGFicmlja3MgQUkgVG9vbHMgKGBkYXRhYnJpY2tzIGFpdG9vbHNgKSBhcmUgRGF0YWJyaWNrcy1vd25lZCBBZ2VudC1Ta2lsbFxuICogKGBTS0lMTC5tZGApIHRyZWVzIGluc3RhbGxlZCBhbmQga2VwdCB1cCB0byBkYXRlIHRocm91Z2ggdGhlIERhdGFicmlja3MgQ0xJLlxuICogVGhlIENMSSBpbnN0YWxscyB0aGVtIGdsb2JhbGx5IHVuZGVyIGB+Ly5kYXRhYnJpY2tzL2FpdG9vbHMvc2tpbGxzLzxuYW1lPi9gXG4gKiAod2l0aCBhIGAuc3RhdGUuanNvbmAgbWFuaWZlc3QpIGFuZCBjYW4gYWxzbyB3cml0ZSBhIHJlc29sdmVkLCBhZ2VudC1hZ25vc3RpY1xuICogY29weSB0byBhbnkgZGlyZWN0b3J5IHdpdGggYGRhdGFicmlja3MgYWl0b29scyBpbnN0YWxsIC0tcGF0aCA8ZGlyPlxuICogLS1za2lsbHMtb25seWAuXG4gKlxuICoge0BsaW5rIHByb3Zpc2lvbkRhdGFicmlja3NBSVRvb2xzfSBmb2xkcyB0aG9zZSBza2lsbHMgaW50byBhIE1hc3RyYSB3b3Jrc3BhY2VcbiAqIGFzIGV4dHJhIExPQ0FMIHNraWxsIHNjYW4gcGF0aHMgc28gYW4gYWdlbnQgY2FuIHVzZSB0aGVtIHdpdGhvdXQgYW55b25lXG4gKiBoYW5kLWNvcHlpbmcgYFNLSUxMLm1kYCBmaWxlcy4gSXQgbmV2ZXIgcmVpbXBsZW1lbnRzIHRoZSBDTEkncyBzb3VyY2luZzogaXRcbiAqIGVpdGhlciBwb2ludHMgTWFzdHJhIGF0IHRoZSBhbHJlYWR5LWluc3RhbGxlZCBnbG9iYWwgdHJlZSwgb3Igc2hlbGxzIG91dCB0b1xuICogdGhlIENMSSB0byBtYXRlcmlhbGl6ZSBhIGN1cmF0ZWQgc3Vic2V0IGludG8gYSB0ZW1wIGRpci5cbiAqXG4gKiBUaGUgb3B0aW9uIGlzIGBmYWxzZSB8IHRydWUgfCBcImF1dG9cIiB8IERhdGFicmlja3NBSVRvb2xzT3B0aW9uc2A6XG4gKlxuICogLSBgZmFsc2VgIChkZWZhdWx0KSAtIG9mZi5cbiAqIC0gYFwiYXV0b1wiYCAtIGVuYWJsZSBvbmx5IHdoZW4gdGhlIGdsb2JhbCB0cmVlIGFscmVhZHkgZXhpc3RzIE9SIHRoZVxuICogICBgZGF0YWJyaWNrc2AgQ0xJIGlzIHJlc29sdmFibGU7IG90aGVyd2lzZSBzaWxlbnRseSBuby1vcC4gVGhlIFwib24gaWYgdGhlXG4gKiAgIENMSSBpcyBhcm91bmRcIiBkZWZhdWx0LlxuICogLSBgdHJ1ZWAgLSByZXF1aXJlZDogZmFpbCBzdGFydHVwIChzdWJqZWN0IHRvIGBmYWlsT25FcnJvcmApIHdoZW4gbmVpdGhlciB0aGVcbiAqICAgaW5zdGFsbGVkIHRyZWUgbm9yIHRoZSBDTEkgY2FuIHN1cHBseSBza2lsbHMuXG4gKiAtIGFuIG9wdGlvbnMgYmFnIGZvciBhIHNwZWNpZmljIGBza2lsbHNgIHN1YnNldCwgYGV4cGVyaW1lbnRhbGAgc2tpbGxzLCBhXG4gKiAgIGByZWZyZXNoYCB0aGF0IHJlLXJ1bnMgdGhlIENMSSBldmVuIHdoZW4gYSB0cmVlIGV4aXN0cywgb3IgYW4gZXhwbGljaXRcbiAqICAgYHBhdGhgLlxuICpcbiAqIEBtb2R1bGVcbiAqL1xuXG5pbXBvcnQgeyBta2R0ZW1wIH0gZnJvbSBcIm5vZGU6ZnMvcHJvbWlzZXNcIjtcbmltcG9ydCB7IGV4aXN0c1N5bmMgfSBmcm9tIFwibm9kZTpmc1wiO1xuaW1wb3J0IHsgaG9tZWRpciwgdG1wZGlyIH0gZnJvbSBcIm5vZGU6b3NcIjtcbmltcG9ydCB7IGpvaW4gfSBmcm9tIFwibm9kZTpwYXRoXCI7XG5pbXBvcnQgeyBzcGF3biB9IGZyb20gXCJAZGJ4LXRvb2xzL2NvcmVcIjtcbmltcG9ydCB7IGVycm9yLCBsb2csIHN0cmluZyB9IGZyb20gXCJAZGJ4LXRvb2xzL3NoYXJlZC1jb3JlXCI7XG5cbmNvbnN0IGxvZ2dlciA9IGxvZy5sb2dnZXIoXCJtYXN0cmEvZGF0YWJyaWNrcy1haXRvb2xzXCIpO1xuXG4vKiogR2xvYmFsIHNraWxscyB0cmVlIHRoZSBgZGF0YWJyaWNrcyBhaXRvb2xzYCBDTEkgaW5zdGFsbHMgaW50by4gKi9cbmNvbnN0IEdMT0JBTF9BSVRPT0xTX1NLSUxMU19QQVRIID0gam9pbihob21lZGlyKCksIFwiLmRhdGFicmlja3NcIiwgXCJhaXRvb2xzXCIsIFwic2tpbGxzXCIpO1xuXG4vKiogVGhlIERhdGFicmlja3MgQ0xJIGJpbmFyeSBuYW1lOyByZXNvbHZlZCBvbiBgUEFUSGAuICovXG5jb25zdCBEQVRBQlJJQ0tTX0NMSSA9IFwiZGF0YWJyaWNrc1wiO1xuXG4vKiAtLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLSB0eXBlcyAtLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLSAqL1xuXG4vKiogSG93IGFnZ3Jlc3NpdmVseSB0byBlbmFibGUgRGF0YWJyaWNrcyBBSSBUb29scyBza2lsbHMuICovXG5leHBvcnQgdHlwZSBEYXRhYnJpY2tzQUlUb29sc01vZGUgPSBcImF1dG9cIiB8IFwicmVxdWlyZVwiO1xuXG4vKiogT3B0aW9ucyBmb3Ige0BsaW5rIHByb3Zpc2lvbkRhdGFicmlja3NBSVRvb2xzfS4gKi9cbmV4cG9ydCBpbnRlcmZhY2UgRGF0YWJyaWNrc0FJVG9vbHNPcHRpb25zIHtcbiAgLyoqXG4gICAqIGBcImF1dG9cImAgKGRlZmF1bHQpIGVuYWJsZXMgb25seSB3aGVuIHRoZSBpbnN0YWxsZWQgdHJlZSBleGlzdHMgb3IgdGhlIENMSVxuICAgKiBpcyByZXNvbHZhYmxlOyBgXCJyZXF1aXJlXCJgIGZhaWxzIHN0YXJ0dXAgd2hlbiBuZWl0aGVyIGNhbiBzdXBwbHkgc2tpbGxzLlxuICAgKi9cbiAgbW9kZT86IERhdGFicmlja3NBSVRvb2xzTW9kZTtcbiAgLyoqIEluc3RhbGwgb25seSB0aGVzZSBza2lsbCBuYW1lcyAoY29tbWEvc3BhY2UgbGlzdCBvciBhcnJheSkuICovXG4gIHNraWxscz86IHN0cmluZyB8IHN0cmluZ1tdO1xuICAvKiogSW5jbHVkZSBleHBlcmltZW50YWwgc2tpbGxzIHdoZW4gdGhlIENMSSBoYXMgdG8gZmV0Y2guICovXG4gIGV4cGVyaW1lbnRhbD86IGJvb2xlYW47XG4gIC8qKlxuICAgKiBSZS1ydW4gdGhlIENMSSB0byAocmUpbWF0ZXJpYWxpemUgc2tpbGxzIGludG8gYSBmcmVzaCBkaXIgZXZlbiB3aGVuIHRoZVxuICAgKiBnbG9iYWwgdHJlZSBhbHJlYWR5IGV4aXN0cy4gRGVmYXVsdHMgdG8gYGZhbHNlYCAocmV1c2UgdGhlIGluc3RhbGxlZCB0cmVlKS5cbiAgICovXG4gIHJlZnJlc2g/OiBib29sZWFuO1xuICAvKipcbiAgICogRXhwbGljaXQgZGlyZWN0b3J5IHRvIG1hdGVyaWFsaXplIHNraWxscyBpbnRvIHdpdGggdGhlIENMSS4gRGVmYXVsdHMgdG8gYVxuICAgKiB0ZW1wIGRpci4gSWdub3JlZCB3aGVuIHRoZSBpbnN0YWxsZWQgZ2xvYmFsIHRyZWUgaXMgdXNlZCBhcy1pcy5cbiAgICovXG4gIHBhdGg/OiBzdHJpbmc7XG4gIC8qKiBBYnNvbHV0ZSBwYXRoIHRvIHRoZSBgZGF0YWJyaWNrc2AgQ0xJLiBEZWZhdWx0cyB0byBgZGF0YWJyaWNrc2Agb24gUEFUSC4gKi9cbiAgY2xpPzogc3RyaW5nO1xuICAvKipcbiAgICogRmFpbCBhcHAgc3RhcnR1cCB3aGVuIHJlcXVpcmVkIHNraWxscyBjYW4ndCBiZSBwcm92aXNpb25lZC4gRGVmYXVsdHMgdG9cbiAgICogYHRydWVgIGZvciBgbW9kZTogXCJyZXF1aXJlXCJgIGFuZCBgZmFsc2VgIGZvciBgXCJhdXRvXCJgLlxuICAgKi9cbiAgZmFpbE9uRXJyb3I/OiBib29sZWFuO1xufVxuXG4vKiogVGhlIGBkYXRhYnJpY2tzQUlUb29sc2AgY29uZmlnIG9wdGlvbjogYSB0b2dnbGUsIGBcImF1dG9cImAsIG9yIGFuIG9wdGlvbnMgYmFnLiAqL1xuZXhwb3J0IHR5cGUgRGF0YWJyaWNrc0FJVG9vbHNPcHRpb24gPSBib29sZWFuIHwgXCJhdXRvXCIgfCBEYXRhYnJpY2tzQUlUb29sc09wdGlvbnM7XG5cbi8qKiBXaGF0IHtAbGluayBwcm92aXNpb25EYXRhYnJpY2tzQUlUb29sc30gcmVzb2x2ZWQuICovXG5leHBvcnQgaW50ZXJmYWNlIFByb3Zpc2lvbmVkRGF0YWJyaWNrc0FJVG9vbHMge1xuICAvKiogRXh0cmEgTE9DQUwgc2tpbGwgc2NhbiBwYXRocyB0byBoYW5kIE1hc3RyYS4gRW1wdHkgd2hlbiBkaXNhYmxlZC91bnJlc29sdmVkLiAqL1xuICBsb2NhbFNraWxsUGF0aHM6IHN0cmluZ1tdO1xuICAvKiogSG93IHRoZSBza2lsbHMgd2VyZSBzb3VyY2VkLCBmb3IgbG9nZ2luZy4gKi9cbiAgc291cmNlPzogXCJpbnN0YWxsZWRcIiB8IFwiY2xpXCI7XG59XG5cbi8qIC0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0gaGVscGVycyAtLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tICovXG5cbi8qKiBOb3JtYWxpemUgdGhlIGBkYXRhYnJpY2tzQUlUb29sc2Agb3B0aW9uIGludG8gYSBmbGF0IG9wdGlvbnMgYmFnLCBvciBgdW5kZWZpbmVkYCB3aGVuIG9mZi4gKi9cbmV4cG9ydCBmdW5jdGlvbiBub3JtYWxpemVEYXRhYnJpY2tzQUlUb29sc09wdGlvbihcbiAgb3B0aW9uOiBEYXRhYnJpY2tzQUlUb29sc09wdGlvbiB8IHVuZGVmaW5lZCxcbik6IERhdGFicmlja3NBSVRvb2xzT3B0aW9ucyB8IHVuZGVmaW5lZCB7XG4gIGlmIChvcHRpb24gPT09IHVuZGVmaW5lZCB8fCBvcHRpb24gPT09IGZhbHNlKSByZXR1cm4gdW5kZWZpbmVkO1xuICBpZiAob3B0aW9uID09PSB0cnVlKSByZXR1cm4geyBtb2RlOiBcInJlcXVpcmVcIiB9O1xuICBpZiAob3B0aW9uID09PSBcImF1dG9cIikgcmV0dXJuIHsgbW9kZTogXCJhdXRvXCIgfTtcbiAgcmV0dXJuIHsgbW9kZTogb3B0aW9uLm1vZGUgPz8gXCJhdXRvXCIsIC4uLm9wdGlvbiB9O1xufVxuXG4vKipcbiAqIE1hdGVyaWFsaXplIERhdGFicmlja3MgQUkgVG9vbHMgc2tpbGxzIGF0IGFwcCBzdGFydHVwIGFuZCByZXR1cm4gdGhlbSBhc1xuICogZXh0cmEgbG9jYWwgc2tpbGwgc2NhbiBwYXRocy5cbiAqXG4gKiBSZXNvbHV0aW9uOiByZXVzZSB0aGUgaW5zdGFsbGVkIGdsb2JhbCB0cmVlIHdoZW4gcHJlc2VudCAobm8gQ0xJIGNhbGwpIHVubGVzc1xuICogYHJlZnJlc2hgL2Bza2lsbHNgL2BwYXRoYCBhc2sgZm9yIGEgZnJlc2ggQ0xJIG1hdGVyaWFsaXphdGlvbjsgb3RoZXJ3aXNlIHJ1blxuICogYGRhdGFicmlja3MgYWl0b29scyBpbnN0YWxsIC0tcGF0aCA8ZGlyPiAtLXNraWxscy1vbmx5YCB3aGVuIHRoZSBDTEkgcmVzb2x2ZXMuXG4gKiBBIGBcInJlcXVpcmVcImAgcHJvdmlzaW9uIHRoYXQgeWllbGRzIG5vdGhpbmcgZmFpbHMgc3RhcnR1cCB1bmxlc3NcbiAqIGBmYWlsT25FcnJvcjogZmFsc2VgLlxuICovXG5leHBvcnQgYXN5bmMgZnVuY3Rpb24gcHJvdmlzaW9uRGF0YWJyaWNrc0FJVG9vbHMoXG4gIG9wdGlvbjogRGF0YWJyaWNrc0FJVG9vbHNPcHRpb24gfCB1bmRlZmluZWQsXG4pOiBQcm9taXNlPFByb3Zpc2lvbmVkRGF0YWJyaWNrc0FJVG9vbHM+IHtcbiAgY29uc3Qgb3B0aW9ucyA9IG5vcm1hbGl6ZURhdGFicmlja3NBSVRvb2xzT3B0aW9uKG9wdGlvbik7XG4gIGNvbnN0IGVtcHR5OiBQcm92aXNpb25lZERhdGFicmlja3NBSVRvb2xzID0geyBsb2NhbFNraWxsUGF0aHM6IFtdIH07XG4gIGlmICghb3B0aW9ucykgcmV0dXJuIGVtcHR5O1xuXG4gIGNvbnN0IG1vZGUgPSBvcHRpb25zLm1vZGUgPz8gXCJhdXRvXCI7XG4gIGNvbnN0IGZhaWxPbkVycm9yID0gb3B0aW9ucy5mYWlsT25FcnJvciA/PyBtb2RlID09PSBcInJlcXVpcmVcIjtcbiAgY29uc3Qgd2FudHNDbGlGZXRjaCA9XG4gICAgb3B0aW9ucy5yZWZyZXNoID09PSB0cnVlIHx8XG4gICAgb3B0aW9ucy5wYXRoICE9PSB1bmRlZmluZWQgfHxcbiAgICBzdHJpbmcucGFyc2VMaXN0KG9wdGlvbnMuc2tpbGxzKS5sZW5ndGggPiAwIHx8XG4gICAgb3B0aW9ucy5leHBlcmltZW50YWwgPT09IHRydWU7XG5cbiAgdHJ5IHtcbiAgICAvLyBGYXN0IHBhdGg6IHJldXNlIHRoZSBhbHJlYWR5LWluc3RhbGxlZCBnbG9iYWwgdHJlZSwgbm8gQ0xJIGNhbGwuXG4gICAgaWYgKCF3YW50c0NsaUZldGNoICYmIGluc3RhbGxlZFRyZWVFeGlzdHMoKSkge1xuICAgICAgbG9nZ2VyLmRlYnVnKFwidXNpbmcgaW5zdGFsbGVkIGFpdG9vbHMgdHJlZVwiLCB7IHBhdGg6IEdMT0JBTF9BSVRPT0xTX1NLSUxMU19QQVRIIH0pO1xuICAgICAgcmV0dXJuIHsgbG9jYWxTa2lsbFBhdGhzOiBbR0xPQkFMX0FJVE9PTFNfU0tJTExTX1BBVEhdLCBzb3VyY2U6IFwiaW5zdGFsbGVkXCIgfTtcbiAgICB9XG5cbiAgICBjb25zdCBjbGlQYXRoID0gYXdhaXQgcmVzb2x2ZUNsaShvcHRpb25zLmNsaSk7XG4gICAgaWYgKGNsaVBhdGgpIHtcbiAgICAgIGNvbnN0IGRpciA9IGF3YWl0IG1hdGVyaWFsaXplVmlhQ2xpKGNsaVBhdGgsIG9wdGlvbnMpO1xuICAgICAgaWYgKGRpcikgcmV0dXJuIHsgbG9jYWxTa2lsbFBhdGhzOiBbZGlyXSwgc291cmNlOiBcImNsaVwiIH07XG4gICAgfVxuXG4gICAgLy8gTm90aGluZyBmZXRjaGVkLiBGYWxsIGJhY2sgdG8gdGhlIGluc3RhbGxlZCB0cmVlIGlmIGl0J3MgdGhlcmUuXG4gICAgaWYgKGluc3RhbGxlZFRyZWVFeGlzdHMoKSkge1xuICAgICAgcmV0dXJuIHsgbG9jYWxTa2lsbFBhdGhzOiBbR0xPQkFMX0FJVE9PTFNfU0tJTExTX1BBVEhdLCBzb3VyY2U6IFwiaW5zdGFsbGVkXCIgfTtcbiAgICB9XG5cbiAgICBpZiAoZmFpbE9uRXJyb3IpIHtcbiAgICAgIHRocm93IG5ldyBFcnJvcihcbiAgICAgICAgYERhdGFicmlja3MgQUkgVG9vbHMgcmVxdWVzdGVkIGJ1dCBubyBpbnN0YWxsZWQgc2tpbGxzIHRyZWUgYXQgXCIke0dMT0JBTF9BSVRPT0xTX1NLSUxMU19QQVRIfVwiIGFuZCB0aGUgXCIke0RBVEFCUklDS1NfQ0xJfVwiIENMSSBpcyBub3QgcmVzb2x2YWJsZS4gSW5zdGFsbCB3aXRoIFxcYGRhdGFicmlja3MgYWl0b29scyBpbnN0YWxsXFxgLCBvciBzZXQgZGF0YWJyaWNrc0FJVG9vbHM6IFwiYXV0b1wiLmAsXG4gICAgICApO1xuICAgIH1cbiAgICBsb2dnZXIuZGVidWcoXCJkYXRhYnJpY2tzIGFpdG9vbHMgdW5hdmFpbGFibGU7IHNraXBwaW5nXCIpO1xuICAgIHJldHVybiBlbXB0eTtcbiAgfSBjYXRjaCAoZXJyKSB7XG4gICAgaWYgKGZhaWxPbkVycm9yKSB7XG4gICAgICB0aHJvdyBuZXcgRXJyb3IoXG4gICAgICAgIGBmYWlsZWQgdG8gcHJvdmlzaW9uIERhdGFicmlja3MgQUkgVG9vbHMgc2tpbGxzOiAke2Vycm9yLmVycm9yTWVzc2FnZShlcnIpfWAsXG4gICAgICAgIHsgY2F1c2U6IGVycm9yLnRvRXJyb3IoZXJyKSB9LFxuICAgICAgKTtcbiAgICB9XG4gICAgbG9nZ2VyLndhcm4oXCJza2lwcGVkXCIsIHsgZXJyb3I6IGVycm9yLmVycm9yTWVzc2FnZShlcnIpIH0pO1xuICAgIHJldHVybiBlbXB0eTtcbiAgfVxufVxuXG4vKiogV2hldGhlciB0aGUgQ0xJJ3MgZ2xvYmFsIHNraWxscyB0cmVlIGV4aXN0cyBhbmQgaG9sZHMgYXQgbGVhc3Qgb25lIHNraWxsLiAqL1xuZnVuY3Rpb24gaW5zdGFsbGVkVHJlZUV4aXN0cygpOiBib29sZWFuIHtcbiAgcmV0dXJuIGV4aXN0c1N5bmMoR0xPQkFMX0FJVE9PTFNfU0tJTExTX1BBVEgpO1xufVxuXG4vKiogUmVzb2x2ZSB0aGUgYGRhdGFicmlja3NgIENMSSwgcmV0dXJuaW5nIGl0cyBpbnZvY2F0aW9uIHBhdGggb3IgYHVuZGVmaW5lZGAuICovXG5hc3luYyBmdW5jdGlvbiByZXNvbHZlQ2xpKGNsaTogc3RyaW5nIHwgdW5kZWZpbmVkKTogUHJvbWlzZTxzdHJpbmcgfCB1bmRlZmluZWQ+IHtcbiAgY29uc3QgY29tbWFuZCA9IHN0cmluZy50cmltVG9OdWxsKGNsaSkgPz8gREFUQUJSSUNLU19DTEk7XG4gIHRyeSB7XG4gICAgY29uc3QgcmVzdWx0ID0gYXdhaXQgc3Bhd24oY29tbWFuZCwgW1wiYWl0b29sc1wiLCBcInZlcnNpb25cIl0sIHtcbiAgICAgIHN0ZG91dDogXCJjYXB0dXJlXCIsXG4gICAgICBzdGRlcnI6IFwiY2FwdHVyZVwiLFxuICAgICAgY2hlY2s6IGZhbHNlLFxuICAgIH0pO1xuICAgIGlmIChyZXN1bHQuZXhpdENvZGUgPT09IDApIHJldHVybiBjb21tYW5kO1xuICB9IGNhdGNoIHtcbiAgICAvLyBDTEkgYWJzZW50IG9yIGBhaXRvb2xzYCB1bnN1cHBvcnRlZCAtIGNhbGxlciBmYWxscyBiYWNrIC8gZmFpbHMuXG4gIH1cbiAgcmV0dXJuIHVuZGVmaW5lZDtcbn1cblxuLyoqXG4gKiBSdW4gYGRhdGFicmlja3MgYWl0b29scyBpbnN0YWxsIC0tcGF0aCA8ZGlyPiAtLXNraWxscy1vbmx5YCB0byBtYXRlcmlhbGl6ZSBhXG4gKiByZXNvbHZlZCwgYWdlbnQtYWdub3N0aWMgc2tpbGwgc2V0IGludG8gYSBkaXJlY3RvcnkgKG5vIGFnZW50cywgbm8gc3RhdGUpLlxuICovXG5hc3luYyBmdW5jdGlvbiBtYXRlcmlhbGl6ZVZpYUNsaShcbiAgY2xpOiBzdHJpbmcsXG4gIG9wdGlvbnM6IERhdGFicmlja3NBSVRvb2xzT3B0aW9ucyxcbik6IFByb21pc2U8c3RyaW5nIHwgdW5kZWZpbmVkPiB7XG4gIGNvbnN0IGRpciA9XG4gICAgc3RyaW5nLnRyaW1Ub051bGwob3B0aW9ucy5wYXRoKSA/PyAoYXdhaXQgbWtkdGVtcChqb2luKHRtcGRpcigpLCBcImRhdGFicmlja3MtYWl0b29scy1cIikpKTtcbiAgY29uc3QgYXJncyA9IFtcImFpdG9vbHNcIiwgXCJpbnN0YWxsXCIsIFwiLS1wYXRoXCIsIGRpciwgXCItLXNraWxscy1vbmx5XCJdO1xuICBjb25zdCBza2lsbHMgPSBzdHJpbmcucGFyc2VMaXN0KG9wdGlvbnMuc2tpbGxzKTtcbiAgaWYgKHNraWxscy5sZW5ndGggPiAwKSBhcmdzLnB1c2goXCItLXNraWxsc1wiLCBza2lsbHMuam9pbihcIixcIikpO1xuICBpZiAob3B0aW9ucy5leHBlcmltZW50YWwpIGFyZ3MucHVzaChcIi0tZXhwZXJpbWVudGFsXCIpO1xuXG4gIGNvbnN0IHJlc3VsdCA9IGF3YWl0IHNwYXduKGNsaSwgYXJncywge1xuICAgIHN0ZG91dDogXCJjYXB0dXJlXCIsXG4gICAgc3RkZXJyOiBcImNhcHR1cmVcIixcbiAgICBjaGVjazogZmFsc2UsXG4gIH0pO1xuICBpZiAocmVzdWx0LmV4aXRDb2RlICE9PSAwKSB7XG4gICAgdGhyb3cgbmV3IEVycm9yKFxuICAgICAgYGRhdGFicmlja3MgYWl0b29scyBpbnN0YWxsIGZhaWxlZCAoZXhpdCAke3Jlc3VsdC5leGl0Q29kZX0pOiAke3Jlc3VsdC5zdGRlcnIgfHwgcmVzdWx0LnN0ZG91dH1gLFxuICAgICk7XG4gIH1cbiAgaWYgKCFleGlzdHNTeW5jKGRpcikpIHJldHVybiB1bmRlZmluZWQ7XG4gIGxvZ2dlci5kZWJ1ZyhcIm1hdGVyaWFsaXplZCBhaXRvb2xzIHNraWxscyB2aWEgQ0xJXCIsIHsgcGF0aDogZGlyLCBza2lsbHMgfSk7XG4gIHJldHVybiBkaXI7XG59XG4iXX0=