titan_router/
segment.rs

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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
#[derive(Debug, Hash, Clone, PartialEq, Eq)]
pub(crate) enum Segment {
  Slash,
  Exact(String),
  Param(String),
}

pub(crate) trait CompareSegment {
  fn eq(&self, may_be_dynamic: &Self) -> CompareSegmentOut;
}

#[derive(Debug, PartialEq, Eq)]
pub(crate) enum CompareSegmentOut {
  NoMatch,
  Match(Option<(String, String)>),
}

impl CompareSegment for Segment {
  fn eq(&self, from_request: &Self) -> CompareSegmentOut {
    match self {
      Segment::Slash => {
        if matches!(from_request, Segment::Slash) {
          CompareSegmentOut::Match(None)
        } else {
          CompareSegmentOut::NoMatch
        }
      }

      Segment::Exact(str) => match from_request {
        Segment::Exact(str_contract) if str_contract == str => {
          CompareSegmentOut::Match(None)
        }

        _ => CompareSegmentOut::NoMatch,
      },
      Segment::Param(param_name) => match from_request {
        Segment::Exact(param_value) => CompareSegmentOut::Match(Some((
          param_name.clone(),
          param_value.clone(),
        ))),
        _ => CompareSegmentOut::NoMatch,
      },
    }
  }
}

#[cfg(test)]
mod testing {
  use super::CompareSegmentOut;

  use super::{CompareSegment, Segment};

  #[test]
  fn test_compare_segment_param_value() {
    let contract = Segment::Param("variable".to_string());
    let request = Segment::Exact("value".to_string());
    let request2 = Segment::Slash;

    assert_eq!(
      CompareSegment::eq(&contract, &request),
      CompareSegmentOut::Match(Some((
        "variable".to_string(),
        "value".to_string(),
      )))
    );

    assert_eq!(
      CompareSegment::eq(&contract, &request2),
      CompareSegmentOut::NoMatch
    );
  }

  #[test]
  fn test_compare_segment_exact() {
    let contract = Segment::Exact("value".to_string());
    let request = Segment::Exact("value".to_string());

    assert_eq!(
      CompareSegment::eq(&contract, &request),
      CompareSegmentOut::Match(None)
    );

    let contract = Segment::Exact("value1".to_string());
    let request = Segment::Exact("value".to_string());

    assert_eq!(
      CompareSegment::eq(&contract, &request),
      CompareSegmentOut::NoMatch,
    );
  }
}