schema

Author: rambleraptor

src/common/api/api.test.ts

@@ -1,6 +1,6 @@
 /// <reference types="jest" />
 import { API, Contact, OpenAPI, Resource, Schema } from "./types.js";
-import {APIClient} from "./api.js";
+import { APIClient } from "./api.js";
 import { fetchOpenAPI, OpenAPIImpl } from "../openapi/openapi.js";
 import { resourceToZodSchema } from "../../schema.js";
 

src/common/api/openapi.test.ts

@@ -1,5 +1,15 @@
-import { API, Contact, OpenAPI, Resource, Schema, ListMethod } from "./types.js";
-import { convertToOpenAPI, generateParentPatternsWithParams } from "./openapi.js";
+import {
+  API,
+  Contact,
+  OpenAPI,
+  Resource,
+  Schema,
+  ListMethod,
+} from "./types.js";
+import {
+  convertToOpenAPI,
+  generateParentPatternsWithParams,
+} from "./openapi.js";
 
 describe("convertToOpenAPI", () => {
   // Common example API used across tests

src/common/openapi/openapi.ts

@@ -8,7 +8,7 @@ import {
   RequestBody,
 } from "./types.js";
 
-import yaml from 'js-yaml';
+import yaml from "js-yaml";
 
 export class OpenAPIImpl {
   private api: OpenAPI;

src/schema.ts

@@ -2,10 +2,13 @@ import { z } from "zod";
 import { Schema } from "./common/api/types.js";
 import { OpenAPIImpl } from "./common/openapi/openapi.js";
 
-export async function resourceToZodSchema(schema: Schema, oas: OpenAPIImpl): Promise<z.ZodTypeAny> {
-  if(schema.$ref) {
-    const newSchema = await oas.dereferenceSchema(schema)
-    return resourceToZodSchema(newSchema, oas)
+export async function resourceToZodSchema(
+  schema: Schema,
+  oas: OpenAPIImpl
+): Promise<z.ZodTypeAny> {
+  if (schema.$ref) {
+    const newSchema = await oas.dereferenceSchema(schema);
+    return resourceToZodSchema(newSchema, oas);
   }
   switch (schema.type) {
     case "boolean":
@@ -33,7 +36,11 @@ export async function resourceToZodSchema(schema: Schema, oas: OpenAPIImpl): Pro
 
     default:
       // TODO: handle ref!
-      console.warn(`could not find item type ${schema.type} for schema ${JSON.stringify(schema)}`);
+      console.warn(
+        `could not find item type ${schema.type} for schema ${JSON.stringify(
+          schema
+        )}`
+      );
       return z.string();
   }
 }

src/server.ts

@@ -45,12 +45,17 @@ async function main() {
   for (const [resourceName, resource] of Object.entries(resources)) {
     const zodSchema = await resourceToZodSchema(resource.schema, oas);
     if (!(zodSchema instanceof z.ZodObject)) {
-      throw new Error(`Expected object schema but got ${zodSchema.constructor.name}`);
+      throw new Error(
+        `Expected object schema but got ${zodSchema.constructor.name}`
+      );
     }
     server.tool(
       resourceName,
       zodSchema._def.shape(),
-      async (args: Record<string, unknown>, extra: RequestHandlerExtra): Promise<Response> => {
+      async (
+        args: Record<string, unknown>,
+        extra: RequestHandlerExtra
+      ): Promise<Response> => {
         console.log(args);
         return {
           content: [{ type: "text", text: "Success" }],