#!/usr/bin/env python3

""" Tool to factory reset a Crownstone. """

import asyncio
import logging
from os import path

from bluenet_logs import BluenetLogs
from crownstone_ble import CrownstoneBle, BleEventBus, BleTopics
from crownstone_core.util.JsonFileStore import JsonFileStore
from crownstone_uart import CrownstoneUart

from tools.util.config import getToolConfig, loadKeysFromConfig, setupDefaultCommandLineArguments, macFilterPassed

tool_version = "1.0.0"

argParser = setupDefaultCommandLineArguments('Setup a Crownstone.')
argParser.add_argument('-a', '--bleAddress',
                       default=None,
                       help='To upload via BLE: the MAC address/handle of the Crownstone you want to connect to. Example: 11:22:33:AA:BB:CC')
argParser.add_argument('-d', '--device',
                       default=None,
                       help='The UART device of the Crownstone you want to connect to. Example: /dev/ttyACM0')
argParser.add_argument('--verbose', '-v',
                       dest="verbose",
                       action='store_true',
                       help='Show verbose output')
argParser.add_argument('--logStringsFile', '-l',
                       default="",
                       dest='logStringsFileName',
                       metavar='path',
                       type=str,
                       help='The path of the file with the extracted firmware logs on your system.')

try:
    file_path = path.dirname(path.realpath(__file__))
    [tool_config, args] = getToolConfig(file_path, argParser)
except Exception as e:
    print("ERROR", e)
    quit()

if args.verbose:
    logging.basicConfig(format='%(asctime)s %(levelname)-7s: %(message)s', level=logging.DEBUG)

core = None
bluenetLogs = None
useBle = False

if args.bleAddress is not None:
    # Use BLE for communication.
    useBle = True

    # Create the library instance.
    print(f'Initializing tool with bleAdapterAddress={tool_config["bleAdapterAddress"]}')
    core = CrownstoneBle(bleAdapterAddress=tool_config["bleAdapterAddress"])

    # Load the encryption keys into the library.
    loadKeysFromConfig(core, tool_config)

else:
    # Use UART for communication.

    # Create the library instance.
    print(f'Initializing tool with device={args.device}')
    core = CrownstoneUart()
    core.initialize_usb_sync(port=args.device)

    if args.logStringsFileName:
        bluenetLogs = BluenetLogs()
        bluenetLogs.setLogStringsFile(args.logStringsFileName)

async def main():

    if useBle:
        await core.connect(args.bleAddress)
        await core.control.commandFactoryReset()
        await core.disconnect()
        await core.shutDown()
    else:
        await core.control.factoryReset()
    print("Done!")

try:
    # asyncio.run does not work here.
    loop = asyncio.get_event_loop()
    loop.run_until_complete(main())
except KeyboardInterrupt:
    print("Stopping")

if not useBle:
    core.stop()
