mirror of
https://github.com/navidrome/navidrome.git
synced 2026-02-05 12:31:10 -05:00
e8863ed147
* feat: implement raw binary framing for host function responses Signed-off-by: Deluan <deluan@navidrome.org> * feat: add CallRaw method for Subsonic API to handle binary responses Signed-off-by: Deluan <deluan@navidrome.org> * test: add tests for raw=true methods and binary framing generation Signed-off-by: Deluan <deluan@navidrome.org> * fix: improve error message for malformed raw responses to indicate incomplete header Signed-off-by: Deluan <deluan@navidrome.org> * fix: add wasm_import_module attribute for raw methods and improve content-type handling Signed-off-by: Deluan <deluan@navidrome.org> --------- Signed-off-by: Deluan <deluan@navidrome.org>
102 lines
3.3 KiB
Python
102 lines
3.3 KiB
Python
# Code generated by ndpgen. DO NOT EDIT.
|
|
#
|
|
# This file contains client wrappers for the SubsonicAPI host service.
|
|
# It is intended for use in Navidrome plugins built with extism-py.
|
|
#
|
|
# IMPORTANT: Due to a limitation in extism-py, you cannot import this file directly.
|
|
# The @extism.import_fn decorators are only detected when defined in the plugin's
|
|
# main __init__.py file. Copy the needed functions from this file into your plugin.
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Any, Tuple
|
|
|
|
import extism
|
|
import json
|
|
import struct
|
|
|
|
|
|
class HostFunctionError(Exception):
|
|
"""Raised when a host function returns an error."""
|
|
pass
|
|
|
|
|
|
@extism.import_fn("extism:host/user", "subsonicapi_call")
|
|
def _subsonicapi_call(offset: int) -> int:
|
|
"""Raw host function - do not call directly."""
|
|
...
|
|
|
|
|
|
@extism.import_fn("extism:host/user", "subsonicapi_callraw")
|
|
def _subsonicapi_callraw(offset: int) -> int:
|
|
"""Raw host function - do not call directly."""
|
|
...
|
|
|
|
|
|
def subsonicapi_call(uri: str) -> str:
|
|
"""Call executes a Subsonic API request and returns the JSON response.
|
|
|
|
The uri parameter should be the Subsonic API path without the server prefix,
|
|
e.g., "getAlbumList2?type=random&size=10". The response is returned as raw JSON.
|
|
|
|
Args:
|
|
uri: str parameter.
|
|
|
|
Returns:
|
|
str: The result value.
|
|
|
|
Raises:
|
|
HostFunctionError: If the host function returns an error.
|
|
"""
|
|
request = {
|
|
"uri": uri,
|
|
}
|
|
request_bytes = json.dumps(request).encode("utf-8")
|
|
request_mem = extism.memory.alloc(request_bytes)
|
|
response_offset = _subsonicapi_call(request_mem.offset)
|
|
response_mem = extism.memory.find(response_offset)
|
|
response = json.loads(extism.memory.string(response_mem))
|
|
|
|
if response.get("error"):
|
|
raise HostFunctionError(response["error"])
|
|
|
|
return response.get("responseJson", "")
|
|
|
|
|
|
def subsonicapi_call_raw(uri: str) -> Tuple[str, bytes]:
|
|
"""CallRaw executes a Subsonic API request and returns the raw binary response.
|
|
Optimized for binary endpoints like getCoverArt and stream that return
|
|
non-JSON data. The response is returned as raw bytes without JSON encoding overhead.
|
|
|
|
Args:
|
|
uri: str parameter.
|
|
|
|
Returns:
|
|
Tuple of (content_type, data) with the raw binary response.
|
|
|
|
Raises:
|
|
HostFunctionError: If the host function returns an error.
|
|
"""
|
|
request = {
|
|
"uri": uri,
|
|
}
|
|
request_bytes = json.dumps(request).encode("utf-8")
|
|
request_mem = extism.memory.alloc(request_bytes)
|
|
response_offset = _subsonicapi_callraw(request_mem.offset)
|
|
response_mem = extism.memory.find(response_offset)
|
|
response_bytes = response_mem.bytes()
|
|
|
|
if len(response_bytes) == 0:
|
|
raise HostFunctionError("empty response from host")
|
|
if response_bytes[0] == 0x01:
|
|
raise HostFunctionError(response_bytes[1:].decode("utf-8"))
|
|
if response_bytes[0] != 0x00:
|
|
raise HostFunctionError("unknown response status")
|
|
if len(response_bytes) < 5:
|
|
raise HostFunctionError("malformed raw response: incomplete header")
|
|
ct_len = struct.unpack(">I", response_bytes[1:5])[0]
|
|
if len(response_bytes) < 5 + ct_len:
|
|
raise HostFunctionError("malformed raw response: content-type overflow")
|
|
content_type = response_bytes[5:5 + ct_len].decode("utf-8")
|
|
data = response_bytes[5 + ct_len:]
|
|
return content_type, data
|