Grouped `--help` output is not shown for `__main__.py` CLIs (`python -m foo`) / add support for `prog_name`

Given a basic package called foo, with:

foo/_cli.py

from rich_click.utils import CommandGroupDict
import rich_click as click

COMMAND_GROUPS = dict(
    foo=[
        CommandGroupDict(
            name="Group 1",
            commands=["bar"],
        ),
        CommandGroupDict(
            name="Group 2",
            commands=["baz", "quux"],
        ),
    ],
)

@click.rich_config(
    help_config=click.RichHelpConfiguration(command_groups=COMMAND_GROUPS),
)
@click.group(context_settings=dict(help_option_names=["--help", "-h"]))
def main():
    pass

@main.command()
def bar():
    pass

@main.command()
def baz():
    pass

`foo/main.py`

from foo import _cli
_cli.main

pyproject.toml

[build-system]
requires = ["hatchling", "hatch-vcs"]
build-backend = "hatchling.build"

[project]
name = "foo"
version = "0.1.0"
description = ""
requires-python = ">=3.10"
readme = "README.rst"
license = {text = "MIT"}
keywords = []
authors = [{name = "Julian Berman"}]
dependencies = ["rich-click"]

[project.scripts]
foo = "foo._cli:main"

i.e. a __main__.py for python -m foo, the results of python -m foo --help does not contain the command groups, even though foo --help will:

~/Desktop/foo is a git repository on main 
⊙  venv/bin/foo --help                                                                                                                                                                                                                                                        julian@Airm ●

 Usage: foo [OPTIONS] COMMAND [ARGS]...                                                                                                                                                                                                                                                     

╭─ Options ────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ --help  -h    Show this message and exit.                                                                                                                                                                                                                                                │
╰──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Group 1 ────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ bar                                                                                                                                                                                                                                                                                      │
╰──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Group 2 ────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ baz                                                                                                                                                                                                                                                                                      │
│ quux                                                                                                                                                                                                                                                                                     │
╰──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯

~/Desktop/foo is a git repository on main 
⊙  venv/bin/python -m foo --help                                                                                                                                                                                                                                              julian@Airm ●

 Usage: python -m foo [OPTIONS] COMMAND [ARGS]...                                                                                                                                                                                                                                           

╭─ Options ────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ --help  -h    Show this message and exit.                                                                                                                                                                                                                                                │
╰──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Commands ───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ bar                                                                                                                                                                                                                                                                                      │
│ baz                                                                                                                                                                                                                                                                                      │
│ quux                                                                                                                                                                                                                                                                                     │
╰──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯

In "native" click, the prog_name option exists in a few places to explicitly say what the program name is -- in my personal experience I always set this, as otherwise it's simply incorrect when the program is used via python -m. Perhaps something like that is needed here as well in order to tell rich-click that when I say foo in COMMAND_GROUPS that we still are talking about the same program?

Not sure if there's another obvious solution here. Thanks again!

ewels / rich-click