@dremio/js-sdk 0.1.0

package/dist/enterprise/catalog/catalogRetrieve.js

@@ -0,0 +1,119 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * 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.
+ */
+import { batch } from "../../common/batch.js";
+import { Err, Ok } from "ts-results-es";
+import { catalogObjectFromEntity } from "./catalogObjectFromEntity.js";
+import { HttpError } from "../../common/HttpError.js";
+import { unableToRetrieveProblem } from "../../oss/catalog/catalogErrors.js";
+export const baseRetrieve = (config) => batch(async (ids) => {
+    const results = new Map();
+    if (ids.size === 1) {
+        const id = Array.from(ids).at(0);
+        return config
+            .sonarV3Request(`catalog/${id}?maxChildren=0`)
+            .then((res) => res.json())
+            .then((response) => {
+            results.set(id, Ok(catalogObjectFromEntity(config, baseRetrieveByPath(config))(response)));
+            return results;
+        })
+            .catch((e) => {
+            if (e instanceof HttpError && e.body?.detail) {
+                results.set(id, Err(unableToRetrieveProblem(e.body.detail)));
+            }
+            else {
+                results.set(id, Err(unableToRetrieveProblem()));
+            }
+            return results;
+        });
+    }
+    const idsArray = Array.from(ids);
+    const chunks = [];
+    while (idsArray.length > 0) {
+        chunks.push(idsArray.splice(0, Math.min(50, idsArray.length)));
+    }
+    const catalogItems = await Promise.all(chunks.map((chunk) => config
+        .sonarV3Request(`catalog/by-ids?maxChildren=0`, {
+        body: JSON.stringify(chunk),
+        headers: { "Content-Type": "application/json" },
+        method: "POST",
+    })
+        .then((res) => res.json())
+        .then((response) => response.data))).then((chunks) => chunks.flat());
+    for (const catalogItem of catalogItems) {
+        results.set(catalogItem.id, Ok(catalogObjectFromEntity(config, baseRetrieveByPath(config))(catalogItem)));
+    }
+    for (const id of ids) {
+        if (!results.has(id)) {
+            results.set(id, Err(unableToRetrieveProblem()));
+        }
+    }
+    return results;
+});
+export const baseRetrieveByPath = (config) => batch(async (paths) => {
+    const results = new Map();
+    // Because the key is a reference type (array), we need to store the
+    // original path reference
+    const originalKeyRef = new Map();
+    for (const key of paths) {
+        originalKeyRef.set(JSON.stringify(key), key);
+    }
+    if (paths.size === 1) {
+        const path = Array.from(paths).at(0);
+        return config
+            .sonarV3Request(`catalog/by-path/${path.map(encodeURIComponent).join("/")}?maxChildren=0`)
+            .then((res) => res.json())
+            .then((response) => {
+            results.set(path, Ok(catalogObjectFromEntity(config, baseRetrieveByPath(config))(response)));
+            return results;
+        })
+            .catch((e) => {
+            if (e instanceof HttpError && e.body?.detail) {
+                results.set(path, Err(unableToRetrieveProblem(e.body.detail)));
+            }
+            else {
+                results.set(path, Err(unableToRetrieveProblem()));
+            }
+            return results;
+        });
+    }
+    const pathsArray = Array.from(paths);
+    const chunks = [];
+    while (pathsArray.length > 0) {
+        chunks.push(pathsArray.splice(0, Math.min(50, pathsArray.length)));
+    }
+    const catalogItems = await Promise.all(chunks.map((chunk) => config
+        .sonarV3Request(`catalog/by-paths?maxChildren=0`, {
+        body: JSON.stringify(chunk),
+        headers: { "Content-Type": "application/json" },
+        method: "POST",
+    })
+        .then((res) => res.json())
+        .then((response) => response.data))).then((chunks) => chunks.flat());
+    for (const catalogItem of catalogItems) {
+        const resolved = catalogObjectFromEntity(config, baseRetrieveByPath(config))(catalogItem);
+        const originalKey = originalKeyRef.get(JSON.stringify(resolved.path));
+        if (originalKey) {
+            results.set(originalKey, Ok(resolved));
+        }
+    }
+    for (const path of paths) {
+        if (!results.has(path)) {
+            results.set(path, Err(unableToRetrieveProblem()));
+        }
+    }
+    return results;
+});
+//# sourceMappingURL=catalogRetrieve.js.map

