1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

use candid::CandidType;
use core::fmt;
use serde::{Deserialize, Serialize};
use std::ops::{Range, RangeBounds};

#[derive(Serialize, Deserialize, CandidType, Copy, Clone, Default, PartialEq, Eq, Hash)]
#[doc(alias = "..")]
pub struct CandidRange<Idx: CandidType> {
    pub start: Idx,
    pub end: Idx,
}

impl<Idx: fmt::Debug + CandidType> fmt::Debug for CandidRange<Idx> {
    fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
        self.start.fmt(fmt)?;
        write!(fmt, "..")?;
        self.end.fmt(fmt)?;
        Ok(())
    }
}

impl<Idx: PartialOrd<Idx> + CandidType> RangeBounds<Idx> for CandidRange<Idx> {
    fn start_bound(&self) -> std::ops::Bound<&Idx> {
        std::ops::Bound::Included(&self.start)
    }

    fn end_bound(&self) -> std::ops::Bound<&Idx> {
        std::ops::Bound::Excluded(&self.end)
    }
}

impl<Idx: PartialOrd<Idx> + CandidType> CandidRange<Idx> {
    pub fn contains<U>(&self, item: &U) -> bool
    where
        Idx: PartialOrd<U>,
        U: ?Sized + PartialOrd<Idx>,
    {
        <Self as RangeBounds<Idx>>::contains(self, item)
    }

    pub fn is_empty(&self) -> bool {
        !(self.start < self.end)
    }
}

impl<Idx: PartialOrd<Idx> + CandidType, Idx2> From<CandidRange<Idx>> for Range<Idx2>
where
    Idx: Into<Idx2>,
{
    fn from(range: CandidRange<Idx>) -> Self {
        Range {
            start: range.start.into(),
            end: range.end.into(),
        }
    }
}

impl<Idx, Idx2: PartialOrd<Idx> + CandidType> From<Range<Idx>> for CandidRange<Idx2>
where
    Idx: Into<Idx2>,
{
    fn from(range: Range<Idx>) -> Self {
        CandidRange {
            start: range.start.into(),
            end: range.end.into(),
        }
    }
}