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

use std::collections::HashMap;

use thruster_core::context::Context;
use thruster_core::middleware::{MiddlewareReturnValue};

pub trait HasQueryParams {
  fn set_query_params(&mut self, query_params: HashMap<String, String>);
  fn route(&self) -> &str;
}

pub fn query_params<T: 'static + Context + HasQueryParams + Send>(mut context: T, next: impl Fn(T) -> MiddlewareReturnValue<T>  + Send) -> MiddlewareReturnValue<T> {
  let mut query_param_hash = HashMap::new();

  {
    let route: &str = &context.route();

    let mut iter = route.split('?');
    // Get rid of first bit (the non-query string part)
    let _ = iter.next().unwrap();

    if let Some(query_string) = iter.next() {
      for query_piece in query_string.split('&') {
        let mut query_iterator = query_piece.split('=');
        let key = query_iterator.next().unwrap().to_owned();

        match query_iterator.next() {
          Some(val) => query_param_hash.insert(key, val.to_owned()),
          None => query_param_hash.insert(key, "true".to_owned())
        };
      }
    }
  }

  context.set_query_params(query_param_hash);

  next(context)
}