ballista-cli 55.0.0

Command Line Client for Ballista distributed query engine.
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

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

mod cancel_result_popup;
mod components;
mod footer;
mod header;
mod help_overlay;
mod main;
mod scheduler_info_popup;
mod search_box;
mod vertical_scrollbar;

use super::{app::App, ui::header::render_header};
use footer::render_footer;
#[cfg(feature = "web")]
pub(crate) use main::dot_parser;
pub use main::{
    executor_details_popup, job_config_popup, job_dot_popup, job_plan_popup,
    job_stages_popup, render_executors, render_jobs, render_metrics, stage_plan_popup,
    stage_tasks_popup,
};
#[cfg(not(feature = "web"))]
pub use main::{
    load_executor_details_popup, load_executors_data, load_job_config_popup,
    load_job_details, load_job_dot, load_job_stages_popup, load_jobs_data,
    load_metrics_data, load_stage_plan,
};

use ratatui::widgets::Block;
use ratatui::{
    Frame,
    layout::{Constraint, Direction, Layout, Rect},
};

pub(crate) fn render(f: &mut Frame, app: &App) {
    let chunks = Layout::default()
        .direction(Direction::Vertical)
        .constraints([
            Constraint::Length(6), // Header
            Constraint::Min(0),    // Main view
            Constraint::Length(2), // Footer
        ])
        .split(f.area());

    let block = Block::default().style(app.theme.app_background);
    f.render_widget(block, f.area());

    render_header(f, chunks[0], app);
    render_main_view(f, app, chunks[1]);
    render_footer(f, chunks[2], app);

    // Overlay help if active
    if app.show_help {
        help_overlay::render_help_overlay(f, app);
    } else if app.show_scheduler_info {
        scheduler_info_popup::render_scheduler_info(f, app);
    } else if app.cancel_job_result.is_some() {
        cancel_result_popup::render_cancel_result_popup(f, app);
    } else if let Some(popup) = &app.job_stages_popup {
        if popup.is_tasks_view() {
            stage_tasks_popup::render_stage_tasks_popup(f, app);
        } else if popup.is_plan_view() {
            stage_plan_popup::render_stage_plan_popup(f, app);
        } else {
            job_stages_popup::render_job_stages_popup(f, app);
        }
    } else if app.job_dot_popup.is_some() {
        job_dot_popup::render_job_dot_popup(f, app);
    } else if app.job_plan_popup.is_some() {
        job_plan_popup::render_job_plan_popup(f, app);
    } else if app.job_config_popup.is_some() {
        job_config_popup::render_job_config_popup(f, app);
    } else if app.executor_details_popup.is_some() {
        executor_details_popup::render_executor_details_popup(f, app);
    }
}

fn render_main_view(f: &mut Frame, app: &App, area: Rect) {
    if app.is_executors_view() {
        render_executors(f, area, app);
    } else if app.is_jobs_view() {
        render_jobs(f, area, app);
    } else if app.is_metrics_view() {
        render_metrics(f, area, app);
    }
}

// Helper functions

pub(crate) fn centered_rect(percent_x: u16, percent_y: u16, r: Rect) -> Rect {
    let popup_layout = Layout::default()
        .direction(Direction::Vertical)
        .constraints([
            Constraint::Percentage((100 - percent_y) / 2),
            Constraint::Percentage(percent_y),
            Constraint::Percentage((100 - percent_y) / 2),
        ])
        .split(r);

    Layout::default()
        .direction(Direction::Horizontal)
        .constraints([
            Constraint::Percentage((100 - percent_x) / 2),
            Constraint::Percentage(percent_x),
            Constraint::Percentage((100 - percent_x) / 2),
        ])
        .split(popup_layout[1])[1]
}