vanta-auditor-api-sdk 0.4.0 → 0.4.2

package/src/mcp-server/server.ts

@@ -1,79 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import { VantaCore } from "../core.js";
-import { SDKOptions } from "../lib/config.js";
-import type { ConsoleLogger } from "./console-logger.js";
-import { createRegisterPrompt } from "./prompts.js";
-import {
-  createRegisterResource,
-  createRegisterResourceTemplate,
-} from "./resources.js";
-import { MCPScope } from "./scopes.js";
-import { createRegisterTool } from "./tools.js";
-import { tool$auditorsCreate } from "./tools/auditorsCreate.js";
-import { tool$auditsCreateCommentForEvidence } from "./tools/auditsCreateCommentForEvidence.js";
-import { tool$auditsCreateCustomControl } from "./tools/auditsCreateCustomControl.js";
-import { tool$auditsCreateCustomEvidenceRequest } from "./tools/auditsCreateCustomEvidenceRequest.js";
-import { tool$auditsGetEvidenceUrls } from "./tools/auditsGetEvidenceUrls.js";
-import { tool$auditsList } from "./tools/auditsList.js";
-import { tool$auditsListComments } from "./tools/auditsListComments.js";
-import { tool$auditsListControls } from "./tools/auditsListControls.js";
-import { tool$auditsListEvidence } from "./tools/auditsListEvidence.js";
-import { tool$auditsUpdateEvidence } from "./tools/auditsUpdateEvidence.js";
-
-export function createMCPServer(deps: {
-  logger: ConsoleLogger;
-  allowedTools?: string[] | undefined;
-  scopes?: MCPScope[] | undefined;
-  serverURL?: string | undefined;
-  bearerAuth?: SDKOptions["bearerAuth"] | undefined;
-  serverIdx?: SDKOptions["serverIdx"] | undefined;
-}) {
-  const server = new McpServer({
-    name: "Vanta",
-    version: "0.4.0",
-  });
-
-  const client = new VantaCore({
-    bearerAuth: deps.bearerAuth,
-    serverURL: deps.serverURL,
-    serverIdx: deps.serverIdx,
-  });
-
-  const scopes = new Set(deps.scopes);
-
-  const allowedTools = deps.allowedTools && new Set(deps.allowedTools);
-  const tool = createRegisterTool(
-    deps.logger,
-    server,
-    client,
-    scopes,
-    allowedTools,
-  );
-  const resource = createRegisterResource(deps.logger, server, client, scopes);
-  const resourceTemplate = createRegisterResourceTemplate(
-    deps.logger,
-    server,
-    client,
-    scopes,
-  );
-  const prompt = createRegisterPrompt(deps.logger, server, client, scopes);
-  const register = { tool, resource, resourceTemplate, prompt };
-  void register; // suppress unused warnings
-
-  tool(tool$auditsList);
-  tool(tool$auditsGetEvidenceUrls);
-  tool(tool$auditsListEvidence);
-  tool(tool$auditsListComments);
-  tool(tool$auditsListControls);
-  tool(tool$auditsCreateCommentForEvidence);
-  tool(tool$auditsUpdateEvidence);
-  tool(tool$auditsCreateCustomEvidenceRequest);
-  tool(tool$auditsCreateCustomControl);
-  tool(tool$auditorsCreate);
-
-  return server;
-}

package/src/mcp-server/shared.ts

@@ -1,74 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import * as z from "zod/v3";
-import { bytesToBase64 } from "../lib/base64.js";
-type BinaryData =
-  | Uint8Array
-  | ArrayBuffer
-  | Blob
-  | ReadableStream
-  | Response
-  | string;
-
-export async function consumeStream(
-  stream: ReadableStream<Uint8Array>,
-): Promise<Uint8Array> {
-  const reader = stream.getReader();
-  const chunks: Uint8Array[] = [];
-
-  try {
-    while (true) {
-      const { done, value } = await reader.read();
-      if (value != null) chunks.push(value);
-      if (done) break;
-    }
-  } finally {
-    reader.releaseLock();
-  }
-
-  return new Uint8Array(await new Blob(chunks).arrayBuffer());
-}
-
-export function isAsyncIterable(
-  value: unknown,
-): value is AsyncIterable<string> {
-  return (
-    typeof value === "object" && value != null && Symbol.asyncIterator in value
-  );
-}
-
-export function isBinaryData(value: unknown): value is BinaryData {
-  return (
-    value instanceof Uint8Array
-    || value instanceof ArrayBuffer
-    || value instanceof Blob
-    || value instanceof ReadableStream
-    || value instanceof Response
-    || typeof value === "string"
-  );
-}
-
-const base64Schema = z.string().base64();
-
-export async function valueToBase64(
-  value: BinaryData | null | undefined,
-): Promise<string | null> {
-  if (value == null) {
-    return null;
-  } else if (value instanceof Uint8Array) {
-    return bytesToBase64(value);
-  } else if (value instanceof ArrayBuffer) {
-    return bytesToBase64(new Uint8Array(value));
-  } else if (value instanceof Response || value instanceof Blob) {
-    return bytesToBase64(new Uint8Array(await value.arrayBuffer()));
-  } else if (value instanceof ReadableStream) {
-    return bytesToBase64(await consumeStream(value));
-  } else if (typeof value === "string") {
-    return base64Schema.parse(value);
-  } else {
-    value satisfies never;
-    throw new Error(`Unsupported image value type: ${typeof value}`);
-  }
-}

