Add log level and callback functions to the python bindings
[deb_shairplay.git] / src / bindings / python / Shairplay.py
index 020eb8fb6d33d013a0bad6abdca43b5eed9534a7..69aec6b9739bb4243978950366dba89418cbcc50 100644 (file)
@@ -1,22 +1,94 @@
+# coding=utf-8
+'''
+Copyright (C) 2012  Juho Vähä-Herttua
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be included
+in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+'''
+
 import os
 import sys
 import platform
 
 from ctypes import *
 
-audio_init_prototype =        CFUNCTYPE(py_object, c_void_p, c_int, c_int, c_int)
-audio_set_volume_prototype =  CFUNCTYPE(None, c_void_p, c_void_p, c_float)
-audio_process_prototype =     CFUNCTYPE(None, c_void_p, c_void_p, c_void_p, c_int)
-audio_flush_prototype =       CFUNCTYPE(None, c_void_p, c_void_p)
-audio_destroy_prototype =     CFUNCTYPE(None, c_void_p, c_void_p)
+class RaopLogLevel:
+       EMERG   = 0
+       ALERT   = 1
+       CRIT    = 2
+       ERR     = 3
+       WARNING = 4
+       NOTICE  = 5
+       INFO    = 6
+       DEBUG   = 7
+       
+raop_log_callback_prototype =   CFUNCTYPE(None, c_int, c_char_p)
+
+audio_init_prototype =          CFUNCTYPE(py_object, c_void_p, c_int, c_int, c_int)
+audio_process_prototype =       CFUNCTYPE(None, c_void_p, c_void_p, c_void_p, c_int)
+audio_destroy_prototype =       CFUNCTYPE(None, c_void_p, c_void_p)
+
+audio_flush_prototype =         CFUNCTYPE(None, c_void_p, c_void_p)
+audio_set_volume_prototype =    CFUNCTYPE(None, c_void_p, c_void_p, c_float)
+audio_set_metadata_prototype =  CFUNCTYPE(None, c_void_p, c_void_p, c_void_p, c_int)
+audio_set_coverart_prototype =  CFUNCTYPE(None, c_void_p, c_void_p, c_void_p, c_int)
 
 class RaopNativeCallbacks(Structure):
-       _fields_ = [("cls",               py_object),
-                   ("audio_init",        audio_init_prototype),
-                   ("audio_set_volume",  audio_set_volume_prototype),
-                   ("audio_process",     audio_process_prototype),
-                   ("audio_flush",       audio_flush_prototype),
-                   ("audio_destroy",     audio_destroy_prototype)]
+       _fields_ = [("cls",                 py_object),
+                   ("audio_init",          audio_init_prototype),
+                   ("audio_process",       audio_process_prototype),
+                   ("audio_destroy",       audio_destroy_prototype),
+                   ("audio_flush",         audio_flush_prototype),
+                   ("audio_set_volume",    audio_set_volume_prototype),
+                   ("audio_set_metadata",  audio_set_metadata_prototype),
+                   ("audio_set_coverart",  audio_set_coverart_prototype)]
+
+def InitShairplay(libshairplay):
+       # Initialize dnssd related functions
+       libshairplay.dnssd_init.restype = c_void_p
+       libshairplay.dnssd_init.argtypes = [POINTER(c_int)]
+       libshairplay.dnssd_register_raop.restype = c_int
+       libshairplay.dnssd_register_raop.argtypes = [c_void_p, c_char_p, c_ushort, POINTER(c_char), c_int, c_int]
+       libshairplay.dnssd_register_airplay.restype = c_int
+       libshairplay.dnssd_register_airplay.argtypes = [c_void_p, c_char_p, c_ushort, POINTER(c_char), c_int]
+       libshairplay.dnssd_unregister_raop.restype = None
+       libshairplay.dnssd_unregister_raop.argtypes = [c_void_p]
+       libshairplay.dnssd_unregister_airplay.restype = None
+       libshairplay.dnssd_unregister_airplay.argtypes = [c_void_p]
+       libshairplay.dnssd_destroy.restype = None
+       libshairplay.dnssd_destroy.argtypes = [c_void_p]
+
+       # Initialize raop related functions
+       libshairplay.raop_init.restype = c_void_p
+       libshairplay.raop_init.argtypes = [c_int, POINTER(RaopNativeCallbacks), c_char_p]
+       libshairplay.raop_set_log_level.restype = None
+       libshairplay.raop_set_log_level.argtypes = [c_void_p, c_int]
+       libshairplay.raop_set_log_callback.restype = None
+       libshairplay.raop_set_log_callback.argtypes = [c_void_p, raop_log_callback_prototype]
+       libshairplay.raop_is_running.restype = c_int
+       libshairplay.raop_is_running.argtypes = [c_void_p]
+       libshairplay.raop_start.restype = c_int
+       libshairplay.raop_start.argtypes = [c_void_p, POINTER(c_ushort), POINTER(c_char), c_int, c_char_p]
+       libshairplay.raop_stop.restype = None
+       libshairplay.raop_stop.argtypes = [c_void_p]
+       libshairplay.raop_destroy.restype = None
+       libshairplay.raop_destroy.argtypes = [c_void_p]
 
 def LoadShairplay(path):
        if sys.maxsize < 2**32:
