move block device discovery out of rebash

1 files changed, 82 insertions, 0 deletions

diff --git a/builder/dnbd3-rootfs/scripts/tools.sh b/builder/dnbd3-rootfs/scripts/tools.sh
new file mode 100644
index 00000000..1a87cff1
--- /dev/null
+++ b/builder/dnbd3-rootfs/scripts/tools.sh
@@ -0,0 +1,82 @@
+#!/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"