bootstrap.py

# -*- coding: utf-8 -*-
# Copyright 2024 Rapyuta Robotics
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

__version__ = "9.5.0"

import os

import click
import rapyuta_io.version
from click import Context
from click_help_colors import HelpColorsGroup
from click_plugins import with_plugins
from pkg_resources import iter_entry_points

from riocli.apply import apply, delete
from riocli.apply.explain import list_examples, explain
from riocli.apply.template import template
from riocli.auth import auth
from riocli.chart import chart
from riocli.completion import completion
from riocli.config import Configuration
from riocli.config.context import cli_context
from riocli.configtree import config_trees
from riocli.constants import Colors, Symbols
from riocli.deployment import deployment
from riocli.device import device
from riocli.disk import disk
from riocli.hwil import hwildevice
from riocli.managedservice import managedservice
from riocli.network import network
from riocli.oauth2 import oauth2
from riocli.organization import organization
from riocli.package import package
from riocli.parameter import parameter
from riocli.project import project
from riocli.rosbag import rosbag
from riocli.secret import secret
from riocli.shell import deprecated_repl, shell
from riocli.static_route import static_route
from riocli.usergroup import usergroup
from riocli.utils import (
    check_for_updates,
    is_pip_installation,
    pip_install_cli,
    update_appimage,
)
from riocli.vpn import vpn


@with_plugins(iter_entry_points("riocli.plugins"))
@click.group(
    invoke_without_command=False,
    cls=HelpColorsGroup,
    help_headers_color=Colors.YELLOW,
    help_options_color=Colors.GREEN,
)
@click.pass_context
def cli(ctx: Context, config: str = None):
    """Manage rapyuta.io features on the command-line"""
    ctx.obj = Configuration(filepath=config)


def safe_cli():
    if os.environ.get("DEBUG", "false").lower() != "true":
        try:
            cli()
        except Exception as e:
            click.secho(str(e), fg=Colors.RED)
            raise SystemExit(1) from e
    else:
        cli()


@cli.command("help")
@click.pass_context
def cli_help(ctx):
    """Print the help message."""
    click.echo(cli.get_help(ctx))


@cli.command()
def version():
    """View installed CLI and SDK versions."""
    click.echo(f"rio {__version__} / SDK {rapyuta_io.__version__}")


@cli.command("update")
@click.option(
    "-f",
    "--force",
    "--silent",
    "silent",
    is_flag=True,
    type=click.BOOL,
    default=False,
    help="Skip confirmation",
)
def update(silent: bool) -> None:
    """Update the CLI to the latest version.

    You can update your existing installation of the CLI to
    its latest version. Based on the installation method, i.e.
    pip or AppImage, the command will update the right
    installation.

    You can skip the confirmation prompt by using the --silent or
    --force or -f flag.
    """
    available, latest = check_for_updates(__version__)
    if not available:
        click.secho("🎉 You are using the latest version", fg=Colors.GREEN)
        return

    click.secho("🎉 A newer version ({}) is available.".format(latest), fg=Colors.GREEN)

    if not silent:
        click.confirm("Do you want to update?", abort=True, default=False)

    try:
        if is_pip_installation():
            pip_install_cli(version=latest)
        else:
            update_appimage(version=latest)
    except Exception as e:
        click.secho("{} Failed to update: {}".format(Symbols.ERROR, e), fg=Colors.RED)
        raise SystemExit(1) from e

    click.secho("{} Update successful!".format(Symbols.SUCCESS), fg=Colors.GREEN)


cli.add_command(apply)
cli.add_command(chart)
cli.add_command(explain)
cli.add_command(list_examples)
cli.add_command(delete)
cli.add_command(auth)
cli.add_command(project)
cli.add_command(device)
cli.add_command(secret)
cli.add_command(package)
cli.add_command(deployment)
cli.add_command(static_route)
cli.add_command(rosbag)
cli.add_command(network)
cli.add_command(completion)
cli.add_command(parameter)
cli.add_command(disk)
cli.add_command(shell)
cli.add_command(deprecated_repl)
cli.add_command(managedservice)
cli.add_command(template)
cli.add_command(organization)
cli.add_command(vpn)
cli.add_command(usergroup)
cli.add_command(config_trees)
cli.add_command(hwildevice)
cli.add_command(cli_context)
cli.add_command(oauth2)