limactl.bash

# SPDX-FileCopyrightText: Copyright The Lima Authors
# SPDX-License-Identifier: Apache-2.0

# Create a dummy Lima instance for testing purposes. It cannot be started because it doesn't have an actual image.
# This function intentionally doesn't use create/editflags, but modifies the template with yq instead.
create_dummy_instance() {
    local name=$1
    local expr=${2:-}

    # Template does not validate without an image, and the image must point to a file that exists (for clonefile).
    local template="{images: [location: /etc/profile]}"
    if [[ -n $expr ]]; then
        template="$(limactl yq "$expr" <<<"$template")"
    fi
    limactl create --name "$name" - <<<"$template"
}

# Ensure a Lima instance exists. When LIMA_BATS_REUSE_INSTANCE is set, reuse an
# existing running instance. Otherwise delete and recreate it.
# The instance configuration is determined by its name; add a case below for new names.
# Close file handles 3 and 4 so the host agent doesn't block BATS from exiting.
ensure_instance() {
    local instance=$1
    if [[ -n "${LIMA_BATS_REUSE_INSTANCE:-}" ]]; then
        run limactl list --format '{{.Status}}' "$instance"
        [[ $status == 0 ]] && [[ $output == "Running" ]] && return
    fi
    limactl unprotect "$instance" || :
    limactl delete --force "$instance" || :
    case "$instance" in
        bats)          limactl start --yes --name "$instance" template:default 3>&- 4>&- ;;
        bats-nomount)  limactl start --yes --name "$instance" --mount-none template:default 3>&- 4>&- ;;
        bats-dummy)    create_dummy_instance "$instance" '.disk = "1M"' ;;
        bats-portfwd)
            local tmpconfig="$BATS_FILE_TMPDIR/config"
            mkdir -p "$tmpconfig"
            local tmpfile="$tmpconfig/bats-portfwd.yaml"
            limactl tmpl copy "template:default" "$tmpfile"
            # Remove existing portForwards section: skip from "portForwards:" until the next top-level key
            local cleaned
            cleaned=$(awk '
                /^portForwards:/ { skip=1; next }
                skip && /^[a-z]/ { skip=0 }
                !skip { print }
            ' "$tmpfile")
            echo "$cleaned" >"$tmpfile"
            # shellcheck source=../tests/port-forwarding-config.bash
            source "$PATH_BATS_ROOT/tests/port-forwarding-config.bash"
            generate_port_forwards_yaml "$(get_host_ipv4)" >>"$tmpfile"
            limactl start --yes --name "$instance" "$tmpfile" 3>&- 4>&-
            ;;
        *)
            echo "ensure_instance: unknown instance name '$instance'" >&2
            return 1
            ;;
    esac
}

# Delete the given Lima instance unless LIMA_BATS_REUSE_INSTANCE is set.
delete_instance() {
    local instance=$1
    if [[ -z "${LIMA_BATS_REUSE_INSTANCE:-}" ]]; then
        limactl unprotect "$instance" || :
        limactl delete --force "$instance" || :
    fi
}