objectiveai-cli 2.0.11

ObjectiveAI command-line interface and embeddable library
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

use clap::Subcommand;
use objectiveai_sdk::agent::favorites::{Action, ChangedNotification};

#[derive(Subcommand)]
pub enum Commands {
    /// Get all favorites
    Get,
    /// Add a favorite
    Add {
        #[command(flatten)]
        args: crate::favorite::AddFavorite,
    },
    /// Delete a favorite by name
    Del { name: String },
    /// Edit a favorite
    Edit {
        #[command(flatten)]
        args: crate::favorite::EditFavorite,
    },
}

impl Commands {
    pub async fn handle(self, cli_config: &crate::Config, handle: &objectiveai_sdk::cli::output::Handle) -> Result<(), crate::error::Error> {
        let (client, mut config) = crate::config::read(cli_config).await?;
        match self {
            Commands::Get => {
                crate::config::emit_value(&config.agents().get_favorites(), handle).await;
                Ok(())
            }
            Commands::Add { args } => {
                let favorite = args.into_favorite()?;
                let name = favorite.get_name().to_string();
                config.agents().add_favorite(favorite);
                crate::config::write(&client, &config, cli_config).await?;
                notify(&mut config, Action::Added, name).await;
                emit_ok(handle).await;
                Ok(())
            }
            Commands::Del { name } => {
                config.agents().del_favorite(&name)?;
                crate::config::write(&client, &config, cli_config).await?;
                notify(&mut config, Action::Removed, name).await;
                emit_ok(handle).await;
                Ok(())
            }
            Commands::Edit { args } => {
                let name = args.name.clone();
                let favorite = config.agents().edit_favorite(&args.name)?;
                args.apply(favorite)?;
                crate::config::write(&client, &config, cli_config).await?;
                notify(&mut config, Action::Edited, name).await;
                emit_ok(handle).await;
                Ok(())
            }
        }
    }
}

async fn notify(
    config: &mut objectiveai_sdk::filesystem::config::Config,
    action: Action,
    name: String,
) {
    let client = crate::api::client::build_viewer_client(config);
    client.send_agents_favorites_changed(
        None,
        None,
        ChangedNotification { action, name },
    );
    client.flush().await;
}

async fn emit_ok(handle: &objectiveai_sdk::cli::output::Handle) {
    objectiveai_sdk::cli::output::Output::<objectiveai_sdk::cli::output::Ok>::Notification(
        objectiveai_sdk::cli::output::Notification {
            agent_id: None,
            value: objectiveai_sdk::cli::output::OK,
        },
    )
    .emit(handle)
    .await;
}