nvda-addon-voicemeeter/addon/globalPlugins/voicemeeter/cdll.py

import ctypes as ct
import platform
import winreg
from pathlib import Path

from .error import VMError

BITS = 64 if ct.sizeof(ct.c_voidp) == 8 else 32

if platform.system() != 'Windows':
    raise VMError('Only Windows OS supported')


VM_KEY = 'VB:Voicemeeter {17359A74-1236-5467}'
REG_KEY = '\\'.join(
    filter(
        None,
        (
            'SOFTWARE',
            'WOW6432Node' if BITS == 64 else '',
            'Microsoft',
            'Windows',
            'CurrentVersion',
            'Uninstall',
        ),
    )
)


def get_vmpath():
    with winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, r'{}'.format('\\'.join((REG_KEY, VM_KEY)))) as vm_key:
        return winreg.QueryValueEx(vm_key, r'UninstallString')[0]


try:
    vm_parent = Path(get_vmpath()).parent
except FileNotFoundError as e:
    raise VMError('Unable to fetch DLL path from the registry') from e

DLL_NAME = f'VoicemeeterRemote{"64" if BITS == 64 else ""}.dll'

dll_path = vm_parent.joinpath(DLL_NAME)
if not dll_path.is_file():
    raise VMError(f'Could not find {dll_path}')

if BITS == 64:
    libc = ct.CDLL(str(dll_path))
else:
    libc = ct.WinDLL(str(dll_path))