#!/usr/bin/env python

from foundation.utils import Workers
import argparse

parser = argparse.ArgumentParser(description="Start an HCI worker.")
parser.add_argument('-a', '--advertise_addr', default=None, help="Advertise address.")
args = parser.parse_args()

workers = Workers(swarm_advertise_addr=args.advertise_addr)
workers.stop_all_workers()

# # Basic services
# workers.swarm.start_ntp(restart=True)
# workers.swarm.start_jupyterlab(restart=True)
# workers.swarm.start_kafka(restart=True)
# workers.swarm.start_kafka_logs(restart=True)
# workers.swarm.start_timescaledb(restart=True)

# # Basic workers
workers.start_worker('timescaledb_api', service_name='timescaledb_api', endpoint='/timescaledbapp/', restart=True)
workers.start_worker('main_app', service_name='main_app', restart=True)

