fyrer 0.1.9

A lightweight tool to run multiple dev servers concurrently
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

use crate::config::Service;
use crate::kill_process::kill_process;
use crate::spawn_service::spawn_service;
use colored::Colorize;
use globset::GlobSetBuilder;
use notify::{Config, RecommendedWatcher, RecursiveMode, Watcher};
use std::fs;
use std::path::Path;
use tokio::process::Child;
use tokio::{
    sync::mpsc,
    time::{Duration, sleep},
};

pub async fn run_with_watch(service: Service, color: colored::Color, max_name_len: usize) {
    let (tx, mut rx) = mpsc::channel(1);
    let watch_dir = Path::new(&service.dir);

    let abs_service_dir =
        fs::canonicalize(&service.dir).unwrap_or_else(|_| Path::new(&service.dir).to_path_buf());
    let ignore = service.clone().ignore.unwrap_or_else(Vec::new);

    let mut builder = GlobSetBuilder::new();
    for pattern in &ignore {
        if let Ok(glob) = globset::Glob::new(pattern) {
            builder.add(glob);
        } else {
            eprintln!("Invalid ignore pattern: {}", pattern);
        }
    }
    let glob_set = builder.build().unwrap();

    let _watcher = {
        let mut watcher = RecommendedWatcher::new(
            move |res: Result<notify::Event, notify::Error>| match res {
                Ok(event) => {
                    if !event.paths.is_empty() {
                        let _ = tx.try_send(event.paths.clone());
                    }
                }
                Err(e) => {
                    eprintln!("Watch error: {:?}", e);
                }
            },
            Config::default(),
        )
        .unwrap_or_else(|e| {
            eprintln!("Failed to create file watcher: {}", e);
            std::process::exit(1);
        });

        watcher
            .watch(watch_dir, RecursiveMode::Recursive)
            .unwrap_or_else(|e| {
                eprintln!("Failed to watch directory {}: {}", service.dir, e);
                std::process::exit(1);
            });
        watcher
    };

    let out_prefix = format!("[{}]", service.name);
    let padded_name = format!("{:<width$}", out_prefix.clone(), width = max_name_len);

    println!(
        "├─{} ➤ Watching {} for changes...",
        padded_name.color(color).bold(),
        service.dir
    );

    let mut child: Option<Child> = spawn_service(&service, color, false, max_name_len).await;

    loop {
        tokio::select! {
            changed_files = rx.recv() => {
                if let Some(paths) = changed_files {
                    let filtered_paths: Vec<_> = paths.into_iter()
                        .filter(|path| {
                            let rel_path = path.strip_prefix(&abs_service_dir).unwrap_or(path);
                            !glob_set.is_match(rel_path)
                        })
                        .collect();

                    if filtered_paths.is_empty() {
                        continue;
                    }

                    if let Some(mut c) = child.take() {
                        let padded_name = format!("{:<width$}", out_prefix.clone(), width = max_name_len);
                        println!(
                            "├─{} ➤ File changed, restarting service...",
                            padded_name.color(color).bold()
                        );
                        kill_process(&mut c).await;
                        sleep(Duration::from_millis(500)).await;
                    }

                    child = spawn_service(&service, color, false, max_name_len).await;
                    sleep(Duration::from_millis(500)).await;

                    // drain remaining events
                    while rx.try_recv().is_ok() {}
                }
            }

            _ = async {
                if let Some(ref mut c) = child {
                    c.wait().await
                } else {
                    std::future::pending().await
                }
            } => {
                let padded_name = format!("{:<width$}", out_prefix.clone(), width = max_name_len);
                println!(
                    "├─{} ➤ Service exited, restarting...",
                    padded_name.color(color).bold()
                );
                sleep(Duration::from_millis(1000)).await;
                child = spawn_service(&service, color, false, max_name_len).await;
            }
        }
    }
}