# Copyright 2023 Oliver Smith
# SPDX-License-Identifier: GPL-3.0-or-later
import os
from pathlib import Path
from typing import Optional
from pmb.core import get_context


def find_path(codename: str, file='') -> Optional[Path]:
    """Find path to device APKBUILD under `device/*/device-`.

    :param codename: device codename
    :param file: file to look for (e.g. APKBUILD or deviceinfo), may be empty
    :returns: path to APKBUILD
    """
    g = list((get_context().config.aports / "device").glob(f"*/device-{codename}/{file}"))
    if not g:
        return None

    if len(g) != 1:
        raise RuntimeError(codename + " found multiple times in the device"
                           " subdirectory of pmaports")

    return g[0]


def list_codenames(aports: Path, vendor=None, archived=True):
    """Get all devices, for which aports are available.

    :param vendor: vendor name to choose devices from, or None for all vendors
    :param archived: include archived devices
    :returns: ["first-device", "second-device", ...]
    """
    ret = []
    for path in aports.glob("device/*/device-*"):
        if not archived and 'archived' in path.parts:
            continue
        device = os.path.basename(path).split("-", 1)[1]
        if (vendor is None) or device.startswith(vendor + '-'):
            ret.append(device)
    return ret


def list_vendors(aports: Path):
    """Get all device vendors, for which aports are available.

    :returns: {"vendor1", "vendor2", ...}
    """
    ret = set()
    for path in (aports / "device").glob("*/device-*"):
        vendor = path.name.split("-", 2)[1]
        ret.add(vendor)
    return ret