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

mod server;
mod stdin;
mod watcher;
use watcher::*;

use crate::{database, opt::Mode, Config, Opt};
use derive_more::{Display, From};
use entity::DatabaseRc;

/// Alias for a result with a program error
pub type ProgramResult<T, E = ProgramError> = std::result::Result<T, E>;

#[derive(Debug, Display, From)]
pub enum ProgramError {
    #[display(fmt = "Could not load database: {:?}", _0)]
    LoadDatabase(async_graphql::Error),
    #[display(fmt = "Could not start file watcher: {}", _0)]
    FileWatcher(notify::Error),
}

pub struct Program;

impl Program {
    /// Runs our program
    pub async fn run(opt: Opt, config: Config) -> ProgramResult<()> {
        // Load our database using the provided opturation and any
        // cached data from a previous run
        let database = database::load(&opt, &config)
            .await
            .map_err(ProgramError::from)?;

        // Initialize our watcher to update the database based on changes
        // that occur in wikis and standalone files
        let _watcher =
            Watcher::initialize(&config, DatabaseRc::clone(&database))
                .await
                .map_err(ProgramError::from)?;

        match opt.mode {
            Mode::Stdin => stdin::run(opt).await,
            Mode::Http => server::run(opt).await,
        }

        Ok(())
    }
}