rft-cli 0.5.2

Zero-config Docker Compose isolation for git worktrees
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

use std::path::PathBuf;
use std::time::Duration;

use notify::{EventKind, RecursiveMode, Watcher};
use owo_colors::OwoColorize;
use tokio::sync::mpsc;

use crate::compose::ComposeFile;
use crate::context::build_context;
use crate::error::Result;
use crate::sync::files::extract_dockerfile_paths;

const DEBOUNCE_MS: u64 = 500;

pub async fn run(indices: Vec<usize>) -> Result<()> {
    let context = build_context().await?;

    println!(
        "{} Starting stacks and watching for changes...",
        "Watch".cyan().bold()
    );

    super::start::run(indices.clone(), false).await?;

    let paths_to_watch = collect_watch_paths(&context.compose_file);

    println!(
        "{} Watching {} path(s) for changes. Press Ctrl+C to stop.",
        "Watch".cyan().bold(),
        paths_to_watch.len()
    );

    for path in &paths_to_watch {
        println!("  {} {}", "→".dimmed(), path.display().to_string().dimmed());
    }

    let (fs_sender, mut fs_receiver) = mpsc::channel::<()>(16);

    let _watcher = create_file_watcher(&paths_to_watch, fs_sender)?;

    loop {
        tokio::select! {
            _ = tokio::signal::ctrl_c() => {
                println!(
                    "\n{} Stopping all stacks...",
                    "Watch".cyan().bold()
                );
                let _ = super::stop::run(indices.clone()).await;
                return Ok(());
            }
            _ = wait_for_change(&mut fs_receiver) => {
                println!(
                    "\n{} File change detected, restarting stacks...",
                    "Watch".cyan().bold()
                );
                let _ = super::restart::run(indices.clone()).await;
            }
        }
    }
}

fn collect_watch_paths(compose_file: &ComposeFile) -> Vec<PathBuf> {
    let mut paths = Vec::new();

    paths.push(compose_file.compose_path.clone());

    for dockerfile_path in extract_dockerfile_paths(compose_file) {
        if let Some(parent) = dockerfile_path.parent() {
            let parent_buf = parent.to_path_buf();
            if !paths.contains(&parent_buf) {
                paths.push(parent_buf);
            }
        }
    }

    paths
}

fn create_file_watcher(paths: &[PathBuf], sender: mpsc::Sender<()>) -> Result<impl Watcher> {
    let mut watcher = notify::recommended_watcher(move |event: notify::Result<notify::Event>| {
        if let Ok(event) = event
            && matches!(
                event.kind,
                EventKind::Modify(_) | EventKind::Create(_) | EventKind::Remove(_)
            )
        {
            let _ = sender.try_send(());
        }
    })
    .map_err(|error| {
        crate::error::RftError::Config(format!("failed to create watcher: {error}"))
    })?;

    for path in paths {
        if path.exists() {
            let mode = if path.is_dir() {
                RecursiveMode::Recursive
            } else {
                RecursiveMode::NonRecursive
            };
            if let Err(error) = watcher.watch(path, mode) {
                eprintln!(
                    "{}",
                    format!("warning: cannot watch {}: {error}", path.display()).yellow()
                );
            }
        }
    }

    Ok(watcher)
}

async fn wait_for_change(receiver: &mut mpsc::Receiver<()>) {
    receiver.recv().await;

    // Debounce: drain additional events within the window
    let deadline = tokio::time::Instant::now() + Duration::from_millis(DEBOUNCE_MS);
    loop {
        tokio::select! {
            _ = receiver.recv() => {}
            _ = tokio::time::sleep_until(deadline) => break,
        }
    }
}