amtrack_rs 0.1.2

A fully open-source Rust-based API and Package for fetching live train and station data from Amtrak.
Documentation 
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
92
93
94

use amtk::decrypt;
use redis::Commands;
#[path = "./structs/mod.rs"]
mod structs;
#[path = "./utils/mod.rs"]
mod utils;
use crate::structs::raw::RawResponse::RawResponse;
use std::str;
use utils::ConvertRawTrains::ConvertRawTrains;
#[macro_use]
extern crate rocket;
use std::fs;
extern crate redis;
use chrono::Utc;
use lazy_static::lazy_static;

lazy_static! {
    static ref REDIS_CLIENT: redis::Client = redis::Client::open("redis://127.0.0.1:6379").unwrap();
}

#[get("/")]
async fn index() -> rocket::response::content::RawHtml<&'static str> {
    rocket::response::content::RawHtml(
        r#"
        <title>AmTrack API</title>
        <p style='font-family:monospace; font-size:1.5em'>
        The API that powers AmTrack: Track Amtrak Trains. Download today on the Apple App Store and Google Play Store. See <a href='https://github.com/qpxdesign/amtrack_rs'>https://github.com/qpxdesign/amtrack_rs</a> for usage terms. </p>
        <p style='font-family:monospace; font-size:1.5em; font-weight:800'> A <a href='https://quinnpatwardhan.com'>https://quinnpatwardhan.com</a> + <a href='https://locomotive.systems'>https://locomotive.systems</a> product.</p>
        </p>
    "#,
    )
}
#[get("/get-trains")]
async fn getTrains() -> String {
    let CACHE_EXPIRY_SECONDS = 30;
    let mut con = REDIS_CLIENT
        .get_connection()
        .expect("Failed to establish connection");
    let cache_time: Option<i64> = con.get("amtrak_api_lastupdated").expect("REDIS FUCKED UP");
    if cache_time.is_some() && Utc::now().timestamp() - cache_time.unwrap() < CACHE_EXPIRY_SECONDS {
        let cached_res: Option<String> = con.get("amtrak_api_gettrains").expect("REDIS FUCKED UP");
        if cached_res.is_some() {
            println!("{}", "Returing Cached");
            return cached_res.unwrap();
        }
    }
    let body = reqwest::get("https://maps.amtrak.com/services/MapDataService/trains/getTrainsData")
        .await
        .unwrap()
        .bytes()
        .await
        .unwrap();
    let a = decrypt(str::from_utf8(&body).unwrap());
    let data: RawResponse = serde_json::from_str(a.unwrap().as_str()).unwrap();
    let c = ConvertRawTrains(data);
    let b = serde_json::to_string(&c).unwrap();
    let _: () = con
        .set("amtrak_api_lastupdated", Utc::now().timestamp())
        .expect("Failed to set");
    let _: () = con
        .set("amtrak_api_gettrains", b.clone())
        .expect("failed to set");
    return b.to_owned();
}

#[get("/get-trains-raw")]
async fn getTrainsRaw() -> String {
    let body = reqwest::get("https://maps.amtrak.com/services/MapDataService/trains/getTrainsData")
        .await
        .unwrap()
        .bytes()
        .await
        .unwrap();
    let a = decrypt(str::from_utf8(&body).unwrap());
    return a.unwrap();
    //let data: RawResponse = serde_json::from_str(a.unwrap().as_str()).unwrap();
    //return serde_json::to_string(&data).unwrap();
}

#[get("/get-stations")]
async fn getStations() -> String {
    let file_contents = fs::read_to_string("static/get-stations.json");
    if file_contents.is_ok() {
        return file_contents.unwrap();
    } else {
        println!("{}", file_contents.err().unwrap());
        return "".to_string();
    }
}

#[launch]
fn rocket() -> _ {
    rocket::build().mount("/", routes![index, getTrains, getStations, getTrainsRaw])
}