Add login command to set API token

Author: max-lt

README.md

@@ -43,6 +43,16 @@ On first run, a `default` alias pointing to `https://dash.openworkers.com/api/v1
 
 ## Commands
 
+### Login
+
+```bash
+# Set token on default alias
+ow login
+
+# Set token on specific alias
+ow dev login
+```
+
 ### Workers
 
 ```bash

src/commands/login.rs

@@ -0,0 +1,55 @@
+use crate::config::{AliasConfig, Config, ConfigError};
+use colored::Colorize;
+use std::io::{self, Write};
+
+pub fn run(alias_name: &str) -> Result<(), ConfigError> {
+    let mut config = Config::load()?;
+
+    // Get existing alias
+    let existing = config
+        .get_alias(alias_name)
+        .ok_or_else(|| ConfigError::AliasNotFound(alias_name.to_string()))?;
+
+    // Must be an API alias
+    let (url, insecure) = match existing {
+        AliasConfig::Api { url, insecure, .. } => (url.clone(), *insecure),
+        AliasConfig::Db { .. } => {
+            eprintln!(
+                "{} Alias '{}' is a database alias, not an API alias.",
+                "Error:".red(),
+                alias_name
+            );
+            return Ok(());
+        }
+    };
+
+    // Prompt for token
+    print!("Enter API token: ");
+    io::stdout().flush().unwrap();
+
+    let mut token = String::new();
+    io::stdin().read_line(&mut token).unwrap();
+    let token = token.trim().to_string();
+
+    if token.is_empty() {
+        eprintln!("{} Token cannot be empty.", "Error:".red());
+        return Ok(());
+    }
+
+    // Update alias with token
+    config.set_alias(
+        alias_name,
+        AliasConfig::api(url, Some(token), insecure),
+        true,
+    )?;
+
+    config.save()?;
+
+    println!(
+        "{} Token saved for alias '{}'.",
+        "Success:".green(),
+        alias_name.bold()
+    );
+
+    Ok(())
+}

src/commands/mod.rs

@@ -1,3 +1,4 @@
 pub mod alias;
 pub mod db;
+pub mod login;
 pub mod workers;

src/main.rs

@@ -30,6 +30,9 @@ enum Commands {
         command: AliasCommand,
     },
 
+    /// Set API token for an alias
+    Login,
+
     /// Database operations (requires db alias)
     Db {
         #[command(subcommand)]
@@ -59,6 +62,7 @@ fn extract_alias_from_args() -> (Option<String>, Vec<String>) {
 
     let known_commands = [
         "alias",
+        "login",
         "db",
         "workers",
         "help",
@@ -144,6 +148,13 @@ async fn main() {
 
     let result = match cli.command {
         Commands::Alias { command } => command.run().map_err(|e| e.to_string()),
+        Commands::Login => (|| {
+            let config = Config::load().map_err(|e| e.to_string())?;
+            let alias_name = alias
+                .or(config.default.clone())
+                .ok_or("No alias specified and no default configured".to_string())?;
+            commands::login::run(&alias_name).map_err(|e| e.to_string())
+        })(),
         Commands::Db { command } => command.run(alias).await.map_err(|e| e.to_string()),
         Commands::Workers { command } => run_workers_command(alias, command).await,
     };