#[macro_use]
extern crate envconfig_derive;

mod api;
mod common;
mod env;
mod middleware;

use actix::prelude::*;
use actix_redis::RedisActor;
use actix_web::{http::header, middleware::cors::Cors, server, App, HttpResponse};
use env::ServerConfig;
use env_logger::Builder;
use envconfig::Envconfig;
use log::info;
use std::net::SocketAddr;

const ENV_LOG_VARIABLE: &str = "STREAMING_API_LOG";

#[derive(Clone)]
pub struct AppState {
    redis: Addr<RedisActor>,
}

fn main() {
    Builder::from_env(ENV_LOG_VARIABLE).init();

    info!("starting streaming api server");

    let server_cfg = ServerConfig::init().expect("failed to obtain server environment");
    let addr = SocketAddr::new(server_cfg.address, server_cfg.port);

    let sys = System::new("streaming-api-server");

    let redis_addr = RedisActor::start("127.0.0.1:6379");

    let app_state = AppState {
        redis: redis_addr.clone(),
    };

    server::new(move || endpoints(&app_state)).bind(addr).unwrap().start();

    sys.run();
}

fn endpoints(app_state: &AppState) -> App<AppState> {
    use api::http;
    use api::ws;

    App::with_state(app_state.clone())
        .prefix("/api/v1")
        .resource("/streaming", |r| r.with(ws::index))
        .resource("/streaming/health", |r| {
            r.middleware(cors_middleware());
            r.get().f(|_| HttpResponse::Ok())
        })
        .resource("/streaming/user", |r| {
            r.middleware(cors_middleware());
            r.get().with(http::user::index)
        })
        .resource("/streaming/public", |r| {
            r.middleware(cors_middleware());
            r.get().with(http::public::index)
        })
        .resource("/streaming/public/local", |r| {
            r.middleware(cors_middleware());
            r.get().with(http::public::local)
        })
        .resource("/streaming/direct", |r| {
            r.middleware(cors_middleware());
            r.get().with(http::direct::index)
        })
        .resource("/streaming/hashtag", |r| {
            r.middleware(cors_middleware());
            r.get().with(http::hashtag::index)
        })
        .resource("/streaming/hashtag/local", |r| {
            r.middleware(cors_middleware());
            r.get().with(http::hashtag::local)
        })
        .resource("/streaming/list", |r| {
            r.middleware(cors_middleware());
            r.get().with(http::list::index)
        })
}

fn cors_middleware() -> Cors {
    Cors::build()
        .allowed_origin("*")
        .allowed_methods(vec!["GET", "OPTIONS"])
        .allowed_headers(vec![header::AUTHORIZATION, header::ACCEPT, header::CACHE_CONTROL])
        .finish()
}