vipune 0.9.1

A minimal memory layer for AI agents
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

//! `vipune project merge <from> <to>` handler.
//!
//! Moves all rows from one project_id to another within a single transaction.

use crate::errors::Error;
use crate::output::{MergeResponse, print_json};
use crate::sqlite::Database;
use std::path::Path;
use std::process::ExitCode;
use std::time::Duration;

/// Wrap a database error, converting SQLITE_BUSY into the actionable MCP-server message.
fn wrap_busy<T>(result: Result<T, Error>) -> Result<T, Error> {
    match result {
        Ok(v) => Ok(v),
        Err(Error::SqliteModule(msg)) if msg.contains("database is locked") => {
            Err(Error::Config(
                "Database is locked. Another process (likely the MCP server) is holding a lock. Stop the MCP server and retry.".to_string()
            ))
        }
        Err(e) => Err(e),
    }
}

/// Build the MCP restart notice text.
///
/// This is a pure function so tests can verify the message content without
/// capturing stderr.
pub(crate) fn build_mcp_restart_notice() -> String {
    "Note: If a vipune MCP server is running, it holds its project_id from startup. Restart the MCP server to see rows under the new project id.".to_string()
}

/// Run the project merge operation.
///
/// # Arguments
///
/// * `db_path` - Path to the SQLite database
/// * `from_project_id` - Source project id to move rows from
/// * `to_project_id` - Target project id to move rows to
/// * `json` - If true, output JSON; otherwise human-readable
///
/// # Errors
///
/// Returns error if the database cannot be opened or the merge fails.
pub fn handle_merge(
    db_path: &Path,
    from_project_id: &str,
    to_project_id: &str,
    json: bool,
) -> Result<ExitCode, Error> {
    // Open database
    let mut db = Database::open(db_path).map_err(|e| {
        let err_msg = e.to_string();
        if err_msg.contains("database is locked") {
            return Error::Config(
                "Database is locked. Another process (likely the MCP server) is holding a lock. Stop the MCP server and retry.".to_string()
            );
        }
        Error::Config(err_msg)
    })?;

    // Set busy timeout to 0ms for fast-fail behavior on database locks
    wrap_busy(db.set_busy_timeout(Duration::ZERO).map_err(Error::from))?;

    let rows_moved = wrap_busy(
        db.merge_project_ids(from_project_id, to_project_id)
            .map_err(Error::from),
    )?;

    let response = MergeResponse {
        from: from_project_id.to_string(),
        to: to_project_id.to_string(),
        rows_moved,
    };

    if json {
        print_json(&response);
    } else {
        println!(
            "Merged {} row(s) from '{}' to '{}'",
            response.rows_moved, response.from, response.to
        );
    }

    // Warn that a running MCP server must be restarted to see the merged rows.
    // Only emitted when rows were actually moved — if nothing changed, no restart is needed.
    if response.rows_moved > 0 {
        eprintln!("\n{}", build_mcp_restart_notice());
    }

    Ok(ExitCode::SUCCESS)
}