from dataclasses import dataclass
import os
import sys
import threading
import time

import KSR as KSR

# global variables corresponding to defined values (e.g., flags) in kamailio.cfg
FLT_ACC = 1
FLT_ACCMISSED = 2
FLT_ACCFAILED = 3
FLT_NATS = 5

FLB_NATB = 6
FLB_NATSIPPING = 7


@dataclass
class Script:
    rank: int
    pid: int


def child_task(kamailio: Script):
    KSR.info(f"===== child task thread {kamailio.pid}\n")
    ## asyncio.run(child_task_async(kamailio))
    count = 0
    while True:
        KSR.info(f"===== child task thread {kamailio.pid}:{count}\n")
        time.sleep(10.0)
        count += 1


def ksr_script_child_init(rank: int):
    if rank <= 0:
        return 1
    kamailio = Script(rank, os.getpid())


    flags = os.environ.get("TEST_THREADS", False)
    if flags:
        KSR.info("===== testing threads\n")
        child_thread = threading.Thread(target=child_task, args=[kamailio], daemon=True)
        child_thread.start()
        KSR.info("===== thread started\n")
    KSR.info("===== kamailio.child_init(%d:%d)\n" % (rank, kamailio.pid))
    return 1

# etc etc