From 2331cd4d18b0a4cca80f7bff80ed24ce528def56 Mon Sep 17 00:00:00 2001
From: hz <1532246395@qq.com>
Date: Sat, 8 Mar 2025 19:02:23 +0800
Subject: [PATCH] =?UTF-8?q?modify:=20=E7=9F=AD=E9=93=BE=E5=92=8C=E5=AF=86?=
 =?UTF-8?q?=E7=A0=81=E4=BF=AE=E6=94=B9=E4=B8=BA=E5=90=8E=E7=AB=AF=E7=94=9F?=
 =?UTF-8?q?=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/handler/short_link.go | 20 +++++++---
 api/static/index.html     |  4 +-
 api/static/index.js       | 77 ++++++++++++++++++++++++++++++++-------
 3 files changed, 80 insertions(+), 21 deletions(-)

diff --git a/api/handler/short_link.go b/api/handler/short_link.go
index 61699bd..650b052 100644
--- a/api/handler/short_link.go
+++ b/api/handler/short_link.go
@@ -33,6 +33,7 @@ func GenerateLinkHandler(c *gin.Context) {
 	}
 
 	var hash string
+	var password string
 	var err error
 	
 	if params.CustomID != "" {
@@ -43,22 +44,29 @@ func GenerateLinkHandler(c *gin.Context) {
 			return
 		}
 		if exists {
-			respondWithError(c, http.StatusBadRequest, "自定义ID已存在")
+			respondWithError(c, http.StatusBadRequest, "短链已存在")
 			return
 		}
 		hash = params.CustomID
+		password = params.Password
 	} else {
+		// 自动生成短链ID和密码
 		hash, err = generateUniqueHash()
 		if err != nil {
 			respondWithError(c, http.StatusInternalServerError, "生成短链接失败")
 			return
 		}
+		if params.Password == "" {
+			password = common.RandomString(8) // 生成8位随机密码
+		} else {
+			password = params.Password
+		}
 	}
 
 	shortLink := model.ShortLink{
 		Hash:     hash,
 		Url:      params.Url,
-		Password: params.Password,
+		Password: password,
 	}
 
 	if err := database.SaveShortLink(&shortLink); err != nil {
@@ -66,10 +74,12 @@ func GenerateLinkHandler(c *gin.Context) {
 		return
 	}
 
-	if params.Password != "" {
-		hash += "?password=" + params.Password
+	// 返回生成的短链ID和密码
+	response := map[string]string{
+		"hash": hash,
+		"password": password,
 	}
-	c.String(http.StatusOK, hash)
+	c.JSON(http.StatusOK, response)
 }
 
 func generateUniqueHash() (string, error) {
diff --git a/api/static/index.html b/api/static/index.html
index d80d409..3bb78f4 100644
--- a/api/static/index.html
+++ b/api/static/index.html
@@ -147,13 +147,13 @@
         <div class="form-group mb-5">
             <label for="apiLink">配置链接:</label>
             <div class="input-group mb-2">
-                <input class="form-control" id="apiLink" type="text" placeholder="链接" />
+                <input class="form-control bg-light" id="apiLink" type="text" placeholder="链接" readonly style="cursor: not-allowed;" />
                 <button class="btn btn-primary" onclick="copyToClipboard('apiLink',this)" type="button">
                     复制链接
                 </button>
             </div>
             <div class="input-group mb-2">
-                <input class="form-control" id="customId" type="text" placeholder="自定义ID（可选）" />
+                <input class="form-control" id="customId" type="text" placeholder="短链ID（可选）" />
                 <input class="form-control" id="password" type="text" placeholder="密码（可选）" />
                 <button class="btn btn-primary" onclick="generateShortLink()" type="button">
                     生成短链
diff --git a/api/static/index.js b/api/static/index.js
index 70bed2f..6ce9c1d 100644
--- a/api/static/index.js
+++ b/api/static/index.js
@@ -1,3 +1,15 @@
+function setInputReadOnly(input, readonly) {
+  if (readonly) {
+    input.readOnly = true;
+    input.classList.add('bg-light');
+    input.style.cursor = 'not-allowed';
+  } else {
+    input.readOnly = false;
+    input.classList.remove('bg-light');
+    input.style.cursor = 'auto';
+  }
+}
+
 function clearExistingValues() {
   // 清除简单输入框和复选框的值
   document.getElementById("endpoint").value = "clash";
@@ -12,7 +24,23 @@ function clearExistingValues() {
   document.getElementById("remove").value = "";
   document.getElementById("apiLink").value = "";
   document.getElementById("apiShortLink").value = "";
-  document.getElementById("password").value = "";
+  
+  // 恢复短链ID和密码输入框状态
+  const customIdInput = document.getElementById("customId");
+  const passwordInput = document.getElementById("password");
+  const generateButton = document.querySelector('button[onclick="generateShortLink()"]');
+  
+  customIdInput.value = "";
+  setInputReadOnly(customIdInput, false);
+  
+  passwordInput.value = "";
+  setInputReadOnly(passwordInput, false);
+  
+  // 恢复生成短链按钮状态
+  generateButton.disabled = false;
+  generateButton.classList.remove('btn-secondary');
+  generateButton.classList.add('btn-primary');
+  
   document.getElementById("nodeList").checked = false;
 
   // 清除由 createRuleProvider, createReplace, 和 createRule 创建的所有额外输入组
@@ -188,9 +216,32 @@ async function parseInputURL() {
     try {
       const response = await axios.get("./short?" + q.toString());
       url = new URL(window.location.href + response.data);
-      document.querySelector("#apiShortLink").value = inputURL;
-      document.querySelector("#password").value = password;
-      document.querySelector("#customId").value = hash;
+      
+      // 回显配置链接
+      const apiLinkInput = document.querySelector("#apiLink");
+      apiLinkInput.value = `${window.location.origin}${window.location.pathname}${response.data}`;
+      setInputReadOnly(apiLinkInput, true);
+      
+      // 回显短链相关信息
+      const apiShortLinkInput = document.querySelector("#apiShortLink");
+      apiShortLinkInput.value = inputURL;
+      setInputReadOnly(apiShortLinkInput, true);
+      
+      // 设置短链ID和密码，并设置为只读
+      const customIdInput = document.querySelector("#customId");
+      const passwordInput = document.querySelector("#password");
+      const generateButton = document.querySelector('button[onclick="generateShortLink()"]');
+      
+      customIdInput.value = hash;
+      setInputReadOnly(customIdInput, true);
+      
+      passwordInput.value = password;
+      setInputReadOnly(passwordInput, true);
+      
+      // 禁用生成短链按钮
+      generateButton.disabled = true;
+      generateButton.classList.add('btn-secondary');
+      generateButton.classList.remove('btn-primary');
     } catch (error) {
       console.log(error);
       alert("获取短链失败，请检查密码！");
@@ -440,6 +491,7 @@ function generateURL() {
     return;
   }
   apiLink.value = `${window.location.origin}${window.location.pathname}${uri}`;
+  setInputReadOnly(apiLink, true);
 }
 
 function generateShortLink() {
@@ -451,14 +503,6 @@ function generateShortLink() {
     return;
   }
 
-  // 如果用户没有输入自定义ID和密码，自动生成
-  if (!customId.value && !password.value) {
-    const randomId = Math.random().toString(36).substring(2, 8);
-    const randomPassword = Math.random().toString(36).substring(2, 8);
-    customId.value = randomId;
-    password.value = randomPassword;
-  }
-
   axios
     .post(
       "./short",
@@ -474,7 +518,12 @@ function generateShortLink() {
       }
     )
     .then((response) => {
-      apiShortLink.value = `${window.location.origin}${window.location.pathname}s/${response.data}`;
+      // 设置返回的短链ID和密码
+      customId.value = response.data.hash;
+      password.value = response.data.password;
+      // 生成完整的短链接
+      const shortLink = `${window.location.origin}${window.location.pathname}s/${response.data.hash}?password=${response.data.password}`;
+      apiShortLink.value = shortLink;
     })
     .catch((error) => {
       console.log(error);
@@ -517,7 +566,7 @@ function updateShortLink() {
       }
     )
     .then((response) => {
-      alert("更新短链成功！");
+      alert(`短链 ${hash} 更新成功！`);
     })
     .catch((error) => {
       console.log(error);