#!/bin/sh

command -v getargs > /dev/null || . /lib/dracut-lib.sh

_md_start() {
    local _udevinfo
    local _path_s
    local _path_d
    local _md="$1"

    _udevinfo="$(udevadm info --query=property --name="${_md}")"
    strstr "$_udevinfo" "MD_LEVEL=container" && return 0
    strstr "$_udevinfo" "DEVTYPE=partition" && return 0

    _path_s="/sys/$(udevadm info -q path -n "${_md}")/md/array_state"
    [ ! -r "$_path_s" ] && return 0

    # inactive ?
    [ "$(cat "$_path_s")" != "inactive" ] && return 0

    mdadm -R "${_md}" 2>&1 | vinfo

    # still inactive ?
    [ "$(cat "$_path_s")" = "inactive" ] && return 0

    _path_d="${_path_s%/*}/degraded"
    [ ! -r "$_path_d" ] && return 0
    : > "$hookdir"/initqueue/work
}

_md_force_run() {
    local _md
    local _UUID
    local _MD_UUID
    local _LEFT_UUID
    local _uuid

    _MD_UUID=$(getargs rd.md.uuid -d rd_MD_UUID=)
    [ -n "$_MD_UUID" ] || getargbool 0 rd.auto || return

    if [ -n "$_MD_UUID" ]; then
        _LEFT_UUID=$(str_replace "$_MD_UUID" "-" ":")
        _MD_UUID=$(str_replace "$_MD_UUID" "-" "")
        _MD_UUID=$(str_replace "$_MD_UUID" ":" "")

        for _md in /dev/md[0-9_]*; do
            [ -b "$_md" ] || continue
            _UUID=$(
                /sbin/mdadm -D --export "$_md" \
                    | while read -r line || [ -n "$line" ]; do
                        str_starts "$line" "MD_UUID=" || continue
                        printf "%s" "${line#MD_UUID=}"
                    done
            )

            [ -z "$_UUID" ] && continue
            # remove the UUID already assembled from the list
            _LEFT_UUID=$(str_replace "$_LEFT_UUID" "$_UUID" "")

            _UUID=$(str_replace "$_UUID" ":" "")

            # check if we should handle this device
            strstr "$_MD_UUID" "$_UUID" || continue

            _md_start "${_md}"
        done
    else
        # try to force-run anything not running yet
        for _md in /dev/md[0-9_]*; do
            [ -b "$_md" ] || continue
            _md_start "${_md}"
        done
    fi

    # try to assemble all the left md devices that should be assembled
    # probably failed because of timing issue in the first beginning
    if [ -n "$_LEFT_UUID" ]; then
        for _uuid in ${_LEFT_UUID}; do
            mdadm --assemble --scan --uuid="$_uuid"
        done
    fi
}

_md_force_run
