From 1c5e17a4abec5a34b74508ee6a7baea8205385bf Mon Sep 17 00:00:00 2001
From: nite07 <admin@nite07.com>
Date: Wed, 13 Mar 2024 16:28:40 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E8=A7=A3=E6=9E=90=E7=9F=AD=E9=93=BE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/handler/short_link.go | 25 ++++++++++++++++-
 api/route.go              | 14 ++++++----
 api/static/index.html     | 12 ++++----
 api/static/index.js       | 59 ++++++++++++++++++++++++++++++++-------
 4 files changed, 87 insertions(+), 23 deletions(-)

diff --git a/api/handler/short_link.go b/api/handler/short_link.go
index 1d82b42..de00e49 100644
--- a/api/handler/short_link.go
+++ b/api/handler/short_link.go
@@ -70,7 +70,30 @@ func ShortLinkGenHandler(c *gin.Context) {
 	c.String(200, hash)
 }
 
-func ShortLinkGetHandler(c *gin.Context) {
+func ShortLinkGetUrlHandler(c *gin.Context) {
+	var params validator.ShortLinkGetValidator
+	if err := c.ShouldBindQuery(&params); err != nil {
+		c.String(400, "参数错误: "+err.Error())
+		return
+	}
+	if strings.TrimSpace(params.Hash) == "" {
+		c.String(400, "参数错误")
+		return
+	}
+	var shortLink model.ShortLink
+	result := database.FindShortLinkByHash(params.Hash, &shortLink)
+	if result.Error != nil {
+		c.String(404, "未找到短链接")
+		return
+	}
+	if shortLink.Password != "" && shortLink.Password != params.Password {
+		c.String(403, "密码错误")
+		return
+	}
+	c.String(200, shortLink.Url)
+}
+
+func ShortLinkGetConfigHandler(c *gin.Context) {
 	// 获取动态路由
 	hash := c.Param("hash")
 	password := c.Query("password")
diff --git a/api/route.go b/api/route.go
index 4c02550..6d5c940 100644
--- a/api/route.go
+++ b/api/route.go
@@ -53,14 +53,18 @@ func SetRoute(r *gin.Engine) {
 			handler.SubHandler(c)
 		},
 	)
-	r.POST(
-		"/short", func(c *gin.Context) {
-			handler.ShortLinkGenHandler(c)
+	r.GET(
+		"/s/:hash", func(c *gin.Context) {
+			handler.ShortLinkGetConfigHandler(c)
 		},
 	)
 	r.GET(
-		"/s/:hash", func(c *gin.Context) {
-			handler.ShortLinkGetHandler(c)
+		"/short", func(c *gin.Context) {
+			handler.ShortLinkGetUrlHandler(c)
+		})
+	r.POST(
+		"/short", func(c *gin.Context) {
+			handler.ShortLinkGenHandler(c)
 		},
 	)
 	r.PUT(
diff --git a/api/static/index.html b/api/static/index.html
index 68fb48f..27d6d65 100644
--- a/api/static/index.html
+++ b/api/static/index.html
@@ -13,7 +13,6 @@
         src="https://cdn.jsdelivr.net/npm/bootstrap@5.3.2/dist/js/bootstrap.bundle.min.js"></script>
     <!-- Axios -->
     <script src="https://cdn.jsdelivr.net/npm/axios@latest/dist/axios.min.js"></script>
-    <script src="./static/index.js"></script>
     <style>
         .container {
             max-width: 800px;
@@ -48,6 +47,7 @@
             <label for="apiLink">解析链接:</label>
             <div class="input-group mb-2">
                 <input class="form-control" id="urlInput" type="text" placeholder="通过生成的链接重新填写下方设置" />
+                <input class="form-control" id="urlshortLinkPasswdInput" type="text" placeholder="短链密码" />
                 <button class="btn btn-primary" onclick="parseInputURL()" type="button">
                     解析
                 </button>
@@ -142,16 +142,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" />
-                <button class="btn btn-primary" onclick="generateURL()" type="button">
-                    生成链接
-                </button>
+                <input class="form-control" id="apiLink" type="text" placeholder="链接" />
                 <button class="btn btn-primary" onclick="copyToClipboard('apiLink',this)" type="button">
                     复制链接
                 </button>
             </div>
             <div class="input-group">
-                <input class="form-control" id="apiShortLink" type="text" />
+                <input class="form-control" id="apiShortLink" type="text" placeholder="链接" />
                 <input class="form-control" id="password" type="text" placeholder="密码" />
                 <button class="btn btn-primary" onclick="generateShortLink()" type="button">
                     生成短链
@@ -164,6 +161,7 @@
                 </button>
             </div>
         </div>
+
         <!-- footer-->
         <footer>
             <p class="text-center">
@@ -174,5 +172,5 @@
         </footer>
     </div>
 </body>
-
+<script src="./static/index.js"></script>
 </html>
\ No newline at end of file
diff --git a/api/static/index.js b/api/static/index.js
index 8be5407..baa28ea 100644
--- a/api/static/index.js
+++ b/api/static/index.js
@@ -63,7 +63,7 @@ function generateURI() {
     noProxy = true;
   }
   if (noSub && noProxy) {
-    alert("订阅链接和节点分享链接不能同时为空！");
+    // alert("订阅链接和节点分享链接不能同时为空！");
     return "";
   }
   // 获取复选框的值
@@ -103,7 +103,7 @@ function generateURI() {
       prepend.trim() === "" ||
       name.trim() === ""
     ) {
-      alert("Rule Provider 中存在空值，请检查后重试！");
+      // alert("Rule Provider 中存在空值，请检查后重试！");
       return "";
     }
     providers.push(`[${behavior},${url},${group},${prepend},${name}]`);
@@ -118,7 +118,7 @@ function generateURI() {
       let group = rules[i * 3 + 2].value;
       // 是否存在空值
       if (rule.trim() === "" || prepend.trim() === "" || group.trim() === "") {
-        alert("Rule 中存在空值，请检查后重试！");
+        // alert("Rule 中存在空值，请检查后重试！");
         return "";
       }
       ruleList.push(`[${rule},${prepend},${group}]`);
@@ -143,7 +143,7 @@ function generateURI() {
     let replaceStr = `<${replaces[i * 2].value}>`;
     let replaceTo = `<${replaces[i * 2 + 1].value}>`;
     if (replaceStr.trim() === "") {
-      alert("重命名设置中存在空值，请检查后重试！");
+      // alert("重命名设置中存在空值，请检查后重试！");
       return "";
     }
     replaceList.push(`[${replaceStr},${replaceTo}]`);
@@ -154,9 +154,12 @@ function generateURI() {
 }
 
 // 将输入框中的 URL 解析为参数
-function parseInputURL() {
+async function parseInputURL() {
   // 获取输入框中的 URL
   const inputURL = document.getElementById("urlInput").value;
+  const urlshortLinkPasswdInput = document.getElementById(
+    "urlshortLinkPasswdInput"
+  ).value;
 
   if (!inputURL) {
     alert("请输入有效的链接！");
@@ -170,13 +173,23 @@ function parseInputURL() {
     alert("无效的链接！");
     return;
   }
-
+  if (url.pathname.includes("/s/")) {
+    let hash = url.pathname.substring(url.pathname.lastIndexOf("/s/") + 3);
+    let q = new URLSearchParams();
+    q.append("hash", hash);
+    q.append("password", urlshortLinkPasswdInput);
+    try {
+      const response = await axios.get("./short?" + q.toString());
+      url = new URL(window.location.href + response.data);
+    } catch (error) {
+      console.log(error);
+      alert("获取短链失败，请检查密码！");
+    }
+  }
+  let params = new URLSearchParams(url.search);
   // 清除现有的输入框值
   clearExistingValues();
 
-  // 获取查询参数
-  const params = new URLSearchParams(url.search);
-
   // 分配值到对应的输入框
   const pathSections = url.pathname.split("/");
   const lastSection = pathSections[pathSections.length - 1];
@@ -360,19 +373,43 @@ function createRule() {
   return div;
 }
 
+function listenInput() {
+  let selectElements = document.querySelectorAll("select");
+  let inputElements = document.querySelectorAll("input");
+  let textAreaElements = document.querySelectorAll("textarea");
+  inputElements.forEach(function (element) {
+    element.addEventListener("input", function () {
+      generateURL();
+    });
+  });
+  textAreaElements.forEach(function (element) {
+    element.addEventListener("input", function () {
+      generateURL();
+    });
+  });
+  selectElements.forEach(function (element) {
+    element.addEventListener("change", function () {
+      generateURL();
+    });
+  });
+}
+
 function addRuleProvider() {
   const div = createRuleProvider();
   document.getElementById("ruleProviderGroup").appendChild(div);
+  listenInput();
 }
 
 function addRule() {
   const div = createRule();
   document.getElementById("ruleGroup").appendChild(div);
+  listenInput();
 }
 
 function addReplace() {
   const div = createReplace();
   document.getElementById("replaceGroup").appendChild(div);
+  listenInput();
 }
 
 function removeElement(button) {
@@ -423,7 +460,7 @@ function updateShortLink() {
   let hash = apiShortLink.value;
   if (hash.startsWith("http")) {
     let u = new URL(hash);
-    hash = u.pathname.replace("/s/", "");
+    hash = u.pathname.substring(u.pathname.lastIndexOf("/s/") + 3);
   }
   if (password.value.trim() === "") {
     alert("请输入密码！");
@@ -455,3 +492,5 @@ function updateShortLink() {
       alert(error.response.data);
     });
 }
+
+listenInput();