ragit-server 0.4.2

poc server implementation of ragit
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

use super::{HandleError, RawResponse, get_pool, handler};
use crate::models::{archive, repo};
use crate::models::repo::RepoOperation;
use warp::reply::{Reply, json, with_header};

pub async fn get_archive_list(user: String, repo: String, api_key: Option<String>) -> Box<dyn Reply> {
    handler(get_archive_list_(user, repo, api_key).await)
}

async fn get_archive_list_(user: String, repo: String, api_key: Option<String>) -> RawResponse {
    let pool = get_pool().await;
    let repo_id = repo::get_id(&user, &repo, pool).await.handle_error(404)?;
    repo::check_auth(repo_id, RepoOperation::Clone, api_key, pool).await.handle_error(500)?.handle_error(404)?;
    let session_id = repo::get_session_id(repo_id, pool).await.handle_error(404)?;
    let archive_list = match session_id {
        Some(session_id) => archive::get_list(&session_id, pool).await.handle_error(500)?,

        // It's not an error to clone an empty repository (github allows that)
        None => vec![],
    };
    Ok(Box::new(json(&archive_list)))
}

pub async fn get_archive(user: String, repo: String, archive_id: String, api_key: Option<String>) -> Box<dyn Reply> {
    handler(get_archive_(user, repo, archive_id, api_key).await)
}

async fn get_archive_(user: String, repo: String, archive_id: String, api_key: Option<String>) -> RawResponse {
    let pool = get_pool().await;
    let repo_id = repo::get_id(&user, &repo, pool).await.handle_error(404)?;
    repo::check_auth(repo_id, RepoOperation::Clone, api_key, pool).await.handle_error(500)?.handle_error(404)?;
    let session_id = repo::get_session_id(repo_id, pool).await.handle_error(404)?;
    let Some(session_id) = session_id else {
        return Err((400, format!("Nothing's pushed to `{user}/{repo}` yet!")));
    };

    let bytes = archive::get_archive(&session_id, &archive_id, pool).await.handle_error(404)?;

    // It's easy to count pushes: there's `finalize-push` api.
    // But clone operations use `/archive` and `/archive-list` apis, which are too general to count as a clone.
    // So we use a naive heuristic: we count downloading the first archive as a clone operation
    if archive::is_first_archive(&session_id, &archive_id, pool).await.handle_error(500)? {
        archive::increment_clone_count(repo_id, pool).await.handle_error(500)?;
    }

    Ok(Box::new(with_header(
        bytes,
        "Content-Type",
        "application/octet-stream",
    )))
}