@@ -37,30 +109,7 @@ def LoadShairplay(path):
        except:
                raise RuntimeError("Couldn't load shairplay library " + libname)
 
-       # Initialize dnssd related functions
-       libshairplay.dnssd_init.restype = c_void_p
-       libshairplay.dnssd_init.argtypes = [POINTER(c_char), c_int, POINTER(c_int)]
-       libshairplay.dnssd_register_raop.restype = c_int
-       libshairplay.dnssd_register_raop.argtypes = [c_void_p, c_char_p, c_ushort]
-       libshairplay.dnssd_register_airplay.restype = c_int
-       libshairplay.dnssd_register_airplay.argtypes = [c_void_p, c_char_p, c_ushort]
-       libshairplay.dnssd_unregister_raop.restype = None
-       libshairplay.dnssd_unregister_raop.argtypes = [c_void_p]
-       libshairplay.dnssd_unregister_airplay.restype = None
-       libshairplay.dnssd_unregister_airplay.argtypes = [c_void_p]
-       libshairplay.dnssd_destroy.restype = None
-       libshairplay.dnssd_destroy.argtypes = [c_void_p]
-
-       # Initialize raop related functions
-       libshairplay.raop_init.restype = c_void_p
-       libshairplay.raop_init.argtypes = [POINTER(RaopNativeCallbacks), c_char_p]
-       libshairplay.raop_start.restype = c_int
-       libshairplay.raop_start.argtypes = [c_void_p, POINTER(c_ushort), POINTER(c_char), c_int]
-       libshairplay.raop_stop.restype = None
-       libshairplay.raop_stop.argtypes = [c_void_p]
-       libshairplay.raop_destroy.restype = None
-       libshairplay.raop_destroy.argtypes = [c_void_p]
-
+       InitShairplay(libshairplay)
        return libshairplay
 
 RSA_KEY = """
@@ -93,69 +142,114 @@ class RaopCallbacks:
        def audio_init(self, bits, channels, samplerate):
                raise NotImplementedError()
 
-       def audio_set_volume(self, session, volume):
-               pass
-
        def audio_process(self, session, buffer):
                raise NotImplementedError()
 
+       def audio_destroy(self, session):
+               raise NotImplementedError()
+
+       def audio_set_volume(self, session, volume):
+               pass
+
        def audio_flush(self, session):
                pass
 
-       def audio_destroy(self, session):
+       def audio_set_metadata(self, session, buffer):
+               pass
+
+       def audio_set_coverart(self, session, buffer):
                pass
 
 class RaopService:
+       def log_callback_cb(self, level, message):
+               if self.log_callback != None:
+                       self.log_callback(level, message)
+               else:
+                       print message
+
        def audio_init_cb(self, cls, bits, channels, samplerate):
                session = self.callbacks.audio_init(bits, channels, samplerate)
-               return py_object(session)
-
-       def audio_set_volume_cb(self, cls, sessionptr, volume):
-               session = cast(sessionptr, py_object).value
-               self.callbacks.audio_set_volume(session, volume)
+               self.sessions.append(session)
+               return session
 
        def audio_process_cb(self, cls, sessionptr, buffer, buflen):
                session = cast(sessionptr, py_object).value
                strbuffer = string_at(buffer, buflen)
                self.callbacks.audio_process(session, strbuffer)
 
+       def audio_destroy_cb(self, cls, sessionptr):
+               session = cast(sessionptr, py_object).value
+               self.callbacks.audio_destroy(session)
+               if session in self.sessions:
+                       self.sessions.remove(session)
+
        def audio_flush_cb(self, cls, sessionptr):
                session = cast(sessionptr, py_object).value
                self.callbacks.audio_flush(session)
 
-       def audio_destroy_cb(self, cls, sessionptr):
+       def audio_set_volume_cb(self, cls, sessionptr, volume):
                session = cast(sessionptr, py_object).value
-               self.callbacks.audio_destroy(session)
+               self.callbacks.audio_set_volume(session, volume)
+
+       def audio_set_metadata_cb(self, cls, sessionptr, buffer, buflen):
+               session = cast(sessionptr, py_object).value
+               strbuffer = string_at(buffer, buflen)
+               self.callbacks.audio_set_metadata(session, strbuffer)
 
+       def audio_set_coverart_cb(self, cls, sessionptr, buffer, buflen):
+               session = cast(sessionptr, py_object).value
+               strbuffer = string_at(buffer, buflen)
+               self.callbacks.audio_set_coverart(session, strbuffer)
 
-       def __init__(self, libshairplay, callbacks):
+
+       def __init__(self, libshairplay, max_clients, callbacks):
                self.libshairplay = libshairplay
                self.callbacks = callbacks
+               self.sessions = []
                self.instance = None
 
                # We need to hold a reference to native_callbacks
                self.native_callbacks = RaopNativeCallbacks()
                self.native_callbacks.audio_init = audio_init_prototype(self.audio_init_cb)
-               self.native_callbacks.audio_set_volume = audio_set_volume_prototype(self.audio_set_volume_cb)
                self.native_callbacks.audio_process = audio_process_prototype(self.audio_process_cb)
-               self.native_callbacks.audio_flush = audio_flush_prototype(self.audio_flush_cb)
                self.native_callbacks.audio_destroy = audio_destroy_prototype(self.audio_destroy_cb)
+               self.native_callbacks.audio_flush = audio_flush_prototype(self.audio_flush_cb)
+               self.native_callbacks.audio_set_volume = audio_set_volume_prototype(self.audio_set_volume_cb)
+               self.native_callbacks.audio_set_metadata = audio_set_metadata_prototype(self.audio_set_metadata_cb)
+               self.native_callbacks.audio_set_coverart = audio_set_coverart_prototype(self.audio_set_coverart_cb)
 
                # Initialize the raop instance with our callbacks
-               self.instance = self.libshairplay.raop_init(pointer(self.native_callbacks), RSA_KEY)
+               self.instance = self.libshairplay.raop_init(max_clients, pointer(self.native_callbacks), RSA_KEY)
                if self.instance == None:
                        raise RuntimeError("Initializing library failed")
 
+               # We need to hold a reference to the log callback wrapper
+               self.log_callback_pointer = raop_log_callback_prototype(self.log_callback_cb)
+               self.libshairplay.raop_set_log_callback(self.instance, self.log_callback_pointer)
+               self.log_callback = None
+
        def __del__(self):
                if self.instance != None:
                        self.libshairplay.raop_destroy(self.instance)
                self.instance = None
 
-       def start(self, port, hwaddrstr):
+       def is_running(self):
+               if self.libshairplay.raop_is_running(self.instance):
+                       return True
+               else:
+                       return False
+
+       def set_log_level(self, level):
+               self.libshairplay.raop_set_log_level(self.instance, level)
+
+       def set_log_callback(self, log_callback):
+               self.log_callback = log_callback
+
+       def start(self, port, hwaddrstr, password=None):
                port = c_ushort(port)
                hwaddr = create_string_buffer(hwaddrstr, len(hwaddrstr))
 
-               ret = self.libshairplay.raop_start(self.instance, pointer(port), hwaddr, c_int(len(hwaddr)))
+               ret = self.libshairplay.raop_start(self.instance, pointer(port), hwaddr, c_int(len(hwaddr)), password)
                if ret < 0:
                        raise RuntimeError("Starting RAOP instance failed")
                return port.value
@@ -168,10 +262,9 @@ class DnssdService:
                self.libshairplay = libshairplay
                self.instance = None
 
-               hwaddr = (c_char * 6)()
                error = c_int(0)
 
-               self.instance = self.libshairplay.dnssd_init(hwaddr, c_int(len(hwaddr)), pointer(error))
+               self.instance = self.libshairplay.dnssd_init(pointer(error))
                if self.instance == None:
                        raise RuntimeError("Initializing library failed: " + str(error.value))
 
@@ -180,14 +273,19 @@ class DnssdService:
                        self.libshairplay.dnssd_destroy(self.instance)
                self.instance = None
 
-       def register_raop(self, name, port):
-               self.libshairplay.dnssd_register_raop(self.instance, name, c_ushort(port))
+       def register_raop(self, name, port, hwaddrstr, password=False):
+               hwaddr = create_string_buffer(hwaddrstr, len(hwaddrstr))
+               use_pw = c_int(0)
+               if password:
+                       use_pw = c_int(1)
+               self.libshairplay.dnssd_register_raop(self.instance, name, c_ushort(port), hwaddr, len(hwaddr), use_pw)
 
        def unregister_raop(self):
                self.libshairplay.dnssd_unregister_raop(self.instance)
 
-       def register_airplay(self, name, port):
-               self.libshairplay.dnssd_register_airplay(self.instance, name, c_ushort(port))
+       def register_airplay(self, name, port, hwaddrstr):
+               hwaddr = create_string_buffer(hwaddrstr, len(hwaddrstr))
+               self.libshairplay.dnssd_register_airplay(self.instance, name, c_ushort(port), hwaddr, len(hwaddr))
 
        def unregister_airplay(self):
                self.libshairplay.dnssd_unregister_airplay(self.instance)