enhance: follow calender toolkit

Author: a7m-1st

src/components/AddWorker/index.tsx

@@ -164,7 +164,7 @@ export function AddWorker({
 		// call ToolSelect's install method
 		if (toolSelectRef.current) {
 			try {
-				if (activeMcp.key === "EXA Search" || activeMcp.key === "Google Calendar" || activeMcp.key === "Google Gmail") {
+				if (activeMcp.key === "EXA Search" || activeMcp.key === "Google Calendar" || activeMcp.key === "Google Gmail MCP") {
 					await toolSelectRef.current.installMcp(
 						activeMcp.id,
 						{ ...envValues },
@@ -179,7 +179,7 @@ export function AddWorker({
 		}
 
 		// For Google Calendar, close dialog after installMcp completes
-		if (activeMcp.key === "Google Calendar") {
+		if (activeMcp.key === "Google Calendar" || activeMcp.key === "Google Gmail MCP") {
 			setShowEnvConfig(false);
 		}
 

src/components/IntegrationList/index.tsx

@@ -108,6 +108,17 @@ export default function IntegrationList({
 					setShowEnvConfig(true);
 				}
 				return;
+			} 
+			
+			if (item.key === "Google Gmail MCP") {
+				const mcp = createMcpFromItem(item, 15);
+				if (isSelectMode) {
+					onShowEnvConfig?.(mcp);
+				} else {
+					setActiveMcp(mcp);
+					setShowEnvConfig(true);
+				}
+				return;
 			}
 
 			if (installed[item.key]) return;
@@ -199,6 +210,64 @@ export default function IntegrationList({
 				console.log("[IntegrationList onConnect] Polling timeout");
 				return;
 			}
+		} else if (mcp.key === "Google Gmail MCP") {
+			console.log(
+				"[IntegrationList onConnect] Google Gmail detected, starting auth flow"
+			);
+
+			// Trigger install/authorization
+			const gmailItem = items.find((item) => item.key === "Google Gmail MCP");
+			try {
+				if (gmailItem && gmailItem.onInstall) {
+					await gmailItem.onInstall();
+				} else {
+					await fetchPost("/install/tool/google_gmail");
+				}
+			} catch (_) {}
+
+			// Select mode: poll OAuth status
+			if (isSelectMode) {
+				console.log(
+					"[IntegrationList onConnect] Starting OAuth status polling"
+				);
+
+				const start = Date.now();
+				const timeoutMs = 5 * 60 * 1000; // 5 minutes
+				while (Date.now() - start < timeoutMs) {
+					try {
+						const statusRes: any = await fetchGet(
+							"/oauth/status/google_gmail"
+						);
+						console.log(
+							"[IntegrationList onConnect] OAuth status:",
+							statusRes?.status
+						);
+
+						if (statusRes?.status === "success") {
+							console.log(
+								"[IntegrationList onConnect] Success! Closing dialog"
+							);
+							await fetchInstalled();
+							onClose();
+							return;
+						}
+						if (
+							statusRes?.status === "failed" ||
+							statusRes?.status === "cancelled"
+						) {
+							console.log(
+								"[IntegrationList onConnect] Failed/cancelled, keeping dialog open"
+							);
+							return;
+						}
+					} catch (err) {
+						console.log("[IntegrationList onConnect] Polling error:", err);
+					}
+					await new Promise((r) => setTimeout(r, 1500));
+				}
+				console.log("[IntegrationList onConnect] Polling timeout");
+				return;
+			}
 		}
 
 		// Select mode: add to tools and close