package/dist/enterprise/catalog/catalogRetrieve.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"catalogRetrieve.js","sourceRoot":"","sources":["../../../src/enterprise/catalog/catalogRetrieve.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAC9C,OAAO,EAAE,GAAG,EAAE,EAAE,EAAe,MAAM,eAAe,CAAC;AAErD,OAAO,EAAE,uBAAuB,EAAE,MAAM,8BAA8B,CAAC;AACvE,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAEtD,OAAO,EAAE,uBAAuB,EAAE,MAAM,oCAAoC,CAAC;AAE7E,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,MAAqB,EAAE,EAAE,CACpD,KAAK,CAAC,KAAK,EAAE,GAAgB,EAAE,EAAE;IAC/B,MAAM,OAAO,GAAG,IAAI,GAAG,EAAoD,CAAC;IAE5E,IAAI,GAAG,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QACnB,MAAM,EAAE,GAAG,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE,CAAC;QAClC,OAAO,MAAM;aACV,cAAc,CAAC,WAAW,EAAE,gBAAgB,CAAC;aAC7C,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;aACzB,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;YACjB,OAAO,CAAC,GAAG,CACT,EAAE,EACF,EAAE,CAAC,uBAAuB,CAAC,MAAM,EAAE,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAQ,CAAC,CACjF,CAAC;YACF,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;YACX,IAAI,CAAC,YAAY,SAAS,IAAK,CAAC,CAAC,IAAY,EAAE,MAAM,EAAE,CAAC;gBACtD,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,uBAAuB,CAAE,CAAC,CAAC,IAAY,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YACxE,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC;YAClD,CAAC;YAED,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC,CAAC;IACP,CAAC;IAED,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACjC,MAAM,MAAM,GAAG,EAAE,CAAC;IAElB,OAAO,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CACpC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACnB,MAAM;SACH,cAAc,CAAC,8BAA8B,EAAE;QAC9C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;QAC3B,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;QAC/C,MAAM,EAAE,MAAM;KACf,CAAC;SACD,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;SACzB,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CACrC,CACF,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;IAElC,KAAK,MAAM,WAAW,IAAI,YAAY,EAAE,CAAC;QACvC,OAAO,CAAC,GAAG,CACT,WAAW,CAAC,EAAE,EACd,EAAE,CAAC,uBAAuB,CAAC,MAAM,EAAE,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW,CAAQ,CAAC,CACpF,CAAC;IACJ,CAAC;IAED,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;QACrB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;YACrB,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC,CAAC;AAEL,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,MAAqB,EAAE,EAAE,CAC1D,KAAK,CAAC,KAAK,EAAE,KAAoB,EAAE,EAAE;IACnC,MAAM,OAAO,GAAG,IAAI,GAAG,EAAsD,CAAC;IAE9E,oEAAoE;IACpE,0BAA0B;IAC1B,MAAM,cAAc,GAAG,IAAI,GAAG,EAAoB,CAAC;IACnD,KAAK,MAAM,GAAG,IAAI,KAAK,EAAE,CAAC;QACxB,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,CAAC;IAC/C,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QACrB,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAE,CAAC;QACtC,OAAO,MAAM;aACV,cAAc,CAAC,mBAAmB,IAAI,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,gBAAgB,CAAC;aACzF,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;aACzB,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;YACjB,OAAO,CAAC,GAAG,CACT,IAAI,EACJ,EAAE,CAAC,uBAAuB,CAAC,MAAM,EAAE,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAQ,CACjF,CAAC;YACF,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;YACX,IAAI,CAAC,YAAY,SAAS,IAAK,CAAC,CAAC,IAAY,EAAE,MAAM,EAAE,CAAC;gBACtD,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,uBAAuB,CAAE,CAAC,CAAC,IAAY,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC1E,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC;YACpD,CAAC;YACD,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC,CAAC;IACP,CAAC;IAED,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,EAAE,CAAC;IAElB,OAAO,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IACrE,CAAC;IAED,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CACpC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACnB,MAAM;SACH,cAAc,CAAC,gCAAgC,EAAE;QAChD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;QAC3B,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;QAC/C,MAAM,EAAE,MAAM;KACf,CAAC;SACD,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;SACzB,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CACrC,CACF,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;IAElC,KAAK,MAAM,WAAW,IAAI,YAAY,EAAE,CAAC;QACvC,MAAM,QAAQ,GAAG,uBAAuB,CAAC,MAAM,EAAE,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;QAC1F,MAAM,WAAW,GAAG,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;QACtE,IAAI,WAAW,EAAE,CAAC;YAChB,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC,QAAQ,CAAQ,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YACvB,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { batch } from \"../../common/batch.ts\";\nimport { Err, Ok, type Result } from \"ts-results-es\";\nimport type { SonarV3Config } from \"../../common/Config.ts\";\nimport { catalogObjectFromEntity } from \"./catalogObjectFromEntity.ts\";\nimport { HttpError } from \"../../common/HttpError.ts\";\nimport type { EnterpriseCatalogObject } from \"./CatalogObjects/index.ts\";\nimport { unableToRetrieveProblem } from \"../../oss/catalog/catalogErrors.ts\";\n\nexport const baseRetrieve = (config: SonarV3Config) =>\n  batch(async (ids: Set<string>) => {\n    const results = new Map<string, Result<EnterpriseCatalogObject, unknown>>();\n\n    if (ids.size === 1) {\n      const id = Array.from(ids).at(0)!;\n      return config\n        .sonarV3Request(`catalog/${id}?maxChildren=0`)\n        .then((res) => res.json())\n        .then((response) => {\n          results.set(\n            id,\n            Ok(catalogObjectFromEntity(config, baseRetrieveByPath(config))(response) as any),\n          );\n          return results;\n        })\n        .catch((e) => {\n          if (e instanceof HttpError && (e.body as any)?.detail) {\n            results.set(id, Err(unableToRetrieveProblem((e.body as any).detail)));\n          } else {\n            results.set(id, Err(unableToRetrieveProblem()));\n          }\n\n          return results;\n        });\n    }\n\n    const idsArray = Array.from(ids);\n    const chunks = [];\n\n    while (idsArray.length > 0) {\n      chunks.push(idsArray.splice(0, Math.min(50, idsArray.length)));\n    }\n\n    const catalogItems = await Promise.all(\n      chunks.map((chunk) =>\n        config\n          .sonarV3Request(`catalog/by-ids?maxChildren=0`, {\n            body: JSON.stringify(chunk),\n            headers: { \"Content-Type\": \"application/json\" },\n            method: \"POST\",\n          })\n          .then((res) => res.json())\n          .then((response) => response.data),\n      ),\n    ).then((chunks) => chunks.flat());\n\n    for (const catalogItem of catalogItems) {\n      results.set(\n        catalogItem.id,\n        Ok(catalogObjectFromEntity(config, baseRetrieveByPath(config))(catalogItem) as any),\n      );\n    }\n\n    for (const id of ids) {\n      if (!results.has(id)) {\n        results.set(id, Err(unableToRetrieveProblem()));\n      }\n    }\n\n    return results;\n  });\n\nexport const baseRetrieveByPath = (config: SonarV3Config) =>\n  batch(async (paths: Set<string[]>) => {\n    const results = new Map<string[], Result<EnterpriseCatalogObject, unknown>>();\n\n    // Because the key is a reference type (array), we need to store the\n    // original path reference\n    const originalKeyRef = new Map<string, string[]>();\n    for (const key of paths) {\n      originalKeyRef.set(JSON.stringify(key), key);\n    }\n\n    if (paths.size === 1) {\n      const path = Array.from(paths).at(0)!;\n      return config\n        .sonarV3Request(`catalog/by-path/${path.map(encodeURIComponent).join(\"/\")}?maxChildren=0`)\n        .then((res) => res.json())\n        .then((response) => {\n          results.set(\n            path,\n            Ok(catalogObjectFromEntity(config, baseRetrieveByPath(config))(response)) as any,\n          );\n          return results;\n        })\n        .catch((e) => {\n          if (e instanceof HttpError && (e.body as any)?.detail) {\n            results.set(path, Err(unableToRetrieveProblem((e.body as any).detail)));\n          } else {\n            results.set(path, Err(unableToRetrieveProblem()));\n          }\n          return results;\n        });\n    }\n\n    const pathsArray = Array.from(paths);\n    const chunks = [];\n\n    while (pathsArray.length > 0) {\n      chunks.push(pathsArray.splice(0, Math.min(50, pathsArray.length)));\n    }\n\n    const catalogItems = await Promise.all(\n      chunks.map((chunk) =>\n        config\n          .sonarV3Request(`catalog/by-paths?maxChildren=0`, {\n            body: JSON.stringify(chunk),\n            headers: { \"Content-Type\": \"application/json\" },\n            method: \"POST\",\n          })\n          .then((res) => res.json())\n          .then((response) => response.data),\n      ),\n    ).then((chunks) => chunks.flat());\n\n    for (const catalogItem of catalogItems) {\n      const resolved = catalogObjectFromEntity(config, baseRetrieveByPath(config))(catalogItem);\n      const originalKey = originalKeyRef.get(JSON.stringify(resolved.path));\n      if (originalKey) {\n        results.set(originalKey, Ok(resolved) as any);\n      }\n    }\n\n    for (const path of paths) {\n      if (!results.has(path)) {\n        results.set(path, Err(unableToRetrieveProblem()));\n      }\n    }\n\n    return results;\n  });\n"]}

