diff --git a/lib/spack/spack/cmd/extensions.py b/lib/spack/spack/cmd/extensions.py index e834d7fd18646b1f51eacb71e70054f75decc17d..b4fb7faa1f9afcdbceadddef2de9ff30e8a988d7 100644 --- a/lib/spack/spack/cmd/extensions.py +++ b/lib/spack/spack/cmd/extensions.py @@ -4,6 +4,7 @@ # SPDX-License-Identifier: (Apache-2.0 OR MIT) import argparse +import sys import llnl.util.tty as tty from llnl.util.tty.colify import colify @@ -21,6 +22,8 @@ def setup_parser(subparser): + subparser.epilog = 'If called without argument returns ' \ + 'the list of all valid extendable packages' arguments.add_common_arguments(subparser, ['long', 'very_long']) subparser.add_argument('-d', '--deps', action='store_true', help='output dependencies along with found specs') @@ -42,7 +45,19 @@ def setup_parser(subparser): def extensions(parser, args): if not args.spec: - tty.die("extensions requires a package spec.") + # If called without arguments, list all the extendable packages + isatty = sys.stdout.isatty() + if isatty: + tty.info('Extendable packages:') + + extendable_pkgs = [] + for name in spack.repo.all_package_names(): + pkg = spack.repo.get(name) + if pkg.extendable: + extendable_pkgs.append(name) + + colify(extendable_pkgs, indent=4) + return # Checks spec = cmd.parse_specs(args.spec) diff --git a/lib/spack/spack/test/cmd/extensions.py b/lib/spack/spack/test/cmd/extensions.py index 505573a7bc3600584db025562aeba206a7769aab..7fc56593ebe0fbb6fee75606cf5ada74df710d9d 100644 --- a/lib/spack/spack/test/cmd/extensions.py +++ b/lib/spack/spack/test/cmd/extensions.py @@ -69,6 +69,11 @@ def check_output(ni, na): check_output(1, 1) +def test_extensions_no_arguments(mock_packages): + out = extensions() + assert 'python' in out + + def test_extensions_raises_if_not_extendable(mock_packages): with pytest.raises(SpackCommandError): extensions("flake8") diff --git a/share/spack/bash/spack-completion.in b/share/spack/bash/spack-completion.in index ca15b8dfb27fca101af7f2e3fc82c32087f0aa71..164fc5c5b3db5b50a16ecf89f6a4987a5da04dbb 100755 --- a/share/spack/bash/spack-completion.in +++ b/share/spack/bash/spack-completion.in @@ -226,7 +226,7 @@ _config_sections() { _extensions() { if [[ -z "${SPACK_EXTENSIONS:-}" ]] then - SPACK_EXTENSIONS="aspell go-bootstrap go icedtea jdk kim-api lua matlab mofem-cephas octave openjdk perl python r ruby rust tcl yorick" + SPACK_EXTENSIONS="$(spack extensions)" fi SPACK_COMPREPLY="$SPACK_EXTENSIONS" } diff --git a/share/spack/spack-completion.bash b/share/spack/spack-completion.bash index b17733e1bf77cf0d006d3f648311650270e59f15..8f677adbf986b4fdd29dbb0b705f3b965914fd1a 100755 --- a/share/spack/spack-completion.bash +++ b/share/spack/spack-completion.bash @@ -226,7 +226,7 @@ _config_sections() { _extensions() { if [[ -z "${SPACK_EXTENSIONS:-}" ]] then - SPACK_EXTENSIONS="aspell go-bootstrap go icedtea jdk kim-api lua matlab mofem-cephas octave openjdk perl python r ruby rust tcl yorick" + SPACK_EXTENSIONS="$(spack extensions)" fi SPACK_COMPREPLY="$SPACK_EXTENSIONS" }