Simplify the set_log_callback, it is better for thread safety
[deb_shairplay.git] / src / bindings / python / Shairplay.py
CommitLineData
1f0de05c 1# coding=utf-8
094e4ad1 2'''
6b27160f 3Copyright (C) 2012 Juho Vähä-Herttua
094e4ad1 4
6b27160f
JVH
5Permission is hereby granted, free of charge, to any person obtaining
6a copy of this software and associated documentation files (the
7"Software"), to deal in the Software without restriction, including
8without limitation the rights to use, copy, modify, merge, publish,
9distribute, sublicense, and/or sell copies of the Software, and to
10permit persons to whom the Software is furnished to do so, subject to
11the following conditions:
094e4ad1 12
6b27160f
JVH
13The above copyright notice and this permission notice shall be included
14in all copies or substantial portions of the Software.
15
16THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
17EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
19IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
20CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
21TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
22SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
094e4ad1
JVH
23'''
24
8be1424a
JVH
25import os
26import sys
27import platform
28
29from ctypes import *
30
10ceb254
JVH
31class RaopLogLevel:
32 EMERG = 0
33 ALERT = 1
34 CRIT = 2
35 ERR = 3
36 WARNING = 4
37 NOTICE = 5
38 INFO = 6
39 DEBUG = 7
40
41raop_log_callback_prototype = CFUNCTYPE(None, c_int, c_char_p)
42
23ab90af
JVH
43audio_init_prototype = CFUNCTYPE(py_object, c_void_p, c_int, c_int, c_int)
44audio_process_prototype = CFUNCTYPE(None, c_void_p, c_void_p, c_void_p, c_int)
45audio_destroy_prototype = CFUNCTYPE(None, c_void_p, c_void_p)
46
47audio_flush_prototype = CFUNCTYPE(None, c_void_p, c_void_p)
48audio_set_volume_prototype = CFUNCTYPE(None, c_void_p, c_void_p, c_float)
49audio_set_metadata_prototype = CFUNCTYPE(None, c_void_p, c_void_p, c_void_p, c_int)
50audio_set_coverart_prototype = CFUNCTYPE(None, c_void_p, c_void_p, c_void_p, c_int)
8be1424a
JVH
51
52class RaopNativeCallbacks(Structure):
23ab90af
JVH
53 _fields_ = [("cls", py_object),
54 ("audio_init", audio_init_prototype),
55 ("audio_process", audio_process_prototype),
56 ("audio_destroy", audio_destroy_prototype),
57 ("audio_flush", audio_flush_prototype),
58 ("audio_set_volume", audio_set_volume_prototype),
59 ("audio_set_metadata", audio_set_metadata_prototype),
60 ("audio_set_coverart", audio_set_coverart_prototype)]
8be1424a 61
c1b1e2c3 62def InitShairplay(libshairplay):
8be1424a
JVH
63 # Initialize dnssd related functions
64 libshairplay.dnssd_init.restype = c_void_p
094e4ad1 65 libshairplay.dnssd_init.argtypes = [POINTER(c_int)]
8be1424a 66 libshairplay.dnssd_register_raop.restype = c_int
a68fedbb 67 libshairplay.dnssd_register_raop.argtypes = [c_void_p, c_char_p, c_ushort, POINTER(c_char), c_int, c_int]
8be1424a 68 libshairplay.dnssd_register_airplay.restype = c_int
094e4ad1 69 libshairplay.dnssd_register_airplay.argtypes = [c_void_p, c_char_p, c_ushort, POINTER(c_char), c_int]
8be1424a
JVH
70 libshairplay.dnssd_unregister_raop.restype = None
71 libshairplay.dnssd_unregister_raop.argtypes = [c_void_p]
72 libshairplay.dnssd_unregister_airplay.restype = None
73 libshairplay.dnssd_unregister_airplay.argtypes = [c_void_p]
74 libshairplay.dnssd_destroy.restype = None
75 libshairplay.dnssd_destroy.argtypes = [c_void_p]
76
77 # Initialize raop related functions
78 libshairplay.raop_init.restype = c_void_p
23ab90af 79 libshairplay.raop_init.argtypes = [c_int, POINTER(RaopNativeCallbacks), c_char_p]
10ceb254
JVH
80 libshairplay.raop_set_log_level.restype = None
81 libshairplay.raop_set_log_level.argtypes = [c_void_p, c_int]
82 libshairplay.raop_set_log_callback.restype = None
83 libshairplay.raop_set_log_callback.argtypes = [c_void_p, raop_log_callback_prototype]
2528f9b0
JVH
84 libshairplay.raop_is_running.restype = c_int
85 libshairplay.raop_is_running.argtypes = [c_void_p]
8be1424a 86 libshairplay.raop_start.restype = c_int
a68fedbb 87 libshairplay.raop_start.argtypes = [c_void_p, POINTER(c_ushort), POINTER(c_char), c_int, c_char_p]
8be1424a
JVH
88 libshairplay.raop_stop.restype = None
89 libshairplay.raop_stop.argtypes = [c_void_p]
90 libshairplay.raop_destroy.restype = None
91 libshairplay.raop_destroy.argtypes = [c_void_p]
92
c1b1e2c3
JVH
93def LoadShairplay(path):
94 if sys.maxsize < 2**32:
95 libname = "shairplay32"
96 else:
97 libname = "shairplay64"
98
99 if platform.system() == "Windows":
100 libname = libname + ".dll"
101 elif platform.system() == "Darwin":
102 libname = "lib" + libname + ".dylib"
103 else:
104 libname = "lib" + libname + ".so"
105
106 try:
107 fullpath = os.path.join(path, libname)
108 libshairplay = cdll.LoadLibrary(fullpath)
109 except:
110 raise RuntimeError("Couldn't load shairplay library " + libname)
111
112 InitShairplay(libshairplay)
8be1424a
JVH
113 return libshairplay
114
115RSA_KEY = """
116-----BEGIN RSA PRIVATE KEY-----
117MIIEpQIBAAKCAQEA59dE8qLieItsH1WgjrcFRKj6eUWqi+bGLOX1HL3U3GhC/j0Qg90u3sG/1CUt
118wC5vOYvfDmFI6oSFXi5ELabWJmT2dKHzBJKa3k9ok+8t9ucRqMd6DZHJ2YCCLlDRKSKv6kDqnw4U
119wPdpOMXziC/AMj3Z/lUVX1G7WSHCAWKf1zNS1eLvqr+boEjXuBOitnZ/bDzPHrTOZz0Dew0uowxf
120/+sG+NCK3eQJVxqcaJ/vEHKIVd2M+5qL71yJQ+87X6oV3eaYvt3zWZYD6z5vYTcrtij2VZ9Zmni/
121UAaHqn9JdsBWLUEpVviYnhimNVvYFZeCXg/IdTQ+x4IRdiXNv5hEewIDAQABAoIBAQDl8Axy9XfW
122BLmkzkEiqoSwF0PsmVrPzH9KsnwLGH+QZlvjWd8SWYGN7u1507HvhF5N3drJoVU3O14nDY4TFQAa
123LlJ9VM35AApXaLyY1ERrN7u9ALKd2LUwYhM7Km539O4yUFYikE2nIPscEsA5ltpxOgUGCY7b7ez5
124NtD6nL1ZKauw7aNXmVAvmJTcuPxWmoktF3gDJKK2wxZuNGcJE0uFQEG4Z3BrWP7yoNuSK3dii2jm
125lpPHr0O/KnPQtzI3eguhe0TwUem/eYSdyzMyVx/YpwkzwtYL3sR5k0o9rKQLtvLzfAqdBxBurciz
126aaA/L0HIgAmOit1GJA2saMxTVPNhAoGBAPfgv1oeZxgxmotiCcMXFEQEWflzhWYTsXrhUIuz5jFu
127a39GLS99ZEErhLdrwj8rDDViRVJ5skOp9zFvlYAHs0xh92ji1E7V/ysnKBfsMrPkk5KSKPrnjndM
128oPdevWnVkgJ5jxFuNgxkOLMuG9i53B4yMvDTCRiIPMQ++N2iLDaRAoGBAO9v//mU8eVkQaoANf0Z
129oMjW8CN4xwWA2cSEIHkd9AfFkftuv8oyLDCG3ZAf0vrhrrtkrfa7ef+AUb69DNggq4mHQAYBp7L+
130k5DKzJrKuO0r+R0YbY9pZD1+/g9dVt91d6LQNepUE/yY2PP5CNoFmjedpLHMOPFdVgqDzDFxU8hL
131AoGBANDrr7xAJbqBjHVwIzQ4To9pb4BNeqDndk5Qe7fT3+/H1njGaC0/rXE0Qb7q5ySgnsCb3DvA
132cJyRM9SJ7OKlGt0FMSdJD5KG0XPIpAVNwgpXXH5MDJg09KHeh0kXo+QA6viFBi21y340NonnEfdf
13354PX4ZGS/Xac1UK+pLkBB+zRAoGAf0AY3H3qKS2lMEI4bzEFoHeK3G895pDaK3TFBVmD7fV0Zhov
13417fegFPMwOII8MisYm9ZfT2Z0s5Ro3s5rkt+nvLAdfC/PYPKzTLalpGSwomSNYJcB9HNMlmhkGzc
1351JnLYT4iyUyx6pcZBmCd8bD0iwY/FzcgNDaUmbX9+XDvRA0CgYEAkE7pIPlE71qvfJQgoA9em0gI
136LAuE4Pu13aKiJnfft7hIjbK+5kyb3TysZvoyDnb3HOKvInK7vXbKuU4ISgxB2bB3HcYzQMGsz1qJ
1372gG0N5hvJpzwwhbhXqFKA4zaaSrw622wDniAK5MlIE0tIAKKP4yxNGjoD2QYjhBGuhvkWKY=
138-----END RSA PRIVATE KEY-----
139"""
140
141class RaopCallbacks:
142 def audio_init(self, bits, channels, samplerate):
143 raise NotImplementedError()
144
8be1424a
JVH
145 def audio_process(self, session, buffer):
146 raise NotImplementedError()
147
23ab90af
JVH
148 def audio_destroy(self, session):
149 raise NotImplementedError()
150
151 def audio_set_volume(self, session, volume):
152 pass
153
8be1424a
JVH
154 def audio_flush(self, session):
155 pass
156
23ab90af
JVH
157 def audio_set_metadata(self, session, buffer):
158 pass
159
160 def audio_set_coverart(self, session, buffer):
8be1424a
JVH
161 pass
162
163class RaopService:
164 def audio_init_cb(self, cls, bits, channels, samplerate):
165 session = self.callbacks.audio_init(bits, channels, samplerate)
1f0de05c
JVH
166 self.sessions.append(session)
167 return session
8be1424a 168
8be1424a
JVH
169 def audio_process_cb(self, cls, sessionptr, buffer, buflen):
170 session = cast(sessionptr, py_object).value
171 strbuffer = string_at(buffer, buflen)
172 self.callbacks.audio_process(session, strbuffer)
173
8be1424a
JVH
174 def audio_destroy_cb(self, cls, sessionptr):
175 session = cast(sessionptr, py_object).value
176 self.callbacks.audio_destroy(session)
1f0de05c
JVH
177 if session in self.sessions:
178 self.sessions.remove(session)
8be1424a 179
23ab90af
JVH
180 def audio_flush_cb(self, cls, sessionptr):
181 session = cast(sessionptr, py_object).value
182 self.callbacks.audio_flush(session)
183
184 def audio_set_volume_cb(self, cls, sessionptr, volume):
185 session = cast(sessionptr, py_object).value
186 self.callbacks.audio_set_volume(session, volume)
8be1424a 187
23ab90af
JVH
188 def audio_set_metadata_cb(self, cls, sessionptr, buffer, buflen):
189 session = cast(sessionptr, py_object).value
190 strbuffer = string_at(buffer, buflen)
191 self.callbacks.audio_set_metadata(session, strbuffer)
192
193 def audio_set_coverart_cb(self, cls, sessionptr, buffer, buflen):
194 session = cast(sessionptr, py_object).value
195 strbuffer = string_at(buffer, buflen)
196 self.callbacks.audio_set_coverart(session, strbuffer)
197
198
199 def __init__(self, libshairplay, max_clients, callbacks):
8be1424a
JVH
200 self.libshairplay = libshairplay
201 self.callbacks = callbacks
1f0de05c 202 self.sessions = []
8be1424a
JVH
203 self.instance = None
204
205 # We need to hold a reference to native_callbacks
206 self.native_callbacks = RaopNativeCallbacks()
207 self.native_callbacks.audio_init = audio_init_prototype(self.audio_init_cb)
8be1424a 208 self.native_callbacks.audio_process = audio_process_prototype(self.audio_process_cb)
8be1424a 209 self.native_callbacks.audio_destroy = audio_destroy_prototype(self.audio_destroy_cb)
23ab90af
JVH
210 self.native_callbacks.audio_flush = audio_flush_prototype(self.audio_flush_cb)
211 self.native_callbacks.audio_set_volume = audio_set_volume_prototype(self.audio_set_volume_cb)
212 self.native_callbacks.audio_set_metadata = audio_set_metadata_prototype(self.audio_set_metadata_cb)
213 self.native_callbacks.audio_set_coverart = audio_set_coverart_prototype(self.audio_set_coverart_cb)
8be1424a
JVH
214
215 # Initialize the raop instance with our callbacks
23ab90af 216 self.instance = self.libshairplay.raop_init(max_clients, pointer(self.native_callbacks), RSA_KEY)
8be1424a
JVH
217 if self.instance == None:
218 raise RuntimeError("Initializing library failed")
219
220 def __del__(self):
221 if self.instance != None:
222 self.libshairplay.raop_destroy(self.instance)
223 self.instance = None
224
2528f9b0
JVH
225 def is_running(self):
226 if self.libshairplay.raop_is_running(self.instance):
227 return True
228 else:
229 return False
230
10ceb254
JVH
231 def set_log_level(self, level):
232 self.libshairplay.raop_set_log_level(self.instance, level)
233
234 def set_log_callback(self, log_callback):
ea824022
JVH
235 # We need to hold a reference to the log callback instance
236 self.log_callback = raop_log_callback_prototype(log_callback)
237 self.libshairplay.raop_set_log_callback(self.instance, self.log_callback)
10ceb254 238
a68fedbb 239 def start(self, port, hwaddrstr, password=None):
8be1424a
JVH
240 port = c_ushort(port)
241 hwaddr = create_string_buffer(hwaddrstr, len(hwaddrstr))
242
a68fedbb 243 ret = self.libshairplay.raop_start(self.instance, pointer(port), hwaddr, c_int(len(hwaddr)), password)
8be1424a
JVH
244 if ret < 0:
245 raise RuntimeError("Starting RAOP instance failed")
246 return port.value
247
248 def stop(self):
249 self.libshairplay.raop_stop(self.instance)
250
251class DnssdService:
252 def __init__(self, libshairplay):
253 self.libshairplay = libshairplay
254 self.instance = None
255
8be1424a
JVH
256 error = c_int(0)
257
094e4ad1 258 self.instance = self.libshairplay.dnssd_init(pointer(error))
8be1424a
JVH
259 if self.instance == None:
260 raise RuntimeError("Initializing library failed: " + str(error.value))
261
262 def __del__(self):
263 if self.instance != None:
264 self.libshairplay.dnssd_destroy(self.instance)
265 self.instance = None
266
a68fedbb 267 def register_raop(self, name, port, hwaddrstr, password=False):
094e4ad1 268 hwaddr = create_string_buffer(hwaddrstr, len(hwaddrstr))
a68fedbb
JVH
269 use_pw = c_int(0)
270 if password:
271 use_pw = c_int(1)
272 self.libshairplay.dnssd_register_raop(self.instance, name, c_ushort(port), hwaddr, len(hwaddr), use_pw)
8be1424a
JVH
273
274 def unregister_raop(self):
275 self.libshairplay.dnssd_unregister_raop(self.instance)
276
094e4ad1
JVH
277 def register_airplay(self, name, port, hwaddrstr):
278 hwaddr = create_string_buffer(hwaddrstr, len(hwaddrstr))
279 self.libshairplay.dnssd_register_airplay(self.instance, name, c_ushort(port), hwaddr, len(hwaddr))
8be1424a
JVH
280
281 def unregister_airplay(self):
282 self.libshairplay.dnssd_unregister_airplay(self.instance)
283