package/src/mcp-server/tools/auditorsCreate.ts

@@ -1,37 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { auditorsCreate } from "../../funcs/auditorsCreate.js";
-import * as components from "../../models/components/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: components.AddAuditorInput$inboundSchema,
-};
-
-export const tool$auditorsCreate: ToolDefinition<typeof args> = {
-  name: "auditors-create",
-  description: `Create an auditor
-
-Create an auditor in Vanta.`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await auditorsCreate(
-      client,
-      args.request,
-      { fetchOptions: { signal: ctx.signal } },
-    ).$inspect();
-
-    if (!result.ok) {
-      return {
-        content: [{ type: "text", text: result.error.message }],
-        isError: true,
-      };
-    }
-
-    const value = result.value;
-
-    return formatResult(value, apiCall);
-  },
-};

package/src/mcp-server/tools/auditsCreateCommentForEvidence.ts

@@ -1,38 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { auditsCreateCommentForEvidence } from "../../funcs/auditsCreateCommentForEvidence.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.CreateCommentForAuditEvidenceRequest$inboundSchema,
-};
-
-export const tool$auditsCreateCommentForEvidence: ToolDefinition<typeof args> =
-  {
-    name: "audits-create-comment-for-evidence",
-    description: `Create a comment for audit evidence
-
-Create a comment in Vanta for a piece of evidence.`,
-    args,
-    tool: async (client, args, ctx) => {
-      const [result, apiCall] = await auditsCreateCommentForEvidence(
-        client,
-        args.request,
-        { fetchOptions: { signal: ctx.signal } },
-      ).$inspect();
-
-      if (!result.ok) {
-        return {
-          content: [{ type: "text", text: result.error.message }],
-          isError: true,
-        };
-      }
-
-      const value = result.value;
-
-      return formatResult(value, apiCall);
-    },
-  };

package/src/mcp-server/tools/auditsCreateCustomControl.ts

@@ -1,37 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { auditsCreateCustomControl } from "../../funcs/auditsCreateCustomControl.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.CreateCustomControlRequest$inboundSchema,
-};
-
-export const tool$auditsCreateCustomControl: ToolDefinition<typeof args> = {
-  name: "audits-create-custom-control",
-  description: `Create a custom control for an audit
-
-Create a custom control for an audit.`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await auditsCreateCustomControl(
-      client,
-      args.request,
-      { fetchOptions: { signal: ctx.signal } },
-    ).$inspect();
-
-    if (!result.ok) {
-      return {
-        content: [{ type: "text", text: result.error.message }],
-        isError: true,
-      };
-    }
-
-    const value = result.value;
-
-    return formatResult(value, apiCall);
-  },
-};

package/src/mcp-server/tools/auditsCreateCustomEvidenceRequest.ts

@@ -1,39 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { auditsCreateCustomEvidenceRequest } from "../../funcs/auditsCreateCustomEvidenceRequest.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.CreateCustomEvidenceRequestRequest$inboundSchema,
-};
-
-export const tool$auditsCreateCustomEvidenceRequest: ToolDefinition<
-  typeof args
-> = {
-  name: "audits-create-custom-evidence-request",
-  description: `Create a custom evidence request for an audit
-
-Create a custom evidence request for an audit.`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await auditsCreateCustomEvidenceRequest(
-      client,
-      args.request,
-      { fetchOptions: { signal: ctx.signal } },
-    ).$inspect();
-
-    if (!result.ok) {
-      return {
-        content: [{ type: "text", text: result.error.message }],
-        isError: true,
-      };
-    }
-
-    const value = result.value;
-
-    return formatResult(value, apiCall);
-  },
-};

