dev 02011127

util_ext.py

@@ -0,0 +1,135 @@
+# coding: utf8
+import os
+import sys
+import json
+from pathlib import Path
+from dataclasses import dataclass, field
+from global_vars import get_with_chained_keys
+
+
+PLAT = sys.platform
+USER_PATH = os.path.expanduser("~")
+
+USER_DATA_PATH_MAP = {
+    "win32": {
+        "Chrome": Path(USER_PATH, "AppData", "Local", "Google", "Chrome", "User Data"),
+        "Edge": Path(USER_PATH, "AppData", "Local", "Microsoft", "Edge", "User Data"),
+        "Brave": Path(USER_PATH, "AppData", "Local", "BraveSoftware", "Brave-Browser", "User Data"),
+    },
+    "darwin": {
+        "Chrome": Path(USER_PATH, "Library", "Application Support", "Google", "Chrome"),
+        "Edge": Path(USER_PATH, "Library", "Application Support", "Microsoft Edge"),
+        "Brave": Path(USER_PATH, "Library", "Application Support", "BraveSoftware", "Brave-Browser"),
+    },
+}
+
+USER_DATA_PATH = USER_DATA_PATH_MAP[PLAT]
+
+
+@dataclass
+class ProfileNode(object):
+    gaia_given_name: str
+    gaia_name: str
+    name: str
+    shortcut_name: str
+    user_name: str
+
+
+ProfilesData = dict[str, ProfileNode]
+
+
+def scan_profiles(browser: str) -> ProfilesData:
+    local_state_path = Path(USER_DATA_PATH[browser], "Local State")
+    local_state_data: dict = json.loads(local_state_path.read_text(encoding="utf8"))
+    info_cache_data: dict = get_with_chained_keys(local_state_data, ["profile", "info_cache"])
+
+    profiles_data: ProfilesData = {}
+
+    for profile_id in info_cache_data:
+        profile_data: dict = info_cache_data[profile_id]
+        profile_node = ProfileNode(
+            gaia_given_name=profile_data.get("gaia_given_name", ""),
+            gaia_name=profile_data.get("gaia_name", ""),
+            name=profile_data.get("name", ""),
+            shortcut_name=profile_data.get("shortcut_name", ""),
+            user_name=profile_data.get("user_name"),
+        )
+        profiles_data[profile_id] = profile_node
+
+    return profiles_data
+
+
+@dataclass
+class ExtensionNode(object):
+    icon: str
+    name: str
+    # path: str
+    profiles: list[str] = field(default_factory=list)
+
+
+ExtensionsData = dict[str, ExtensionNode]
+
+
+def get_extension_icon_path(ext_icons: dict[str, str], ext_path: str, profile_path: Path) -> str:
+    if len(ext_icons) == 0:
+        return ""
+
+    if "128" in ext_icons:
+        icon_file = ext_icons["128"]
+    else:
+        icon_file = ext_icons[str(max(map(lambda x: int(x), ext_icons.keys())))]
+    # 如果路径以 / 开头，会被认为是根而忽略其他，因此需要检查一下
+    if icon_file.startswith("/"):
+        icon_file = icon_file[1:]
+
+    full_path = Path(profile_path, "Extensions", ext_path, icon_file)
+    return str(full_path)
+
+
+def scan_extensions(browser: str, is_chrome_compat=False) -> ExtensionsData:
+    profile_data = scan_profiles(browser)
+    extensions_data: ExtensionsData = {}
+
+    browser_data_path = USER_DATA_PATH[browser]
+    if browser == "Chrome" and is_chrome_compat:
+        pref_file = "Preferences"
+    else:
+        pref_file = "Secure Preferences"
+
+    for profile_id in profile_data:
+        profile_path = Path(browser_data_path, profile_id)
+        secure_pref_path = Path(profile_path, pref_file)
+        secure_pref_data: dict = json.loads(secure_pref_path.read_text(encoding="utf8"))
+        ext_settings_data: dict = get_with_chained_keys(secure_pref_data, ["extensions", "settings"], dict())
+
+        for ext_id in ext_settings_data:
+            ext_data: dict = ext_settings_data[ext_id]
+            # 这里插件有几种情况
+            ext_path: str = ext_data.get("path", "")
+            if len(ext_path) == 0:
+                # 如果 path 是空，则该插件可能是一些内部插件，不予收集
+                continue
+            elif not (ext_path.startswith(ext_id) or os.path.exists(ext_path)):
+                # 如果 path 以插件 ID 开头，则为商店安装的插件，
+                # 如果不是以插件 ID 开头，但是是一个存在的路径，则为离线安装的插件
+                # 否则，可能也是内部插件，不予收集
+                continue
+            ext_manifest: dict = ext_data.get("manifest", {})
+            if len(ext_manifest) == 0:
+                # 如果 manifest 为空，则该插件可能是离线插件，需要额外找它的 manifest
+                ext_manifest_path = Path(ext_path, "manifest.json")
+                ext_manifest: dict = json.loads(ext_manifest_path.read_text(encoding="utf8"))
+
+            if ext_id not in extensions_data:
+                ext_node = ExtensionNode(
+                    icon=get_extension_icon_path(ext_manifest.get("icons", {}), ext_path, profile_path),
+                    name=ext_manifest.get("name", ""),
+                    # path=ext_data.get("path", ""),
+                    profiles=[profile_id],
+                )
+                extensions_data[ext_id] = ext_node
+            else:
+                ext_node = extensions_data[ext_id]
+                ext_node.profiles += [profile_id]
+
+    return extensions_data