From 5d575216a8ce2b5bda51f4100531bafc0e75eb81 Mon Sep 17 00:00:00 2001
From: fanchenio <fanchenio@163.com>
Date: Tue, 7 Nov 2023 10:27:18 +0800
Subject: [PATCH] Support parsing URL files

---
 electron/main/worker.ts    | 42 ++++++++++++++++++++++++++++++++------
 src/pages/item/AddEdit.vue |  6 ++----
 2 files changed, 38 insertions(+), 10 deletions(-)

diff --git a/electron/main/worker.ts b/electron/main/worker.ts
index b32122d..5cc6f23 100644
--- a/electron/main/worker.ts
+++ b/electron/main/worker.ts
@@ -5,7 +5,7 @@ import {
   newItem,
 } from "../../commons/utils/common";
 import { CommonItem, Item } from "../../types/item";
-import { parse, join } from "node:path";
+import { parse, join, extname } from "node:path";
 import { readdirSync, readFileSync, statSync, writeFileSync } from "node:fs";
 import { execSync } from "node:child_process";
 import xml2js from "xml2js";
@@ -655,14 +655,26 @@ async function getDropItemInfo(
           }
         }
       }
-      // 文件类型
-      let stats = statSync(item.data.target);
-      // 路径
-      item.type = stats.isFile() ? 0 : 1;
       // 获取图标
       item.data.icon = getFileIcon(item.data.target);
+      // 获取后缀，判断是否是url
+      let ext = extname(item.data.target);
+      if (ext && ext.toLowerCase() === ".url") {
+        // url
+        let url = parseUrlFileContent(readFileSync(item.data.target, "utf-8"));
+        if (url && url.trim() !== "") {
+          item.data.target = url;
+          item.type = 2;
+        } else {
+          continue;
+        }
+      } else {
+        // 文件类型
+        let stats = statSync(item.data.target);
+        item.type = stats.isFile() ? 0 : 1;
+      }
       // 去掉后缀
-      if (item.type === 0) {
+      if (item.type === 0 || item.type === 2) {
         item.name = deleteExtname(item.name);
       }
       // push
@@ -672,6 +684,24 @@ async function getDropItemInfo(
   return itemList;
 }
 
+/**
+ * 解析.url文件内容以获取URL
+ * @param content
+ * @returns
+ */
+function parseUrlFileContent(content: string) {
+  if (content) {
+    const lines = content.split("\n");
+    for (const line of lines) {
+      if (line.startsWith("URL=")) {
+        const url = line.substring(4).trim();
+        return url;
+      }
+    }
+  }
+  return null;
+}
+
 /**
  * 刷新项目图标
  * @param itemList
diff --git a/src/pages/item/AddEdit.vue b/src/pages/item/AddEdit.vue
index 0766b8f..f4ac3f8 100644
--- a/src/pages/item/AddEdit.vue
+++ b/src/pages/item/AddEdit.vue
@@ -628,10 +628,8 @@ let getURLInfoLoading = ref<boolean>(false);
 function getURLInfo() {
   if (form.data.target) {
     getURLInfoLoading.value = true;
-    if (
-      form.data.target.indexOf("http://") < 0 &&
-      form.data.target.indexOf("https://") < 0
-    ) {
+    const regex = /^.+:\/\/.*/;
+    if (!regex.test(form.data.target)) {
       form.data.target = "http://" + form.data.target;
     }
     // 获取网址信息