feat(mcp): update reject

Author: taltultc

doit-mcp-server/src/utils.ts

@@ -313,6 +313,9 @@ export const renderCustomerContextScreen = async (
   oauthReqInfo: AuthRequest | null,
   apiKey: string
 ) => {
+  if (action === "reject") {
+    return renderAuthorizationRejectedContent(oauthReqInfo?.redirectUri || "/");
+  }
   return html`
     <div class="max-w-md mx-auto bg-white p-8 rounded-lg shadow-md">
       <h1 class="text-2xl font-heading font-bold mb-6 text-gray-900">

src/tools/changeCustomer.ts

@@ -5,6 +5,7 @@ import {
   formatZodError,
   handleGeneralError,
 } from "../utils/util.js";
+import { handleValidateUserRequest } from "./validateUser.js";
 
 // Schema definition
 export const ChangeCustomerArgumentsSchema = z.object({
@@ -57,14 +58,20 @@ export async function handleChangeCustomerRequest(
       await updateCustomerContext(newContext);
     }
 
+    // Verify that the new context is valid
+    const newCustomerDomain = await handleValidateUserRequest(
+      { customerContext: newContext }, // Validate doers
+      token
+    );
+
+    const domain = newCustomerDomain.content[0].text.split("Domain: ")[1];
+
     // Create response
     const response: ChangeCustomerResponse = {
       success: true,
       previousContext,
       newContext,
-      message: `Customer context successfully changed${
-        previousContext ? ` from '${previousContext}'` : ""
-      } to '${newContext}'`,
+      message: `Customer context successfully changed to '${domain}'`,
     };
 
     return createSuccessResponse(response.message);