use core::cmp::Ordering;
#[derive(Debug)]
pub(crate) struct PanicOnDrop {
id: u32,
armed: bool,
}
impl PanicOnDrop {
pub(crate) fn new(id: u32) -> Self {
Self { id, armed: false }
}
pub(crate) fn armed(id: u32) -> Self {
Self { id, armed: true }
}
pub(crate) fn id(&self) -> u32 {
self.id
}
}
impl Drop for PanicOnDrop {
fn drop(&mut self) {
assert!(!self.armed, "PanicOnDrop({}) destructor fired", self.id);
}
}
impl PartialEq for PanicOnDrop {
fn eq(&self, other: &Self) -> bool {
self.id == other.id
}
}
impl Eq for PanicOnDrop {}
impl PartialOrd for PanicOnDrop {
fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
Some(self.cmp(other))
}
}
impl Ord for PanicOnDrop {
fn cmp(&self, other: &Self) -> Ordering {
self.id.cmp(&other.id)
}
}
pub(crate) fn bombs(count: u32, armed: u32) -> impl Iterator<Item = PanicOnDrop> {
(0..count).map(move |i| {
if i == armed {
PanicOnDrop::armed(i)
} else {
PanicOnDrop::new(i)
}
})
}