from . import PlayerModule, log95, Track
import socket, re

DEBUG = False

name_table_path = "/home/user/mixes/name_table.txt"

rds_base = "Gramy: {} - {}"
rds_default_artist = "radio95"

udp_host = ("127.0.0.1", 5000)

from typing import TextIO
_log_out: TextIO

logger_level = log95.log95Levels.DEBUG if DEBUG else log95.log95Levels.CRITICAL_ERROR
assert _log_out # pyright: ignore[reportUnboundVariable]
logger = log95.log95("RDS-MODULE", logger_level, output=_log_out)

def load_dict_from_custom_format(file_path: str) -> dict[str, str]:
    try:
        result_dict = {}
        with open(file_path, 'r') as file:
            for line in file:
                if line.strip() == "" or line.startswith(";"): continue
                key, value = line.split(':', 1)
                result_dict[key.strip()] = value.strip()
        return result_dict
    except FileNotFoundError:
        logger.error(f"{name_table_path} does not exist, or could not be accesed")
        return {}

def update_rds(track_name: str):
    name_table = load_dict_from_custom_format(name_table_path)
    try:
        name = name_table[track_name]
        has_name = True
    except KeyError:
        has_name = False
        name = track_name.rsplit(".", 1)[0]

#    name = re.sub(r'^\s*\d+\s*[-.]?\s*', '', name)

    if " - " in name:
        count = name.count(" - ")
        while count != 1: # youtube reuploads, to avoid things like ilikedick123 - Micheal Jackson - Smooth Criminal
            name = name.split(" - ", 1)[1]
            count = name.count(" - ")
        artist = name.split(" - ", 1)[0]
        title = name.split(" - ", 1)[1]
    else:
        artist = rds_default_artist
        title = name
        if not has_name: logger.warning(f"File does not have a alias in the name table ({track_name})")

#    title = re.sub(r'\s*[\(\[][^\(\)\[\]]*[\)\]]', '', title) # there might be junk

    prt = rds_base.format(artist, title)[:64]
    rtp = []
    rtp.append(1) # type 2
    rtp.append(prt.find(title)) # start 2
    rtp.append(len(title) - 1) # len 2
    rtp.append(4) # type 1
    rtp.append(prt.find(artist)) # start 1
    rtp.append(len(artist) - 1) # len 1

    rtp = [j_size if i_rt > j_size else i_rt for i_rt,j_size in zip(rtp, [255,0x3f,0x3f,255,0x3f,0x1f])]

    rtp = ','.join(list(map(str, rtp)))

    try:
        f = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        f.settimeout(1.0)
        data = f"TEXT={prt}\r\nRTP={rtp}\r\n".encode()
        f.sendto(data, udp_host)
        logger.debug("Sending", str(data))
        f.close()
    except Exception as e: logger.error(f"Error updating RDS: {e}")

    return prt, rtp

class Module(PlayerModule):
    def on_new_track(self, index: int, track: Track, next_track: Track | None):
        if track.official:
            rds_rt, rds_rtp = update_rds(track.path.name)
            logger.info(f"RT set to '{rds_rt}'")
            logger.debug(f"{rds_rtp=}")

module = Module()