From 64dad6ed635f7b29748a3e37c893ae464bd6259a Mon Sep 17 00:00:00 2001
From: luke <yewunetemail@163.com>
Date: Wed, 28 May 2025 23:37:08 +0800
Subject: [PATCH] =?UTF-8?q?=E7=A7=BB=E9=99=A4AnythingLLM=E5=AF=86=E7=A0=81?=
 =?UTF-8?q?=E6=98=8E=E6=96=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pages/login.html |  1 +
 pages/login.js   | 15 ++++++++++++++-
 pages/popup.js   | 14 +++++++++++---
 3 files changed, 26 insertions(+), 4 deletions(-)

diff --git a/pages/login.html b/pages/login.html
index bdc66c4..6733d5f 100644
--- a/pages/login.html
+++ b/pages/login.html
@@ -51,6 +51,7 @@
 <div class="form-group">
     <input id="username" placeholder="用户名" />
     <input id="password" type="password" placeholder="密码" />
+    <input type="password" id="llmPassword" placeholder="请输入AnythingLLM的密码" autocomplete="off" />
     <button id="loginBtn">登录</button>
 </div>
 <div class="error" id="errorBox"></div>
diff --git a/pages/login.js b/pages/login.js
index d8134f7..718c0d1 100644
--- a/pages/login.js
+++ b/pages/login.js
@@ -1,3 +1,12 @@
+// 自动回显 AnythingLLM 密码
+document.addEventListener("DOMContentLoaded", function () {
+    chrome.storage.local.get(["llmPassword"], function (res) {
+        if (res.llmPassword) {
+            document.getElementById("llmPassword").value = res.llmPassword;
+        }
+    });
+});
+
 document.getElementById("loginBtn").addEventListener("click", doLogin);
 document.addEventListener("keydown", (e) => {
     if (e.key === "Enter") doLogin();
@@ -6,6 +15,7 @@ document.addEventListener("keydown", (e) => {
 function doLogin() {
     const username = document.getElementById("username").value.trim();
     const password = document.getElementById("password").value.trim();
+    const llmPassword = document.getElementById("llmPassword").value;
     const errorBox = document.getElementById("errorBox");
     errorBox.textContent = "";
 
@@ -14,6 +24,9 @@ function doLogin() {
         return;
     }
 
+    // 保存 llmPassword（登录不依赖LLM密码是否填写，只是单独存储）
+    chrome.storage.local.set({ llmPassword });
+
     fetch("http://app.wisdompulse.cn/api/v1/user/login", {
         method: "POST",
         headers: { "Content-Type": "application/json" },
@@ -26,7 +39,7 @@ function doLogin() {
                 chrome.storage.local.set({
                     token: body.token,
                     username: body.username || username,
-                    loginAt: Date.now() // 毫秒时间戳
+                    loginAt: Date.now()
                 }, () => {
                     window.location.href = chrome.runtime.getURL("pages/popup.html");
                 });
diff --git a/pages/popup.js b/pages/popup.js
index 7943793..f3bf93e 100644
--- a/pages/popup.js
+++ b/pages/popup.js
@@ -25,12 +25,20 @@ document.addEventListener("DOMContentLoaded", async () => {
   const TOKEN_EXPIRE_MS = 7 * 24 * 60 * 60 * 1000; // 7天有效期
 
   // === AnythingLLM 配置 ===
-  const LLM_PASSWORD = "lukeye@6";
   const LLM_TOKEN_API = "http://llm.wisdompulse.cn/api/request-token";
   const LLM_WORKSPACES_API = "http://llm.wisdompulse.cn/api/workspaces";
   const LLM_STREAM_API_BASE = "http://llm.wisdompulse.cn/api/workspace";
   const LLM_TOKEN_EXPIRE_MS = 23 * 60 * 60 * 1000;
 
+  // 动态获取 AnythingLLM 密码
+  async function getLLMPassword() {
+    return new Promise((resolve) => {
+      chrome.storage.local.get(["llmPassword"], (res) => {
+        resolve(res.llmPassword || "");
+      });
+    });
+  }
+
   // 获取 LLM Token
   async function getLLMToken(forceRefresh = false) {
     return new Promise((resolve, reject) => {
@@ -46,7 +54,8 @@ document.addEventListener("DOMContentLoaded", async () => {
           resolve(res.llmToken);
         } else {
           try {
-            console.log("开始请求 LLM token，POST", LLM_TOKEN_API, "参数：", { password: LLM_PASSWORD });
+            const LLM_PASSWORD = await getLLMPassword();
+            console.log("开始请求 LLM token，POST", LLM_TOKEN_API, "参数：", { password: "******" });
             const response = await fetch(LLM_TOKEN_API, {
               method: "POST",
               headers: { "Content-Type": "application/json" },
@@ -221,7 +230,6 @@ document.addEventListener("DOMContentLoaded", async () => {
     }
   }
 
-
   function log(...args) {
     const version = chrome?.runtime?.getManifest?.().version || "dev";
     console.log("[v" + version + "]", ...args);