# Copyright (c) 2018 Adam Karpierz
# SPDX-License-Identifier: Zlib

from ctypes.wintypes import HMODULE
from pathlib import Path

__all__ = ('dll_path', 'python_dll_path')


def dll_path(handle: HMODULE | int) -> Path | None:
    """Retrieves the fully qualified path for the file that contains the specified module.

    The module must have been loaded by the current process.
    """
    import ctypes
    from ctypes.wintypes import HMODULE, LPWSTR, DWORD
    MAX_PATH = 520
    GetModuleFileNameW = ctypes.windll.kernel32.GetModuleFileNameW
    GetModuleFileNameW.restype  = DWORD
    GetModuleFileNameW.argtypes = [HMODULE, LPWSTR, DWORD]
    buf = ctypes.create_unicode_buffer(MAX_PATH)
    # print("LENNNN", len(buf))
    result = GetModuleFileNameW(handle, buf, len(buf))
    dll_path = buf.value
    # print("@@@@@@@@@@@", handle, result, dll_path)
    return (Path(dll_path) if handle != 0 and result != 0
            and dll_path and Path(dll_path).exists() else None)


def python_dll_path() -> Path | None:
    """Retrieves the fully qualified path for the file that contains Python dll module.

    The module must have been loaded by the current process.
    """
    try:
        from ctypes import pythonapi
    except ImportError:  # pragma: no cover
        from sys import dllhandle
    else:
        dllhandle = pythonapi._handle
    return dll_path(dllhandle)


del HMODULE