cheshire_chess 0.3.2

#![allow(dead_code)]

use std::io::{Read, Write};
use std::net::{TcpStream, ToSocketAddrs};
use std::sync::{mpsc, Arc};
use std::thread;
use std::time::Duration;

use serde::{Deserialize, Serialize};

#[allow(dead_code)]
const TRACKER_HOST: &str = "chess.virtualraremedia.com";
#[allow(dead_code)]
const TRACKER_PORT: u16 = 443;

#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct RemoteServer {
    pub host: String,
    pub port: u16,
    pub name: String,
    pub players: u32,
}

fn tls_config() -> Arc<rustls::ClientConfig> {
    let root_store = rustls::RootCertStore::from_iter(
        webpki_roots::TLS_SERVER_ROOTS.iter().cloned(),
    );
    let config = rustls::ClientConfig::builder()
        .with_root_certificates(root_store)
        .with_no_client_auth();
    Arc::new(config)
}

fn connect_tracker() -> Option<rustls::StreamOwned<rustls::ClientConnection, TcpStream>> {
    let addr = format!("{TRACKER_HOST}:{TRACKER_PORT}");
    let resolved = addr.to_socket_addrs().ok()?.next()?;
    let tcp = TcpStream::connect_timeout(&resolved, Duration::from_secs(5)).ok()?;
    tcp.set_read_timeout(Some(Duration::from_secs(10))).ok();

    let config = tls_config();
    let server_name = TRACKER_HOST.to_string().try_into().ok()?;
    let conn = rustls::ClientConnection::new(config, server_name).ok()?;
    Some(rustls::StreamOwned::new(conn, tcp))
}

/// Fetch the server list from the tracker. Blocking.
pub fn fetch_servers() -> Vec<RemoteServer> {
    let mut stream = match connect_tracker() {
        Some(s) => s,
        None => return Vec::new(),
    };

    let request = format!(
        "GET /servers HTTP/1.1\r\nHost: {TRACKER_HOST}\r\nConnection: close\r\n\r\n"
    );
    if stream.write_all(request.as_bytes()).is_err() {
        return Vec::new();
    }

    let mut response = String::new();
    let _ = stream.read_to_string(&mut response);

    let body = response.split("\r\n\r\n").nth(1).unwrap_or("[]");
    serde_json::from_str(body).unwrap_or_default()
}

/// Register this server with the tracker. Blocking.
pub fn register(host: &str, port: u16, name: &str, players: u32) {
    let mut stream = match connect_tracker() {
        Some(s) => s,
        None => return,
    };

    let body = serde_json::to_string(&serde_json::json!({
        "host": host,
        "port": port,
        "name": name,
        "players": players,
    })).unwrap();

    let request = format!(
        "POST /register HTTP/1.1\r\nHost: {TRACKER_HOST}\r\nContent-Type: application/json\r\nContent-Length: {}\r\nConnection: close\r\n\r\n{}",
        body.len(), body
    );
    let _ = stream.write_all(request.as_bytes());
}

/// Get this machine's public IP.
pub fn get_public_ip() -> String {
    let addrs = "api.ipify.org:443".to_socket_addrs();
    if let Ok(mut resolved) = addrs {
        if let Some(addr) = resolved.next() {
            if let Ok(tcp) = TcpStream::connect_timeout(&addr, Duration::from_secs(3)) {
                tcp.set_read_timeout(Some(Duration::from_secs(3))).ok();
                let config = tls_config();
                if let Ok(server_name) = "api.ipify.org".to_string().try_into() {
                    if let Ok(conn) = rustls::ClientConnection::new(config, server_name) {
                        let mut stream = rustls::StreamOwned::new(conn, tcp);
                        let req = "GET / HTTP/1.1\r\nHost: api.ipify.org\r\nConnection: close\r\n\r\n";
                        if stream.write_all(req.as_bytes()).is_ok() {
                            let mut resp = String::new();
                            let _ = stream.read_to_string(&mut resp);
                            if let Some(body) = resp.split("\r\n\r\n").nth(1) {
                                let ip = body.trim().to_string();
                                if !ip.is_empty() {
                                    return ip;
                                }
                            }
                        }
                    }
                }
            }
        }
    }
    "127.0.0.1".to_string()
}

/// Fetch puzzles from the server. Blocking.
pub fn fetch_puzzles(
    client_id: &str,
    theme: &str,
    max_rating: u16,
    limit: usize,
    offset: usize,
) -> Vec<crate::puzzle::Puzzle> {
    let mut stream = match connect_tracker() {
        Some(s) => s,
        None => return Vec::new(),
    };
    let path = format!(
        "/puzzles?theme={theme}&max_rating={max_rating}&offset={offset}&limit={limit}"
    );
    let request = format!(
        "GET {path} HTTP/1.1\r\nHost: {TRACKER_HOST}\r\nX-Cheshire-ID: {client_id}\r\nConnection: close\r\n\r\n"
    );
    if stream.write_all(request.as_bytes()).is_err() {
        return Vec::new();
    }

    let mut response = String::new();
    let _ = stream.read_to_string(&mut response);

    let body = response.split("\r\n\r\n").nth(1).unwrap_or("[]");
    serde_json::from_str(body).unwrap_or_default()
}

/// Fetch theme list and counts from the server. Blocking.
pub fn fetch_themes(client_id: &str) -> Vec<(String, String, usize)> {
    let mut stream = match connect_tracker() {
        Some(s) => s,
        None => return Vec::new(),
    };

    let request = format!(
        "GET /themes HTTP/1.1\r\nHost: {TRACKER_HOST}\r\nX-Cheshire-ID: {client_id}\r\nConnection: close\r\n\r\n"
    );
    if stream.write_all(request.as_bytes()).is_err() {
        return Vec::new();
    }

    let mut response = String::new();
    let _ = stream.read_to_string(&mut response);

    let body = response.split("\r\n\r\n").nth(1).unwrap_or("[]");
    serde_json::from_str(body).unwrap_or_default()
}

/// Start a background heartbeat thread that re-registers every 30 seconds.
/// Check if a newer version is available. Returns Some(latest_version) if update available.
pub fn check_update() -> Option<String> {
    let mut stream = connect_tracker()?;
    let request = format!(
        "GET /version HTTP/1.1\r\nHost: {TRACKER_HOST}\r\nConnection: close\r\n\r\n"
    );
    stream.write_all(request.as_bytes()).ok()?;
    let mut response = String::new();
    let _ = stream.read_to_string(&mut response);
    let body = response.split("\r\n\r\n").nth(1)?;
    let latest: String = serde_json::from_str(body).ok()?;
    let current = env!("CARGO_PKG_VERSION");
    if latest != current { Some(latest) } else { None }
}

pub fn start_heartbeat(host: String, port: u16, name: String) -> mpsc::Sender<u32> {
    let (tx, rx) = mpsc::channel::<u32>();

    thread::spawn(move || {
        let mut players = 0u32;
        loop {
            while let Ok(count) = rx.try_recv() {
                players = count;
            }
            register(&host, port, &name, players);
            thread::sleep(Duration::from_secs(30));
        }
    });

    tx
}