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
95
96
97
98
99

#![feature(proc_macro_hygiene, decl_macro)]
/**
 * Interface layer
 * Magnus Fredriksson
 */
#[macro_use] extern crate rocket;
#[cfg(test)]
mod integration_tests;
extern crate futures;

/**
 * Persistence layer
 */
mod persistance;
/**
 * App configuration
 */
mod environment;
/**
 * App model, database / request to struct mapping
 */
mod model;
/**
 * App business logic
 */
mod services;
/**
 * App public facins API endpoints
 */
mod endpoints;

use rocket::http::Method;
use rocket_cors;
use rocket::routes;
use rocket_cors::{AllowedHeaders, AllowedOrigins};
use rocket::{Request, Response};
use rocket::fairing::{Fairing, Info, Kind};

/**
 *  Program entrypoint, initializes rocket with the public endpoints
 */ 
fn main() {
    check_env();
    rocket().launch();
}

#[derive(Default)]
struct ResponsePostProcessor {}

impl Fairing for ResponsePostProcessor {
    fn info(&self) -> Info {
        Info {
            name: "Allow origin header",
            kind: Kind::Request | Kind::Response
        }
    }
    fn on_response(&self, _request: &Request, response: &mut Response) {
        response.set_raw_header("Access-Control-Allow-Origin", "*");
    }
}

fn rocket() -> rocket::Rocket {
    rocket::ignite()
    .attach(ResponsePostProcessor{})
    .mount("/", routes![
        endpoints::default, 
        endpoints::log_message,
        endpoints::devices_endpoints::register, 
        endpoints::devices_endpoints::register_json, 
        endpoints::devices_endpoints::get_tracker, 
        endpoints::devices_endpoints::unregister, 
        endpoints::devices_endpoints::unregister_json, 
        endpoints::videos_endpoints::register_view,
        endpoints::videos_endpoints::get_video]) 
    .register( catchers![
        endpoints::catchers::not_found, 
        endpoints::catchers::bad_request, 
        endpoints::catchers::unproc_request])
    .mount("/", rocket_cors::catch_all_options_routes())
    .manage(cors())
}

fn cors() -> rocket_cors::Cors {
    let allowed_origins = AllowedOrigins::all();
    rocket_cors::CorsOptions {
        allowed_origins,
        allowed_methods: vec![Method::Get].into_iter().map(From::from).collect(),
        allowed_headers: AllowedHeaders::some(&["Authorization", "Accept"]),
        allow_credentials: true,
        ..Default::default()
    }.to_cors().unwrap()
}

fn check_env() {
    match String::from(environment::PRODUCTION_STRING) == environment::get_current_env() {
        false => colour::yellow!("\n### USING STAGING ENVIRONMENT (not an error) ###\n\n"),
        true =>  colour::dark_red!("\n### WARNING! USING PRODUCTION ENVIRONMENT ###\n\n")
    }
}