Module materialize.cli.scratch.ssh

# Copyright Materialize, Inc. and contributors. All rights reserved.
#
# Use of this software is governed by the Business Source License
# included in the LICENSE file at the root of this repository.
#
# As of the Change Date specified in that file, in accordance with
# the Business Source License, use of this software will be governed
# by the Apache License, Version 2.0.

import argparse

import boto3

from materialize.cli.scratch import check_required_vars
from materialize.scratch import mssh


def configure_parser(parser: argparse.ArgumentParser) -> None:
    check_required_vars()

    parser.add_argument("instance", help="The ID of the instance to connect to")
    parser.add_argument("command", nargs="*", help="The command to run via SSH, if any")


def run(args: argparse.Namespace) -> None:
    # SSH will join together multiple arguments with spaces, so we don't lose
    # anything by doing the same.
    instance = boto3.resource("ec2").Instance(args.instance)
    mssh(instance, " ".join(args.command))