Add: _uuid module (#3974)

rimi0108 · web-flow · commit 42290b103e6c · 2022-08-07T15:27:53.000+09:00
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/stdlib/Cargo.toml b/stdlib/Cargo.toml
@@ -66,6 +66,12 @@ num_enum = "0.5.7"
 ascii = "1.0.0"
 parking_lot = "0.12.0"
 
+# uuid
+[target.'cfg(not(any(target_os = "ios", target_os = "android", target_os = "windows", target_arch = "wasm32")))'.dependencies]
+mac_address = "1.1.3"
+once_cell = "1.13.0"
+uuid = { version = "1.1.2", features = ["v1", "fast-rng", "macro-diagnostics"] }
+
 # mmap
 [target.'cfg(all(unix, not(target_arch = "wasm32")))'.dependencies]
 memmap2 = "0.5.4"
diff --git a/stdlib/src/lib.rs b/stdlib/src/lib.rs
@@ -55,6 +55,13 @@ mod select;
 mod ssl;
 #[cfg(all(unix, not(target_os = "redox"), not(target_os = "ios")))]
 mod termios;
+#[cfg(not(any(
+    target_os = "android",
+    target_os = "ios",
+    target_os = "windows",
+    target_arch = "wasm32"
+)))]
+mod uuid;
 
 use rustpython_common as common;
 use rustpython_vm as vm;
@@ -149,5 +156,9 @@ pub fn get_module_inits() -> impl Iterator<Item = (Cow<'static, str>, StdlibInit
         {
             "_scproxy" => scproxy::make_module,
         }
+        #[cfg(not(any(target_os = "android", target_os = "ios", target_os = "windows", target_arch = "wasm32")))]
+        {
+            "_uuid" => uuid::make_module,
+        }
     }
 }
diff --git a/stdlib/src/uuid.rs b/stdlib/src/uuid.rs
@@ -0,0 +1,46 @@
+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
+    }
+}

Original file line number	Diff line number	Diff line change
`@@ -55,6 +55,13 @@ mod select;`
`55`	`55`	`mod ssl;`
`56`	`56`	`#[cfg(all(unix, not(target_os = "redox"), not(target_os = "ios")))]`
`57`	`57`	`mod termios;`
	`58`	`+#[cfg(not(any(`
	`59`	`+ target_os = "android",`
	`60`	`+ target_os = "ios",`
	`61`	`+ target_os = "windows",`
	`62`	`+ target_arch = "wasm32"`
	`63`	`+)))]`
	`64`	`+mod uuid;`
`58`	`65`
`59`	`66`	`use rustpython_common as common;`
`60`	`67`	`use rustpython_vm as vm;`
`@@ -149,5 +156,9 @@ pub fn get_module_inits() -> impl Iterator<Item = (Cow<'static, str>, StdlibInit`
`149`	`156`	`{`
`150`	`157`	`"_scproxy" => scproxy::make_module,`
`151`	`158`	`}`
	`159`	`+ #[cfg(not(any(target_os = "android", target_os = "ios", target_os = "windows", target_arch = "wasm32")))]`
	`160`	`+ {`
	`161`	`+ "_uuid" => uuid::make_module,`
	`162`	`+ }`
`152`	`163`	`}`
`153`	`164`	`}`