package/dist/enterprise/credentials/exchangePat.d.ts

@@ -0,0 +1,12 @@
+import type { Config } from "../../common/Config.ts";
+export declare const exchangePat: (endpoint: URL, config: Config) => (subject_token: string) => Promise<{
+    access_token: string;
+    expires_in: number;
+    token_type: string;
+    issued_token_type: string;
+    scope: string;
+}>;
+export declare const invalidSubjectToken: {
+    readonly title: "The provided subject_token is invalid.";
+    readonly type: "https://api.dremio.dev/problems/oauth2/invalid-subject-token";
+};

package/dist/enterprise/credentials/exchangePat.js

@@ -0,0 +1,55 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * 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.
+ */
+import { HttpError } from "../../common/HttpError.js";
+import {} from "../../common/Problem.js";
+export const exchangePat = (endpoint, config) => (subject_token) => {
+    const body = new URLSearchParams({
+        grant_type: "urn:ietf:params:oauth:grant-type:token-exchange",
+        scope: "dremio.all",
+        subject_token,
+        subject_token_type: "urn:ietf:params:oauth:token-type:dremio:personal-access-token",
+    });
+    return (config.fetch || globalThis.fetch)(endpoint.toString(), {
+        body,
+        headers: {
+            Accept: "application/json",
+            "Content-Type": "application/x-www-form-urlencoded",
+        },
+        method: "POST",
+    })
+        .then(async (res) => {
+        if (!res.ok) {
+            throw await HttpError.fromResponse(res);
+        }
+        return res.json();
+    })
+        .catch((e) => {
+        if (e instanceof HttpError) {
+            if (e.status === 401) {
+                config.logger?.warn(invalidSubjectToken);
+                throw new Error(invalidSubjectToken.title, {
+                    cause: invalidSubjectToken,
+                });
+            }
+        }
+        throw e;
+    });
+};
+export const invalidSubjectToken = {
+    title: "The provided subject_token is invalid.",
+    type: "https://api.dremio.dev/problems/oauth2/invalid-subject-token",
+};
+//# sourceMappingURL=exchangePat.js.map

