memex-cli 0.2.1

mod commands;
mod editor;
mod git;
mod models;
mod store;

use anyhow::Result;
use clap::{Parser, Subcommand};

use commands::context::OutputFormat;
use models::NodeStatus;

#[derive(Parser)]
#[command(
    name = "memex",
    about = "Organize development work into a versioned, navigable DAG of conversation nodes",
    version
)]
struct Cli {
    #[command(subcommand)]
    command: Commands,
}

#[derive(Subcommand)]
enum Commands {
    /// Initialize a new memex in the current directory
    Init,

    /// Manage conversation nodes
    Node {
        #[command(subcommand)]
        subcommand: NodeCommands,
    },

    /// View the conversation graph
    Graph {
        #[command(subcommand)]
        subcommand: GraphCommands,
    },

    /// Generate a context payload for the next conversation
    Context {
        /// Node ID (defaults to active node)
        id: Option<String>,

        /// Output format: markdown, xml, or plain
        #[arg(long, default_value = "markdown")]
        format: String,

        /// Max number of ancestor nodes to include (0 = none)
        #[arg(long, default_value_t = 2)]
        depth: usize,
    },

    /// Search node summaries
    Search {
        /// Query string to search for
        query: String,
    },

    /// Render a PR-comment payload listing memex nodes whose key_artifacts match the given files
    PrContext {
        /// Changed files (repo-relative paths). Repeatable.
        #[arg(long = "file", num_args = 1, action = clap::ArgAction::Append)]
        files: Vec<String>,

        /// HTML marker comment used by callers to find/update the comment idempotently
        #[arg(long, default_value = commands::pr_context::DEFAULT_MARKER)]
        marker: String,

        /// Cap on the number of nodes rendered (most recent first; 0 = unlimited)
        #[arg(long, default_value_t = commands::pr_context::DEFAULT_LIMIT)]
        limit: usize,
    },
}

#[derive(Subcommand)]
enum NodeCommands {
    /// Create a new conversation node
    Create {
        /// Parent node ID (defaults to active node)
        #[arg(long)]
        parent: Option<String>,

        /// Git ref (branch/SHA) to associate with this node
        #[arg(long, name = "git-ref")]
        git_ref: Option<String>,

        /// Tags to attach to this node
        #[arg(long, name = "tag", num_args = 1)]
        tags: Vec<String>,

        /// Node goal (skips editor; required for non-interactive use)
        #[arg(long)]
        goal: Option<String>,
    },

    /// Edit a node's summary in $EDITOR
    Edit {
        /// Node ID (defaults to active node)
        id: Option<String>,

        /// Summary as a TOML string (skips editor; for non-interactive/agent use)
        #[arg(long)]
        summary: Option<String>,

        /// Overwrite the goal text
        #[arg(long)]
        goal: Option<String>,

        /// Append a decision (repeatable)
        #[arg(long, num_args = 1, action = clap::ArgAction::Append)]
        decision: Vec<String>,

        /// Append a key artifact path or name (repeatable)
        #[arg(long, num_args = 1, action = clap::ArgAction::Append)]
        artifact: Vec<String>,

        /// Append an open thread (repeatable)
        #[arg(long = "open-thread", num_args = 1, action = clap::ArgAction::Append)]
        open_thread: Vec<String>,

        /// Append a rejected approach as TOML with description and reason fields (repeatable)
        #[arg(long, num_args = 1, action = clap::ArgAction::Append)]
        rejected: Vec<String>,
    },

    /// Apply agent-authored draft additions from a JSON payload on stdin
    Auto {
        /// Node ID (defaults to active node)
        id: Option<String>,

        /// Read the JSON payload from stdin (required)
        #[arg(long = "from-stdin")]
        from_stdin: bool,

        /// Write the merged additions to the node (mutually exclusive with --dry-run)
        #[arg(long, conflicts_with = "dry_run")]
        apply: bool,

        /// Print the diff against the active node without writing (default)
        #[arg(long = "dry-run", conflicts_with = "apply")]
        dry_run: bool,
    },

    /// Show a node's full summary
    Show {
        /// Node ID (defaults to active node)
        id: Option<String>,
    },

    /// List all nodes
    List,

    /// Mark a node as resolved
    Resolve {
        /// Node ID (defaults to active node)
        id: Option<String>,

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

    /// Mark a node as abandoned
    Abandon {
        /// Node ID (defaults to active node)
        id: Option<String>,

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

    /// Reopen a resolved or abandoned node (set back to Active)
    Reopen {
        /// Node ID (defaults to active node)
        id: Option<String>,
    },
}

#[derive(Subcommand)]
enum GraphCommands {
    /// Render the graph as an ASCII tree
    View,
}

fn main() {
    let cli = Cli::parse();
    if let Err(e) = run(cli) {
        eprintln!("Error: {:#}", e);
        std::process::exit(1);
    }
}

fn run(cli: Cli) -> Result<()> {
    match cli.command {
        Commands::Init => commands::init::run(),

        Commands::Node { subcommand } => match subcommand {
            NodeCommands::Create {
                parent,
                git_ref,
                tags,
                goal,
            } => commands::node::create(
                parent.as_deref(),
                git_ref.as_deref(),
                &tags,
                goal.as_deref(),
            ),
            NodeCommands::Edit {
                id,
                summary,
                goal,
                decision,
                artifact,
                open_thread,
                rejected,
            } => commands::node::edit(
                id.as_deref(),
                summary.as_deref(),
                goal.as_deref(),
                &decision,
                &artifact,
                &open_thread,
                &rejected,
            ),
            NodeCommands::Auto {
                id,
                from_stdin,
                apply,
                dry_run: _,
            } => commands::node::auto(id.as_deref(), from_stdin, apply),
            NodeCommands::Show { id } => commands::node::show(id.as_deref()),
            NodeCommands::List => commands::node::list(),
            NodeCommands::Resolve { id, force } => {
                commands::node::set_status(id.as_deref(), NodeStatus::Resolved, force)
            }
            NodeCommands::Abandon { id, force } => {
                commands::node::set_status(id.as_deref(), NodeStatus::Abandoned, force)
            }
            NodeCommands::Reopen { id } => {
                commands::node::set_status(id.as_deref(), NodeStatus::Active, true)
            }
        },

        Commands::Graph { subcommand } => match subcommand {
            GraphCommands::View => commands::graph::view(),
        },

        Commands::Context { id, format, depth } => {
            let fmt = OutputFormat::from_str(&format)?;
            commands::context::run(id.as_deref(), fmt, depth)
        }

        Commands::Search { query } => commands::search::run(&query),

        Commands::PrContext {
            files,
            marker,
            limit,
        } => commands::pr_context::run(&files, &marker, limit),
    }
}