From 34b5105f170c0c8f10efe889c5246708ba0e028c Mon Sep 17 00:00:00 2001
From: Brett Beutell <brbeut@gmail.com>
Date: Tue, 3 Dec 2024 18:03:53 +0100
Subject: [PATCH] Validate openapi operation in the requestpanel

---
 packages/types/src/openapi.ts                       |  2 +-
 .../RequestorPage/RequestPanel/RequestPanel.tsx     | 13 +++----------
 .../RequestPanel/RouteDocumentation/openapi.ts      | 12 +++++++++++-
 3 files changed, 15 insertions(+), 12 deletions(-)

diff --git a/packages/types/src/openapi.ts b/packages/types/src/openapi.ts
index 1930878fa..ec05c8d96 100644
--- a/packages/types/src/openapi.ts
+++ b/packages/types/src/openapi.ts
@@ -77,7 +77,7 @@ const OpenAPIRequestBodySchema = z.object({
   content: z.record(ContentSchema),
 });
 
-const OpenAPIOperationSchema = z.object({
+export const OpenAPIOperationSchema = z.object({
   title: z.string().optional(),
   summary: z.string().optional(),
   description: z.string().optional(),
diff --git a/studio/src/pages/RequestorPage/RequestPanel/RequestPanel.tsx b/studio/src/pages/RequestorPage/RequestPanel/RequestPanel.tsx
index e914c7c69..acc6d6a86 100644
--- a/studio/src/pages/RequestorPage/RequestPanel/RequestPanel.tsx
+++ b/studio/src/pages/RequestorPage/RequestPanel/RequestPanel.tsx
@@ -1,7 +1,7 @@
 import { Button } from "@/components/ui/button";
 import { Tabs } from "@/components/ui/tabs";
 import { useToast } from "@/components/ui/use-toast";
-import { cn, objectWithKey } from "@/utils";
+import { cn } from "@/utils";
 import { EraserIcon, InfoCircledIcon } from "@radix-ui/react-icons";
 import { type Dispatch, type SetStateAction, memo, useMemo } from "react";
 import { FormDataForm } from "../FormDataForm";
@@ -22,6 +22,7 @@ import {
 } from "@/components/CodeMirrorEditor";
 import { useRequestorStore } from "../store";
 import { RouteDocumentation } from "./RouteDocumentation/RouteDocumentation";
+import { isOpenApiOperation } from "./RouteDocumentation/openapi";
 
 type RequestPanelProps = {
   aiEnabled: boolean;
@@ -37,14 +38,6 @@ type RequestPanelProps = {
   onSubmit: () => void;
 };
 
-function isValidOpenApiSpec(openApiSpec: unknown): boolean {
-  return (
-    objectWithKey(openApiSpec, "parameters") ||
-    objectWithKey(openApiSpec, "requestBody") ||
-    objectWithKey(openApiSpec, "responses")
-  );
-}
-
 export const RequestPanel = memo(function RequestPanel(
   props: RequestPanelProps,
 ) {
@@ -116,7 +109,7 @@ export const RequestPanel = memo(function RequestPanel(
       return null;
     }
   }, [activeRoute?.openApiSpec]);
-  const shouldShowDocs = isValidOpenApiSpec(openApiSpec);
+  const shouldShowDocs = isOpenApiOperation(openApiSpec);
 
   return (
     <Tabs
diff --git a/studio/src/pages/RequestorPage/RequestPanel/RouteDocumentation/openapi.ts b/studio/src/pages/RequestorPage/RequestPanel/RouteDocumentation/openapi.ts
index 0538ea218..c2b68b3de 100644
--- a/studio/src/pages/RequestorPage/RequestPanel/RouteDocumentation/openapi.ts
+++ b/studio/src/pages/RequestorPage/RequestPanel/RouteDocumentation/openapi.ts
@@ -1,4 +1,9 @@
-import { type OpenApiSpec, OpenApiSpecSchema } from "@fiberplane/fpx-types";
+import {
+  type OpenAPIOperation,
+  OpenAPIOperationSchema,
+  type OpenApiSpec,
+  OpenApiSpecSchema,
+} from "@fiberplane/fpx-types";
 
 export type { OpenAPIOperation } from "@fiberplane/fpx-types";
 export type { OpenAPISchema } from "@fiberplane/fpx-types";
@@ -14,3 +19,8 @@ export function isOpenApiSpec(value: unknown): value is OpenApiSpec {
   }
   return result.success;
 }
+
+export function isOpenApiOperation(value: unknown): value is OpenAPIOperation {
+  const result = OpenAPIOperationSchema.safeParse(value);
+  return result.success;
+}