#!/usr/bin/env python3
import argparse

from iottb.subcommands.capture import setup_capture_parser
from iottb.subcommands.add_device import setup_init_device_root_parser
from iottb.utils.tcpdump_utils import list_interfaces

######################
# Argparse setup
######################
def setup_argparse():
    # create top level parser
    root_parser = argparse.ArgumentParser(prog='iottb')
    subparsers = root_parser.add_subparsers(title='subcommands', required=True, dest='command')

    # shared options
    root_parser.add_argument('--verbose', '-v', action='count', default=0)
    # configure subcommands
    setup_capture_parser(subparsers)
    setup_init_device_root_parser(subparsers)

    # Utility to list interfaces directly with iottb instead of relying on external tooling

    interfaces_parser = subparsers.add_parser('list-interfaces', aliases=['li', 'if'])
    interfaces_parser.set_defaults(func=list_interfaces)

    return root_parser


def main():
    parser = setup_argparse()
    args = parser.parse_args()
    print(args)
    if args.command:
        try:
            args.func(args)
        except KeyboardInterrupt:
            print('Received keyboard interrupt. Exiting...')
            exit(1)
        except Exception as e:
            print(f'Error: {e}')
    # create_capture_directory(args.device_name)


if __name__ == '__main__':
    main()