From dc31386cf208fdaa9110a43fc348d84fa8e368ab Mon Sep 17 00:00:00 2001
From: syui <syui@syui.ai>
Date: Tue, 3 Mar 2026 15:56:33 +0900
Subject: [PATCH] fix ver

---
 src/main.rs | 27 ++++++++++++++++++++++-----
 1 file changed, 22 insertions(+), 5 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 1e4990a..2ff69c2 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -8,15 +8,22 @@ use aigpt::mcp::MCPServer;
 #[derive(Parser)]
 #[command(name = "aigpt")]
 #[command(about = "AI memory MCP server")]
-#[command(version)]
+#[command(disable_version_flag = true)]
 struct Cli {
+    #[arg(short = 'v', long = "version")]
+    version: bool,
+
     #[command(subcommand)]
     command: Option<Commands>,
 }
 
 #[derive(Subcommand)]
 enum Commands {
-    /// Initial setup: clone repo, link config, register MCP
+    /// Show version
+    #[command(name = "v")]
+    Version,
+
+    /// Initial setup: link config, register MCP
     Setup,
 
     /// Start MCP server (JSON-RPC over stdio)
@@ -38,8 +45,18 @@ enum Commands {
 fn main() -> Result<()> {
     let cli = Cli::parse();
 
-    if let Some(Commands::Setup) = &cli.command {
-        return run_setup();
+    if cli.version {
+        println!("{}", env!("CARGO_PKG_VERSION"));
+        return Ok(());
+    }
+
+    match &cli.command {
+        Some(Commands::Version) => {
+            println!("{}", env!("CARGO_PKG_VERSION"));
+            return Ok(());
+        }
+        Some(Commands::Setup) => return run_setup(),
+        _ => {}
     }
 
     config::init();
@@ -75,7 +92,7 @@ fn main() -> Result<()> {
             println!("Saved. ({} records)", reader::memory_count());
         }
 
-        Some(Commands::Setup) => unreachable!(),
+        Some(Commands::Version) | Some(Commands::Setup) => unreachable!(),
     }
 
     Ok(())