zilliz 1.4.3

TUI and CLI tool for managing Zilliz Cloud clusters and Milvus operations
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

pub mod app;
pub mod event;
pub mod handler;
pub mod home_fetch;
pub mod render;
pub mod views;
pub mod widgets;
pub mod wizard;

use anyhow::Result;
use crossterm::{
    event::{DisableMouseCapture, EnableMouseCapture},
    execute,
    terminal::{disable_raw_mode, enable_raw_mode, EnterAlternateScreen, LeaveAlternateScreen},
};
use ratatui::prelude::*;

use crate::config::manager::ConfigManager;
use crate::model::loader::Models;

use self::app::App;
use self::event::{Event, EventLoop};

/// Restores the terminal (raw mode, alternate screen, mouse capture) on every
/// exit path — including an error return or a panic inside the render/event
/// loop — so the user is never left with a corrupted terminal.
struct TerminalGuard;

impl Drop for TerminalGuard {
    fn drop(&mut self) {
        let _ = disable_raw_mode();
        let _ = execute!(std::io::stdout(), LeaveAlternateScreen, DisableMouseCapture);
    }
}

/// Main entry point for TUI mode.
pub async fn run(models: Models, config_mgr: ConfigManager) -> Result<()> {
    enable_raw_mode()?;
    let mut stdout = std::io::stdout();
    execute!(stdout, EnterAlternateScreen, EnableMouseCapture)?;
    // Past this point the guard owns terminal teardown; it runs on normal
    // return, `?` early-return, and unwinding panics alike.
    let _guard = TerminalGuard;
    let backend = CrosstermBackend::new(stdout);
    let mut terminal = Terminal::new(backend)?;

    let mut app = App::new(models, config_mgr);
    if app.auth.is_signed_in() {
        app.spawn_home_counts_fetch();
    }
    let event_loop = EventLoop::new(250);

    let result = run_loop(&mut terminal, &mut app, &event_loop);

    let _ = terminal.show_cursor();

    result
}

fn run_loop(
    terminal: &mut Terminal<CrosstermBackend<std::io::Stdout>>,
    app: &mut App,
    event_loop: &EventLoop,
) -> Result<()> {
    loop {
        terminal.draw(|frame| render::render(frame, app))?;

        // Drain any wizard messages (device-code ready, token received, errors)
        // before blocking on the next crossterm event so the UI updates promptly.
        while let Ok(msg) = app.msg_rx.try_recv() {
            handler::handle_wizard_msg(app, msg);
        }
        // Same for home-counts background updates.
        while let Ok(msg) = app.home_rx.try_recv() {
            app.apply_home_msg(msg);
        }

        match event_loop.next()? {
            Event::Key(key) => {
                handler::handle_key(app, key);
            }
            Event::Tick | Event::Resize(_, _) => {
                if app.auth.is_signed_in() && app.home_counts_refresh_due() {
                    app.spawn_home_counts_fetch();
                }
            }
        }

        if app.should_quit {
            // Cancel any in-flight wizard tasks so they don't run past quit.
            handler::cancel_wizard(app);
            app.abort_home_counts_fetch();
            break;
        }
    }

    Ok(())
}