systemprompt-cli 0.1.22

systemprompt.io OS - CLI for agent orchestration, AI operations, and system management
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

use clap::Args;
use systemprompt_extension::{ExtensionRegistry, SchemaSource};

use crate::CliConfig;
use crate::commands::plugins::types::{SchemaWithExtension, SchemasListOutput};
use crate::shared::CommandResult;

#[derive(Debug, Clone, Args)]
pub struct SchemasArgs {
    #[arg(long, help = "Filter by extension ID")]
    pub extension: Option<String>,
}

pub fn execute(args: &SchemasArgs, _config: &CliConfig) -> CommandResult<SchemasListOutput> {
    let registry = ExtensionRegistry::discover();

    let mut schemas: Vec<SchemaWithExtension> = registry
        .extensions()
        .iter()
        .filter(|ext| args.extension.as_ref().is_none_or(|f| ext.id().contains(f)))
        .flat_map(|ext| {
            ext.schemas()
                .iter()
                .map(|schema| {
                    let source = match &schema.sql {
                        SchemaSource::Inline(_) => "inline".to_string(),
                        SchemaSource::File(path) => path.display().to_string(),
                    };

                    SchemaWithExtension {
                        extension_id: ext.id().to_string(),
                        extension_name: ext.name().to_string(),
                        table: schema.table.clone(),
                        source,
                        migration_weight: ext.migration_weight(),
                    }
                })
                .collect::<Vec<_>>()
        })
        .collect();

    schemas.sort_by_key(|s| s.migration_weight);

    let total = schemas.len();

    let output = SchemasListOutput { schemas, total };

    CommandResult::table(output)
        .with_title("Schemas Across Extensions")
        .with_columns(vec![
            "extension_id".to_string(),
            "table".to_string(),
            "migration_weight".to_string(),
            "source".to_string(),
        ])
}