124 lines
3.9 KiB
Python
124 lines
3.9 KiB
Python
"""将用户选中的 .lnk / .url 解析为实际要存储的路径（目标程序或 URL）。"""
|
|
import base64
|
|
import json
|
|
import os
|
|
import re
|
|
import subprocess
|
|
import sys
|
|
|
|
|
|
def _looks_like_shell_lnk(path: str) -> bool:
|
|
"""Windows Shell Link 文件头为 4C 00 00 00（扩展名异常时仍可识别）。"""
|
|
try:
|
|
with open(path, "rb") as f:
|
|
return f.read(4) == b"L\x00\x00\x00"
|
|
except OSError:
|
|
return False
|
|
|
|
|
|
def _read_internet_shortcut_url(path: str) -> str | None:
|
|
try:
|
|
with open(path, "r", encoding="utf-8", errors="ignore") as f:
|
|
for line in f:
|
|
s = line.strip()
|
|
if s[:4].upper() == "URL=":
|
|
return s[4:].strip()
|
|
except OSError:
|
|
pass
|
|
return None
|
|
|
|
|
|
def _resolve_windows_lnk(lnk_path: str) -> str | None:
|
|
if sys.platform != "win32":
|
|
return None
|
|
env = os.environ.copy()
|
|
env["LNK_B64"] = base64.b64encode(os.path.normpath(lnk_path).encode("utf-8")).decode(
|
|
"ascii"
|
|
)
|
|
ps = (
|
|
"$p=[Text.Encoding]::UTF8.GetString([Convert]::FromBase64String($env:LNK_B64));"
|
|
"$sc=(New-Object -ComObject WScript.Shell).CreateShortcut($p);"
|
|
"@{ t=[string]$sc.TargetPath; a=[string]$sc.Arguments }|ConvertTo-Json -Compress"
|
|
)
|
|
try:
|
|
r = subprocess.run(
|
|
[
|
|
"powershell",
|
|
"-NoProfile",
|
|
"-Sta",
|
|
"-Command",
|
|
ps,
|
|
],
|
|
capture_output=True,
|
|
text=True,
|
|
encoding="utf-8",
|
|
errors="replace",
|
|
env=env,
|
|
timeout=15,
|
|
creationflags=getattr(subprocess, "CREATE_NO_WINDOW", 0),
|
|
)
|
|
raw = (r.stdout or "").strip().lstrip("\ufeff")
|
|
if not raw:
|
|
return None
|
|
data = json.loads(raw)
|
|
target = (data.get("t") or "").strip()
|
|
args = (data.get("a") or "").strip()
|
|
if not target:
|
|
return None
|
|
low = target.lower()
|
|
if low.endswith("cmd.exe") or low.endswith("\\cmd.exe"):
|
|
for m in re.finditer(
|
|
r"([a-zA-Z]:[^\"'\s]+\.(?:bat|cmd))|\x22([a-zA-Z]:[^\"]+\.(?:bat|cmd))\x22",
|
|
args,
|
|
re.IGNORECASE,
|
|
):
|
|
cand = (m.group(1) or m.group(2) or "").strip('"')
|
|
if cand and os.path.isfile(cand):
|
|
return os.path.normpath(cand)
|
|
if os.path.isfile(target) or low.startswith("\\\\"):
|
|
return os.path.normpath(target)
|
|
return os.path.normpath(target)
|
|
except Exception:
|
|
return None
|
|
|
|
|
|
def path_for_storage(local_path: str) -> str:
|
|
"""
|
|
拖拽/选择文件时写入数据库的 path：.lnk/.url 解析为目标，其它保持原路径。
|
|
解析失败时退回本地路径。
|
|
"""
|
|
if not local_path:
|
|
return local_path
|
|
local_path = os.path.normpath(local_path.strip())
|
|
|
|
ext = os.path.splitext(local_path)[1].lower()
|
|
if ext == ".url":
|
|
try:
|
|
if os.path.isfile(local_path):
|
|
url = _read_internet_shortcut_url(local_path)
|
|
if url:
|
|
return url
|
|
except (OSError, ValueError):
|
|
pass
|
|
return local_path
|
|
|
|
try:
|
|
is_file = os.path.isfile(local_path)
|
|
except (OSError, ValueError):
|
|
is_file = False
|
|
|
|
is_lnk = ext in (".lnk", ".ink") or (
|
|
is_file and _looks_like_shell_lnk(local_path)
|
|
)
|
|
if is_lnk and is_file:
|
|
resolved = _resolve_windows_lnk(local_path)
|
|
if resolved:
|
|
return resolved
|
|
return local_path
|
|
|
|
|
|
def item_name_from_sources(local_path: str, store_path: str) -> str:
|
|
"""列表显示名：仍用快捷方式/所选文件的文件名（无扩展名），与原来行为一致。"""
|
|
base = os.path.splitext(os.path.basename(local_path))[0]
|
|
return base or os.path.basename(store_path)
|