Add WebDAV provider

music_assistant/providers/webdav/__init__.py

@@ -0,0 +1,108 @@
+"""WebDAV filesystem provider for Music Assistant."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from music_assistant_models.config_entries import ConfigEntry, ConfigValueOption
+from music_assistant_models.enums import ConfigEntryType, ProviderFeature
+
+from music_assistant.constants import CONF_PASSWORD, CONF_USERNAME
+
+from .constants import CONF_CONTENT_TYPE, CONF_URL, CONF_VERIFY_SSL
+from .provider import WebDAVFileSystemProvider
+
+if TYPE_CHECKING:
+    from music_assistant_models.config_entries import ConfigValueType, ProviderConfig
+    from music_assistant_models.provider import ProviderManifest
+
+    from music_assistant.mass import MusicAssistant
+    from music_assistant.models import ProviderInstanceType
+
+
+# Supported features
+SUPPORTED_FEATURES = {
+    ProviderFeature.BROWSE,
+    ProviderFeature.SEARCH,
+}
+
+
+async def setup(
+    mass: MusicAssistant, manifest: ProviderManifest, config: ProviderConfig
+) -> ProviderInstanceType:
+    """Initialize provider(instance) with given configuration."""
+    prov = WebDAVFileSystemProvider(mass, manifest, config)
+    await prov.handle_async_init()
+    return prov
+
+
+async def get_config_entries(
+    mass: MusicAssistant,
+    instance_id: str | None = None,
+    action: str | None = None,
+    values: dict[str, ConfigValueType] | None = None,
+) -> tuple[ConfigEntry, ...]:
+    """Return Config entries to setup this provider."""
+    # ruff: noqa: ARG001
+    return (
+        ConfigEntry(
+            key=CONF_CONTENT_TYPE,
+            type=ConfigEntryType.STRING,
+            label="Content type",
+            options=[
+                ConfigValueOption("Music", "music"),
+                ConfigValueOption("Audiobooks", "audiobooks"),
+                ConfigValueOption("Podcasts", "podcasts"),
+            ],
+            default_value="music",
+            description="The type of content stored on this WebDAV server",
+            hidden=instance_id is not None,
+        ),
+        ConfigEntry(
+            key=CONF_URL,
+            type=ConfigEntryType.STRING,
+            label="WebDAV URL",
+            required=True,
+            description="The base URL of your WebDAV server (e.g., https://example.com/webdav)",
+        ),
+        ConfigEntry(
+            key=CONF_USERNAME,
+            type=ConfigEntryType.STRING,
+            label="Username",
+            required=False,
+            description="Username for WebDAV authentication (leave empty for no authentication)",
+        ),
+        ConfigEntry(
+            key=CONF_PASSWORD,
+            type=ConfigEntryType.SECURE_STRING,
+            label="Password",
+            required=False,
+            description="Password for WebDAV authentication",
+        ),
+        ConfigEntry(
+            key=CONF_VERIFY_SSL,
+            type=ConfigEntryType.BOOLEAN,
+            label="Verify SSL certificate",
+            default_value=False,
+            description="Verify SSL certificates when connecting to HTTPS WebDAV servers",
+        ),
+        ConfigEntry(
+            key="missing_album_artist_action",
+            type=ConfigEntryType.STRING,
+            label="Action when album artist tag is missing",
+            options=[
+                ConfigValueOption("Use track artist(s)", "track_artist"),
+                ConfigValueOption("Use folder name", "folder_name"),
+                ConfigValueOption("Use 'Various Artists'", "various_artists"),
+            ],
+            default_value="various_artists",
+            description="What to do when a track is missing the album artist tag",
+        ),
+        ConfigEntry(
+            key="ignore_album_playlists",
+            type=ConfigEntryType.BOOLEAN,
+            label="Ignore playlists in album folders",
+            default_value=True,
+            description="Ignore playlist files found in album subdirectories",
+        ),
+    )

music_assistant/providers/webdav/constants.py

@@ -0,0 +1,54 @@
+"""Constants for WebDAV filesystem provider."""
+
+from __future__ import annotations
+
+from typing import Final
+
+# Config keys
+CONF_URL = "url"
+CONF_VERIFY_SSL = "verify_ssl"
+CONF_CONTENT_TYPE = "content_type"
+
+# File extensions - reuse from filesystem_local patterns
+TRACK_EXTENSIONS: Final[tuple[str, ...]] = (
+    "mp3",
+    "flac",
+    "wav",
+    "m4a",
+    "aac",
+    "ogg",
+    "wma",
+    "opus",
+    "mp4",
+    "m4p",
+)
+
+PLAYLIST_EXTENSIONS: Final[tuple[str, ...]] = ("m3u", "m3u8", "pls")
+
+AUDIOBOOK_EXTENSIONS: Final[tuple[str, ...]] = (
+    "mp3",
+    "flac",
+    "m4a",
+    "m4b",
+    "aac",
+    "ogg",
+    "opus",
+    "wav",
+)
+
+PODCAST_EPISODE_EXTENSIONS: Final[tuple[str, ...]] = TRACK_EXTENSIONS
+
+IMAGE_EXTENSIONS: Final[tuple[str, ...]] = ("jpg", "jpeg", "png", "gif", "bmp", "webp", "tiff")
+
+SUPPORTED_EXTENSIONS: Final[tuple[str, ...]] = (
+    *TRACK_EXTENSIONS,
+    *PLAYLIST_EXTENSIONS,
+    *AUDIOBOOK_EXTENSIONS,
+    *IMAGE_EXTENSIONS,
+)
+
+# WebDAV specific constants
+WEBDAV_TIMEOUT: Final[int] = 30
+
+# Concurrent processing limit
+MAX_CONCURRENT_TASKS: Final[int] = 3

music_assistant/providers/webdav/helpers.py

@@ -0,0 +1,198 @@
+"""WebDAV helper functions for Music Assistant."""
+
+from __future__ import annotations
+
+import contextlib
+import logging
+from dataclasses import dataclass
+from urllib.parse import unquote, urljoin
+
+import aiohttp
+from defusedxml import ElementTree
+from music_assistant_models.errors import LoginFailed, SetupFailedError
+
+LOGGER = logging.getLogger(__name__)
+
+DAV_NAMESPACE = {"d": "DAV:"}
+
+
+@dataclass
+class WebDAVItem:
+    """Representation of a WebDAV resource."""
+
+    href: str
+    name: str
+    is_dir: bool
+    size: int | None = None
+    last_modified: str | None = None
+
+    @property
+    def ext(self) -> str | None:
+        """Return file extension."""
+        if self.is_dir:
+            return None
+        try:
+            return self.name.rsplit(".", 1)[1].lower()
+        except IndexError:
+            return None
+
+
+async def webdav_propfind(
+    session: aiohttp.ClientSession,
+    url: str,
+    depth: int = 1,
+    timeout: int = 30,
+    auth: aiohttp.BasicAuth | None = None,
+) -> list[WebDAVItem]:
+    """
+    Execute a PROPFIND request on a WebDAV resource.
+
+    Args:
+        session: Active HTTP session with auth configured
+        url: WebDAV URL to query
+        depth: Depth level (0=properties only, 1=immediate children)
+        timeout: Request timeout in seconds
+
+    Returns:
+        List of WebDAVItem objects
+
+    Raises:
+        LoginFailed: Authentication failed
+        SetupFailedError: Connection or other setup issues
+    """
+    headers = {
+        "Depth": str(depth),
+        "Content-Type": "application/xml; charset=utf-8",
+    }
+
+    body = """<?xml version="1.0" encoding="utf-8"?>
+<d:propfind xmlns:d="DAV:">
+    <d:prop>
+        <d:resourcetype/>
+        <d:getcontentlength/>
+        <d:getlastmodified/>
+        <d:displayname/>
+    </d:prop>
+</d:propfind>"""
+
+    try:
+        async with session.request(
+            "PROPFIND",
+            url,
+            headers=headers,
+            data=body,
+            auth=auth,
+            timeout=aiohttp.ClientTimeout(total=timeout),
+        ) as resp:
+            if resp.status == 401:
+                raise LoginFailed(f"Authentication failed for WebDAV server: {url}")
+            if resp.status == 403:
+                raise LoginFailed(f"Access forbidden for WebDAV server: {url}")
+            if resp.status == 404:
+                LOGGER.debug("WebDAV resource not found: %s", url)
+                return []
+            elif resp.status >= 400:
+                raise SetupFailedError(
+                    f"WebDAV PROPFIND failed with status {resp.status} for {url}"
+                )
+
+            response_text = await resp.text()
+            return _parse_propfind_response(response_text, url)
+
+    except TimeoutError as err:
+        raise SetupFailedError(f"WebDAV connection timeout: {url}") from err
+    except aiohttp.ClientError as err:
+        raise SetupFailedError(f"WebDAV connection error: {err}") from err
+
+
+def _parse_propfind_response(response_text: str, base_url: str) -> list[WebDAVItem]:
+    """Parse WebDAV PROPFIND XML response."""
+    try:
+        root = ElementTree.fromstring(response_text)
+    except ElementTree.ParseError as err:
+        LOGGER.warning("Failed to parse WebDAV PROPFIND response: %s", err)
+        return []
+
+    items: list[WebDAVItem] = []
+
+    for response_elem in root.findall("d:response", DAV_NAMESPACE):
+        href_elem = response_elem.find("d:href", DAV_NAMESPACE)
+        if href_elem is None or not href_elem.text:
+            continue
+
+        href = unquote(href_elem.text.rstrip("/"))
+
+        # Skip the base directory itself
+        if href.rstrip("/") == base_url.rstrip("/"):
+            continue
+
+        # Extract properties
+        propstat = response_elem.find("d:propstat", DAV_NAMESPACE)
+        if propstat is None:
+            continue
+
+        prop = propstat.find("d:prop", DAV_NAMESPACE)
+        if prop is None:
+            continue
+
+        # Check if it's a directory
+        resourcetype = prop.find("d:resourcetype", DAV_NAMESPACE)
+        is_collection = (
+            resourcetype is not None
+            and resourcetype.find("d:collection", DAV_NAMESPACE) is not None
+        )
+
+        # Get size
+        size = None
+        if not is_collection:
+            contentlength = prop.find("d:getcontentlength", DAV_NAMESPACE)
+            if contentlength is not None and contentlength.text:
+                with contextlib.suppress(ValueError):
+                    size = int(contentlength.text)
+
+        # Get last modified
+        lastmodified = prop.find("d:getlastmodified", DAV_NAMESPACE)
+        last_modified = lastmodified.text if lastmodified is not None else None
+
+        # Get display name or extract from href
+        displayname = prop.find("d:displayname", DAV_NAMESPACE)
+        if displayname is not None and displayname.text:
+            name = displayname.text
+        else:
+            name = href.split("/")[-1] or href.split("/")[-2]
+
+        items.append(
+            WebDAVItem(
+                href=href,
+                name=name,
+                is_dir=is_collection,
+                size=size,
+                last_modified=last_modified,
+            )
+        )
+
+    return items
+
+
+async def webdav_test_connection(
+    session: aiohttp.ClientSession,  # Pass session in instead of creating
+    url: str,
+    auth: aiohttp.BasicAuth | None = None,
+    timeout: int = 10,
+) -> None:
+    """Test WebDAV connection and authentication."""
+    await webdav_propfind(session, url, depth=0, timeout=timeout, auth=auth)
+
+
+def normalize_webdav_url(url: str) -> str:
+    """Normalize WebDAV URL by ensuring it ends with a slash."""
+    if not url.endswith("/"):
+        return f"{url}/"
+    return url
+
+
+def build_webdav_url(base_url: str, path: str) -> str:
+    """Build a WebDAV URL by joining base URL with path."""
+    normalized_base = normalize_webdav_url(base_url)
+    path = path.removeprefix("/")
+    return urljoin(normalized_base, path)