systemprompt-cli 0.14.1

Unified CLI for systemprompt.io AI governance: agent orchestration, MCP governance, analytics, profiles, cloud deploy, and self-hosted operations.
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

use anyhow::{Context, Result, anyhow};
use std::sync::Arc;
use systemprompt_database::services::DatabaseProvider;
use systemprompt_database::{Database, MigrationService};
use systemprompt_extension::ExtensionRegistry;
use systemprompt_logging::CliService;
use systemprompt_models::Config;
use systemprompt_runtime::DatabaseContext;

use crate::cli_settings::CliConfig;
use crate::shared::{CommandResult, render_result};

use super::admin_migrate::select_extensions;
use super::types::MigrateMarkAppliedOutput;

#[derive(Clone, Copy)]
pub(super) struct MarkAppliedArgs<'a> {
    pub extension: &'a str,
    pub version: u32,
    pub json: bool,
}

pub(super) async fn execute_migrate_mark_applied(
    config: &CliConfig,
    args: MarkAppliedArgs<'_>,
) -> Result<()> {
    let sys_config = Config::get()?;
    let database = Arc::new(
        Database::from_config_with_write(
            &sys_config.database_type,
            &sys_config.database_url,
            sys_config.database_write_url.as_deref(),
        )
        .await
        .context("Failed to connect to database")?,
    );

    run_mark_applied(
        database.write(),
        &ExtensionRegistry::discover()?,
        args,
        config,
    )
    .await
}

pub(super) async fn execute_migrate_mark_applied_standalone(
    db_ctx: &DatabaseContext,
    config: &CliConfig,
    args: MarkAppliedArgs<'_>,
) -> Result<()> {
    run_mark_applied(
        db_ctx.db_pool().write(),
        &ExtensionRegistry::discover()?,
        args,
        config,
    )
    .await
}

async fn run_mark_applied(
    write_provider: &dyn DatabaseProvider,
    registry: &ExtensionRegistry,
    args: MarkAppliedArgs<'_>,
    config: &CliConfig,
) -> Result<()> {
    let MarkAppliedArgs {
        extension,
        version,
        json,
    } = args;

    let extensions = select_extensions(registry, Some(extension))?;
    let ext = extensions
        .into_iter()
        .next()
        .ok_or_else(|| anyhow!("Extension '{}' not found", extension))?;

    let migration_service = MigrationService::new(write_provider);
    let outcome = migration_service
        .mark_applied(ext.as_ref(), version)
        .await
        .map_err(|e| anyhow!("Failed to mark migration as applied: {}", e))?;

    let message = format!(
        "Recorded {} v{:03} '{}' as applied (checksum {})",
        outcome.extension_id,
        outcome.version,
        outcome.name,
        &outcome.checksum[..outcome.checksum.len().min(8)]
    );

    let output = MigrateMarkAppliedOutput {
        extension_id: outcome.extension_id,
        version: outcome.version,
        name: outcome.name,
        checksum: outcome.checksum,
        message: message.clone(),
    };

    if json || config.is_json_output() {
        let result = CommandResult::text(output).with_title("Migration Mark Applied");
        render_result(&result);
    } else {
        CliService::success(&message);
    }

    Ok(())
}