package/dist/enterprise/credentials/exchangePat.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"exchangePat.js","sourceRoot":"","sources":["../../../src/enterprise/credentials/exchangePat.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAEtD,OAAO,EAAgB,MAAM,yBAAyB,CAAC;AAEvD,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,QAAa,EAAE,MAAc,EAAE,EAAE,CAAC,CAAC,aAAqB,EAAE,EAAE;IACtF,MAAM,IAAI,GAAG,IAAI,eAAe,CAAC;QAC/B,UAAU,EAAE,iDAAiD;QAC7D,KAAK,EAAE,YAAY;QACnB,aAAa;QACb,kBAAkB,EAAE,+DAA+D;KACpF,CAAC,CAAC;IACH,OAAO,CAAC,MAAM,CAAC,KAAK,IAAI,UAAU,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE;QAC7D,IAAI;QACJ,OAAO,EAAE;YACP,MAAM,EAAE,kBAAkB;YAC1B,cAAc,EAAE,mCAAmC;SACpD;QACD,MAAM,EAAE,MAAM;KACf,CAAC;SACC,IAAI,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QAClB,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,MAAM,MAAM,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;QAC1C,CAAC;QAED,OAAO,GAAG,CAAC,IAAI,EAMb,CAAC;IACL,CAAC,CAAC;SACD,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;QACX,IAAI,CAAC,YAAY,SAAS,EAAE,CAAC;YAC3B,IAAI,CAAC,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACrB,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,mBAAmB,CAAC,CAAC;gBACzC,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,KAAK,EAAE;oBACzC,KAAK,EAAE,mBAAmB;iBAC3B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,MAAM,CAAC,CAAC;IACV,CAAC,CAAC,CAAC;AACP,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,mBAAmB,GAAG;IACjC,KAAK,EAAE,wCAAwC;IAC/C,IAAI,EAAE,8DAA8D;CAC1C,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { HttpError } from \"../../common/HttpError.ts\";\nimport type { Config } from \"../../common/Config.ts\";\nimport { type Problem } from \"../../common/Problem.ts\";\n\nexport const exchangePat = (endpoint: URL, config: Config) => (subject_token: string) => {\n  const body = new URLSearchParams({\n    grant_type: \"urn:ietf:params:oauth:grant-type:token-exchange\",\n    scope: \"dremio.all\",\n    subject_token,\n    subject_token_type: \"urn:ietf:params:oauth:token-type:dremio:personal-access-token\",\n  });\n  return (config.fetch || globalThis.fetch)(endpoint.toString(), {\n    body,\n    headers: {\n      Accept: \"application/json\",\n      \"Content-Type\": \"application/x-www-form-urlencoded\",\n    },\n    method: \"POST\",\n  })\n    .then(async (res) => {\n      if (!res.ok) {\n        throw await HttpError.fromResponse(res);\n      }\n\n      return res.json() as Promise<{\n        access_token: string;\n        expires_in: number;\n        token_type: string;\n        issued_token_type: string;\n        scope: string;\n      }>;\n    })\n    .catch((e) => {\n      if (e instanceof HttpError) {\n        if (e.status === 401) {\n          config.logger?.warn(invalidSubjectToken);\n          throw new Error(invalidSubjectToken.title, {\n            cause: invalidSubjectToken,\n          });\n        }\n      }\n      throw e;\n    });\n};\n\nexport const invalidSubjectToken = {\n  title: \"The provided subject_token is invalid.\",\n  type: \"https://api.dremio.dev/problems/oauth2/invalid-subject-token\",\n} as const satisfies Problem;\n"]}

package/dist/enterprise/credentials/fromPat.d.ts

@@ -0,0 +1,2 @@
+import type { CredentialProvider } from "../../common/CredentialProvider.ts";
+export declare const fromPat: (pat: string) => CredentialProvider;

package/dist/enterprise/credentials/fromPat.js

@@ -0,0 +1,36 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * 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.
+ */
+import { exchangePat } from "./exchangePat.js";
+export const fromPat = (pat) => {
+    let access_token = null;
+    let expires_at = null;
+    return {
+        get: async (config) => {
+            if (!access_token || (expires_at && expires_at <= new Date(Date.now() - 60000))) {
+                try {
+                    const response = await exchangePat(new URL("/oauth/token", config.origin), config)(pat);
+                    access_token = response.access_token;
+                    expires_at = new Date(Date.now() + response.expires_in * 1000);
+                }
+                catch (_e) {
+                    access_token = pat;
+                }
+            }
+            return access_token;
+        },
+    };
+};
+//# sourceMappingURL=fromPat.js.map

package/dist/enterprise/credentials/fromPat.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fromPat.js","sourceRoot":"","sources":["../../../src/enterprise/credentials/fromPat.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAG/C,MAAM,CAAC,MAAM,OAAO,GAAG,CAAC,GAAW,EAAsB,EAAE;IACzD,IAAI,YAAY,GAAkB,IAAI,CAAC;IACvC,IAAI,UAAU,GAAgB,IAAI,CAAC;IACnC,OAAO;QACL,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YACpB,IAAI,CAAC,YAAY,IAAI,CAAC,UAAU,IAAI,UAAU,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC,EAAE,CAAC;gBAChF,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,IAAI,GAAG,CAAC,cAAc,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC;oBAExF,YAAY,GAAG,QAAQ,CAAC,YAAY,CAAC;oBACrC,UAAU,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC,UAAU,GAAG,IAAI,CAAC,CAAC;gBACjE,CAAC;gBAAC,OAAO,EAAW,EAAE,CAAC;oBACrB,YAAY,GAAG,GAAG,CAAC;gBACrB,CAAC;YACH,CAAC;YACD,OAAO,YAAY,CAAC;QACtB,CAAC;KACF,CAAC;AACJ,CAAC,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { exchangePat } from \"./exchangePat.ts\";\nimport type { CredentialProvider } from \"../../common/CredentialProvider.ts\";\n\nexport const fromPat = (pat: string): CredentialProvider => {\n  let access_token: string | null = null;\n  let expires_at: Date | null = null;\n  return {\n    get: async (config) => {\n      if (!access_token || (expires_at && expires_at <= new Date(Date.now() - 60000))) {\n        try {\n          const response = await exchangePat(new URL(\"/oauth/token\", config.origin), config)(pat);\n\n          access_token = response.access_token;\n          expires_at = new Date(Date.now() + response.expires_in * 1000);\n        } catch (_e: unknown) {\n          access_token = pat;\n        }\n      }\n      return access_token;\n    },\n  };\n};\n"]}

package/dist/enterprise/credentials/index.d.ts

@@ -0,0 +1,2 @@
+export * from "../../oss/credentials/index.ts";
+export * from "./fromPat.ts";

