@keboola/api-client 0.1.0 → 1.0.0

package/dist/index.js

@@ -45,6 +45,25 @@ var ServiceUnavailableError = class _ServiceUnavailableError extends Error {
     Object.setPrototypeOf(this, _ServiceUnavailableError.prototype);
   }
 };
+
+// src/constants.ts
+var KeboolaHttpHeader = {
+  STORAGE_API_TOKEN: "X-StorageApi-Token",
+  MANAGEMENT_API_TOKEN: "X-Kbc-Manageapitoken",
+  STORAGE_API_URL: "x-storageapi-url"
+};
+var HttpStatus = {
+  NO_CONTENT: 204,
+  BAD_REQUEST: 400,
+  UNPROCESSABLE_ENTITY: 422,
+  INTERNAL_SERVER_ERROR: 500,
+  BAD_GATEWAY: 502,
+  SERVICE_UNAVAILABLE: 503,
+  GATEWAY_TIMEOUT: 504
+};
+var ENCRYPTED_VALUE_PREFIX = "KBC::";
+
+// src/fetchClient/createFetchClient/utils.ts
 var HttpHeader = {
   CONTENT_TYPE: "content-type",
   CONTENT_DISPOSITION: "content-disposition"
@@ -64,7 +83,7 @@ function removeUndefined(obj) {
   return objCopy;
 }
 var parseData = async (response) => {
-  if (response.status === 204) return null;
+  if (response.status === HttpStatus.NO_CONTENT) return null;
   const contentType = response.headers.get(HttpHeader.CONTENT_TYPE);
   if (contentType && contentType == HttpContentType.JSON) {
     return response.json();
@@ -284,14 +303,6 @@ var createAssetsClient = ({ baseUrl, middlewares }) => {
     getPublishedChangelogPosts
   };
 };
-
-// src/constants.ts
-var KeboolaHttpHeader = {
-  STORAGE_API_TOKEN: "X-StorageApi-Token",
-  MANAGEMENT_API_TOKEN: "X-Kbc-Manageapitoken",
-  STORAGE_API_URL: "x-storageapi-url"
-};
-var ENCRYPTED_VALUE_PREFIX = "KBC::";
 var SUGGESTION_CATEGORIES = ["error", "maintenance", "documentation"];
 var suggestionCategorySchema = z.enum(SUGGESTION_CATEGORIES);
 var suggestionSchema = z.object({
@@ -692,7 +703,7 @@ var createDataScienceClient = ({ baseUrl, middlewares }) => {
         nextLogTimestamp
       };
     } catch (error) {
-      if (isApiError(error) && error.response.status === 400 && TailError.safeParse(error.data).data?.context?.code === "apps.appNotRunning") {
+      if (isApiError(error) && error.response.status === HttpStatus.BAD_REQUEST && TailError.safeParse(error.data).data?.context?.code === "apps.appNotRunning") {
         return { entries: [], nextLogTimestamp: null };
       }
       throw error;
@@ -835,14 +846,6 @@ var createEditorClient = ({ baseUrl, middlewares }) => {
     });
     return data;
   };
-  const getWorkspacePassword = async (id, signal) => {
-    const { data } = await client.get(
-      `/sql/sessions/{id}/workspace-password`,
-      { path: { id } },
-      { signal }
-    );
-    return data;
-  };
   const resetWorkspacePassword = async (id) => {
     const { data } = await client.post(`/sql/sessions/{id}/reset-password`, {
       path: { id }
@@ -867,7 +870,6 @@ var createEditorClient = ({ baseUrl, middlewares }) => {
     tableDefinition,
     load,
     unload,
-    getWorkspacePassword,
     resetWorkspacePassword,
     getSessionCredentials
   };
@@ -996,6 +998,10 @@ var createKaiAgentClient = ({ baseUrl, middlewares }) => {
   const submitToolOutput = async (chatId, body) => {
     await client.post("/api/chat/{id}/ui-tool-result", { path: { id: chatId }, body });
   };
+  const stopChat = async (chatId) => {
+    const { data } = await client.post("/api/chat/{id}/stop", { path: { id: chatId } });
+    return data;
+  };
   const getSuggestions = async (request, signal) => {
     const { data } = await genericClient.post(
       "/api/suggestions",
@@ -1075,6 +1081,7 @@ var createKaiAgentClient = ({ baseUrl, middlewares }) => {
     getUsage,
     approveToolCall,
     submitToolOutput,
+    stopChat,
     getSuggestions,
     getAgentSettings,
     updateAgentSettings,
@@ -1108,6 +1115,7 @@ var createVerifyClient = () => {
 };
 
 // src/clients/management/authMiddleware.ts
+var TOKEN_CACHE_TTL_MS = 6e4;
 var isManagementClientAuthError = (error) => error instanceof ManagementClientAuthError;
 var createAuth = (baseUrl) => {
   const fetchClient = createGenericFetchClient({ baseUrl });
@@ -1153,18 +1161,34 @@ var createAuthMiddleware = ({
   const auth = createAuth(baseUrl);
   const verify = createVerifyClient();
   let token = accessToken;
+  let tokenValidUntil = 0;
+  let inflightAuth = null;
   const isDevelopment = process.env.NODE_ENV === "development";
-  return (next) => async (request) => {
-    if (!!token) {
+  const ensureValidToken = async () => {
+    if (token && Date.now() < tokenValidUntil) return token;
+    if (token) {
       const tokenInfo = await verify.managementApiToken(baseUrl, token);
       const isTokenValid = !tokenInfo.isDisabled && !tokenInfo.isExpired;
-      if (!isTokenValid) {
-        token = isDevelopment ? await auth.createCurrentUserSessionToken(token) : await auth.createAdminAccountSessionToken();
+      if (isTokenValid) {
+        tokenValidUntil = Date.now() + TOKEN_CACHE_TTL_MS;
+        return token;
       }
-    } else {
-      token = await auth.createAdminAccountSessionToken();
+      token = isDevelopment ? await auth.createCurrentUserSessionToken(token) : await auth.createAdminAccountSessionToken();
+      tokenValidUntil = Date.now() + TOKEN_CACHE_TTL_MS;
+      return token;
     }
-    request.request.headers.set(KeboolaHttpHeader.MANAGEMENT_API_TOKEN, token);
+    token = await auth.createAdminAccountSessionToken();
+    tokenValidUntil = Date.now() + TOKEN_CACHE_TTL_MS;
+    return token;
+  };
+  return (next) => async (request) => {
+    if (!inflightAuth) {
+      inflightAuth = ensureValidToken().finally(() => {
+        inflightAuth = null;
+      });
+    }
+    const validToken = await inflightAuth;
+    request.request.headers.set(KeboolaHttpHeader.MANAGEMENT_API_TOKEN, validToken);
     return next(request);
   };
 };
@@ -1368,14 +1392,6 @@ var createMaintainers = (client) => {
       path: { id: maintainerId, invitationId }
     });
   };
-  const getMaintainerOrganizations = async (maintainerId, signal) => {
-    const { data } = await client.get(
-      "/manage/maintainers/{id}/organizations",
-      { path: { id: maintainerId } },
-      { signal }
-    );
-    return data;
-  };
   const createMaintainerOrganization = async ({
     maintainerId,
     ...body
@@ -1425,7 +1441,6 @@ var createMaintainers = (client) => {
     getMaintainerInvitation,
     createMaintainerInvitation,
     deleteMaintainerInvitation,
-    getMaintainerOrganizations,
     createMaintainerOrganization,
     getMaintainerMetadata,
     setMaintainerMetadata,
@@ -1747,13 +1762,18 @@ var createManagementClient = ({
     baseUrl,
     middlewares
   });
+  const verifyToken = async (signal) => {
+    const { data } = await openApiClient.get("/manage/tokens/verify", {}, { signal });
+    return data;
+  };
   return {
     currentUser: createCurrentUser(openApiClient),
     features: createFeatures(client),
     maintainers: createMaintainers(openApiClient),
     organizations: createOrganizations(openApiClient),
     projects: createProjects(client),
-    users: createUsers(client)
+    users: createUsers(client),
+    verifyToken
   };
 };
 
@@ -1933,6 +1953,72 @@ var createMetastoreClient = ({ baseUrl, middlewares }) => {
   };
 };
 
+// src/clients/oauth/oauthClient.ts
+var createOauthClient = ({ baseUrl, middlewares }) => {
+  const client = createGenericFetchClient({ baseUrl, middlewares });
+  const getCredentials = async (query, signal) => {
+    const { data } = await client.get(
+      "/credentials",
+      { query },
+      { signal }
+    );
+    return data;
+  };
+  const getCredential = async (componentId, id, signal) => {
+    const { data } = await client.get(
+      "/credentials/{componentId}/{id}",
+      { path: { componentId, id } },
+      { signal }
+    );
+    return data;
+  };
+  const _createCredentials = async (componentId, body, signal) => {
+    const { data } = await client.post(
+      "/credentials/{componentId}",
+      { path: { componentId }, body },
+      { signal }
+    );
+    return data;
+  };
+  const _createLegacyCredentials = async (componentId, { data: credentialsData, ...rest }, signal) => {
+    const { data } = await client.post(
+      "/credentials/{componentId}",
+      {
+        path: { componentId },
+        body: {
+          ...rest,
+          "#data": JSON.stringify(credentialsData)
+        }
+      },
+      { signal }
+    );
+    return data;
+  };
+  const createCredential = async (componentId, body, signal) => {
+    try {
+      return await _createCredentials(componentId, body, signal);
+    } catch (error) {
+      if (isApiError(error) && error.response.status === HttpStatus.UNPROCESSABLE_ENTITY) {
+        return _createLegacyCredentials(componentId, body, signal);
+      }
+      throw error;
+    }
+  };
+  const deleteCredential = async (componentId, id, signal) => {
+    await client.delete(
+      "/credentials/{componentId}/{id}",
+      { path: { componentId, id } },
+      { signal }
+    );
+  };
+  return {
+    getCredentials,
+    getCredential,
+    createCredential,
+    deleteCredential
+  };
+};
+
 // src/clients/queryService/queryServiceClient.ts
 var createQueryServiceClient = ({ baseUrl, middlewares }) => {
   const client = createOpenapiFetchClient({
@@ -2088,7 +2174,6 @@ var createMergeRequests = (client) => {
   const getMergeRequestWithActivityLog = async (id, signal) => {
     const { data } = await client.get(
       "/v2/storage/merge-request/{id}",
-      // @ts-expect-error backend accepts `?include=activityLog`; OpenAPI schema regen pending (connection#7250).
       { path: { id }, query: { include: "activityLog" } },
       { signal }
     );
@@ -2120,11 +2205,18 @@ var createMergeRequests = (client) => {
   const requestChangesMergeRequest = async (id, reason) => {
     const { data } = await client.put("/v2/storage/merge-request/{id}/request-changes", {
       path: { id },
-      // @ts-expect-error backend accepts optional `{ reason }` body; OpenAPI schema regen pending (connection#7250).
       body: reason ? { reason } : void 0
     });
     return data;
   };
+  const getMergeRequestConflicts = async (id, signal) => {
+    const { data } = await client.get(
+      "/v2/storage/merge-request/{id}/conflicts",
+      { path: { id } },
+      { signal }
+    );
+    return data;
+  };
   return {
     getMergeRequests,
     getMergeRequest,
@@ -2133,7 +2225,8 @@ var createMergeRequests = (client) => {
     updateMergeRequest,
     requestReview,
     approveMergeRequest,
-    requestChangesMergeRequest
+    requestChangesMergeRequest,
+    getMergeRequestConflicts
   };
 };
 
@@ -3648,6 +3741,11 @@ var createClients = async ({
           baseUrl: serviceBaseUrl,
           middlewares: commonMiddlewares
         })
+      },
+      {
+        serviceId: "oauth",
+        clientName: "oauth",
+        clientFn: (serviceBaseUrl) => createOauthClient({ baseUrl: serviceBaseUrl, middlewares: commonMiddlewares })
       }
     ]
   });
@@ -3817,7 +3915,12 @@ var createLoggingMiddleware = (options = {}) => {
 
 // src/middlewares/retryMiddleware.ts
 var DEFAULT_MAX_RETRIES = 5;
-var DEFAULT_RETRYABLE_STATUSES = [500, 502, 503, 504];
+var DEFAULT_RETRYABLE_STATUSES = [
+  HttpStatus.INTERNAL_SERVER_ERROR,
+  HttpStatus.BAD_GATEWAY,
+  HttpStatus.SERVICE_UNAVAILABLE,
+  HttpStatus.GATEWAY_TIMEOUT
+];
 var DEFAULT_BACKOFF_FACTOR = 0.3;
 var defaultShouldRetry = (error) => error instanceof ApiError && DEFAULT_RETRYABLE_STATUSES.includes(error.response.status);
 var createRetryMiddleware = (options) => {