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

use lta_models::prelude::{BusArrivalResp, BusArrivalRespRaw as RawBusArrivalResp};

use crate::Client;
use crate::{
    blocking::{prelude::BusRequests, ClientExt, LTAClient},
    reqwest_blocking::ReqwestBlocking,
    Bus, LTAResult,
};
use concat_string::concat_string;

impl BusRequests<LTAClient<ReqwestBlocking>> for Bus {
    fn get_arrival<'a>(
        client: &LTAClient<ReqwestBlocking>,
        bus_stop_code: u32,
        service_no: impl Into<Option<&'a str>>,
    ) -> LTAResult<BusArrivalResp> {
        let url = concat_string!(client.base_url(), "/BusArrivalv2");

        match service_no.into() {
            Some(srv_no) => client.build_req_with_query::<RawBusArrivalResp, _, _>(&url, |rb| {
                let srv_no = srv_no.as_ref();
                rb.query(&[
                    ("BusStopCode", bus_stop_code.to_string().as_str()),
                    ("ServiceNo", srv_no),
                ])
            }),
            None => client.build_req_with_query::<RawBusArrivalResp, _, _>(&url, |rb| {
                rb.query(&[("BusStopCode", bus_stop_code.to_string())])
            }),
        }
    }
}