From c5ca9c1b92144f1ad9cc41faa70a2c4cf4f92450 Mon Sep 17 00:00:00 2001
From: knowen <1369727119@qq.com>
Date: Fri, 20 Mar 2026 00:43:52 +0800
Subject: [PATCH] fix(mcp): add login check before generating image

---
 src/mcp-server.js | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/mcp-server.js b/src/mcp-server.js
index d8a70c2..4b2429a 100644
--- a/src/mcp-server.js
+++ b/src/mcp-server.js
@@ -33,6 +33,16 @@ server.registerTool(
     try {
       const { ops } = await createGeminiSession();
 
+      // 前置检查：确保已登录
+      const loginCheck = await ops.checkLogin();
+      if (!loginCheck.ok || !loginCheck.loggedIn) {
+        disconnect();
+        return {
+          content: [{ type: "text", text: `Gemini 未登录 Google 账号，请先在浏览器中完成登录后重试` }],
+          isError: true,
+        };
+      }
+
       // 如果有参考图，先上传
       if (referenceImages.length > 0) {
         // 需要先处理新建会话（如果需要），因为 generateImage 内部的 newChat 会在上传之后才执行