from playwright.sync_api import sync_playwright
from playwright.sync_api import Playwright
from playwright.sync_api import Page
import time
import csv
import email
import ssl
import imaplib
from email.header import decode_header, make_header
import re

def mail_receive():
    """
    (1) IMAP4クライアントインスタンスを作成する
    * Gmail, Yahoo!メールともに nego_combo = ("ssl", 993)
    - 143ポートが暗号化なし
    - 暗号化なしとSTARTTLSは動作確認できるサーバがなかった（コードが正しいかも不明）
    """
    host = "imap.gmail.com"
    nego_combo = ("ssl", 993) # ("通信方式", port番号)

    if nego_combo[0] == "no-encrypt":
        imapclient = imaplib.IMAP4(host, nego_combo[1])
    elif nego_combo[0] == "starttls":
        context = ssl.create_default_context()
        imapclient = imaplib.IMAP4(host, nego_combo[1])
        imapclient.starttls(ssl_context=context)
    elif nego_combo[0] == "ssl":
        context = ssl.create_default_context()
        imapclient = imaplib.IMAP4_SSL(host, nego_combo[1], ssl_context=context)
    imapclient.debug = 3  # 各命令をトレースする

    """
    (2) IMAP4サーバにログインする
    """
    username = "XXXXXXX@gmail.com"
    password = "XXXXXXXXX"
    imapclient.login(username, password)


    """
    (3) メール受信（MIMEメッセージを受信）
    """
    imapclient.select() # メールボックスの選択
    typ, data = imapclient.search(None, "ALL")  # data = [b"1 2 3 4 ..."]
    datas = data[0].split()
    fetch_num = 1  # 取得したいメッセージの数
    if (len(datas)-fetch_num) < 0:
        fetch_num = len(datas)
    msg_list = []  # 取得したMIMEメッセージを格納するリスト
    for num in datas[len(datas)-fetch_num::]:
        typ, data = imapclient.fetch(num, '(RFC822)')
        msg = email.message_from_bytes(data[0][1])
        msg_list.append(msg)
    imapclient.close()
    imapclient.logout()


    """
    受信したメール（MIMEメッセージ）の確認
    """
    for msg in msg_list:
        # 各ヘッダ情報はディクショナリのようにアクセスできる
        from_addr = str(make_header(decode_header(msg["From"])))
        subject = str(make_header(decode_header(msg["Subject"])))

        # 本文(payload)を取得する
        for part in msg.walk():
            payload = part.get_payload(decode=True)
            if payload is None:
                continue
            charset = part.get_content_charset()
            if charset is not None:
                payload = payload.decode(charset, "ignore")

            # 括弧 () 内の数字を抽出する正規表現パターン
            pattern = r'\b(\d{6})\b'

            # # payloadから数字を抽出
            # 
            extracted_numbers = re.search(pattern, payload)


            # 抽出された数字を表示
            #for number in extracted_numbers:
                #print("抽出された数字:", number)
            #print(payload)
    return extracted_numbers[0]

profile_directory = "C:/Users/kuron/AppData/Local/Google/Chrome/User Data/postionYou/04new-window" 
with sync_playwright() as p:
        browser = p.chromium.launch_persistent_context(
            profile_directory,
            headless=False,
            channel="chrome",
            args=["--blink-settings=imagesEnabled=false"]
        )

        page = browser.new_page()
        for n in range(151,200):
            print(str(n) + "を始めます")
            page.goto("https://xion.bonusblock.io?r=WVMSdEz5")

            # メール登録
            page.get_by_text("CONNECT").click()
            page.get_by_placeholder("example@email.com").click()
            page.get_by_placeholder("example@email.com").fill("XXXXXXXXXX+" + str(n) + "@gmail.com")
            page.get_by_role("button", name="Continue with email").click()
            time.sleep(20)
            word = mail_receive()
            page.get_by_label("One-time passcode").click()
            page.get_by_label("One-time passcode").fill(word)
            page.get_by_text("Create a new account").click()
            time.sleep(30)
            page.locator("div").filter(has_text=re.compile(r"Personal Account")).nth(5).click()
            time.sleep(20)

            # 連携
            page.locator("#app div").filter(has_text=re.compile(r"^Connect Twitter$")).nth(3).click()
            page.get_by_role("button", name="連携アプリを認証").click()
            page.get_by_role("complementary").get_by_text("Connect Discord").click()
            page.get_by_role("button", name="認証").click()
            # ミント
            page.get_by_role("button", name="Mint \"Welcome\" NFT").click()
            page.get_by_role("button", name="Mint Badge").click()
            # Unlink
            page.locator(".el-box > .el-row").first.click()
            page.get_by_role("button", name="Unlink").click()
            page.locator(".el-box > .el-row").nth(1).click()
            page.get_by_role("button", name="Unlink").click()
            #ログアウト
            time.sleep(5)
            page.locator('rect').first.hover()
            page.locator('li.el-menu-item:has-text("Logout")').first.click()
            page.get_by_role("button", name="Log out").click()
            time.sleep(5)