#!/usr/bin/env python3
# Entrypoint/script to setup and start an LTTng tracing session

import argparse
import time

from tracetools_trace.tools import lttng
from tracetools_trace.tools import names


def main():
    parser = argparse.ArgumentParser(description='Setup and launch an LTTng tracing session.')
    parser.add_argument('--session-name', '-s', dest='session_name',
                        default=f'session-{time.strftime("%Y%m%d%H%M%S")}',
                        help='the name of the tracing session (default: session-YYYYMMDDHHMMSS)')
    parser.add_argument('--path', '-p', dest='path',
                        default='/tmp',
                        help='path of the base directory for trace data (default: %(default)s)')
    parser.add_argument('--ust', '-u', nargs='*', dest='events_ust', default=names.DEFAULT_EVENTS_ROS,
                        help='the ROS UST events to enable (default: all events) '
                             '[to disable all UST events, '
                             'provide this flag without any event name]')
    parser.add_argument('--kernel', '-k', nargs='*', dest='events_kernel',
                        default=names.DEFAULT_EVENTS_KERNEL,
                        help='the kernel events to enable (default: all events) '
                             '[to disable all UST events, '
                             'provide this flag without any event name]')
    parser.add_argument('--list', '-l', dest='list', action='store_true',
                        help='display lists of enabled events (default: %(default)s)')
    args = parser.parse_args()

    session_name = args.session_name
    base_path = args.path
    full_path = base_path + '/' + session_name
    ros_events = args.events_ust
    kernel_events = args.events_kernel

    ust_enabled = len(ros_events) > 0
    kernel_enabled = len(kernel_events) > 0
    if ust_enabled:
        print(f'UST tracing enabled ({len(ros_events)} events)')
        if args.list:
            print(f'\tevents: {ros_events}')
    else:
        print('UST tracing disabled')
    if kernel_enabled:
        print(f'kernel tracing enabled ({len(kernel_events)} events)')
        if args.list:
            print(f'\tevents: {kernel_events}')
    else:
        print('kernel tracing disabled')

    lttng.lttng_setup(session_name, full_path, ros_events=ros_events, kernel_events=kernel_events)
    print(f'writting tracing session to: {full_path}')

    input('press enter to start...')
    lttng.lttng_start(session_name)
    input('press enter to stop...')

    print('stopping & destroying tracing session')
    lttng.lttng_stop(session_name)
    lttng.lttng_destroy(session_name)