"""Python bindings for 0MQ."""

# Copyright (C) PyZMQ Developers
# Distributed under the terms of the Modified BSD License.

import os
import sys
import glob

# load bundled libzmq, if there is one:

here = os.path.dirname(__file__)

bundled = []
bundled_sodium = []
for ext in ('pyd', 'so', 'dll', 'dylib'):
    bundled_sodium.extend(glob.glob(os.path.join(here, 'libsodium*.%s*' % ext)))
    bundled.extend(glob.glob(os.path.join(here, 'libzmq*.%s*' % ext)))

# If we are running in a debug interpreter, load libzmq_d.pyd instead of libzmq.pyd
# hasattr(sys, 'gettotalrefcount') is used to detect whether we are running in a debug interpreter
# Taken from http://stackoverflow.com/questions/646518/python-how-to-detect-debug-interpreter
if os.name == 'nt':
    def is_debug_filename(name):
        # Note this fails for filenames like foo.bar_d.so.x.y.z,
        # but such names should not appear on Windows.
        root, ext = os.path.splitext(name)
        return root.endswith('_d')

    if hasattr(sys, 'gettotalrefcount'):
        bundled_sodium = [x for x in bundled_sodium if is_debug_filename(x)]
        bundled = [x for x in bundled if is_debug_filename(x)]
    else:
        bundled_sodium = [x for x in bundled_sodium if not is_debug_filename(x)]
        bundled = [x for x in bundled if not is_debug_filename(x)]

if bundled:
    import ctypes
    if bundled_sodium:
        if bundled[0].endswith('.pyd'):
            # a Windows Extension
            _libsodium = ctypes.cdll.LoadLibrary(bundled_sodium[0])
        else:
            _libsodium = ctypes.CDLL(bundled_sodium[0], mode=ctypes.RTLD_GLOBAL)
    if bundled[0].endswith('.pyd'):
        # a Windows Extension
        _libzmq = ctypes.cdll.LoadLibrary(bundled[0])
    else:
        _libzmq = ctypes.CDLL(bundled[0], mode=ctypes.RTLD_GLOBAL)
    del ctypes
else:
    import zipimport
    try:
        if isinstance(__loader__, zipimport.zipimporter):
            # a zipped pyzmq egg
            from zmq import libzmq as _libzmq
    except (NameError, ImportError):
        pass
    finally:
        del zipimport

del os, sys, glob, here, bundled, bundled_sodium, ext

# zmq top-level imports

from zmq import backend
from zmq.backend import *
from zmq import sugar
from zmq.sugar import *

def get_includes():
    """Return a list of directories to include for linking against pyzmq with cython."""
    from os.path import join, dirname, abspath, pardir
    base = dirname(__file__)
    parent = abspath(join(base, pardir))
    return [ parent ] + [ join(parent, base, subdir) for subdir in ('utils',) ]


__all__ = ['get_includes'] + sugar.__all__ + backend.__all__

