aegis-platform-sdk 0.1.0

package/dist/index.cjs

@@ -0,0 +1,719 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var index_exports = {};
+__export(index_exports, {
+  AegisAPIError: () => AegisAPIError,
+  AegisClient: () => AegisClient,
+  AuthError: () => AuthError,
+  NotFoundError: () => NotFoundError,
+  PermissionDeniedError: () => PermissionDeniedError
+});
+module.exports = __toCommonJS(index_exports);
+
+// src/errors.ts
+var AegisAPIError = class extends Error {
+  statusCode;
+  detail;
+  payload;
+  constructor(statusCode, detail = "", payload = null) {
+    super(`HTTP ${statusCode}: ${detail}`);
+    this.name = new.target.name;
+    this.statusCode = statusCode;
+    this.detail = detail;
+    this.payload = payload;
+  }
+};
+var AuthError = class extends AegisAPIError {
+};
+var PermissionDeniedError = class extends AegisAPIError {
+};
+var NotFoundError = class extends AegisAPIError {
+};
+
+// src/transport.ts
+function buildUrl(baseUrl, path, params) {
+  const url = new URL(baseUrl.replace(/\/+$/, "") + path);
+  if (params) {
+    for (const [key, value] of Object.entries(params)) {
+      if (value !== void 0 && value !== null) url.searchParams.set(key, String(value));
+    }
+  }
+  return url.toString();
+}
+async function handleResponse(res) {
+  if (res.status === 204) return null;
+  const text = await res.text();
+  let payload = text;
+  try {
+    payload = text ? JSON.parse(text) : null;
+  } catch {
+  }
+  if (res.ok) return payload;
+  let detail = "";
+  if (payload && typeof payload === "object" && "detail" in payload) {
+    detail = String(payload.detail ?? "");
+  } else if (typeof payload === "string") {
+    detail = payload;
+  }
+  if (res.status === 401) throw new AuthError(401, detail || "unauthenticated", payload);
+  if (res.status === 403) throw new PermissionDeniedError(403, detail || "forbidden", payload);
+  if (res.status === 404) throw new NotFoundError(404, detail || "not found", payload);
+  throw new AegisAPIError(res.status, detail || text, payload);
+}
+
+// src/resources/auth.ts
+var AuthResource = class {
+  constructor(t, setToken) {
+    this.t = t;
+    this.setToken = setToken;
+  }
+  t;
+  setToken;
+  async login(username, password, opts = {}) {
+    const body = { username, password };
+    if (opts.totpCode) body.totp_code = opts.totpCode;
+    const pair = await this.t.request("POST", "/auth/login", { json: body });
+    if (opts.attach !== false) this.setToken(pair.access_token);
+    return pair;
+  }
+  me() {
+    return this.t.request("GET", "/auth/me");
+  }
+  /** Activate a vendor dev-mode session; requires the `vendor` role. */
+  enableDevMode(password) {
+    return this.t.request("POST", "/me/dev-mode", { json: { password } });
+  }
+};
+
+// src/resources/iam.ts
+var UsersResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  list() {
+    return this.t.request("GET", "/iam/users");
+  }
+  get(userId) {
+    return this.t.request("GET", `/iam/users/${userId}`);
+  }
+};
+var RolesResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  list() {
+    return this.t.request("GET", "/iam/roles");
+  }
+  get(roleId) {
+    return this.t.request("GET", `/iam/roles/${roleId}`);
+  }
+};
+var GroupsResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  list(opts = {}) {
+    return this.t.request("GET", "/iam/groups", { params: opts });
+  }
+  get(groupId) {
+    return this.t.request("GET", `/iam/groups/${groupId}`);
+  }
+  create(name, description) {
+    return this.t.request("POST", "/iam/groups", { json: { name, description } });
+  }
+  update(groupId, patch) {
+    return this.t.request("PATCH", `/iam/groups/${groupId}`, { json: patch });
+  }
+  delete(groupId) {
+    return this.t.request("DELETE", `/iam/groups/${groupId}`);
+  }
+  listMembers(groupId) {
+    return this.t.request("GET", `/iam/groups/${groupId}/members`);
+  }
+  addMember(groupId, userId) {
+    return this.t.request("POST", `/iam/groups/${groupId}/members`, { json: { user_id: userId } });
+  }
+  removeMember(groupId, userId) {
+    return this.t.request("DELETE", `/iam/groups/${groupId}/members/${userId}`);
+  }
+  listForUser(userId) {
+    return this.t.request("GET", `/iam/users/${userId}/groups`);
+  }
+};
+var PermissionsResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  list(resource) {
+    return this.t.request("GET", "/iam/permissions", { params: { resource } });
+  }
+};
+var NavResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  tree(asRoleId) {
+    return this.t.request("GET", "/iam/nav", { params: { as_role_id: asRoleId } });
+  }
+};
+var IamResource = class {
+  users;
+  roles;
+  groups;
+  permissions;
+  nav;
+  constructor(t) {
+    this.users = new UsersResource(t);
+    this.roles = new RolesResource(t);
+    this.groups = new GroupsResource(t);
+    this.permissions = new PermissionsResource(t);
+    this.nav = new NavResource(t);
+  }
+};
+
+// src/resources/osdk.ts
+var OsdkResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  list() {
+    return this.t.request("GET", "/governance/tokens");
+  }
+  get(tokenId) {
+    return this.t.request("GET", `/governance/tokens/${tokenId}`);
+  }
+  /** Returns the plaintext secret ONCE — stash it like any other secret. */
+  create(name, scopes, expiresAt) {
+    return this.t.request("POST", "/governance/tokens", {
+      json: { name, scopes, expires_at: expiresAt }
+    });
+  }
+  rotate(tokenId) {
+    return this.t.request("POST", `/governance/tokens/${tokenId}/rotate`);
+  }
+  revoke(tokenId) {
+    return this.t.request("DELETE", `/governance/tokens/${tokenId}`);
+  }
+};
+
+// src/resources/operator.ts
+var OperatorTasksResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  list(opts = {}) {
+    return this.t.request("GET", "/operator/tasks", { params: { ...opts } });
+  }
+  get(taskId) {
+    return this.t.request("GET", `/operator/tasks/${taskId}`);
+  }
+  create(input) {
+    return this.t.request("POST", "/operator/tasks", { json: input });
+  }
+  cancel(taskId) {
+    return this.t.request("PATCH", `/operator/tasks/${taskId}`, { json: { action: "cancel" } });
+  }
+  retry(taskId) {
+    return this.t.request("PATCH", `/operator/tasks/${taskId}`, { json: { action: "retry" } });
+  }
+  reprioritize(taskId, priority) {
+    return this.t.request("PATCH", `/operator/tasks/${taskId}`, {
+      json: { action: "reprioritize", priority }
+    });
+  }
+};
+var OperatorResource = class {
+  tasks;
+  constructor(t) {
+    this.tasks = new OperatorTasksResource(t);
+  }
+};
+
+// src/resources/ontology.ts
+var ObjectsResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  get(objectId) {
+    return this.t.request("GET", `/ontology/nodes/${objectId}`);
+  }
+  list(kind, opts = {}) {
+    return this.t.request("GET", "/ontology/nodes", { params: { node_type: kind, ...opts } });
+  }
+  create(kind, properties, name) {
+    return this.t.request("POST", "/ontology/nodes", {
+      json: { kind, properties, name }
+    });
+  }
+  createMany(kind, items, markings) {
+    return this.t.request("POST", "/ontology/nodes/bulk", { json: { kind, items, markings } });
+  }
+  delete(objectId) {
+    return this.t.request("DELETE", `/ontology/nodes/${objectId}`);
+  }
+  bulkDelete(opts) {
+    return this.t.request("POST", "/ontology/nodes/bulk-delete", { json: opts });
+  }
+  links(objectId) {
+    return this.t.request("GET", `/entities/${objectId}/links`);
+  }
+  history(objectId) {
+    return this.t.request("GET", `/lineage/by-asset/${objectId}`);
+  }
+  queryTimeseries(objectId, propertyName, opts = {}) {
+    return this.t.request(
+      "POST",
+      `/ontology/nodes/${objectId}/properties/${propertyName}/timeseries-query`,
+      { json: opts }
+    );
+  }
+  appendTimeseries(objectId, propertyName, points) {
+    return this.t.request(
+      "POST",
+      `/ontology/nodes/${objectId}/properties/${propertyName}/timeseries/append`,
+      { json: { points } }
+    );
+  }
+};
+var GraphResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  expand(seedId, opts = {}) {
+    return this.t.request("GET", `/entities/${seedId}/graph`, { params: opts });
+  }
+};
+var ExplorerResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  histogram(property, filters) {
+    return this.t.request("GET", "/entities/aggregate", {
+      params: { property, ...filters }
+    });
+  }
+  timeline(filters, bin) {
+    return this.t.request("GET", "/entities/timeline", {
+      params: { bin, ...filters }
+    });
+  }
+};
+var ObjectTypesResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  get(kind) {
+    return this.t.request("GET", `/ontology/object-types/${kind}`);
+  }
+  create(kind, input) {
+    return this.t.request("POST", "/ontology/object-types", { json: { kind, ...input } });
+  }
+  update(kind, input) {
+    return this.t.request("PUT", `/ontology/object-types/${kind}`, { json: input });
+  }
+  delete(kind) {
+    return this.t.request("DELETE", `/ontology/object-types/${kind}`);
+  }
+};
+var LinkTypesResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  declare(input) {
+    return this.t.request("POST", "/ontology/link-types", { json: input });
+  }
+  delete(edgeId) {
+    return this.t.request("DELETE", `/ontology/link-types/${edgeId}`);
+  }
+};
+var OntologyResource = class {
+  constructor(t) {
+    this.t = t;
+    this.objects = new ObjectsResource(t);
+    this.graph = new GraphResource(t);
+    this.explorer = new ExplorerResource(t);
+    this.objectTypes = new ObjectTypesResource(t);
+    this.linkTypes = new LinkTypesResource(t);
+  }
+  t;
+  objects;
+  graph;
+  explorer;
+  objectTypes;
+  linkTypes;
+  /** Shortcut — same as `objectTypes.get(kind)`. */
+  objectType(kind) {
+    return this.objectTypes.get(kind);
+  }
+  osdkManifest() {
+    return this.t.request("GET", "/ontology/osdk/manifest");
+  }
+  osdkFunctions() {
+    return this.t.request("GET", "/ontology/osdk/manifest/functions");
+  }
+  osdkGenerate(lang) {
+    return this.t.request("GET", "/ontology/osdk/generate", { params: { lang } });
+  }
+};
+
+// src/resources/aip.ts
+var FlowsResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  list() {
+    return this.t.request("GET", "/campaign/flows");
+  }
+  get(flowId) {
+    return this.t.request("GET", `/campaign/flows/${flowId}`);
+  }
+  runs(flowId) {
+    return this.t.request("GET", `/campaign/flows/${flowId}/runs`);
+  }
+  patch(flowId, payload) {
+    return this.t.request("PATCH", `/campaign/flows/${flowId}`, { json: payload });
+  }
+  run(flowId, dryRun) {
+    return this.t.request("POST", `/campaign/flows/${flowId}/run`, {
+      json: dryRun === void 0 ? {} : { dry_run: dryRun }
+    });
+  }
+};
+var AgentsResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  list() {
+    return this.t.request("GET", "/campaign/agents");
+  }
+  get(agentId) {
+    return this.t.request("GET", `/campaign/agents/${agentId}`);
+  }
+  patch(agentId, config) {
+    return this.t.request("PATCH", `/campaign/agents/${agentId}`, { json: config });
+  }
+  manualRun(agentId) {
+    return this.t.request("POST", `/campaign/agents/${agentId}/runs`);
+  }
+  runs(agentId) {
+    return this.t.request("GET", `/campaign/agents/${agentId}/runs`);
+  }
+};
+var ModelsResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  catalog(provider) {
+    return this.t.request("GET", "/platform/v25/aip/models/catalog", { params: { provider } });
+  }
+  defaults() {
+    return this.t.request("GET", "/platform/v25/aip/models/defaults");
+  }
+  setDefault(input) {
+    return this.t.request("PUT", "/platform/v25/aip/models/defaults", { json: input });
+  }
+};
+var BudgetResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  get() {
+    return this.t.request("GET", "/platform/v23/llm-budget");
+  }
+  set(tokensPerMonth) {
+    return this.t.request("PUT", "/platform/v23/llm-budget", {
+      json: { tokens_per_month: tokensPerMonth }
+    });
+  }
+};
+var AipResource = class {
+  flows;
+  agents;
+  models;
+  budget;
+  constructor(t) {
+    this.flows = new FlowsResource(t);
+    this.agents = new AgentsResource(t);
+    this.models = new ModelsResource(t);
+    this.budget = new BudgetResource(t);
+  }
+};
+
+// src/resources/functions.ts
+var FunctionsResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  list() {
+    return this.t.request("GET", "/platform/v26/functions");
+  }
+  get(slug) {
+    return this.t.request("GET", `/platform/v26/functions/${slug}`);
+  }
+  create(input) {
+    return this.t.request("POST", "/platform/v26/functions", { json: input });
+  }
+  invoke(slug, inputs, opts = {}) {
+    return this.t.request("POST", `/platform/v26/functions/${slug}/invoke`, {
+      json: { inputs: inputs ?? {}, ...opts }
+    });
+  }
+  tests(slug) {
+    return this.t.request("GET", `/platform/v26/functions/${slug}/tests`);
+  }
+  addTest(slug, name, inputs, expectedContains) {
+    return this.t.request("POST", `/platform/v26/functions/${slug}/tests`, {
+      json: { name, inputs, expected_contains: expectedContains }
+    });
+  }
+  runTest(slug, testId) {
+    return this.t.request("POST", `/platform/v26/functions/${slug}/tests/${testId}/run`);
+  }
+  runTests(slug) {
+    return this.t.request("POST", `/platform/v26/functions/${slug}/test`);
+  }
+  publish(slug, branch) {
+    return this.t.request("POST", `/platform/v26/functions/${slug}/publish`, {
+      json: branch ? { branch } : {}
+    });
+  }
+  versions(slug) {
+    return this.t.request("GET", `/platform/v26/functions/${slug}/versions`);
+  }
+  pullRequests(slug, status) {
+    return this.t.request("GET", `/platform/v26/functions/${slug}/pull-requests`, {
+      params: { status }
+    });
+  }
+  openPullRequest(slug, input) {
+    return this.t.request("POST", `/platform/v26/functions/${slug}/pull-requests`, { json: input });
+  }
+  updatePullRequest(slug, prId, patch) {
+    return this.t.request("PATCH", `/platform/v26/functions/${slug}/pull-requests/${prId}`, {
+      json: patch
+    });
+  }
+  runPrCi(slug, prId) {
+    return this.t.request("POST", `/platform/v26/functions/${slug}/pull-requests/${prId}/ci`);
+  }
+  reviewPullRequest(slug, prId, decision, comment) {
+    return this.t.request("POST", `/platform/v26/functions/${slug}/pull-requests/${prId}/reviews`, {
+      json: { decision, comment }
+    });
+  }
+  mergePullRequest(slug, prId) {
+    return this.t.request("POST", `/platform/v26/functions/${slug}/pull-requests/${prId}/merge`);
+  }
+};
+var CodeRepositoriesResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  list() {
+    return this.t.request("GET", "/platform/v26/code-repositories");
+  }
+  create(input) {
+    return this.t.request("POST", "/platform/v26/code-repositories", { json: input });
+  }
+  branches(repoId) {
+    return this.t.request("GET", `/platform/v26/code-repositories/${repoId}/branches`);
+  }
+  createBranch(repoId, name, description) {
+    return this.t.request("POST", `/platform/v26/code-repositories/${repoId}/branches`, {
+      json: { name, description }
+    });
+  }
+  functions(repoId) {
+    return this.t.request("GET", `/platform/v26/code-repositories/${repoId}/functions`);
+  }
+  assignFunction(repoId, slug) {
+    return this.t.request("POST", `/platform/v26/code-repositories/${repoId}/functions`, {
+      json: { slug }
+    });
+  }
+  runCi(repoId) {
+    return this.t.request("POST", `/platform/v26/code-repositories/${repoId}/ci`);
+  }
+  mergeBranch(repoId, sourceBranch, into) {
+    return this.t.request("POST", `/platform/v26/code-repositories/${repoId}/merge`, {
+      json: { source_branch: sourceBranch, into }
+    });
+  }
+};
+
+// src/resources/datasets.ts
+var DatasetsResource = class {
+  constructor(t) {
+    this.t = t;
+  }
+  t;
+  list() {
+    return this.t.request("GET", "/datasets");
+  }
+  get(name) {
+    return this.t.request("GET", `/datasets/${name}`);
+  }
+  sample(name, limit) {
+    return this.t.request("GET", `/datasets/${name}/sample`, { params: { limit } });
+  }
+  listBranches(name) {
+    return this.t.request("GET", `/datasets/${name}/branches`);
+  }
+  createBranch(name, branchName, opts = {}) {
+    return this.t.request("POST", `/datasets/${name}/branches`, {
+      json: { branch_name: branchName, ...opts }
+    });
+  }
+  deactivateBranch(name, branchName) {
+    return this.t.request("DELETE", `/datasets/${name}/branches/${branchName}`);
+  }
+  listTransactions(name, branch) {
+    return this.t.request("GET", `/datasets/${name}/transactions`, { params: { branch } });
+  }
+  beginTransaction(name, kind, branch) {
+    return this.t.request("POST", `/datasets/${name}/transactions`, { json: { kind, branch } });
+  }
+  commitTransaction(name, txId, inputPayload) {
+    return this.t.request("POST", `/datasets/${name}/transactions/${txId}/commit`, {
+      json: inputPayload
+    });
+  }
+  abortTransaction(name, txId, reason) {
+    return this.t.request("POST", `/datasets/${name}/transactions/${txId}/abort`, {
+      json: { reason }
+    });
+  }
+  mergeFastForward(name, source, into) {
+    return this.t.request("POST", `/datasets/${name}/branches/${source}/merge`, {
+      json: { into }
+    });
+  }
+  /** 409 responses carry the conflict body in `AegisAPIError.payload`. */
+  mergeThreeWay(name, source, into, resolutions) {
+    return this.t.request("POST", `/datasets/${name}/branches/${source}/merge3`, {
+      json: { into, resolutions }
+    });
+  }
+  getClassification(name) {
+    return this.t.request("GET", `/datasets/${name}/classification`);
+  }
+  setClassification(name, patch) {
+    return this.t.request("PATCH", `/datasets/${name}/classification`, { json: patch });
+  }
+  listMarkings(name) {
+    return this.t.request("GET", `/datasets/${name}/markings`);
+  }
+  applyMarking(name, markingId) {
+    return this.t.request("POST", `/datasets/${name}/markings`, { json: { marking_id: markingId } });
+  }
+  removeMarking(name, markingId) {
+    return this.t.request("DELETE", `/datasets/${name}/markings/${markingId}`);
+  }
+};
+
+// src/client.ts
+function envVar(name) {
+  if (typeof process !== "undefined" && process.env) return process.env[name];
+  return void 0;
+}
+var AegisClient = class {
+  baseUrl;
+  _token;
+  timeoutMs;
+  _fetch;
+  auth;
+  iam;
+  osdk;
+  operator;
+  ontology;
+  aip;
+  functions;
+  codeRepositories;
+  datasets;
+  constructor(options = {}) {
+    const baseUrl = options.baseUrl ?? envVar("AEGIS_API_URL") ?? "http://localhost:8002";
+    this.baseUrl = baseUrl.replace(/\/+$/, "");
+    this._token = options.token ?? envVar("AEGIS_TOKEN") ?? null;
+    this.timeoutMs = options.timeoutMs ?? 15e3;
+    this._fetch = options.fetch ?? globalThis.fetch.bind(globalThis);
+    this.auth = new AuthResource(this, (t) => this.setToken(t));
+    this.iam = new IamResource(this);
+    this.osdk = new OsdkResource(this);
+    this.operator = new OperatorResource(this);
+    this.ontology = new OntologyResource(this);
+    this.aip = new AipResource(this);
+    this.functions = new FunctionsResource(this);
+    this.codeRepositories = new CodeRepositoriesResource(this);
+    this.datasets = new DatasetsResource(this);
+  }
+  /** Replace the active bearer token (or clear with `null`). */
+  setToken(token) {
+    this._token = token;
+  }
+  get token() {
+    return this._token;
+  }
+  /** Generic escape hatch — call any AEGIS endpoint not yet wrapped. */
+  async request(method, path, opts = {}) {
+    const url = buildUrl(this.baseUrl, path, opts.params);
+    const headers = {};
+    if (this._token) headers.Authorization = `Bearer ${this._token}`;
+    const init = { method, headers };
+    if (opts.json !== void 0) {
+      headers["Content-Type"] = "application/json";
+      init.body = JSON.stringify(opts.json);
+    }
+    if (opts.signal) {
+      init.signal = opts.signal;
+    } else if (typeof AbortSignal !== "undefined" && "timeout" in AbortSignal) {
+      init.signal = AbortSignal.timeout(this.timeoutMs);
+    }
+    const res = await this._fetch(url, init);
+    return handleResponse(res);
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  AegisAPIError,
+  AegisClient,
+  AuthError,
+  NotFoundError,
+  PermissionDeniedError
+});
+//# sourceMappingURL=index.cjs.map