SDK

export class AtlanApiError extends Error {
  constructor(response, body) {
    super(body?.detail ?? `Atlan API returned ${response.status}`);
    this.name = "AtlanApiError";
    this.status = response.status;
    this.code = body?.code;
    this.traceId = body?.trace_id;
  }
}

export function createAtlanJsonClient({ gatewayUrl, token, timeoutMs = 10_000 }) {
  return async function requestJson(path, init = {}) {
    const headers = new Headers(init.headers);
    headers.set("Authorization", `Bearer ${token}`);
    if (init.body != null && !headers.has("Content-Type")) {
      headers.set("Content-Type", "application/json");
    }

    const response = await fetch(new URL(path, gatewayUrl), {
      ...init,
      headers,
      signal: init.signal ?? AbortSignal.timeout(timeoutMs),
    });
    const contentType = response.headers.get("content-type") ?? "";
    const isJson = contentType.includes("application/json") ||
      contentType.includes("application/problem+json");
    const body = isJson ? await response.json() : await response.text();

    if (!response.ok) throw new AtlanApiError(response, body);
    if (!isJson) throw new TypeError("Expected a JSON response");
    return body;
  };
}

const atlan = createAtlanJsonClient({
  gatewayUrl: process.env.ATLAN_GATEWAY_URL,
  token: process.env.ATLAN_TOKEN,
});

const identity = await atlan("/registry/v1/auth/whoami");
console.log(identity);

import { writeFile } from "node:fs/promises";

export async function downloadSkillBundle({ gatewayUrl, token, skillId, version }) {
  const response = await fetch(
    new URL(`/skill/v1/skills/${skillId}/versions/${version}/bundle`, gatewayUrl),
    { headers: { Authorization: `Bearer ${token}` }, signal: AbortSignal.timeout(10_000) },
  );
  if (!response.ok) {
    const body = await response.json().catch(() => undefined);
    throw new AtlanApiError(response, body);
  }
  if (!response.headers.get("content-type")?.includes("application/zip")) {
    throw new TypeError("Expected an application/zip response");
  }
  await writeFile("./skill-bundle.zip", Buffer.from(await response.arrayBuffer()));
}