rust_drission 0.2.5

//! Boss自动打招呼 简易版

use std::time::Duration;

use rust_drission::{utils::sleep_random_ms, BrowserConfig, CdpError, ChromiumPage};

fn main() -> Result<(), CdpError> {
    // 跨平台用户数据目录（Windows: C:\Users\xxx\..., macOS: /Users/xxx/..., Linux: /home/xxx/...）
    let user_data_dir = std::env::temp_dir().join("drission_boss_userdata");
    let user_data_dir = user_data_dir.to_string_lossy().to_string();

    let config = BrowserConfig::new()
        .user_data_dir(user_data_dir)
        .headless(false);
    let mut page = ChromiumPage::new(config)?;

    // 进入到牛人页
    page.get("https://www.zhipin.com/web/chat/recommend")?;

    let card_selector = "#recommend-list .card-item";

    page.wait(card_selector, Duration::from_secs(30))?;

    // 滚动
    const RECOMMEND_FRAME_LOCATOR: &str = "css:iframe[name=recommendFrame]";
    // 记录已经打过招呼的
    let mut sayed: Vec<String> = vec![];
    // 打招呼人数
    let say_limit = 20;
    // 已打招呼的
    let mut sayed_count = 0;

    while sayed_count < say_limit {
        // 牛人卡片 并排除 含  .similar-geek-wrap 的卡片
        let card_item_eles = page.eles(card_selector)?;
        let mut bk = false;
        println!("当前牛人卡片数量: {}", card_item_eles.len());
        for i in 0..card_item_eles.len() {
            let card_item_ele = &card_item_eles[i];

            let similar_ele = card_item_ele.element(".similar-geek-wrap")?;
            if similar_ele.is_some() {
                continue;
            }
            if say_limit - sayed_count <= 0 {
                break;
            }
            if check_vip2(&page)? {
                println!("检测到充值vip弹窗，停止打招呼");
                bk = true;
                break;
            }
            // 获取牛人筛选配置 用于筛选打招呼的牛人
            // 获取牛人id
            let card_inner_ele = card_item_ele.element(".card-inner")?;
            if card_inner_ele.is_none() {
                continue;
            }
            // 已经打过招呼的 就跳过
            let geek_id = card_inner_ele.unwrap().attr("data-geekid")?;
            if sayed.contains(&geek_id) {
                continue;
            }
            // 获取牛人 期望的岗位
            let expect_job_ele = card_item_ele.element(".expect-wrap")?;
            if expect_job_ele.is_none() {
                continue;
            }
            let expect_text = expect_job_ele.unwrap().text()?;

            // 获取牛人信息
            let boss_info_ele = card_item_ele.element(".join-text-wrap.base-info")?;
            if boss_info_ele.is_none() {
                continue;
            }
            let boss_info_text = boss_info_ele.unwrap().text()?;

            // 打招呼按钮
            let say_hello_btn_ele = card_item_ele.element(".btn.btn-greet")?;
            if say_hello_btn_ele.is_none() {
                continue;
            }

            // 点击打招呼
            // say_hello_btn_ele.unwrap().click()?;
            sayed_count += 1;
            sayed.push(geek_id.clone());
            println!(
                "牛人id: {}, 期望岗位: {}, 牛人信息: {}",
                geek_id.clone(),
                expect_text,
                boss_info_text
            );

            println!("打招呼进度: {}/{}", sayed_count, say_limit);

            sleep_random_ms(800, 1200);
        }

        if bk {
            break;
        }
        println!("滚动加载更多牛人...");
        // 获取iframe 用于滚动
        let iframe_ele = page.get_iframe(RECOMMEND_FRAME_LOCATOR)?.unwrap();
        iframe_ele.run_js("window.scrollTo(0, document.body.scrollHeight);")?;
        sleep_random_ms(1200, 1500);
    }

    // 关闭浏览器
    page.close_browser();
    Ok(())
}

// 检测充值vip弹窗
pub fn check_vip2(page: &ChromiumPage) -> Result<bool, CdpError> {
    let vip_layout = page.ele(".vip2-layout")?;
    if vip_layout.is_some() {
        // 关闭窗口
        let close_ele = page.ele(".icon-close")?;
        if let Some(close_ele) = close_ele {
            close_ele.click()?;
        }
    }
    return Ok(vip_layout.is_some());
}