package/src/mcp-server/tools/auditsGetEvidenceUrls.ts

@@ -1,38 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { auditsGetEvidenceUrls } from "../../funcs/auditsGetEvidenceUrls.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.ListAuditEvidenceUrlsRequest$inboundSchema,
-};
-
-export const tool$auditsGetEvidenceUrls: ToolDefinition<typeof args> = {
-  name: "audits-get-evidence-urls",
-  description: `List audit evidence url
-
-Returns a paginated list of evidence urls for an audit. This endpoint should be called whenever an
-evidence is created or has a statusUpdatedAt field that is more recent than the most recent polling event.`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await auditsGetEvidenceUrls(
-      client,
-      args.request,
-      { fetchOptions: { signal: ctx.signal } },
-    ).$inspect();
-
-    if (!result.ok) {
-      return {
-        content: [{ type: "text", text: result.error.message }],
-        isError: true,
-      };
-    }
-
-    const value = result.value;
-
-    return formatResult(value, apiCall);
-  },
-};

package/src/mcp-server/tools/auditsList.ts

@@ -1,37 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { auditsList } from "../../funcs/auditsList.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.ListAuditsRequest$inboundSchema,
-};
-
-export const tool$auditsList: ToolDefinition<typeof args> = {
-  name: "audits-list",
-  description: `List audits
-
-Returns a paginated list of audits scoped to the audit firm.`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await auditsList(
-      client,
-      args.request,
-      { fetchOptions: { signal: ctx.signal } },
-    ).$inspect();
-
-    if (!result.ok) {
-      return {
-        content: [{ type: "text", text: result.error.message }],
-        isError: true,
-      };
-    }
-
-    const value = result.value;
-
-    return formatResult(value, apiCall);
-  },
-};

package/src/mcp-server/tools/auditsListComments.ts

@@ -1,37 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { auditsListComments } from "../../funcs/auditsListComments.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.ListAuditCommentsRequest$inboundSchema,
-};
-
-export const tool$auditsListComments: ToolDefinition<typeof args> = {
-  name: "audits-list-comments",
-  description: `List audit comments
-
-Returns a paginated list of comments for an audit.`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await auditsListComments(
-      client,
-      args.request,
-      { fetchOptions: { signal: ctx.signal } },
-    ).$inspect();
-
-    if (!result.ok) {
-      return {
-        content: [{ type: "text", text: result.error.message }],
-        isError: true,
-      };
-    }
-
-    const value = result.value;
-
-    return formatResult(value, apiCall);
-  },
-};

package/src/mcp-server/tools/auditsListControls.ts

@@ -1,37 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { auditsListControls } from "../../funcs/auditsListControls.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.ListAuditControlsRequest$inboundSchema,
-};
-
-export const tool$auditsListControls: ToolDefinition<typeof args> = {
-  name: "audits-list-controls",
-  description: `List audit controls
-
-Returns a paginated list of controls for an audit.`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await auditsListControls(
-      client,
-      args.request,
-      { fetchOptions: { signal: ctx.signal } },
-    ).$inspect();
-
-    if (!result.ok) {
-      return {
-        content: [{ type: "text", text: result.error.message }],
-        isError: true,
-      };
-    }
-
-    const value = result.value;
-
-    return formatResult(value, apiCall);
-  },
-};

package/src/mcp-server/tools/auditsListEvidence.ts

@@ -1,37 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { auditsListEvidence } from "../../funcs/auditsListEvidence.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.ListAuditEvidenceRequest$inboundSchema,
-};
-
-export const tool$auditsListEvidence: ToolDefinition<typeof args> = {
-  name: "audits-list-evidence",
-  description: `List audit evidence
-
-Returns a paginated list of evidence for an audit.`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await auditsListEvidence(
-      client,
-      args.request,
-      { fetchOptions: { signal: ctx.signal } },
-    ).$inspect();
-
-    if (!result.ok) {
-      return {
-        content: [{ type: "text", text: result.error.message }],
-        isError: true,
-      };
-    }
-
-    const value = result.value;
-
-    return formatResult(value, apiCall);
-  },
-};

package/src/mcp-server/tools/auditsUpdateEvidence.ts

