# -*- mode: sh; indent-tabs-mode: nil; sh-basic-offset: 4; -*-
# vim: et sts=4 sw=4

#  SPDX-License-Identifier: LGPL-2.1+
#
#  Copyright © 2020-2021 Collabora Ltd.
#  Copyright © 2020-2021 Valve Corporation.
#
#  This file is part of steamos-customizations.
#
#  steamos-customizations is free software; you can redistribute it and/or
#  modify it under the terms of the GNU Lesser General Public License as
#  published by the Free Software Foundation; either version 2.1 of the License,
#  or (at your option) any later version.

_steamos-partsets()
{
    local cur prev words cword
    _init_completion || return

    case "$prev" in
        --devices)
            COMPREPLY=($(compgen -W "efi-{A,B,dev} verity-{A,B} rootfs-{A,B,dev} var-{A,B,dev} home" -- "$cur"))
            return
            ;;
    esac

    if [[ "$cur" == -* ]]; then
        COMPREPLY=($(compgen -W "-h --help --devices" -- "$cur"))
        return
    fi

    local args
    _count_args "" "--devices"
    if [[ "$args" -eq 1 ]]; then
        _filedir -d
    fi
} &&
complete -F _steamos-partsets steamos-partsets
