0X57 Firefox安装插件列表

# -*-coding:utf-8-*-

import os
import json

def find_firefox_profile_path():
    # 获取当前用户的配置目录
    # config_dir = os.path.expandvars(r'%APPDATA%\Floorp')
    config_dir = 'C:\\Users\\Administrator\\AppData\\Roaming\\zen\\'
    profiles_ini_path = os.path.join(config_dir, "profiles.ini")

    if not os.path.exists(profiles_ini_path):
        raise FileNotFoundError("Could not find profiles.ini file.")

    with open(profiles_ini_path, "r", encoding="utf-8") as file:
        lines = file.readlines()

    profile_path = None
    for line in lines:
        if line.strip().startswith("Path="):
            profile_path = line.strip().split("=")[1]
            break

    if not profile_path:
        raise FileNotFoundError("Could not find profile path in profiles.ini file.")

    return os.path.join(config_dir, profile_path)

def get_firefox_addons(profile_path):
    # print("profile_path", profile_path)
    addons_file_path = os.path.join(profile_path, "extensions.json")

    if not os.path.exists(addons_file_path):
        raise FileNotFoundError("Could not find extensions.json file.")

    with open(addons_file_path, "r", encoding="utf-8") as file:
        addons_data = json.load(file)

    addons = []
    for addon in addons_data.get("addons", []):
        source_url = ""
        install_telemetry_info = addon.get("installTelemetryInfo")
        if install_telemetry_info:
            source_url = install_telemetry_info.get("sourceURL", "")

        location = addon.get("location", "")

        # 只导出用户安装的插件, 过滤应用内置插件
        if location == "app-profile":
            addons.append({
                "name": addon.get("defaultLocale", {}).get("name", ""),
                "version": addon.get("version", ""),
                "description": addon.get("defaultLocale", {}).get("description", "No description"),
                "manifestVersion": addon.get("manifestVersion", ""),
                "active": addon.get("active", False),
                "sourceURI": addon.get("sourceURI", ""),
                "sourceURL": source_url
            })

    return addons

def export_addons_to_file(addons, output_file):
    with open(output_file, "w") as file:
        enable_count = 0
        disable_count = 0

        enable_addons = """\n## Enabled Extensions\n"""
        disable_addons = """\n## Disabled Extensions\n"""

        for addon in addons:
            # 处理每个插件的信息
            name = addon.get("name", "No name")
            version = addon.get("version", "No version")
            description = addon.get("description", "No description")
            active = addon.get("active", False)
            source_uri = addon.get("sourceURI", "No source URI")
            source_url = addon.get("sourceURL", "No source URL")

            if active:
              enable_count = enable_count+1
              enable_addons += f"{enable_count}. [{name} {version}]({source_uri} \"{description}\")\n"
            else:
              disable_count = disable_count+1
              disable_addons += f"{disable_count}. [{name} {version}]({source_uri} \"{description}\")\n"

        file.write(enable_addons)
        file.write(disable_addons)

def main():
    try:
        profile_path = find_firefox_profile_path()
        addons = get_firefox_addons(profile_path)
        output_file = "firefox_addons.md"
        export_addons_to_file(addons, output_file)
        print(f"Exported Firefox addons to {output_file}")
    except Exception as e:
        print(f"An error occurred: {e}")

if __name__ == "__main__":
    main()