@@ -1,37 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { auditsUpdateEvidence } from "../../funcs/auditsUpdateEvidence.js";
-import * as operations from "../../models/operations/index.js";
-import { formatResult, ToolDefinition } from "../tools.js";
-
-const args = {
-  request: operations.UpdateAuditEvidenceRequest$inboundSchema,
-};
-
-export const tool$auditsUpdateEvidence: ToolDefinition<typeof args> = {
-  name: "audits-update-evidence",
-  description: `Update audit evidence
-
-Update audit evidence.`,
-  args,
-  tool: async (client, args, ctx) => {
-    const [result, apiCall] = await auditsUpdateEvidence(
-      client,
-      args.request,
-      { fetchOptions: { signal: ctx.signal } },
-    ).$inspect();
-
-    if (!result.ok) {
-      return {
-        content: [{ type: "text", text: result.error.message }],
-        isError: true,
-      };
-    }
-
-    const value = result.value;
-
-    return formatResult(value, apiCall);
-  },
-};

package/src/mcp-server/tools.ts

@@ -1,129 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
-import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
-import { objectOutputType, ZodRawShape, ZodTypeAny } from "zod/v3";
-import { VantaCore } from "../core.js";
-import { ConsoleLogger } from "./console-logger.js";
-import { MCPScope } from "./scopes.js";
-import { isAsyncIterable, isBinaryData, valueToBase64 } from "./shared.js";
-
-export type ToolDefinition<Args extends undefined | ZodRawShape = undefined> =
-  Args extends ZodRawShape ? {
-      name: string;
-      description: string;
-      scopes?: MCPScope[];
-      args: Args;
-      tool: (
-        client: VantaCore,
-        args: objectOutputType<Args, ZodTypeAny>,
-        extra: RequestHandlerExtra,
-      ) => CallToolResult | Promise<CallToolResult>;
-    }
-    : {
-      name: string;
-      description: string;
-      scopes?: MCPScope[];
-      args?: undefined;
-      tool: (
-        client: VantaCore,
-        extra: RequestHandlerExtra,
-      ) => CallToolResult | Promise<CallToolResult>;
-    };
-
-// Optional function to assist with formatting tool results
-export async function formatResult(
-  value: unknown,
-  init: { response?: Response | undefined },
-): Promise<CallToolResult> {
-  if (typeof value === "undefined") {
-    return { content: [] };
-  }
-
-  const { response } = init;
-  const contentType = response?.headers.get("content-type") ?? "";
-  let content: CallToolResult["content"] = [];
-
-  if (contentType.search(/\bjson\b/g)) {
-    content = [{ type: "text", text: JSON.stringify(value) }];
-  } else if (
-    contentType.startsWith("text/event-stream")
-    && isAsyncIterable(value)
-  ) {
-    content = await consumeSSE(value);
-  } else if (contentType.startsWith("text/") && typeof value === "string") {
-    content = [{ type: "text", text: value }];
-  } else if (isBinaryData(value) && contentType.startsWith("image/")) {
-    const data = await valueToBase64(value);
-    content = data == null
-      ? []
-      : [{ type: "image", data, mimeType: contentType }];
-  } else {
-    return {
-      content: [{
-        type: "text",
-        text: `Unsupported content type: "${contentType}"`,
-      }],
-      isError: true,
-    };
-  }
-
-  return { content };
-}
-
-async function consumeSSE(
-  value: AsyncIterable<unknown>,
-): Promise<CallToolResult["content"]> {
-  const content: CallToolResult["content"] = [];
-
-  for await (const chunk of value) {
-    if (typeof chunk === "string") {
-      content.push({ type: "text", text: chunk });
-    } else {
-      content.push({ type: "text", text: JSON.stringify(chunk) });
-    }
-  }
-
-  return content;
-}
-
-export function createRegisterTool(
-  logger: ConsoleLogger,
-  server: McpServer,
-  sdk: VantaCore,
-  allowedScopes: Set<MCPScope>,
-  allowedTools?: Set<string>,
-): <A extends ZodRawShape | undefined>(tool: ToolDefinition<A>) => void {
-  return <A extends ZodRawShape | undefined>(tool: ToolDefinition<A>): void => {
-    if (allowedTools && !allowedTools.has(tool.name)) {
-      return;
-    }
-
-    const scopes = tool.scopes ?? [];
-    if (allowedScopes.size > 0 && scopes.length === 0) {
-      return;
-    }
-
-    if (
-      allowedScopes.size > 0
-      && !scopes.every((s: MCPScope) => allowedScopes.has(s))
-    ) {
-      return;
-    }
-
-    if (tool.args) {
-      server.tool(tool.name, tool.description, tool.args, async (args, ctx) => {
-        return tool.tool(sdk, args, ctx);
-      });
-    } else {
-      server.tool(tool.name, tool.description, async (ctx) => {
-        return tool.tool(sdk, ctx);
-      });
-    }
-
-    logger.debug("Registered tool", { name: tool.name });
-  };
-}