package/dist/enterprise/credentials/index.js

@@ -0,0 +1,18 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * 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.
+ */
+export * from "../../oss/credentials/index.js";
+export * from "./fromPat.js";
+//# sourceMappingURL=index.js.map

package/dist/enterprise/credentials/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/enterprise/credentials/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,cAAc,gCAAgC,CAAC;AAC/C,cAAc,cAAc,CAAC","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nexport * from \"../../oss/credentials/index.ts\";\nexport * from \"./fromPat.ts\";\n"]}

package/dist/enterprise/engines/Engine.d.ts

@@ -0,0 +1,78 @@
+import type { SonarV3Config } from "../../common/Config.ts";
+import { type Result } from "ts-results-es";
+export declare class Engine {
+    #private;
+    readonly id: EngineEntity["id"];
+    readonly name: EngineEntity["name"];
+    readonly spec: {
+        c3StorageSize: EngineEntity["spec"]["c3StorageSize"];
+        executorPodMetadata: {
+            annotations: EngineEntity["spec"]["executorPodMetadata"]["annotations"];
+            labels: EngineEntity["spec"]["executorPodMetadata"]["labels"];
+            nodeSelectors: EngineEntity["spec"]["executorPodMetadata"]["nodeSelectors"];
+            tolerations: {
+                key: EngineEntity["spec"]["executorPodMetadata"]["tolerations"][number]["key"];
+                operator: EngineEntity["spec"]["executorPodMetadata"]["tolerations"][number]["operator"];
+                effect: EngineEntity["spec"]["executorPodMetadata"]["tolerations"][number]["effect"];
+                tolerationDuration: Temporal.Duration;
+                value?: EngineEntity["spec"]["executorPodMetadata"]["tolerations"][number]["value"];
+            }[];
+            idleTimeout?: Temporal.Duration;
+            resourceAllocationOffset: EngineEntity["spec"]["executorPodMetadata"]["resourceAllocationOffset"];
+            size: EngineEntity["spec"]["executorPodMetadata"]["size"];
+            spillStorageSize: EngineEntity["spec"]["executorPodMetadata"]["spillStorageSize"];
+            targetCpuCapacity: EngineEntity["spec"]["executorPodMetadata"]["targetCpuCapacity"];
+        };
+    };
+    readonly status: Omit<EngineEntity["status"], "message"> & {
+        message: string | null;
+    };
+    constructor(entity: EngineEntity, config: SonarV3Config);
+    get settled(): boolean;
+    start(): Promise<Result<Engine, unknown>>;
+    stop(): Promise<Result<Engine, unknown>>;
+}
+export type EngineEntity = {
+    id: string;
+    name: string;
+    spec: {
+        c3StorageSize: string;
+        executorPodMetadata: {
+            labels: Record<string, string>;
+            annotations: Record<string, string>;
+            nodeSelectors: Record<string, string>;
+            tolerations: {
+                key: string;
+                operator: "Exists" | "Equal";
+                effect: "NoSchedule" | "PreferNoSchedule" | "NoExecute";
+                tolerationSeconds: number;
+                value?: number;
+            }[];
+            idleTimeout?: string;
+            resourceAllocationOffset: string;
+            size: "2XSmall" | "XSmall" | "Small" | "Medium" | "Large" | "XLarge" | "2XLarge";
+            spillStorageSize: string;
+            targetCpuCapacity: string;
+        };
+    };
+    status: {
+        message: string;
+        replicas: {
+            id: string;
+            executors: {
+                cpu: number;
+                memory: number;
+                name: string;
+                status: string;
+            }[];
+        }[];
+        runningExecutors: number;
+        runningReplicas: number;
+        startingExecutors: number;
+        startingReplicas: number;
+        state: "STARTING" | "RUNNING" | "STOPPING" | "STOPPED" | "WARNING" | "FAILED";
+        stoppingExecutors: number;
+        stoppingReplicas: number;
+    };
+    version: string;
+};

package/dist/enterprise/engines/Engine.js

