cahier 0.1.4

A terminal session recorder and manager.
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

use anyhow::{Context, Result};
use clap::{Parser, Subcommand};
use std::io::Write;
use std::sync::{Arc, Mutex};

use cahier::{common, config, db, export, repl, tui};
use common::DEFAULT_MAX_OUTPUT_SIZE;

#[derive(Parser)]
#[command(author, version, about, long_about = None)]
struct Args {
    #[command(subcommand)]
    command: Option<Commands>,
}

#[derive(Subcommand)]
enum Commands {
    /// Start cahier REPL (default)
    Start {
        /// Maximum output size in bytes before redirecting to file (default: 16384)
        #[arg(long, default_value_t = DEFAULT_MAX_OUTPUT_SIZE)]
        max_output_size: usize,
    },
    /// Export history to markdown
    Export {
        /// Output file path (default: stdout)
        #[arg(short, long)]
        output: Option<String>,

        /// Export only the commands (plain text)
        #[arg(long)]
        only_commands: bool,
    },
    /// Open TUI editor for command history
    Edit,
}

fn main() -> Result<()> {
    common::init_base_dir();

    // Ensure cahier directory exists
    let cahier_dir = common::cahier_dir();
    if !cahier_dir.exists() {
        std::fs::create_dir_all(&cahier_dir).context("Failed to create cahier directory")?;
    }

    #[cfg(unix)]
    {
        use std::os::unix::fs::PermissionsExt;
        let metadata = std::fs::metadata(&cahier_dir)?;
        let mut perms = metadata.permissions();
        if perms.mode() & 0o777 != 0o700 {
            perms.set_mode(0o700);
            std::fs::set_permissions(&cahier_dir, perms)
                .context("Failed to set cahier directory permissions")?;
        }
    }

    let args = Args::parse();
    let db_path = common::db_path();
    let db = db::Database::init(&db_path).context("Failed to initialize database")?;
    let config = config::Config::load().context("Failed to load configuration")?;

    match args.command {
        Some(Commands::Export {
            output,
            only_commands,
        }) => {
            let content = if only_commands {
                export::generate_commands_text(&db).context("Failed to generate commands export")?
            } else {
                export::generate_markdown(&db).context("Failed to generate markdown export")?
            };

            if let Some(path) = output {
                std::fs::write(path, content).context("Failed to write export output")?;
            } else {
                println!("{}", content);
            }
            Ok(())
        }
        Some(Commands::Edit) => {
            if let Some(cmd) = tui::run(db)? {
                // User selected a command to edit in REPL
                // Re-initialize database because tui::run consumed it
                let db =
                    db::Database::init(&db_path).context("Failed to re-initialize database")?;
                let pty_writer = setup_signal_handler()?;
                repl::run_repl(db, DEFAULT_MAX_OUTPUT_SIZE, pty_writer, config, Some(cmd))
            } else {
                Ok(())
            }
        }
        Some(Commands::Start { max_output_size }) => {
            let pty_writer = setup_signal_handler()?;
            repl::run_repl(db, max_output_size, pty_writer, config, None)
        }
        None => {
            // Default behavior: start REPL with default max_output_size
            let pty_writer = setup_signal_handler()?;
            repl::run_repl(db, DEFAULT_MAX_OUTPUT_SIZE, pty_writer, config, None)
        }
    }
}

type PtyWriter = Arc<Mutex<Option<Box<dyn Write + Send>>>>;

/// Sets up the Ctrl+C signal handler and returns the shared PTY writer
fn setup_signal_handler() -> Result<PtyWriter> {
    let pty_writer: PtyWriter = Arc::new(Mutex::new(None));

    let writer_clone = Arc::clone(&pty_writer);
    ctrlc::set_handler(move || {
        if let Ok(mut writer_opt) = writer_clone.lock() {
            if let Some(writer) = writer_opt.as_mut() {
                // Send Ctrl+C (ETX) to the running command
                let _ = writer.write_all(&[3]);
                let _ = writer.flush();
            }
            // If no writer, do nothing (at prompt)
        }
    })
    .context("Failed to set Ctrl+C handler")?;

    Ok(pty_writer)
}