code/default/launcher/autorun.py - master - githubmirror/XX-Net

code/default/launcher/autorun.py
181 строка5 KB
XX-Net.dev
Update UI for flexible APP_NAME.
25 фев 2022, 21:29
25 фев 2022, 21:294f3d633
#!/usr/bin/env python
"""A simple crossplatform autostart helper"""

import os
import sys

from config import app_name
from xlog import getLogger

xlog = getLogger("launcher")

current_path = os.path.dirname(os.path.abspath(__file__))
root_path = os.path.abspath(os.path.join(current_path, os.pardir))
top_path = os.path.abspath(os.path.join(root_path, os.pardir, os.pardir))

if sys.platform == 'win32':
    import winreg

    _registry = winreg.ConnectRegistry(None, winreg.HKEY_CURRENT_USER)


    def get_runonce():
        return winreg.OpenKey(_registry, r"Software\Microsoft\Windows\CurrentVersion\Run", 0, winreg.KEY_ALL_ACCESS)


    def add(name, application):
        """add a new autostart entry"""
        key = get_runonce()
        winreg.SetValueEx(key, name, 0, winreg.REG_SZ, application)
        winreg.CloseKey(key)


    def exists(name):
        """check if an autostart entry exists"""
        key = get_runonce()
        exists = True
        try:
            winreg.QueryValueEx(key, name)
        except:  # WindowsError
            exists = False
        winreg.CloseKey(key)
        return exists


    def remove(name):
        if not exists(name):
            return

        """delete an autostart entry"""
        key = get_runonce()
        winreg.DeleteValue(key, name)
        winreg.CloseKey(key)


    run_cmd = "\"" + os.path.join(top_path, "start.vbs") + "\""
elif sys.platform.startswith('linux'):
    _xdg_config_home = os.environ.get("XDG_CONFIG_HOME", "~/.config")
    home_config_path = os.path.expanduser(_xdg_config_home)
    _xdg_user_autostart = os.path.join(home_config_path, "autostart")


    def getfilename(name):
        """get the filename of an autostart (.desktop) file"""
        return os.path.join(_xdg_user_autostart, name + ".desktop")


    def add(name, application):
        if not os.path.isdir(home_config_path):
            xlog.warn("autorun linux config path not found:%s", home_config_path)
            return

        if not os.path.isdir(_xdg_user_autostart):
            try:
                os.mkdir(_xdg_user_autostart)
            except Exception as e:
                xlog.warn("Enable auto start, create path:%s fail:%r", _xdg_user_autostart, e)
                return

        """add a new autostart entry"""
        desktop_entry = "[Desktop Entry]\n" \
                        "Name=%s\n" \
                        "Exec=%s\n" \
                        "Type=Application\n" \
                        "Terminal=false\n" \
                        "X-GNOME-Autostart-enabled=true" % (name, application)
        with open(getfilename(name), "w") as f:
            f.write(desktop_entry)


    def exists(name):
        """check if an autostart entry exists"""
        return os.path.exists(getfilename(name))


    def remove(name):
        """delete an autostart entry"""
        if (exists(name)):
            os.unlink(getfilename(name))


    run_cmd = os.path.join(top_path, "start")
elif sys.platform == 'darwin':
    plist_template = """<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
<plist version="1.0">
<dict>
	<key>Label</key>
	<string>com.xxnet.launcher</string>

	<key>LimitLoadToSessionType</key>
	<string>Aqua</string>

	<key>ProgramArguments</key>
	<array>
	  <string>%s</string>
	  <string>-hungup</string>
	</array>

	<key>RunAtLoad</key>
	<true/>

	<key>StandardErrorPath</key>
	<string>/dev/null</string>
	<key>StandardOutPath</key>
	<string>/dev/null</string>
</dict>
</plist>"""

    run_cmd = os.path.join(top_path, "start")
    from os.path import expanduser

    home = expanduser("~")
    launch_path = os.path.join(home, "Library/LaunchAgents")

    plist_file_path = os.path.join(launch_path, "com.xxnet.launcher.plist")


    def add(name, cmd):
        file_content = plist_template % cmd
        xlog.info("create file:%s", plist_file_path)

        if not os.path.isdir(launch_path):
            os.mkdir(launch_path, 0o755)

        with open(plist_file_path, "w") as f:
            f.write(file_content)


    def remove(name):
        if (os.path.isfile(plist_file_path)):
            os.unlink(plist_file_path)
            xlog.info("remove file:%s", plist_file_path)
else:
    def add(name, cmd):
        pass


    def remove(name):
        pass


def enable():
    add(app_name, run_cmd)


def disable():
    remove(app_name)


def test():
    assert not exists("test_xxx")
    try:
        add("test_xxx", "test")
        assert exists("test_xxx")
    finally:
        remove("test_xxx")
    assert not exists("test_xxx")


if __name__ == "__main__":
    test()