Windrecorder项目开机自启动功能异常分析与解决方案

问题背景

在Windrecorder项目中，用户报告了一个关于开机自启动功能的严重问题。当用户在Web界面中勾选"开机后自动启动应用"选项并保存设置时，系统会出现卡死现象，甚至导致WebUI异常退出。这个问题影响了用户体验，需要及时解决。

问题分析

经过技术分析，发现问题出在windrecorder\utils.py文件中的make_shortcut(bat_path, folder=startup_folder)函数调用上。该函数用于在Windows启动目录中创建快捷方式，以实现开机自启动功能。

深入研究发现，当前实现使用了第三方库pyshortcuts来创建快捷方式，但在某些情况下会导致Streamlit应用异常退出。这种异常行为可能与COM对象的初始化和释放有关，特别是在多线程环境下使用时。

技术细节

在Windows系统中，创建快捷方式通常需要使用Windows Shell的COM接口。pyshortcuts库虽然提供了便捷的封装，但在某些特定场景下可能无法正确处理COM对象的生命周期管理，导致资源泄漏或异常。

正确的做法应该是显式地初始化和释放COM对象，确保在任何情况下都能正确清理资源。这也是为什么手动创建快捷方式可以正常工作，而通过pyshortcuts则会出现问题的原因。

解决方案

基于上述分析，我们决定弃用pyshortcuts库，改为直接使用Windows API来创建快捷方式。具体实现如下：

import pythoncom
from win32com.client import Dispatch

def change_startup_shortcut(is_create=True):
    startup_folder = os.path.join(
        os.getenv("APPDATA"),
        "Microsoft",
        "Windows",
        "Start Menu",
        "Programs",
        "Startup",
    )
    shortcut_path = os.path.join(startup_folder, "start_app.bat.lnk")

    if is_create:
        if not os.path.exists(shortcut_path):
            current_dir = os.getcwd()
            bat_path = os.path.join(current_dir, "start_app.bat")
            create_shortcut(bat_path, shortcut_path)
    else:
        if os.path.exists(shortcut_path):
            os.remove(shortcut_path)

def create_shortcut(target_path, shortcut_path):
    pythoncom.CoInitialize()
    try:
        shell = Dispatch('WScript.Shell')
        shortcut = shell.CreateShortCut(shortcut_path)
        shortcut.Targetpath = target_path
        shortcut.WorkingDirectory = os.path.dirname(target_path)
        shortcut.save()
    finally:
        pythoncom.CoUninitialize()

实现要点

1. COM对象管理：显式调用pythoncom.CoInitialize()和pythoncom.CoUninitialize()来初始化和释放COM对象，确保资源正确释放。

2. 异常处理：使用try-finally结构确保在任何情况下都能释放COM资源，避免资源泄漏。

3. 路径处理：正确处理目标路径和工作目录，确保快捷方式能够正确指向目标程序。

4. 条件检查：在创建或删除快捷方式前检查文件是否存在，避免不必要的操作或错误。

效果验证

经过测试，新的实现方案能够稳定工作，不再出现卡死或WebUI异常退出的问题。快捷方式的创建和删除操作都能按预期执行，开机自启动功能恢复正常。

总结

通过这次问题解决，我们认识到在Windows开发中，特别是涉及COM接口调用时，需要特别注意资源管理和异常处理。直接使用底层API虽然代码量可能增加，但能提供更可靠的控制和更好的稳定性。这也提醒我们在选择第三方库时需要谨慎评估其适用性和可靠性。