spikes 0.3.2

mod auth;
mod commands;
mod config;
mod error;
mod output;
mod spike;
mod storage;

use clap::{Parser, Subcommand};
use commands::delete::DeleteOptions;
use commands::deploy::DeployOptions;
use commands::export::ExportFormat;
use commands::inject::InjectOptions;
use commands::list::ListOptions;
use commands::login::LoginOptions;
use commands::pull::PullOptions;
use commands::push::PushOptions;
use commands::resolve::ResolveOptions;
use commands::serve::ServeOptions;
use commands::share::ShareOptions;
use commands::shares::SharesOptions;
use commands::unshare::UnshareOptions;
use commands::usage::UsageOptions;

#[derive(Parser)]
#[command(name = "spikes")]
#[command(about = "Feedback collection for static mockups", long_about = None)]
#[command(version)]
struct Cli {
    #[command(subcommand)]
    command: Option<Commands>,

    /// Port for dev server (magic mode)
    #[arg(long, short, default_value = "3847", global = true)]
    port: u16,
}

#[derive(Subcommand)]
enum Commands {
    /// Initialize a .spikes/ directory
    Init {
        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// List all spikes
    List {
        /// Output as JSON
        #[arg(long)]
        json: bool,

        /// Filter by page name
        #[arg(long)]
        page: Option<String>,

        /// Filter by reviewer name
        #[arg(long)]
        reviewer: Option<String>,

        /// Filter by rating (love, like, meh, no)
        #[arg(long)]
        rating: Option<String>,

        /// Show only unresolved spikes
        #[arg(long)]
        unresolved: bool,
    },

    /// Show a single spike by ID
    Show {
        /// Spike ID (or prefix)
        id: String,

        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Export all spikes
    Export {
        /// Output format: json, csv, or jsonl
        #[arg(long, short, default_value = "json")]
        format: String,
    },

    /// Show elements with most feedback
    Hotspots {
        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// List all reviewers who left feedback
    Reviewers {
        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Add widget script tag to HTML files
    Inject {
        /// Directory containing HTML files
        directory: String,

        /// Remove widget script tags instead of adding
        #[arg(long)]
        remove: bool,

        /// URL for widget script (default: /spikes.js for local serve)
        #[arg(long)]
        widget_url: Option<String>,

        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Start local development server
    Serve {
        /// Port to listen on (default: 3847)
        #[arg(long, short, default_value = "3847")]
        port: u16,

        /// Directory to serve (default: current directory)
        #[arg(long, short, default_value = ".")]
        dir: String,

        /// Enable review mode with spike markers on pages
        #[arg(long, short)]
        marked: bool,

        /// Allowed CORS origin (e.g., https://spikes.sh). Without this flag, CORS is disabled (same-origin only).
        #[arg(long)]
        cors_allow_origin: Option<String>,
    },

    /// Deploy backend to Cloudflare
    Deploy {
        #[command(subcommand)]
        backend: DeployBackend,
    },

    /// Fetch spikes from remote and merge with local
    Pull {
        /// Remote endpoint URL (or from .spikes/config.toml)
        #[arg(long)]
        endpoint: Option<String>,

        /// Auth token (or from .spikes/config.toml)
        #[arg(long)]
        token: Option<String>,

        /// Pull from a public share URL (e.g., https://spikes.sh/s/project-slug)
        #[arg(long)]
        from: Option<String>,

        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Upload local spikes to remote
    Push {
        /// Remote endpoint URL (or from .spikes/config.toml)
        #[arg(long)]
        endpoint: Option<String>,

        /// Auth token (or from .spikes/config.toml)
        #[arg(long)]
        token: Option<String>,

        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Sync with remote (pull then push)
    Sync {
        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Manage remote endpoint configuration
    Remote {
        #[command(subcommand)]
        action: RemoteAction,
    },

    /// Show current configuration
    Config {
        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Show version
    Version,

    /// Update spikes CLI and widget to the latest version
    Update,

    /// Log in to spikes.sh hosted service
    Login {
        /// Auth token (or enter interactively)
        #[arg(long)]
        token: Option<String>,

        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Log out from spikes.sh
    Logout {
        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Show current user identity
    Whoami {
        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Upload a directory to spikes.sh for instant sharing
    Share {
        /// Directory to upload
        directory: String,

        /// Custom name for the share URL
        #[arg(long)]
        name: Option<String>,

        /// Password-protect the share (Pro only)
        #[arg(long)]
        password: Option<String>,

        /// Host URL for the API
        #[arg(long, default_value = "https://spikes.sh")]
        host: String,

        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// List your shared projects on spikes.sh
    Shares {
        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Delete a shared project from spikes.sh
    Unshare {
        /// Share slug to delete
        slug: String,

        /// Skip confirmation prompt
        #[arg(long, short)]
        force: bool,

        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Delete a spike from local storage
    Delete {
        /// Spike ID or prefix (minimum 4 characters)
        id: String,

        /// Skip confirmation prompt
        #[arg(long, short)]
        force: bool,

        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Mark a spike as resolved
    Resolve {
        /// Spike ID or prefix (minimum 4 characters)
        id: String,

        /// Mark as unresolved instead
        #[arg(long)]
        unresolve: bool,

        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Open Stripe billing portal in browser
    Billing {
        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Upgrade to Pro subscription via Stripe checkout
    Upgrade {
        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Display current usage statistics
    Usage {
        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// MCP (Model Context Protocol) server for agent integration
    Mcp {
        #[command(subcommand)]
        action: McpAction,
    },

    /// Manage API keys for agent authentication
    Auth {
        #[command(subcommand)]
        action: AuthAction,
    },
}

#[derive(Subcommand)]
enum DeployBackend {
    /// Scaffold Cloudflare Worker + D1 for multi-reviewer sync
    Cloudflare {
        /// Output directory (default: ./spikes-worker)
        #[arg(long)]
        dir: Option<String>,

        /// Output as JSON
        #[arg(long)]
        json: bool,
    },
}

#[derive(Subcommand)]
enum McpAction {
    /// Start the MCP server (stdio transport by default)
    Serve {
        /// Use hosted API instead of local JSONL (requires SPIKES_TOKEN or auth.toml)
        #[arg(long)]
        remote: bool,

        /// Transport mode: stdio or http (default: stdio)
        #[arg(long, value_enum, default_value = "stdio")]
        transport: McpTransport,

        /// Port for HTTP transport (default: 3848, only used with --transport http)
        #[arg(long, default_value = "3848")]
        port: u16,

        /// Bind address for HTTP transport (default: 127.0.0.1, only used with --transport http)
        #[arg(long, default_value = "127.0.0.1")]
        bind: String,
    },

    /// Generate MCP config for Claude Desktop, Cursor, or other clients
    Install {
        /// Output as JSON
        #[arg(long)]
        json: bool,
    },
}

/// MCP transport mode
#[derive(Clone, Debug, clap::ValueEnum)]
enum McpTransport {
    /// Use standard input/output for JSON-RPC (default)
    Stdio,
    /// Use HTTP transport with POST endpoint
    Http,
}

#[derive(Subcommand)]
enum AuthAction {
    /// Create a new API key for agent authentication
    CreateKey {
        /// Optional name/label for the key
        #[arg(long)]
        name: Option<String>,

        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// List all API keys
    ListKeys {
        /// Output as JSON
        #[arg(long)]
        json: bool,
    },

    /// Revoke an API key
    RevokeKey {
        /// The key ID to revoke (e.g. key_abc123)
        key_id: String,

        /// Output as JSON
        #[arg(long)]
        json: bool,
    },
}

#[derive(Subcommand)]
enum RemoteAction {
    /// Add or update remote endpoint
    Add {
        /// Endpoint URL
        endpoint: String,

        /// Auth token
        #[arg(long)]
        token: Option<String>,

        /// Use spikes.sh hosted backend
        #[arg(long)]
        hosted: bool,
    },

    /// Remove remote configuration
    Remove,

    /// Show current remote configuration
    Show {
        /// Output as JSON
        #[arg(long)]
        json: bool,
    },
}

fn main() {
    let cli = Cli::parse();

    let result = match cli.command {
        // Magic mode: no subcommand = auto-serve current directory
        None => commands::magic::run(cli.port),
        Some(Commands::Init { json }) => commands::init::run(json),
        Some(Commands::List {
            json,
            page,
            reviewer,
            rating,
            unresolved,
        }) => commands::list::run(ListOptions {
            json,
            page,
            reviewer,
            rating,
            unresolved,
        }),
        Some(Commands::Show { id, json }) => commands::show::run(&id, json),
        Some(Commands::Export { format }) => {
            let fmt = match format.parse::<ExportFormat>() {
                Ok(f) => f,
                Err(e) => {
                    eprintln!("Error: {}", e);
                    std::process::exit(1);
                }
            };
            commands::export::run(fmt)
        }
        Some(Commands::Hotspots { json }) => commands::hotspots::run(json),
        Some(Commands::Reviewers { json }) => commands::reviewers::run(json),
        Some(Commands::Inject {
            directory,
            remove,
            widget_url,
            json,
        }) => commands::inject::run(InjectOptions {
            directory,
            remove,
            widget_url,
            json,
        }),
        Some(Commands::Serve { port, dir, marked, cors_allow_origin }) => commands::serve::run(ServeOptions {
            port,
            directory: dir,
            marked,
            cors_allow_origin,
        }),
        Some(Commands::Deploy { backend }) => match backend {
            DeployBackend::Cloudflare { dir, json } => {
                commands::deploy::run(DeployOptions { dir, json })
            }
        },
        Some(Commands::Pull {
            endpoint,
            token,
            from,
            json,
        }) => commands::pull::run(PullOptions {
            endpoint,
            token,
            from,
            json,
        }),
        Some(Commands::Push {
            endpoint,
            token,
            json,
        }) => commands::push::run(PushOptions {
            endpoint,
            token,
            json,
        }),
        Some(Commands::Sync { json }) => commands::sync::run(json),
        Some(Commands::Remote { action }) => match action {
            RemoteAction::Add { endpoint, token, hosted } => {
                commands::remote::add(&endpoint, token, hosted)
            }
            RemoteAction::Remove => commands::remote::remove(),
            RemoteAction::Show { json } => commands::remote::show(json),
        },
        Some(Commands::Config { json }) => commands::config_cmd::run(json),
        Some(Commands::Version) => {
            println!("spikes {}", env!("CARGO_PKG_VERSION"));
            Ok(())
        }
        Some(Commands::Update) => commands::update::run(),
        Some(Commands::Login { token, json }) => commands::login::run(LoginOptions { token, json }),
        Some(Commands::Logout { json }) => commands::logout::run(json),
        Some(Commands::Whoami { json }) => commands::whoami::run(json),
        Some(Commands::Share { directory, name, password, host, json }) => {
            commands::share::run(ShareOptions { directory, name, password, host, json })
        }
        Some(Commands::Shares { json }) => commands::shares::run(SharesOptions { json }),
        Some(Commands::Unshare { slug, force, json }) => {
            commands::unshare::run(UnshareOptions { slug, force, json })
        }
        Some(Commands::Delete { id, force, json }) => {
            commands::delete::run(DeleteOptions { id, force, json })
        }
        Some(Commands::Resolve { id, unresolve, json }) => {
            commands::resolve::run(ResolveOptions { id, unresolve, json })
        }
        Some(Commands::Billing { json }) => commands::billing::run(json),
        Some(Commands::Upgrade { json }) => commands::upgrade::run(json),
        Some(Commands::Usage { json }) => commands::usage::run(UsageOptions { json }),
        Some(Commands::Mcp { action }) => match action {
            McpAction::Serve { remote, transport, port, bind } => {
                let transport_mode = match transport {
                    McpTransport::Stdio => commands::mcp::TransportMode::Stdio,
                    McpTransport::Http => commands::mcp::TransportMode::Http { port, bind },
                };
                commands::mcp::run(remote, transport_mode)
            }
            McpAction::Install { json } => {
                commands::mcp::install(json)
            }
        },
        Some(Commands::Auth { action }) => match action {
            AuthAction::CreateKey { name, json } => {
                commands::auth_keys::create_key(name, json)
            }
            AuthAction::ListKeys { json } => {
                commands::auth_keys::list_keys(json)
            }
            AuthAction::RevokeKey { key_id, json } => {
                commands::auth_keys::revoke_key(&key_id, json)
            }
        },
    };

    if let Err(e) = result {
        eprintln!("Error: {}", e);
        std::process::exit(1);
    }
}