import time
from playwright.sync_api import sync_playwright

with sync_playwright() as p:
    browser = p.chromium.connect_over_cdp('http://localhost:9222')
    page = browser.contexts[0].pages[0]
    
    print('Joining targeted California & USA student groups...')
    join_btns = page.locator("div[aria-label*='Join'], div[role='button']:has-text('Join')").all()
    print('Found Join buttons:', len(join_btns))
    
    # Group 2: Hội du học sinh Việt Nam tại bang California (index 1)
    if len(join_btns) >= 2:
        print('Joining Hội du học sinh Việt Nam tại bang California...')
        join_btns[1].click()
        page.wait_for_timeout(3000)
        
    # Group 4: (USA) HỘI DU HỌC SINH VIỆT NAM TẠI MỸ (index 3)
    if len(join_btns) >= 4:
        print('Joining (USA) HỘI DU HỌC SINH VIỆT NAM TẠI MỸ...')
        join_btns[3].click()
        page.wait_for_timeout(3000)
        
    page.screenshot(path=r'C:\Projects\Historical\FacebookAutoPoster\groups_joined.png')
    print('Groups joined screenshot saved!')
    browser.close()