@madewithremy/admin 0.1.5 → 0.1.7

package/dist/index.js

@@ -89,6 +89,16 @@ var AdminTimeoutError = class extends Error {
 
 // src/http.ts
 var REQUEST_TIMEOUT_MS = 3e4;
+var TRANSIENT_GRACE_MS = 2 * 6e4;
+function isTransientError(err) {
+  if (err instanceof AdminApiError) {
+    return err.status >= 500;
+  }
+  if (err instanceof AdminTimeoutError) {
+    return true;
+  }
+  return err instanceof TypeError;
+}
 var REPORT_TIMEOUT_MS = 6e4;
 function seg(value) {
   return encodeURIComponent(String(value));
@@ -350,19 +360,24 @@ function liftFromLive(ctx, params = {}) {
 var dataSources_exports = {};
 __export(dataSources_exports, {
   DEFAULT_WAIT_TIMEOUT_MS: () => DEFAULT_WAIT_TIMEOUT_MS,
+  DOCUMENTS_IDS_MAX: () => DOCUMENTS_IDS_MAX,
   addDocument: () => addDocument,
+  allDocuments: () => allDocuments,
   configGet: () => configGet,
   configSet: () => configSet,
+  create: () => create,
   deleteSource: () => deleteSource,
   documents: () => documents,
   drop: () => drop,
   list: () => list3,
+  move: () => move,
   promote: () => promote,
   revectorize: () => revectorize,
   rm: () => rm,
   search: () => search,
   summarize: () => summarize,
-  waitForIngest: () => waitForIngest
+  waitForIngest: () => waitForIngest,
+  waitForMove: () => waitForMove
 });
 import { createHash } from "crypto";
 
@@ -456,12 +471,26 @@ async function waitForIngest(ctx, params) {
   }
   const wanted = new Set(documentIds);
   const start = Date.now();
+  let transientSince = null;
   for (; ; ) {
-    const { documents: docs } = await call(
-      ctx,
-      "GET",
-      `${base(ctx.appId)}/documents?slug=${encodeURIComponent(slug)}`
-    );
+    let docs;
+    try {
+      docs = await allDocuments(ctx, { slug, ids: documentIds });
+      transientSince = null;
+    } catch (err) {
+      if (!isTransientError(err)) {
+        throw err;
+      }
+      transientSince ??= Date.now();
+      if (Date.now() - transientSince > TRANSIENT_GRACE_MS) {
+        throw err;
+      }
+      onProgress?.(
+        `retrying after a gateway error\u2026 (${Math.round((Date.now() - start) / 1e3)}s)`
+      );
+      await sleep(POLL_MS);
+      continue;
+    }
     const tracked = (docs ?? []).filter((d) => wanted.has(d.id));
     const pending = tracked.filter((d) => d.status === "processing");
     const failed = tracked.filter((d) => d.status === "error");
@@ -486,12 +515,41 @@ async function waitForIngest(ctx, params) {
     await sleep(POLL_MS);
   }
 }
+var DOCUMENTS_IDS_MAX = 200;
 function documents(ctx, params) {
   return call(
     ctx,
     "GET",
-    `${base(ctx.appId)}/documents${qs({ slug: params.slug, candidate: params.candidate || void 0 })}`
-  );
+    `${base(ctx.appId)}/documents${qs({
+      slug: params.slug,
+      candidate: params.candidate || void 0,
+      ids: params.ids?.length ? params.ids.join(",") : void 0,
+      limit: params.limit,
+      cursor: params.cursor
+    })}`
+  );
+}
+async function allDocuments(ctx, params) {
+  const groups = params.ids ? Array.from(
+    { length: Math.ceil(params.ids.length / DOCUMENTS_IDS_MAX) },
+    (_, i) => params.ids.slice(i * DOCUMENTS_IDS_MAX, (i + 1) * DOCUMENTS_IDS_MAX)
+  ) : [void 0];
+  const out = [];
+  for (const ids of groups) {
+    let cursor;
+    do {
+      const page = await documents(ctx, {
+        slug: params.slug,
+        candidate: params.candidate,
+        ids,
+        limit: 1e3,
+        cursor
+      });
+      out.push(...page.documents ?? []);
+      cursor = page.nextCursor ?? void 0;
+    } while (cursor);
+  }
+  return out;
 }
 function list3(ctx) {
   return call(ctx, "GET", base(ctx.appId));
@@ -533,7 +591,7 @@ function configGet(ctx, slug) {
   );
 }
 function configSet(ctx, params) {
-  const { slug, ingest, retrieval } = params;
+  const { slug, ingest, retrieval, placement } = params;
   return call(
     ctx,
     "POST",
@@ -541,10 +599,75 @@ function configSet(ctx, params) {
     {
       slug,
       ...ingest ? { ingest } : {},
-      ...retrieval ? { retrieval } : {}
+      ...retrieval ? { retrieval } : {},
+      ...placement !== void 0 ? { placement } : {}
     }
   );
 }
+function create(ctx, params) {
+  const { slug, name, ingest, placement } = params;
+  return call(ctx, "POST", base(ctx.appId), {
+    slug,
+    ...name ? { name } : {},
+    ...ingest ? { ingest } : {},
+    ...placement !== void 0 ? { placement } : {}
+  });
+}
+function move(ctx, params) {
+  return call(ctx, "POST", `${base(ctx.appId)}/move`, {
+    slug: params.slug,
+    placement: params.placement
+  });
+}
+async function waitForMove(ctx, params) {
+  const timeoutMs = params.timeoutMs ?? DEFAULT_WAIT_TIMEOUT_MS;
+  const start = Date.now();
+  let transientSince = null;
+  for (; ; ) {
+    let source;
+    try {
+      const { dataSources } = await list3(ctx);
+      source = dataSources.find((s) => s.slug === params.slug);
+      transientSince = null;
+    } catch (err) {
+      if (!isTransientError(err)) {
+        throw err;
+      }
+      transientSince ??= Date.now();
+      if (Date.now() - transientSince > TRANSIENT_GRACE_MS) {
+        throw err;
+      }
+      params.onProgress?.(
+        `retrying after a gateway error\u2026 (${Math.round((Date.now() - start) / 1e3)}s)`
+      );
+      await sleep(POLL_MS);
+      continue;
+    }
+    if (!source) {
+      throw new Error(`Data source "${params.slug}" not found.`);
+    }
+    const migration = source.migration;
+    if (!migration) {
+      return { status: "done", source };
+    }
+    if (migration.error) {
+      return { status: "failed", source, error: migration.error };
+    }
+    if (Date.now() - start > timeoutMs) {
+      return {
+        status: "timeout",
+        source,
+        error: `Timed out after ${Math.round(timeoutMs / 1e3)}s with ${migration.copied}/${migration.total} documents copied`
+      };
+    }
+    params.onProgress?.(
+      `moving\u2026 ${migration.copied}/${migration.total} documents (${Math.round(
+        (Date.now() - start) / 1e3
+      )}s)`
+    );
+    await sleep(POLL_MS);
+  }
+}
 function revectorize(ctx, params) {
   const { slug, ingest } = params;
   return call(
@@ -681,7 +804,9 @@ function summarizeRelease(release) {
   }
   return {
     ...summary2,
-    error: buildFailureReason(release) ?? "Build failed (no error entry in the build log)"
+    error: release.failureReason ?? buildFailureReason(release) ?? "Build failed (no error entry in the build log)",
+    ...release.failureKind ? { failureKind: release.failureKind } : {},
+    ...release.failurePhase ? { failurePhase: release.failurePhase } : {}
   };
 }
 var POLL_MS2 = 3e3;
@@ -1101,33 +1226,157 @@ function remove(ctx, ref) {
   );
 }
 
+// src/ops/infra.ts
+var infra_exports = {};
+__export(infra_exports, {
+  DEFAULT_WAIT_TIMEOUT_MS: () => DEFAULT_WAIT_TIMEOUT_MS2,
+  destroy: () => destroy,
+  get: () => get5,
+  hibernate: () => hibernate,
+  list: () => list6,
+  provision: () => provision,
+  rename: () => rename,
+  resize: () => resize,
+  resume: () => resume,
+  waitForPhase: () => waitForPhase
+});
+var DEFAULT_WAIT_TIMEOUT_MS2 = 10 * 60 * 1e3;
+var POLL_MS3 = 3e3;
+function base2(appId) {
+  return `/_internal/v2/apps/${appId}/infra`;
+}
+function list6(ctx, params) {
+  return call(
+    ctx,
+    "GET",
+    `${base2(ctx.appId)}${params?.includeDestroyed ? "?includeDestroyed=true" : ""}`
+  );
+}
+function get5(ctx, params) {
+  return call(
+    ctx,
+    "GET",
+    `${base2(ctx.appId)}/${encodeURIComponent(params.id)}`
+  );
+}
+function provision(ctx, params) {
+  return call(ctx, "POST", base2(ctx.appId), {
+    offeringId: params.offeringId,
+    ...params.name ? { name: params.name } : {}
+  });
+}
+function hibernate(ctx, params) {
+  return call(
+    ctx,
+    "POST",
+    `${base2(ctx.appId)}/${encodeURIComponent(params.id)}/hibernate`
+  );
+}
+function resume(ctx, params) {
+  return call(
+    ctx,
+    "POST",
+    `${base2(ctx.appId)}/${encodeURIComponent(params.id)}/resume`
+  );
+}
+function destroy(ctx, params) {
+  return call(
+    ctx,
+    "POST",
+    `${base2(ctx.appId)}/${encodeURIComponent(params.id)}/destroy`
+  );
+}
+function resize(ctx, params) {
+  return call(
+    ctx,
+    "POST",
+    `${base2(ctx.appId)}/${encodeURIComponent(params.id)}/resize`,
+    { offeringId: params.offeringId }
+  );
+}
+function rename(ctx, params) {
+  return call(
+    ctx,
+    "POST",
+    `${base2(ctx.appId)}/${encodeURIComponent(params.id)}/rename`,
+    { name: params.name }
+  );
+}
+async function waitForPhase(ctx, params) {
+  const timeoutMs = params.timeoutMs ?? DEFAULT_WAIT_TIMEOUT_MS2;
+  const wanted = new Set(params.phases);
+  const start = Date.now();
+  let transientSince = null;
+  for (; ; ) {
+    let resource;
+    try {
+      ({ resource } = await get5(ctx, { id: params.id }));
+      transientSince = null;
+    } catch (err) {
+      if (!isTransientError(err)) {
+        throw err;
+      }
+      transientSince ??= Date.now();
+      if (Date.now() - transientSince > TRANSIENT_GRACE_MS) {
+        throw err;
+      }
+      params.onProgress?.(
+        `retrying after a gateway error\u2026 (${Math.round((Date.now() - start) / 1e3)}s)`
+      );
+      await sleep(POLL_MS3);
+      continue;
+    }
+    if (wanted.has(resource.phase) && (params.until?.(resource) ?? true)) {
+      return { status: "done", resource };
+    }
+    if (resource.phase === "failed") {
+      return {
+        status: "failed",
+        resource,
+        error: resource.lastError ?? "The platform could not complete the change."
+      };
+    }
+    if (Date.now() - start > timeoutMs) {
+      return {
+        status: "timeout",
+        resource,
+        error: `Timed out after ${Math.round(timeoutMs / 1e3)}s; resource is still ${resource.phase}`
+      };
+    }
+    params.onProgress?.(
+      `${resource.phase}${resource.detail ? ` \xB7 ${resource.detail}` : ""}${resource.resizingTo ? ` \xB7 resizing to ${resource.resizingTo.label}` : ""}\u2026 (${Math.round((Date.now() - start) / 1e3)}s)`
+    );
+    await sleep(POLL_MS3);
+  }
+}
+
 // src/ops/issues.ts
 var issues_exports = {};
 __export(issues_exports, {
   close: () => close,
   comment: () => comment,
-  create: () => create,
+  create: () => create2,
   del: () => del,
   edit: () => edit,
-  get: () => get5,
-  list: () => list6,
+  get: () => get6,
+  list: () => list7,
   reopen: () => reopen
 });
-function list6(ctx, params = {}) {
+function list7(ctx, params = {}) {
   return call(
     ctx,
     "GET",
     `/_internal/v2/apps/${ctx.appId}/issues${qs(params)}`
   );
 }
-function get5(ctx, number) {
+function get6(ctx, number) {
   return call(
     ctx,
     "GET",
     `/_internal/v2/apps/${ctx.appId}/issues/${seg(number)}`
   );
 }
-function create(ctx, params) {
+function create2(ctx, params) {
   const requestBody = {
     title: params.title,
     authorKind: "agent"
@@ -1322,9 +1571,9 @@ function grade(ctx, runId) {
 var methods_exports = {};
 __export(methods_exports, {
   invoke: () => invoke,
-  list: () => list7
+  list: () => list8
 });
-async function list7(ctx) {
+async function list8(ctx) {
   const dashboard = await call(
     ctx,
     "GET",
@@ -1381,19 +1630,19 @@ function invalidate(ctx, params = {}) {
 // src/ops/requests.ts
 var requests_exports = {};
 __export(requests_exports, {
-  get: () => get6,
-  list: () => list8,
+  get: () => get7,
+  list: () => list9,
   statsForMethod: () => statsForMethod,
   statsSummary: () => statsSummary
 });
-function list8(ctx, params = {}) {
+function list9(ctx, params = {}) {
   return call(
     ctx,
     "GET",
     `/_internal/v2/apps/${ctx.appId}/requests${qs(params)}`
   );
 }
-function get6(ctx, requestId) {
+function get7(ctx, requestId) {
   return call(
     ctx,
     "GET",
@@ -1419,18 +1668,18 @@ function statsForMethod(ctx, methodId, params = {}) {
 var secrets_exports = {};
 __export(secrets_exports, {
   del: () => del2,
-  get: () => get7,
-  list: () => list9,
+  get: () => get8,
+  list: () => list10,
   set: () => set2
 });
-function list9(ctx) {
+function list10(ctx) {
   return call(
     ctx,
     "GET",
     `/_internal/v2/apps/${ctx.appId}/secrets`
   );
 }
-function get7(ctx, key) {
+function get8(ctx, key) {
   return call(
     ctx,
     "GET",
@@ -1486,11 +1735,11 @@ function updateSettings(ctx, partial) {
 var users_exports = {};
 __export(users_exports, {
   createApiKey: () => createApiKey,
-  list: () => list10,
+  list: () => list11,
   revokeApiKey: () => revokeApiKey,
   setRole: () => setRole
 });
-function list10(ctx, params = {}) {
+function list11(ctx, params = {}) {
   return call(
     ctx,
     "GET",
@@ -1691,6 +1940,7 @@ function buildClient(ctx) {
       removeDomain: (direction, domain) => removeDomain(ctx, direction, domain)
     },
     files: bindOps(files_exports, ctx),
+    infra: bindOps(infra_exports, ctx),
     issues: bindOps(issues_exports, ctx),
     jewels: bindOps(jewels_exports, ctx),
     methods: bindOps(methods_exports, ctx),