import argparse
import logging
import random

from config import build_sender_dependencies
from model import DrillingRealtimeData


logger = logging.getLogger(__name__)


def rand_int(a, b):
    return random.randint(a, b)


def rand_float(a, b, digits=2):
    return round(random.uniform(a, b), digits)


def build_random_payload(equipment_code):
    entity = DrillingRealtimeData.empty(wellid=equipment_code)
    entity = DrillingRealtimeData(
        ts=entity.ts,
        wellid=entity.wellid,
        stknum=rand_int(0, 500),
        recid=rand_int(0, 100000),
        seqid=rand_int(0, 100000),
        actual_date=entity.actual_date,
        actual_time=entity.actual_time,
        actcod=rand_int(0, 9),
        deptbitm=rand_float(0, 5000),
        deptbitv=rand_float(0, 5000),
        deptmeas=rand_float(0, 5000),
        deptvert=rand_float(0, 5000),
        blkpos=rand_float(0, 100),
        ropa=rand_float(0, 200),
        hkla=rand_float(0, 500),
        hklx=rand_float(0, 500),
        woba=rand_float(0, 200),
        wobx=rand_float(0, 200),
        torqa=rand_float(0, 200),
        torqx=rand_float(0, 200),
        rpma=rand_int(0, 300),
        sppa=rand_float(0, 5000),
        chkp=rand_float(0, 5000),
        spm1=rand_int(0, 200),
        spm2=rand_int(0, 200),
        spm3=rand_int(0, 200),
        tvolact=rand_float(0, 20000),
        tvolcact=rand_float(0, 20000),
        mfop=rand_int(0, 1000),
        mfoa=rand_float(0, 1000),
        mfia=rand_float(0, 1000),
        mdoa=rand_float(0, 1000),
        mdia=rand_float(0, 1000),
        mtoa=rand_float(0, 1000),
        mtia=rand_float(0, 1000),
        mcoa=rand_float(0, 1000),
        mcia=rand_float(0, 1000),
        stkc=rand_int(0, 200),
        lagstks=rand_int(0, 200),
        deptretm=rand_float(0, 5000),
        gasa=rand_float(0, 100),
        space1=rand_float(0, 10),
        space2=rand_float(0, 10),
        space3=rand_float(0, 10),
        space4=rand_float(0, 10),
        space5=rand_float(0, 10),
    )
    return entity.to_payload(equipment_code)


def run_sender(args, deps):
    mqtt_config = deps.config.mqtt
    logger.info(
        "mqtt_sender is disabled by current requirements; skipping MQTT publish logic for topic=%s client_id=%s interval=%ss count=%s",
        mqtt_config.pub_topic,
        mqtt_config.sender_client_id,
        args.interval,
        args.count or "forever",
    )



def add_arguments(parser):
    parser.add_argument("--config", default="config.yaml", help="Path to config yaml")
    parser.add_argument("--interval", type=float, default=3.0, help="Publish interval (seconds)")
    parser.add_argument("--count", type=int, default=0, help="Publish count (0 = forever)")



def main(argv=None):
    parser = argparse.ArgumentParser(description="MQTT random data sender")
    add_arguments(parser)
    args = parser.parse_args(argv)
    deps = build_sender_dependencies(args.config)
    run_sender(args, deps)


if __name__ == "__main__":
    main()