#!/usr/bin/env bash
# shellcheck source=./rebash/core.sh
source "/usr/lib/rebash/core.sh"
core.import logging

tools_find_block_device() {
    # shellcheck disable=SC2034,SC2016
    local __doc__='
    >>> lsblk() {
    >>>     if [[ "${@: -1}" == "" ]];then
    >>>         echo "lsblk: : not a block device"
    >>>         return 1
    >>>     fi
    >>>     if [[ "${@: -1}" != "/dev/sdb" ]];then
    >>>         echo "/dev/sda disk"
    >>>         echo "/dev/sda1 part SYSTEM_LABEL 0x7"
    >>>         echo "/dev/sda2 part"
    >>>     fi
    >>>     if [[ "${@: -1}" != "/dev/sda" ]];then
    >>>         echo "/dev/sdb disk"
    >>>         echo "/dev/sdb1 part boot_partition "
    >>>         echo "/dev/sdb2 part system_partition"
    >>>     fi
    >>> }
    >>> blkid() {
    >>>     [[ "${@: -1}" != "/dev/sda2" ]] && return 0
    >>>     echo "gpt"
    >>>     echo "only discoverable by blkid"
    >>>     echo "boot_partition"
    >>>     echo "192d8b9e"
    >>> }
    >>> tools_find_block_device "boot_partition"
    >>> tools_find_block_device "boot_partition" /dev/sda
    >>> tools_find_block_device "discoverable by blkid"
    >>> tools_find_block_device "_partition"
    >>> tools_find_block_device "not matching anything" || echo not found
    >>> tools_find_block_device "" || echo not found
    /dev/sdb1
    /dev/sda2
    /dev/sda2
    /dev/sdb1 /dev/sdb2
    not found
    not found
    '
    local partition_pattern="$1"
    local device="$2"

    [ "$partition_pattern" = "" ] && return 1
    tools_find_block_device_simple() {
        local device_info
        lsblk --noheadings --list --paths --output \
        NAME,TYPE,LABEL,PARTLABEL,UUID,PARTUUID ${device:+"$device"} \
        | sort --unique | while read -r device_info; do
            local current_device
            current_device=$(echo "$device_info" | cut -d' ' -f1)
            if [[ "$device_info" = *"${partition_pattern}"* ]]; then
                echo "$current_device"
            fi
        done
    }
    tools_find_block_device_deep() {
        local device_info
        lsblk --noheadings --list --paths --output NAME ${device:+"$device"} \
        | sort --unique | cut -d' ' -f1 | while read -r current_device; do
            blkid -p -o value "$current_device" \
            | while read -r device_info; do
                if [[ "$device_info" = *"${partition_pattern}"* ]]; then
                    echo "$current_device"
                fi
            done
        done
    }
    local candidates
    candidates=($(tools_find_block_device_simple))
    [ ${#candidates[@]} -eq 0 ] && candidates=($(tools_find_block_device_deep))
    unset -f tools_find_block_device_simple
    unset -f tools_find_block_device_deep
    [ ${#candidates[@]} -eq 0 ] && return 1
    [ ${#candidates[@]} -ne 1 ] && echo "${candidates[@]}" && return 1
    logging.plain "${candidates[0]}"
}
alias tools.find_block_device="tools_find_block_device"