import socket
import ssl
import random
import time
from yt_dlp import YoutubeDL


# TODO eradicate global variables
# TODO add SSL support

reconnect = 0
text_str = 0
hostname = "irc.libera.chat"
channel = "##tech-tangents"
botnick = "notcqst"


context = ssl.create_default_context()

user_str = ("USER " + botnick + " " + botnick + " " + botnick + " :" + botnick + "\n")
nick_str = ("NICK " + botnick + "\n")
channel_string = ("JOIN " + channel + "\n")
send_to_channel_str = ("PRIVMSG " + channel + " :Hello " + "\r\n")


def ping(irc):
    if "PING" in text_str:
        split_str = text_str.split(" ")
        print(split_str)
        print(split_str)
        print(split_str)
        print(split_str)
        print(split_str)
        print("PONG\n")
        irc.send((bytes("PONG :" + split_str[1] + "\r\n", 'utf-8')))


def check_youtube(irc):
    if "https://youtube.com" in text_str or "https://www.youtube.com" in text_str:
        split_str = text_str.split('https://')
        vid_url_www = split_str[1].strip()
        vid_url = ''.join(("https://", vid_url_www))
        with YoutubeDL() as ydl:
            info_dict = ydl.extract_info(vid_url, download=False)
            video_url = info_dict.get("url", None)
            video_id = info_dict.get("id", None)
            video_title = info_dict.get('title', None)
            channel_name = info_dict.get('channel', None)
            irc.send((bytes("PRIVMSG " + channel + " :" + channel_name + ": " + video_title + "\r\n", 'utf-8')))
            return 1
    else:
        return 0


def command_choose(split_spaces, irc):
    choices = split_spaces[2:]
    decision = random.choice(choices)
    irc.send((bytes("PRIVMSG " + channel + " :" + decision + "\r\n", 'utf-8')))


def command_hello(irc):
    irc.send((bytes("PRIVMSG " + channel + " :" + "Hello, " + "\r\n", 'utf-8')))


def unknown_command(irc):
    irc.send((bytes("PRIVMSG " + channel + " :" + "Sorry I don't understand." + "\r\n", 'utf-8')))


def command_flag(irc):
    if ":n!" in text_str:
        split_text = text_str.split(':n!')
        split_spaces = split_text[1].split(' ')
        command = split_spaces[1].strip()
        match command:
            case "choose": command_choose(split_spaces, irc)
            case "hello": command_hello()
            case _: unknown_command()


def parse_cmd(irc):
    ping(irc)
    command_flag(irc)
    check_youtube(irc)


def connect_checked():
    if reconnect >= 3:
        exit(1)
    else:
        # irc.connect((hostname, 6697))
        with socket.create_connection((hostname, 6697)) as sock:
            with context.wrap_socket(sock, server_hostname=hostname) as irc:
                irc.send(bytes(user_str, 'utf-8'))
                irc.send(bytes(nick_str, 'utf-8'))
                irc.send(bytes(channel_string, 'utf-8'))
                irc.send(bytes(send_to_channel_str, 'utf-8'))
                connect_loop(irc)


def connect_loop(irc):
    while 1:
        text = irc.recv(2040)
        if not text:
            break
        print(text)
        global text_str
        text_str = text.decode("utf-8")
        parse_cmd(irc)
    irc.close()
    global reconnect
    reconnect += 1
    time.sleep(10)
    connect_checked()


print("connecting to: " + hostname)
connect_checked()