import socket
import random
import time
from yt_dlp import YoutubeDL

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

irc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
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 check_youtube():
    if "https://youtube.com" in text_str or "https://www.youtube.com" in text_str:
        split_str = text_str.split('https://')
        vid_url = split_str[1].strip()
        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):
    print(split_spaces)
    arr_hack = len(split_spaces)
    arr_hack -= 2
    choices = []
    for x in range(arr_hack):
        choices.append(split_spaces[x+2])
        print(choices)
    decision = random.choice(choices)
    irc.send((bytes("PRIVMSG " + channel + " :" + decision + "\r\n", 'utf-8')))


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


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


def command_flag():
    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)
            case "hello": command_hello()
            case _: unknown_command()


def parse_cmd():
    command_flag()
    check_youtube()


def connect_checked():
    if reconnect >= 3:
        exit(1)
    else:
        irc.connect((server, 6667))
        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()


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


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