EN605.607.81.SP22_ASDM_Project/back_end/src/main.rs

mod ai;
mod constants;
mod db_handler;
mod game_logic;
mod json_handlers;
mod random_helper;
mod state;

const SQLITE_DB_PATH: &str = "./fourLineDropper.db";

use db_handler::DBHandlerRequest;

use std::sync::mpsc::sync_channel;

use db_handler::start_db_handler_thread;
use tokio::sync::oneshot;
use warp::{Filter, Rejection};

#[tokio::main]
async fn main() {
    let (db_tx, db_rx) = sync_channel::<DBHandlerRequest>(32);
    let db_tx_clone = db_tx.clone();

    let (shutdown_tx, shutdown_rx) = oneshot::channel::<()>();

    // Required because shutdown_tx is not cloneable, and its "send" consumes
    // itself.
    let (s_helper_tx, s_helper_rx) = sync_channel::<()>(1);

    std::thread::spawn(move || {
        if let Ok(_unused_value) = s_helper_rx.recv() {
            shutdown_tx
                .send(())
                .expect("Should be able to send shutdown signal");
        }
    });

    start_db_handler_thread(db_rx, SQLITE_DB_PATH.into(), s_helper_tx.clone());

    let route = warp::body::content_length_limit(1024 * 32)
        .and(warp::body::bytes())
        .and_then(move |bytes: bytes::Bytes| {
            let db_tx_clone = db_tx_clone.clone();
            let s_helper_tx_clone = s_helper_tx.clone();
            async move {
                let body_str_result = std::str::from_utf8(bytes.as_ref());
                if let Ok(body_str) = body_str_result {
                    let json_result = serde_json::from_str(body_str);
                    if let Ok(json_value) = json_result {
                        Ok(
                            json_handlers::handle_json(json_value, db_tx_clone, s_helper_tx_clone)
                                .unwrap_or_else(|e| e),
                        )
                    } else {
                        Ok(String::from("{\"type\": \"invalid_syntax\"}"))
                    }
                } else {
                    Ok::<String, Rejection>(String::from("{\"type\": \"invalid_syntax\"}"))
                }
            }
        });

    let (_addr, server) =
        warp::serve(route).bind_with_graceful_shutdown(([0, 0, 0, 0], 1237), async move {
            shutdown_rx.await.ok();
        });

    tokio::task::spawn(server).await.unwrap();
}