knowledge-base-cli 0.1.0

Command-line interface for a file-based knowledge base
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

pub mod entity;
pub mod entity_context;
pub mod entity_type;
pub mod property;
pub mod reference;
mod validate;

use clap::Subcommand;
use knowledge_base_crud::{Error, KnowledgeBase};
use std::fmt;
use std::io::{self, Write};
use std::path::Path;
use std::process::ExitCode;

#[derive(Debug, Subcommand)]
pub enum Command {
    /// Validate the configured knowledge base.
    Validate,
    /// Work with entities.
    Entity {
        #[command(subcommand)]
        command: entity::Command,
    },
    /// Work with entity types.
    EntityType {
        #[command(subcommand)]
        command: entity_type::Command,
    },
    /// Work with properties.
    Property {
        #[command(subcommand)]
        command: property::Command,
    },
    /// Work with references.
    Reference {
        #[command(subcommand)]
        command: reference::Command,
    },
    /// Work with entity context documents.
    EntityContext {
        #[command(subcommand)]
        command: entity_context::Command,
    },
}

#[derive(Debug)]
pub enum CommandError {
    Crud(Error),
    InvalidFilter(String),
    Serialization(serde_yaml::Error),
    Output(io::Error),
}

impl fmt::Display for CommandError {
    fn fmt(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::Crud(error) => error.fmt(formatter),
            Self::InvalidFilter(message) => formatter.write_str(message),
            Self::Serialization(error) => write!(formatter, "cannot serialize command output: {error}"),
            Self::Output(error) => write!(formatter, "cannot write command output: {error}"),
        }
    }
}

impl From<Error> for CommandError {
    fn from(error: Error) -> Self {
        Self::Crud(error)
    }
}

pub fn execute(command: Command, root: &Path) -> Result<ExitCode, CommandError> {
    let knowledge_base = KnowledgeBase::new(root);
    match command {
        Command::Validate => Ok(validate::execute(root)),
        Command::Entity { command } => entity::execute(command, &knowledge_base),
        Command::EntityType { command } => entity_type::execute(command, &knowledge_base),
        Command::Property { command } => property::execute(command, &knowledge_base),
        Command::Reference { command } => reference::execute(command, &knowledge_base),
        Command::EntityContext { command } => entity_context::execute(command, &knowledge_base),
    }
}

fn write_content(content: &str) -> Result<ExitCode, CommandError> {
    io::stdout().lock().write_all(content.as_bytes()).map_err(CommandError::Output)?;
    Ok(ExitCode::SUCCESS)
}