import argparse
import os, sys

from install import install
from publish import publish
from remove import remove
from list import list_cmd
from uninstall import uninstall


def parse_arguments():
    # create our subcommands and args
    arg_parser = argparse.ArgumentParser(prog="Digpkg", description="The package manager for the Ground programming language.")
    sub_parsers = arg_parser.add_subparsers(dest="command")
    
    # install command
    install_command = sub_parsers.add_parser(name="install", description="install a mineral")
    install_command.add_argument("name", help="name of the mineral to install")
    install_command.add_argument("--max-retries", help="max number of download retries before giving up", default=3, type=int)
    
    # uninstall command
    uninstall_command = sub_parsers.add_parser(name="uninstall", description="uninstall a mineral")
    uninstall_command.add_argument("name", help="name of the mineral to uninstall")
    
    # list command
    list_command = sub_parsers.add_parser(name="list", description="list all minerals installed in the current environment")
    
    # publish command
    publish_command = sub_parsers.add_parser(name="publish", description="publish a package to the repository")
    publish_command.add_argument("name", help="name and version of the package")
    publish_command.add_argument("folder_path", help="path to the folder that will be uploaded")
    
    # remove command
    remove_command = sub_parsers.add_parser(name="remove", description="remove a published package from the repository")
    remove_command.add_argument("name", help="name and version of the package")
    
    # env command
    
    args = arg_parser.parse_args()
    
    if not args.command:
        arg_parser.print_help()
        sys.exit(0)
        
    if args.command == "install":
        install(args)
    elif args.command == "publish":
        publish(args)
    elif args.command == "remove":
        remove(args)
    elif args.command == "list":
        list_cmd(args)
    elif args.command == "uninstall":
        uninstall(args)

def main():
    parse_arguments()


if __name__ == "__main__":
    main()