From 28f019b72c1027207ec8dc82844401512712464d Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 20 Mar 2023 16:51:06 +0800
Subject: feat: add `--info` (#79)

---
 src/cli.rs  | 3 +++
 src/main.rs | 5 +++++
 2 files changed, 8 insertions(+)

(limited to 'src')

diff --git a/src/cli.rs b/src/cli.rs
index 9211878..fd56e8b 100644
--- a/src/cli.rs
+++ b/src/cli.rs
@@ -24,6 +24,9 @@ pub struct Cli {
     /// Select a role
     #[clap(short, long)]
     pub role: Option<String>,
+    /// Print information
+    #[clap(long)]
+    pub info: bool,
     /// Input text
     text: Vec<String>,
 }
diff --git a/src/main.rs b/src/main.rs
index 589e865..58fa954 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -60,6 +60,11 @@ fn main() -> Result<()> {
     if let Some(prompt) = &cli.prompt {
         config.write().add_prompt(prompt)?;
     }
+    if cli.info {
+        let info = config.read().info()?;
+        println!("{info}");
+        exit(0);
+    }
     let no_stream = cli.no_stream;
     let client = ChatGptClient::init(config.clone())?;
     if atty::isnt(atty::Stream::Stdin) {
-- 
cgit v1.2.3