from typing import Optional, Tuple
import sys
import re
import subprocess
import asyncio


def get_bpftrace_version(bpftrace_path: str) -> Optional[Tuple]:
    version_str = subprocess.check_output([bpftrace_path, '--version'], encoding='utf8').strip()
    version_str = version_str[version_str.find(' ')+1:]  # remove 'bpftrace' word
    re_version = re.match(r'v(\d+)\.(\d+)\.(\d+)', version_str)
    if re_version:
        # regex enforces exactly 3 integers
        return version_str, tuple(map(int, re_version.groups()))  # (major, minor, patch)
    return version_str, (999, 999, 999)


def get_tracepoints_csv(bpftrace_path: str) -> str:
    # use a set to prevent duplicate probes
    # see https://github.com/iovisor/bpftrace/issues/1581
    tracepoints = frozenset(subprocess.check_output([bpftrace_path, '-l'], encoding='utf8').split())
    return ','.join(tracepoints)


def asyncio_get_all_tasks(loop):
    if sys.version_info >= (3, 7):
        return asyncio.all_tasks(loop)
    else:
        return asyncio.Task.all_tasks(loop) # pylint: disable=no-member