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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124

use std::convert::Infallible;

use eyre::Result;
use warp::{hyper::StatusCode, Filter};

use atuin_common::api::SyncHistoryRequest;

use super::handlers;
use super::{database::Database, database::Postgres};
use crate::models::User;
use crate::settings::Settings;

fn with_settings(
    settings: Settings,
) -> impl Filter<Extract = (Settings,), Error = Infallible> + Clone {
    warp::any().map(move || settings.clone())
}

fn with_db(
    db: impl Database + Clone + Send + Sync,
) -> impl Filter<Extract = (impl Database + Clone,), Error = Infallible> + Clone {
    warp::any().map(move || db.clone())
}

fn with_user(
    postgres: Postgres,
) -> impl Filter<Extract = (User,), Error = warp::Rejection> + Clone {
    warp::header::<String>("authorization").and_then(move |header: String| {
        // async closures are still buggy :(
        let postgres = postgres.clone();

        async move {
            let header: Vec<&str> = header.split(' ').collect();

            let token;

            if header.len() == 2 {
                if header[0] != "Token" {
                    return Err(warp::reject());
                }

                token = header[1];
            } else {
                return Err(warp::reject());
            }

            let user = postgres
                .get_session_user(token)
                .await
                .map_err(|_| warp::reject())?;

            Ok(user)
        }
    })
}

pub async fn router(
    settings: &Settings,
) -> Result<impl Filter<Extract = impl warp::Reply, Error = Infallible> + Clone> {
    let postgres = Postgres::new(settings.db_uri.as_str()).await?;
    let index = warp::get().and(warp::path::end()).map(handlers::index);

    let count = warp::get()
        .and(warp::path("sync"))
        .and(warp::path("count"))
        .and(warp::path::end())
        .and(with_user(postgres.clone()))
        .and(with_db(postgres.clone()))
        .and_then(handlers::history::count);

    let sync = warp::get()
        .and(warp::path("sync"))
        .and(warp::path("history"))
        .and(warp::query::<SyncHistoryRequest>())
        .and(warp::path::end())
        .and(with_user(postgres.clone()))
        .and(with_db(postgres.clone()))
        .and_then(handlers::history::list);

    let add_history = warp::post()
        .and(warp::path("history"))
        .and(warp::path::end())
        .and(warp::body::json())
        .and(with_user(postgres.clone()))
        .and(with_db(postgres.clone()))
        .and_then(handlers::history::add);

    let user = warp::get()
        .and(warp::path("user"))
        .and(warp::path::param::<String>())
        .and(warp::path::end())
        .and(with_db(postgres.clone()))
        .and_then(handlers::user::get);

    let register = warp::post()
        .and(warp::path("register"))
        .and(warp::path::end())
        .and(warp::body::json())
        .and(with_settings(settings.clone()))
        .and(with_db(postgres.clone()))
        .and_then(handlers::user::register);

    let login = warp::post()
        .and(warp::path("login"))
        .and(warp::path::end())
        .and(warp::body::json())
        .and(with_db(postgres))
        .and_then(handlers::user::login);

    let r = warp::any()
        .and(
            index
                .or(count)
                .or(sync)
                .or(add_history)
                .or(user)
                .or(register)
                .or(login)
                .or(warp::any().map(|| warp::reply::with_status("☕", StatusCode::IM_A_TEAPOT))),
        )
        .with(warp::filters::log::log("atuin::api"));

    Ok(r)
}