#! /usr/bin/env python

import argparse
from colabcode import ColabCode

if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description="ColabCode: Run VS Code On Colab / Kaggle Notebooks"
    )
    parser._action_groups.pop()
    required = parser.add_argument_group("required arguments")
    optional = parser.add_argument_group("optional arguments")

    required.add_argument(
        "--port",
        type=int,
        help="the port you want to run code-server on",
        required=True,
    )
    optional.add_argument(
        "--password",
        type=str,
        help="password to protect your code-server from unauthorized access",
        default=None,
    )
    optional.add_argument(
        "--mount_drive",
        action="store_true",
        help="if you use --mount_drive, your google drive will be mounted",
    )

    args = parser.parse_args()

    ColabCode(port=args.port, password=args.password, mount_drive=args.mount_drive)
