updated README

Author: geelen

README.md

@@ -393,53 +393,53 @@ const server = new McpServer({
   version: "1.0.0"
 });
 
-server.tool(
+const listMessageTool = server.tool(
   "listMessages",
   { channel: z.string() },
   async ({ channel }) => ({
     content: [{ type: "text", text: await listMessages(channel) }]
   })
 );
 
-server.tool(
+const putMessageTool = server.tool(
+  "putMessage",
+  { channel: z.string(), message: z.string() },
+  async ({ channel, message }) => ({
+    content: [{ type: "text", text: await putMessage(channel, string) }]
+  })
+);
+// Until we upgrade auth, `putMessage` is disabled (won't show up in listTools)
+putMessageTool.disable()
+
+const upgradeAuthTool = server.tool(
   "upgradeAuth",
   { permission: z.enum(["write', vadmin"])},
-  upgradeAuth
-)
-
-// Connect with the existing set of tools
-const transport = new StdioServerTransport();
-await server.connect(transport);
-
-// Any mutations after connection result in `listChanged` notifications so the client knows to refresh
-async function upgradeAuth({permission}) {
-  const { ok, err, previous } = await upgradeAuthAndStoreToken(permission)
-
-  if (!ok) return {content: [{ type: "text", text: `Error: ${err}` }]}
+  // Any mutations here will automatically emit `listChanged` notifications
+  async ({ permission }) => {
+    const { ok, err, previous } = await upgradeAuthAndStoreToken(permission)
+    if (!ok) return {content: [{ type: "text", text: `Error: ${err}` }]}
+
+    // If we previously had read-only access, 'putMessage' is now available
+    if (previous === "read") {
+      putMessageTool.enable()
+    }
 
-  // If we previously had read-only access, we need to add 'putMessage' now we can use it
-  if (previous === "read") {
-    server.tool(
-      "putMessage",
-      { channel: z.string(), message: z.string() },
-      async ({ channel, message }) => ({
-        content: [{ type: "text", text: await putMessage(channel, string) }]
+    if (permission === 'write') {
+      // If we've just upgraded to 'write' permissions, we can still call 'upgradeAuth' 
+      // but can only upgrade to 'admin'. 
+      upgradeAuthTool.update({
+        paramSchema: { permission: z.enum(["admin"]) }, // change validation rules
       })
-    );
+    } else {
+      // If we're now an admin, we no longer have anywhere to upgrade to, so fully remove that tool
+      upgradeAuthTool.remove()
+    }
   }
+)
 
-  // If we've just upgraded to 'write' permissions, we can still call 'upgradeAuth' but can only upgrade to 'admin' 
-  if (permission === 'write') {
-    server.updateTool(
-      "upgradeAuth",
-      { permission: z.enum(["admin"])}, // change param validation
-      upgradeAuth
-    )
-  } else {
-    // If we're on admin, we no longer have anywhere to upgrade to
-    server.removeTool("upgradeAuth")
-  }
-}
+// Connect as normal
+const transport = new StdioServerTransport();
+await server.connect(transport);
 ```
 
 ### Low-Level Server