promptpro 0.1.1

PromptPro is an advanced prompt management system with versioning, tagging, encryption, and TUI interface. The ultimate tool for prompt engineers and AI developers.
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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144

use clap::{Parser, Subcommand};

mod commands;
mod storage;
mod tui;
mod types;

use anyhow::Result;

#[derive(Parser)]
#[command(author, version, about, long_about = None)]
struct Cli {
    #[command(subcommand)]
    command: Commands,
}

#[derive(Subcommand)]
enum Commands {
    /// Initialize a new prompt vault
    Init {
        /// Path to the vault directory (default: ~/promptpro/default_vault)
        #[arg(long)]
        path: Option<String>,
    },
    /// Add a new prompt
    Add {
        /// Content of the prompt
        content: String,
    },
    /// Update an existing prompt
    Update {
        /// Key of the prompt to update
        key: String,
        /// New content of the prompt
        content: String,
        /// Optional message for the update
        #[arg(short, long)]
        message: Option<String>,
    },
    /// Get a prompt by key and selector
    Get {
        /// Key of the prompt
        key: String,
        /// Selector (version, tag, latest)
        selector: Option<String>,
        /// Output to file instead of stdout
        #[arg(short, long)]
        output: Option<String>,
    },
    /// Show history of a prompt
    History {
        /// Key of the prompt
        key: String,
    },
    /// Tag a specific version of a prompt
    Tag {
        /// Key of the prompt
        key: String,
        /// Tag name
        tag: String,
        /// Version number (optional, defaults to latest)
        version: Option<u64>,
    },
    /// Promote a tag to the latest version
    Promote {
        /// Key of the prompt
        key: String,
        /// Tag name to promote
        tag: String,
    },
    /// Open TUI editor
    Tui,
    /// Edit a prompt in TUI mode
    Edit {
        /// Key of the prompt to edit
        key: String,
    },
    /// Dump the vault to a binary file
    Dump {
        /// Output file path for the dump
        output: String,
        /// Password to encrypt the dump (optional)
        #[arg(long)]
        password: Option<String>,
    },
    /// Restore/Resume the vault from a binary file
    Resume {
        /// Input file path to restore from
        input: String,
        /// Password to decrypt the dump (optional)
        #[arg(long)]
        password: Option<String>,
    },
}

/// Main function for the CLI binary
#[tokio::main]
async fn main() -> Result<()> {
    let cli = Cli::parse();

    match cli.command {
        Commands::Init { path } => commands::init(path).await,
        Commands::Add { content } => commands::add(content).await,
        Commands::Update { key, content, message } => commands::update(key, content, message).await,
        Commands::Get { key, selector, output } => commands::get(key, selector, output).await,
        Commands::History { key } => commands::history(key).await,
        Commands::Tag { key, tag, version } => commands::tag(key, tag, version).await,
        Commands::Promote { key, tag } => commands::promote(key, tag).await,
        Commands::Tui => commands::tui().await,
        Commands::Edit { key } => commands::edit(key).await,
        Commands::Dump { output, password } => commands::dump(output, password).await,
        Commands::Resume { input, password } => commands::resume(input, password).await,
    }
}

/// Function that can be called from Python to run CLI commands with arguments
pub fn run_cli_from_args(args: Vec<String>) -> Result<()> {
    // Skip the first argument since it's typically the program name
    let cli_args = if !args.is_empty() {
        args
    } else {
        vec!["promptpro".to_string()] // Default to showing help if no args
    };
    
    // Parse the arguments using clap
    let cli = Cli::try_parse_from(cli_args)?;
    
    // Execute the command based on the parsed arguments
    tokio::runtime::Runtime::new()?.block_on(async {
        match cli.command {
            Commands::Init { path } => commands::init(path).await,
            Commands::Add { content } => commands::add(content).await,
            Commands::Update { key, content, message } => commands::update(key, content, message).await,
            Commands::Get { key, selector, output } => commands::get(key, selector, output).await,
            Commands::History { key } => commands::history(key).await,
            Commands::Tag { key, tag, version } => commands::tag(key, tag, version).await,
            Commands::Promote { key, tag } => commands::promote(key, tag).await,
            Commands::Tui => commands::tui().await,
            Commands::Edit { key } => commands::edit(key).await,
            Commands::Dump { output, password } => commands::dump(output, password).await,
            Commands::Resume { input, password } => commands::resume(input, password).await,
        }
    })
}