# Code generated by ndpgen. DO NOT EDIT.
#
# This file contains client wrappers for the Search 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

import extism
import json


class HostFunctionError(Exception):
    """Raised when a host function returns an error."""
    pass


@extism.import_fn("extism:host/user", "search_find")
def _search_find(offset: int) -> int:
    """Raw host function - do not call directly."""
    ...


@dataclass
class SearchFindResult:
    """Result type for search_find."""
    results: Any
    total: int


def search_find(query: str) -> SearchFindResult:
    """Call the search_find host function.

    Args:
        query: str parameter.

    Returns:
        SearchFindResult containing results, total,.

    Raises:
        HostFunctionError: If the host function returns an error.
    """
    request = {
        "query": query,
    }
    request_bytes = json.dumps(request).encode("utf-8")
    request_mem = extism.memory.alloc(request_bytes)
    response_offset = _search_find(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 SearchFindResult(
        results=response.get("results", None),
        total=response.get("total", 0),
    )