cli.rs

use std::{collections::HashSet, time::Duration};

use alkonost::{Alkonost, AlkonostInMessage, AlkonostOutMessage, RequestSettings};
use tokio::time::sleep;
use tracing::Level;

#[tokio::main]
pub async fn main() {
    let subscriber = tracing_subscriber::FmtSubscriber::builder()
        .with_max_level(Level::INFO)
        .finish();

    tracing::subscriber::set_global_default(subscriber).expect("setting default subscriber failed");

    let request_settings = RequestSettings {
        browser_name: "Firefox".to_string(),
        browser_version: "90.0".to_string(),
        user_agent:
            r#"Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:90.0) Gecko/20100101 Firefox/90.0"#
                .to_string(),
    };
    let poll_interval = Duration::from_secs(90);

    let (actor, mut result_rx) = match Alkonost::init(request_settings, poll_interval) {
        Ok(r) => r,
        Err(e) => {
            tracing::error!("Error initializing alkonost: {}", &e);
            return;
        }
    };

    let mut actor_tx = actor.tx;
    let actor_handle = actor.join_handle;

    let rx_reader = tokio::spawn(async move {
        while let Some(message) = result_rx.recv().await {
            match message {
                AlkonostOutMessage::NewChat { channel, video_id } => {
                    tracing::info!("New stream <{}> on channel <{}>", video_id, channel);
                }
                AlkonostOutMessage::ChatClosed { channel, video_id } => {
                    tracing::info!("Stream <{}> from channel <{}> has ended", video_id, channel);
                }
                AlkonostOutMessage::DetectorResult {
                    video_id,
                    decisions,
                    processed_messages: _,
                } => {
                    tracing::info!("<{}>: {:?}", video_id, decisions);
                }
            }
        }

        tracing::info!("rx_reader has been closed");
    });

    let mut channels = HashSet::new();
    channels.insert("UCtMVHI3AJD4Qk4hcbZnI9ZQ"); // SomeOrdinaryGamers
    channels.insert("UC-lHJZR3Gqxm24_Vd_AJ5Yw"); // PewDiePie
    channels.insert("UCqNH56x9g4QYVpzmWTzqVYg"); // Dynamo Gaming
    channels.insert("UCam8T03EOFBsNdR0thrFHdQ"); // VEGETTA777
    channels.insert("UCaHEdZtk6k7SVP-umnzifmQ"); // TheDonato
    channels.insert("UC5c9VlYTSvBSCaoMu_GI6gQ"); // Total Gaming
    channels.insert("UChXi_PlJkRMPYFQBOJ3MpxA"); // Gyan Gaming
    channels.insert("UCSJ4gkVC6NrvII8umztf0Ow"); // Lofi Girl
    channels.insert("UC2wKfjlioOCLP4xQMOWNcgg"); // Typical Gamer
    channels.insert("UCw7FkXsC00lH2v2yB5LQoYA"); // jackfrags
    channels.insert("UCsjTQnlZcSB6fSiP7ht_0OQ"); // Hacks Busters

    for channel_id in channels {
        let message = AlkonostInMessage::AddChannel(channel_id.to_string());
        match actor_tx.send(message).await {
            Ok(_r) => {}
            Err(e) => {
                tracing::error!("Couldn't send message to a stream finder: {}", &e);
                return;
            }
        }
    }

    sleep(Duration::from_secs(130)).await;
    tracing::info!("Closing...");
    match actor_tx.send(AlkonostInMessage::Close).await {
        Ok(_r) => {}
        Err(e) => {
            tracing::error!("Couldn't send message to a stream finder: {}", &e);
            return;
        }
    }

    let _ = actor_handle.await;
    let _ = rx_reader.await;

    tracing::info!("Closed");
}