import os
import pytchat
from colorama import Fore
from DrissionPage import WebPage, ChromiumOptions

options = ChromiumOptions()
options.set_argument('--headless')

stream_id = ""

def stream_url_pars():
    global stream_id
   
    try:
        page = WebPage(mode='d',chromium_options=options)

        url = 'https://www.youtube.com/@qqchieftain'
        page.get(url)

        links = page.eles('tag:a')

        for link in links:
            href = link.attr('href')
            if '/watch?v=' in str(href):
                print(f"Stream last: {href}")
                stream_link = href
                break

        stream_link = stream_link.split("=")
        print(f"Stream-ID: {stream_link[1]}")
        main(stream_link[1])

        page.quit()

    except Exception as e:
        print(e)

def copy_text(text):
    command = 'echo ' + str(text).strip() + '| clip'
    os.system(command)
    print("Ключ скопирован")

def main(stream_id):
    try:
        chat = pytchat.create(video_id=stream_id)
        print("Подлюченно к чату")
        while chat.is_alive():
            for messages in chat.get().sync_items():
                if "chieftain.cc" in messages.message:
                    print(Fore.RED + f"Ключ извлечён: {c.message}")
                    key = str(c.message).split(" ")
                    copy_text(key[0])

                else:
                    continue
    except Exception as r:
        print(r)

if __name__ == '__main__':
    stream_url_pars()
    main(stream_id)