summaryrefslogtreecommitdiffstats
path: root/crates/rebel-runner/src/util/steal.rs
diff options
context:
space:
mode:
authorMatthias Schiffer <mschiffer@universe-factory.net>2024-04-20 14:28:05 +0200
committerMatthias Schiffer <mschiffer@universe-factory.net>2024-04-20 14:38:17 +0200
commite9bf0fc40c0eb7e9d4228b804d62f31b0a136528 (patch)
tree7872f587782d5635eadbf82ae861d474d4da2efe /crates/rebel-runner/src/util/steal.rs
parent35e68444dd5e9d3d5fc39409c48be6eb3fa05e07 (diff)
downloadrebel-e9bf0fc40c0eb7e9d4228b804d62f31b0a136528.tar
rebel-e9bf0fc40c0eb7e9d4228b804d62f31b0a136528.zip
Rename directories to match crate names
Diffstat (limited to 'crates/rebel-runner/src/util/steal.rs')
-rw-r--r--crates/rebel-runner/src/util/steal.rs40
1 files changed, 40 insertions, 0 deletions
diff --git a/crates/rebel-runner/src/util/steal.rs b/crates/rebel-runner/src/util/steal.rs
new file mode 100644
index 0000000..91b2cdf
--- /dev/null
+++ b/crates/rebel-runner/src/util/steal.rs
@@ -0,0 +1,40 @@
+use std::ops::{Deref, DerefMut};
+
+#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
+pub struct Steal<T>(pub Option<T>);
+
+impl<T> Steal<T> {
+ pub fn new(value: T) -> Steal<T> {
+ Steal(Some(value))
+ }
+
+ pub fn steal(&mut self) -> T {
+ self.0
+ .take()
+ .expect("Attempted to steal already stoken value")
+ }
+}
+
+impl<T> From<T> for Steal<T> {
+ fn from(value: T) -> Self {
+ Steal::new(value)
+ }
+}
+
+impl<T> Deref for Steal<T> {
+ type Target = T;
+
+ fn deref(&self) -> &Self::Target {
+ self.0
+ .as_ref()
+ .expect("Attempted to dereference stolen value")
+ }
+}
+
+impl<T> DerefMut for Steal<T> {
+ fn deref_mut(&mut self) -> &mut Self::Target {
+ self.0
+ .as_mut()
+ .expect("Attempted to dereference stolen value")
+ }
+}