mini-apm 0.0.0

Minimal APM for Rails
Documentation 
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

use tower_cookies::Cookies;

use crate::{DbPool, models::project::Project};

pub const PROJECT_COOKIE: &str = "miniapm_project";

/// Extracts current project context from cookie
#[derive(Clone, Debug)]
pub struct WebProjectContext {
    pub current_project: Option<Project>,
    pub projects: Vec<Project>,
    pub projects_enabled: bool,
}

impl WebProjectContext {
    pub fn project_id(&self) -> Option<i64> {
        self.current_project.as_ref().map(|p| p.id)
    }

    /// Check if the given project ID is the current project (for template use)
    pub fn is_current_project(&self, id: &i64) -> bool {
        self.current_project.as_ref().map(|p| p.id) == Some(*id)
    }

    /// Returns true if project selector should be shown (more than 1 project)
    pub fn show_selector(&self) -> bool {
        self.projects_enabled && self.projects.len() > 1
    }
}

pub fn get_project_context(pool: &DbPool, cookies: &Cookies) -> WebProjectContext {
    let projects_enabled = std::env::var("ENABLE_PROJECTS")
        .map(|v| v == "1" || v.to_lowercase() == "true")
        .unwrap_or(false);

    if !projects_enabled {
        return WebProjectContext {
            current_project: None,
            projects: vec![],
            projects_enabled: false,
        };
    }

    let projects = crate::models::project::list_all(pool).unwrap_or_default();

    // Get project slug from cookie
    let project_slug = cookies.get(PROJECT_COOKIE).map(|c| c.value().to_string());

    let current_project = match project_slug {
        Some(slug) => projects.iter().find(|p| p.slug == slug).cloned(),
        None => projects.first().cloned(),
    };

    WebProjectContext {
        current_project,
        projects,
        projects_enabled,
    }
}