snapd/tests/lib/external/snapd-testing-tools/tools/snaps.name

#!/bin/bash

show_help() {
    echo "usage: snaps.name gadget, kernel, core, snap-suffix"
    echo ""
    echo "get the name of a specific snap for the current system"
}

gadget_name() {
    snap list | grep -E '(gadget$|gadget,)' | awk '{ print $1 }'
}

kernel_name(){
    # TODO this may not always be true during remodel scenarios
    # if the old kernel remains
    snap list | grep -E '(kernel$|kernel,)' | awk '{ print $1 }'
}

core_name(){
    local core_name
    core_name="$(snap model --verbose | grep -Po "^base:\\s+\\K.*" || true)"
    if [ -z "$core_name" ]; then
        core_name="core"
    fi
    echo "$core_name"
}

snap_suffix(){
    if os.query is-core18; then
        echo "-core18"
    elif os.query is-core20; then
        echo "-core20"
    elif os.query is-core22; then
        echo "-core22"
    elif os.query is-core24; then
        echo "-core24"
    fi
}

main() {
    if [ $# -eq 0 ]; then
        show_help
        exit 0
    fi

    while [ $# -gt 0 ]; do
        case "$1" in
            -h|--help)
                show_help
                exit 0
                ;;
            gadget)
                gadget_name
                exit
                ;;
            kernel)
                kernel_name
                exit
                ;;
            core)
                core_name
                exit
                ;;
            snap-suffix)
                snap_suffix
                exit
                ;;
            *)
                echo "snaps.name: unknown snap $1" >&2
                exit 1
                ;;
        esac
    done

}

main "$@"