uuid.rs

pub(crate) use _uuid::make_module;

#[pymodule]
mod _uuid {
    use crate::{builtins::PyNone, vm::VirtualMachine};
    use mac_address::get_mac_address;
    use once_cell::sync::OnceCell;
    use rand::Rng;
    use std::time::{Duration, SystemTime};
    use uuid::{
        v1::{Context, Timestamp},
        Uuid,
    };

    fn get_node_id() -> [u8; 6] {
        match get_mac_address() {
            Ok(Some(_ma)) => get_mac_address().unwrap().unwrap().bytes(),
            _ => rand::thread_rng().gen::<[u8; 6]>(),
        }
    }

    pub fn now_unix_duration() -> Duration {
        use std::time::UNIX_EPOCH;

        let now = SystemTime::now();
        now.duration_since(UNIX_EPOCH)
            .expect("SystemTime before UNIX EPOCH!")
    }

    #[pyfunction]
    fn generate_time_safe() -> (Vec<u8>, PyNone) {
        static CONTEXT: Context = Context::new(0);
        let now = now_unix_duration();
        let ts = Timestamp::from_unix(&CONTEXT, now.as_secs(), now.subsec_nanos());

        static NODE_ID: OnceCell<[u8; 6]> = OnceCell::new();
        let unique_node_id = NODE_ID.get_or_init(get_node_id);

        (Uuid::new_v1(ts, unique_node_id).as_bytes().to_vec(), PyNone)
    }

    #[pyattr]
    fn has_uuid_generate_time_safe(_vm: &VirtualMachine) -> u32 {
        0
    }
}