handmade-revolt-backend/crates/quark/src/presence/entry.rs

use std::env;

use serde::{Deserialize, Serialize};

lazy_static! {
    pub static ref REGION_ID: u16 = env::var("REGION_ID")
        .unwrap_or_else(|_| "0".to_string())
        .parse()
        .unwrap();
    pub static ref REGION_KEY: String = format!("region{}", &*REGION_ID);
}

/// Compact presence information for a user
#[derive(Serialize, Deserialize, Debug)]
pub struct PresenceEntry {
    /// Region this session exists in
    ///
    /// We can have up to 65535 regions
    pub region_id: u16,

    /// Unique session ID
    pub session_id: u8,

    /// Known flags about session
    pub flags: u8,
}

impl PresenceEntry {
    /// Create a new presence entry from a given session ID and known flags
    pub fn from(session_id: u8, flags: u8) -> Self {
        Self {
            region_id: *REGION_ID,
            session_id,
            flags,
        }
    }
}

pub trait PresenceOp {
    /// Find next available session ID
    fn find_next_id(&self) -> u8;
}

impl PresenceOp for Vec<PresenceEntry> {
    fn find_next_id(&self) -> u8 {
        // O(n^2) scan algorithm
        // should be relatively fast at low numbers anyways
        for i in 0..255 {
            let mut found = false;
            for entry in self {
                if entry.session_id == i {
                    found = true;
                    break;
                }
            }

            if !found {
                return i;
            }
        }

        255
    }
}