working

librespot_example.py

@@ -0,0 +1,121 @@
+import os
+import re
+import socket
+import threading
+
+from librespot.audio.decoders import AudioQuality, VorbisOnlyAudioQuality
+from librespot.core import Session
+from librespot.metadata import TrackId
+
+session: Session
+sock: socket
+
+
+def handler(client: socket.socket, address: str):
+    req_raw = client.recv(1024 * 1024)
+    if len(req_raw) == 0:
+        return
+    req_arr = req_raw.split(b"\r\n")
+    req_http_raw = req_arr[0]
+    req_header_str = req_raw.split(b"\r\n\r\n")[0]
+    req_body_str = req_raw.split(b"\r\n\r\n")[1]
+    req_http_arr = req_http_raw.split(b" ")
+    req_method = req_http_arr[0]
+    req_uri = req_http_arr[1]
+    req_http_version = req_http_arr[2]
+    req_header = {}
+    for header in req_header_str.split(b"\r\n"):
+        try:
+            key, value = header.split(b": ")
+        except ValueError:
+            continue
+        else:
+            req_header[key.decode().lower()] = value.decode()
+    status, headers, content, manually = response(client, req_uri.decode(),
+                                                  req_header, req_body_str)
+    if not manually:
+        client.send(req_http_version + b" " + status.encode() + b"\r\n")
+        client.send(b"Access-Control-Allow-Origin: *\r\n")
+        for header in headers:
+            client.send(header.encode() + "\r\n")
+        client.send(b"\r\n")
+        client.send(content)
+    client.close()
+
+
+class HttpCode:
+    http_200 = "200 OK"
+    http_204 = "204 No Content"
+    http_400 = "400 Bad Request"
+    http_403 = "403 Forbidden"
+    http_404 = "404 Not Found"
+    http_500 = "500 Internal Server Error"
+
+
+def main():
+    global session, sock
+    session = None
+    if os.path.isfile("credentials.json"):
+        try:
+            session = Session.Builder().stored_file().create()
+        except RuntimeError:
+            pass
+    if session is None or not session.is_valid():
+        username = input("Username: ")
+        password = input("Password: ")
+        session = Session.Builder().user_pass(username, password).create()
+    if not session.is_valid():
+        return
+    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    sock.bind(("127.0.0.1", 8080))
+    sock.listen(5)
+    while True:
+        threading.Thread(target=handler, args=sock.accept()).start()
+
+
+def response(client: socket.socket, uri: str, header: dict,
+             body: bytes) -> tuple[str, list, bytes, bool]:
+    if re.search(r"^/audio/track/([0-9a-zA-Z]{22})$", uri) is not None:
+        track_id_search = re.search(
+            r"^/audio/track/(?P<TrackID>[0-9a-zA-Z]{22})$", uri)
+        track_id_str = track_id_search.group("TrackID")
+        track_id = TrackId.from_base62(track_id_str)
+        stream = session.content_feeder().load(
+            track_id, VorbisOnlyAudioQuality(AudioQuality.VERY_HIGH), False,
+            None)
+        start = 0
+        end = stream.input_stream.stream().size()
+        if header.get("range") is not None:
+            range_search = re.search(
+                "^bytes=(?P<start>[0-9]+?)-(?P<end>[0-9]+?)$",
+                header.get("range"))
+            if range_search is not None:
+                start = int(range_search.group("start"))
+                end = (int(range_search.group("end"))
+                       if int(range_search.group("end")) <=
+                       stream.input_stream.stream().size() else
+                       stream.input_stream.stream().size())
+                stream.input_stream.stream().skip(start)
+        client.send(b"HTTP/1.0 200 OK\r\n")
+        client.send(b"Access-Control-Allow-Origin: *\r\n")
+        client.send(b"Content-Length: " +
+                    (str(stream.input_stream.stream().size()).encode() if
+                     stream.input_stream.stream().size() == end else "{}-{}/{}"
+                     .format(start, end,
+                             stream.input_stream.stream().size()).encode()) +
+                    b"\r\n")
+        client.send(b"Content-Type: audio/ogg\r\n")
+        client.send(b"\r\n")
+        while True:
+            if (stream.input_stream.stream().pos() >=
+                    stream.input_stream.stream().size()):
+                break
+            byte = stream.input_stream.stream().read(1)
+            client.send(byte)
+        return "", [], b"", True
+    else:
+        return HttpCode.http_404, [], HttpCode.http_404.encode(), False
+
+
+if __name__ == "__main__":
+    main()