| """AstrBot CLI entry point""" |
|
|
| import sys |
|
|
| import click |
|
|
| from . import __version__ |
| from .commands import conf, init, plug, run |
|
|
| logo_tmpl = r""" |
| ___ _______.___________..______ .______ ______ .___________. |
| / \ / | || _ \ | _ \ / __ \ | | |
| / ^ \ | (----`---| |----`| |_) | | |_) | | | | | `---| |----` |
| / /_\ \ \ \ | | | / | _ < | | | | | | |
| / _____ \ .----) | | | | |\ \----.| |_) | | `--' | | | |
| /__/ \__\ |_______/ |__| | _| `._____||______/ \______/ |__| |
| """ |
|
|
|
|
| @click.group() |
| @click.version_option(__version__, prog_name="AstrBot") |
| def cli() -> None: |
| """The AstrBot CLI""" |
| click.echo(logo_tmpl) |
| click.echo("Welcome to AstrBot CLI!") |
| click.echo(f"AstrBot CLI version: {__version__}") |
|
|
|
|
| @click.command() |
| @click.argument("command_name", required=False, type=str) |
| def help(command_name: str | None) -> None: |
| """Display help information for commands |
| |
| If COMMAND_NAME is provided, display detailed help for that command. |
| Otherwise, display general help information. |
| """ |
| ctx = click.get_current_context() |
| if command_name: |
| |
| command = cli.get_command(ctx, command_name) |
| if command: |
| |
| click.echo(command.get_help(ctx)) |
| else: |
| click.echo(f"Unknown command: {command_name}") |
| sys.exit(1) |
| else: |
| |
| click.echo(cli.get_help(ctx)) |
|
|
|
|
| cli.add_command(init) |
| cli.add_command(run) |
| cli.add_command(help) |
| cli.add_command(plug) |
| cli.add_command(conf) |
|
|
| if __name__ == "__main__": |
| cli() |
|
|