@@ -0,0 +1,75 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * 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.
+ */
+import parseMilliseconds from "parse-ms";
+import { Err, Ok } from "ts-results-es";
+export class Engine {
+    id;
+    name;
+    spec;
+    status;
+    #config;
+    // eslint-disable-next-line no-unused-private-class-members
+    #tag;
+    constructor(entity, config) {
+        this.id = entity.id;
+        this.name = entity.name;
+        this.spec = {
+            ...entity.spec,
+            executorPodMetadata: {
+                ...entity.spec.executorPodMetadata,
+                idleTimeout: entity.spec.executorPodMetadata.idleTimeout
+                    ? Temporal.Duration.from(entity.spec.executorPodMetadata.idleTimeout)
+                    : undefined,
+                tolerations: entity.spec.executorPodMetadata.tolerations.map(({ tolerationSeconds, ...toleration }) => ({
+                    ...toleration,
+                    tolerationDuration: Temporal.Duration.from(parseMilliseconds(tolerationSeconds * 1000)),
+                })),
+            },
+        };
+        this.status = {
+            ...entity.status,
+            message: entity.status.message?.length ? entity.status.message : null,
+        };
+        this.#tag = entity.version;
+        this.#config = config;
+    }
+    get settled() {
+        return this.status.state !== "STARTING" && this.status.state !== "STOPPING";
+    }
+    start() {
+        return this.#config
+            .sonarV3Request(`engines/${this.id}/start`, {
+            headers: { Accept: "application/json" },
+            keepalive: true,
+            method: "PUT",
+        })
+            .then((res) => res.json())
+            .then((entity) => Ok(new Engine(entity, this.#config)))
+            .catch((e) => Err(e));
+    }
+    stop() {
+        return this.#config
+            .sonarV3Request(`engines/${this.id}/stop`, {
+            headers: { Accept: "application/json" },
+            keepalive: true,
+            method: "PUT",
+        })
+            .then((res) => res.json())
+            .then((entity) => Ok(new Engine(entity, this.#config)))
+            .catch((e) => Err(e));
+    }
+}
+//# sourceMappingURL=Engine.js.map

package/dist/enterprise/engines/Engine.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Engine.js","sourceRoot":"","sources":["../../../src/enterprise/engines/Engine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,iBAAiB,MAAM,UAAU,CAAC;AAEzC,OAAO,EAAE,GAAG,EAAE,EAAE,EAAe,MAAM,eAAe,CAAC;AAErD,MAAM,OAAO,MAAM;IACD,EAAE,CAAqB;IACvB,IAAI,CAAuB;IAC3B,IAAI,CAmBlB;IACc,MAAM,CAEpB;IAEO,OAAO,CAAgB;IAEhC,2DAA2D;IAClD,IAAI,CAAS;IAEtB,YAAY,MAAoB,EAAE,MAAqB;QACrD,IAAI,CAAC,EAAE,GAAG,MAAM,CAAC,EAAE,CAAC;QACpB,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QACxB,IAAI,CAAC,IAAI,GAAG;YACV,GAAG,MAAM,CAAC,IAAI;YACd,mBAAmB,EAAE;gBACnB,GAAG,MAAM,CAAC,IAAI,CAAC,mBAAmB;gBAClC,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,WAAW;oBACtD,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,WAAW,CAAC;oBACrE,CAAC,CAAC,SAAS;gBACb,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,WAAW,CAAC,GAAG,CAC1D,CAAC,EAAE,iBAAiB,EAAE,GAAG,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;oBACzC,GAAG,UAAU;oBACb,kBAAkB,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,GAAG,IAAI,CAAC,CAAC;iBACxF,CAAC,CACH;aACF;SACF,CAAC;QACF,IAAI,CAAC,MAAM,GAAG;YACZ,GAAG,MAAM,CAAC,MAAM;YAChB,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI;SACtE,CAAC;QACF,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC;QAC3B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,KAAK,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,KAAK,UAAU,CAAC;IAC9E,CAAC;IAED,KAAK;QACH,OAAO,IAAI,CAAC,OAAO;aAChB,cAAc,CAAC,WAAW,IAAI,CAAC,EAAE,QAAQ,EAAE;YAC1C,OAAO,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE;YACvC,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,KAAK;SACd,CAAC;aACD,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAA2B,CAAC;aAClD,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;aACtD,KAAK,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACnC,CAAC;IAED,IAAI;QACF,OAAO,IAAI,CAAC,OAAO;aAChB,cAAc,CAAC,WAAW,IAAI,CAAC,EAAE,OAAO,EAAE;YACzC,OAAO,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE;YACvC,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,KAAK;SACd,CAAC;aACD,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAA2B,CAAC;aAClD,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;aACtD,KAAK,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACnC,CAAC;CACF","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport parseMilliseconds from \"parse-ms\";\nimport type { SonarV3Config } from \"../../common/Config.ts\";\nimport { Err, Ok, type Result } from \"ts-results-es\";\n\nexport class Engine {\n  public readonly id: EngineEntity[\"id\"];\n  public readonly name: EngineEntity[\"name\"];\n  public readonly spec: {\n    c3StorageSize: EngineEntity[\"spec\"][\"c3StorageSize\"];\n    executorPodMetadata: {\n      annotations: EngineEntity[\"spec\"][\"executorPodMetadata\"][\"annotations\"];\n      labels: EngineEntity[\"spec\"][\"executorPodMetadata\"][\"labels\"];\n      nodeSelectors: EngineEntity[\"spec\"][\"executorPodMetadata\"][\"nodeSelectors\"];\n      tolerations: {\n        key: EngineEntity[\"spec\"][\"executorPodMetadata\"][\"tolerations\"][number][\"key\"];\n        operator: EngineEntity[\"spec\"][\"executorPodMetadata\"][\"tolerations\"][number][\"operator\"];\n        effect: EngineEntity[\"spec\"][\"executorPodMetadata\"][\"tolerations\"][number][\"effect\"];\n        tolerationDuration: Temporal.Duration;\n        value?: EngineEntity[\"spec\"][\"executorPodMetadata\"][\"tolerations\"][number][\"value\"];\n      }[];\n      idleTimeout?: Temporal.Duration;\n      resourceAllocationOffset: EngineEntity[\"spec\"][\"executorPodMetadata\"][\"resourceAllocationOffset\"];\n      size: EngineEntity[\"spec\"][\"executorPodMetadata\"][\"size\"];\n      spillStorageSize: EngineEntity[\"spec\"][\"executorPodMetadata\"][\"spillStorageSize\"];\n      targetCpuCapacity: EngineEntity[\"spec\"][\"executorPodMetadata\"][\"targetCpuCapacity\"];\n    };\n  };\n  public readonly status: Omit<EngineEntity[\"status\"], \"message\"> & {\n    message: string | null;\n  };\n\n  readonly #config: SonarV3Config;\n\n  // eslint-disable-next-line no-unused-private-class-members\n  readonly #tag: string;\n\n  constructor(entity: EngineEntity, config: SonarV3Config) {\n    this.id = entity.id;\n    this.name = entity.name;\n    this.spec = {\n      ...entity.spec,\n      executorPodMetadata: {\n        ...entity.spec.executorPodMetadata,\n        idleTimeout: entity.spec.executorPodMetadata.idleTimeout\n          ? Temporal.Duration.from(entity.spec.executorPodMetadata.idleTimeout)\n          : undefined,\n        tolerations: entity.spec.executorPodMetadata.tolerations.map(\n          ({ tolerationSeconds, ...toleration }) => ({\n            ...toleration,\n            tolerationDuration: Temporal.Duration.from(parseMilliseconds(tolerationSeconds * 1000)),\n          }),\n        ),\n      },\n    };\n    this.status = {\n      ...entity.status,\n      message: entity.status.message?.length ? entity.status.message : null,\n    };\n    this.#tag = entity.version;\n    this.#config = config;\n  }\n\n  get settled() {\n    return this.status.state !== \"STARTING\" && this.status.state !== \"STOPPING\";\n  }\n\n  start(): Promise<Result<Engine, unknown>> {\n    return this.#config\n      .sonarV3Request(`engines/${this.id}/start`, {\n        headers: { Accept: \"application/json\" },\n        keepalive: true,\n        method: \"PUT\",\n      })\n      .then((res) => res.json() as Promise<EngineEntity>)\n      .then((entity) => Ok(new Engine(entity, this.#config)))\n      .catch((e: unknown) => Err(e));\n  }\n\n  stop(): Promise<Result<Engine, unknown>> {\n    return this.#config\n      .sonarV3Request(`engines/${this.id}/stop`, {\n        headers: { Accept: \"application/json\" },\n        keepalive: true,\n        method: \"PUT\",\n      })\n      .then((res) => res.json() as Promise<EngineEntity>)\n      .then((entity) => Ok(new Engine(entity, this.#config)))\n      .catch((e: unknown) => Err(e));\n  }\n}\n\nexport type EngineEntity = {\n  id: string;\n  name: string;\n  spec: {\n    c3StorageSize: string;\n    executorPodMetadata: {\n      labels: Record<string, string>;\n      annotations: Record<string, string>;\n      nodeSelectors: Record<string, string>;\n      tolerations: {\n        key: string;\n        operator: \"Exists\" | \"Equal\";\n        effect: \"NoSchedule\" | \"PreferNoSchedule\" | \"NoExecute\";\n        tolerationSeconds: number;\n        value?: number;\n      }[];\n      idleTimeout?: string;\n      resourceAllocationOffset: string;\n      size: \"2XSmall\" | \"XSmall\" | \"Small\" | \"Medium\" | \"Large\" | \"XLarge\" | \"2XLarge\";\n      spillStorageSize: string;\n      targetCpuCapacity: string;\n    };\n  };\n  status: {\n    message: string;\n    replicas: {\n      id: string;\n      executors: {\n        cpu: number;\n        memory: number;\n        name: string;\n        status: string;\n      }[];\n    }[];\n    runningExecutors: number;\n    runningReplicas: number;\n    startingExecutors: number;\n    startingReplicas: number;\n    state: \"STARTING\" | \"RUNNING\" | \"STOPPING\" | \"STOPPED\" | \"WARNING\" | \"FAILED\";\n    stoppingExecutors: number;\n    stoppingReplicas: number;\n  };\n  version: string;\n};\n"]}

package/dist/enterprise/engines/EnginesResource.d.ts

@@ -0,0 +1,10 @@
+import type { ResourceConfig, SonarV3Config } from "../../common/Config.ts";
+import type { SignalParam } from "../../common/Params.ts";
+import { Engine } from "./Engine.ts";
+export declare class EnginesResource {
+    #private;
+    constructor(config: ResourceConfig & SonarV3Config);
+    list(): {
+        data({ signal }?: SignalParam): AsyncGenerator<Engine, void, unknown>;
+    };
+}

package/dist/enterprise/engines/EnginesResource.js

@@ -0,0 +1,34 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * 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.
+ */
+import { Engine } from "./Engine.js";
+export class EnginesResource {
+    #config;
+    constructor(config) {
+        this.#config = config;
+    }
+    list() {
+        const getPage = ({ signal } = {}) => this.#config
+            .sonarV3Request("engines", { headers: { Accept: "application/json" }, signal })
+            .then((res) => res.json())
+            .then((res) => res.data.map((entity) => new Engine(entity, this.#config)));
+        return {
+            async *data({ signal } = {}) {
+                yield* await getPage({ signal });
+            },
+        };
+    }
+}
+//# sourceMappingURL=EnginesResource.js.map

package/dist/enterprise/engines/EnginesResource.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"EnginesResource.js","sourceRoot":"","sources":["../../../src/enterprise/engines/EnginesResource.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,EAAE,MAAM,EAAqB,MAAM,aAAa,CAAC;AAExD,MAAM,OAAO,eAAe;IACjB,OAAO,CAAiC;IAEjD,YAAY,MAAsC;QAChD,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED,IAAI;QACF,MAAM,OAAO,GAAG,CAAC,EAAE,MAAM,KAAkB,EAAE,EAAE,EAAE,CAC/C,IAAI,CAAC,OAAO;aACT,cAAc,CAAC,SAAS,EAAE,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE,EAAE,MAAM,EAAE,CAAC;aAC9E,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,EAAuC,CAAC;aAC9D,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAC/E,OAAO;YACL,KAAK,CAAC,CAAC,IAAI,CAAC,EAAE,MAAM,KAAkB,EAAE;gBACtC,KAAK,CAAC,CAAC,MAAM,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;YACnC,CAAC;SACF,CAAC;IACJ,CAAC;CACF","sourcesContent":["/*\n * Copyright (C) 2024-2025 Dremio Corporation\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { ResourceConfig, SonarV3Config } from \"../../common/Config.ts\";\nimport type { SignalParam } from \"../../common/Params.ts\";\nimport { Engine, type EngineEntity } from \"./Engine.ts\";\n\nexport class EnginesResource {\n  readonly #config: ResourceConfig & SonarV3Config;\n\n  constructor(config: ResourceConfig & SonarV3Config) {\n    this.#config = config;\n  }\n\n  list() {\n    const getPage = ({ signal }: SignalParam = {}) =>\n      this.#config\n        .sonarV3Request(\"engines\", { headers: { Accept: \"application/json\" }, signal })\n        .then((res) => res.json() as Promise<{ data: EngineEntity[] }>)\n        .then((res) => res.data.map((entity) => new Engine(entity, this.#config)));\n    return {\n      async *data({ signal }: SignalParam = {}) {\n        yield* await getPage({ signal });\n      },\n    };\n  }\n}\n"]}

package/dist/enterprise/index.d.ts

@@ -0,0 +1,60 @@
+import type { Config } from "../common/Config.ts";
+import { Resources } from "./resources.ts";
+export declare const Dremio: (config: Config) => {
+    _request: import("../common/Config.ts").RequestFn;
+    _sonarResourceConfig: {
+        logger: {
+            debug: typeof console.debug;
+            error: typeof console.error;
+            info: typeof console.info;
+            warn: typeof console.warn;
+        } | undefined;
+        origin: string;
+        request: import("../common/Config.ts").RequestFn;
+        sonarV2Request: (path: string, init: RequestInit | undefined) => Promise<Response>;
+        sonarV3Request: (path: string, init: RequestInit | undefined) => Promise<Response>;
+        v3Request: (path: string, init: RequestInit | undefined) => Promise<Response>;
+    };
+    _sonarV2Request: (path: string, init: RequestInit | undefined) => Promise<Response>;
+    _sonarV3Request: (path: string, init: RequestInit | undefined) => Promise<Response>;
+    catalog: {
+        _catalogReferenceFromEntity: (entity: unknown) => import("./interfaces.ts").EnterpriseCatalogReference;
+        list: () => {
+            data(): AsyncGenerator<import("./interfaces.ts").EnterpriseFunctionCatalogReference | import("./interfaces.ts").EnterpriseHomeCatalogReference | import("./interfaces.ts").EnterpriseSourceCatalogReference | import("./interfaces.ts").EnterpriseSpaceCatalogReference, void, unknown>;
+        };
+        retrieve: (key: string) => Promise<import("ts-results-es").Result<import("./interfaces.ts").EnterpriseCatalogObject, unknown>>;
+        retrieveByPath: (key: string[]) => Promise<import("ts-results-es").Result<import("./interfaces.ts").EnterpriseCatalogObject, unknown>>;
+    };
+    engines: import("./engines/EnginesResource.ts").EnginesResource;
+    jobs: {
+        create: (query: import("../common/Query.ts").Query) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<string>>;
+        observe: (id: string) => import("rxjs").Observable<import("../oss/jobs/Job.ts").JobResult>;
+        retrieve: (id: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("../oss/jobs/Job.ts").JobResult>;
+    };
+    roles: {
+        retrieve: (id: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("ts-results-es").Result<import("./interfaces.ts").Role, unknown>>;
+        retrieveByName: (name: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("ts-results-es").Result<import("./interfaces.ts").Role, unknown>>;
+    };
+    scripts: {
+        list(): {
+            data({ signal }?: import("../common/Params.ts").SignalParam): AsyncGenerator<import("./interfaces.ts").EnterpriseScript, void, unknown>;
+        };
+        retrieve: (id: string, { signal }?: import("../common/Params.ts").SignalParam) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<import("./interfaces.ts").EnterpriseScript>>;
+        store: (properties: {
+            name: string;
+            query: import("../common/Query.ts").Query;
+        }) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<import("./interfaces.ts").EnterpriseScript>>;
+    };
+    users: {
+        retrieveById: (id: string, { signal }?: {
+            signal?: AbortSignal;
+        }) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<import("./interfaces.ts").EnterpriseUser>>;
+        retrieveByName: (name: string, { signal }?: {
+            signal?: AbortSignal;
+        }) => Promise<import("ts-results-es").Err<unknown> | import("ts-results-es").Ok<import("./interfaces.ts").EnterpriseUser>>;
+    };
+};
+export * from "../common/sharedExports.ts";
+export * from "./credentials/index.ts";
+export * from "./interfaces.ts";
+export { Resources as _Resources };

package/dist/enterprise/index.js

@@ -0,0 +1,43 @@
+/*
+ * Copyright (C) 2024-2025 Dremio Corporation
+ *
+ * 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.
+ */
+import { createRequest } from "../common/createRequest.js";
+import { Resources } from "./resources.js";
+const getSonarResourceConfig = (config) => {
+    const request = createRequest(config);
+    return {
+        logger: config.logger,
+        origin: config.origin,
+        request,
+        sonarV2Request: (path, init) => request(`/apiv2/${path}`, init),
+        sonarV3Request: (path, init) => request(`/api/v3/${path}`, init),
+        v3Request: (path, init) => request(`/api/v3/${path}`, init),
+    };
+};
+export const Dremio = (config) => {
+    const sonarResourceConfig = getSonarResourceConfig(config);
+    return {
+        ...Resources(sonarResourceConfig),
+        _request: sonarResourceConfig.request,
+        _sonarResourceConfig: sonarResourceConfig,
+        _sonarV2Request: sonarResourceConfig.sonarV2Request,
+        _sonarV3Request: sonarResourceConfig.sonarV3Request,
+    };
+};
+export * from "../common/sharedExports.js";
+export * from "./credentials/index.js";
+export * from "./interfaces.js";
+export { Resources as _Resources };
+//# sourceMappingURL=index.js.map