konarr 0.5.1

Konarr
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

//! # Konarr Models

use geekorm::ConnectionManager;

pub mod auth;
pub mod cache;
pub mod components;
pub mod dependencies;
pub mod projects;
pub mod security;
pub mod settings;

pub use auth::sessions::{SessionState, SessionType, Sessions};
pub use auth::users::{UserRole, Users};
pub use cache::DbCache;
pub use components::{Component, ComponentManager, ComponentType, ComponentVersion};
pub use dependencies::Dependencies;
pub use dependencies::snapshots::{Snapshot, SnapshotMetadata, SnapshotMetadataKey, SnapshotState};
pub use projects::{ProjectSnapshots, ProjectStatus, ProjectType, Projects};
pub use security::advisories::AdvisoriesMetadata;
pub use security::{Advisories, Alerts, SecurityState};
pub use settings::{ServerSettings, Setting};

use crate::tasks::alerts::AlertCalculatorTask;
use crate::tasks::projects::ProjectsTask;
use crate::tasks::server::ServerTask;
use crate::tasks::statistics::StatisticsTask;
use crate::tasks::{CatalogueTask, TaskTrait};
use crate::{Config, KonarrError, db};

/// Initialize the database with the necessary tables.
///
/// - Create / Migrate Database
/// - Initiale data
/// - Update Statistics
/// - Update Security Data
pub async fn database_initialise(config: &mut Config) -> Result<ConnectionManager, KonarrError> {
    log::info!("Initialising Database");
    let database = config.database().await?;

    db::init(&database.acquire().await).await?;

    {
        log::debug!("Initialising Models...");
        let connection = database.acquire().await;
        // Initialise the models
        ServerSettings::init(&connection).await?;
        Component::init(&connection).await?;
        SnapshotMetadata::init(&connection).await?;

        // Store the server setting into the config file
        if let Ok(token) = ServerSettings::fetch_by_name(&connection, Setting::AgentKey).await {
            if config.agent.token != Some(token.value.clone()) {
                log::info!("Updating Agent Token");
                config.agent.token = Some(token.value);
                config.autosave()?;
            }
        }
        log::debug!("Initialised Models :: {}", connection.count());
    }

    // Load the server settings from the config
    ServerSettings::load_config(&database.acquire().await, config).await?;
    log::debug!("Loaded Server Settings from Config");

    // Update Stats
    ServerTask::spawn(&database).await?;
    StatisticsTask::spawn(&database).await?;
    ProjectsTask::spawn(&database).await?;
    AlertCalculatorTask::spawn(&database).await?;
    CatalogueTask::spawn(&database).await?;

    log::info!("Database Initialised!");
    Ok(database)
}