bcode_tui 0.0.1-alpha.0

Terminal user interface for bcode
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

//! Eval TUI plugin launcher.

use std::io::Write;
use std::path::PathBuf;

use bmux_tui::terminal::Terminal;

use crate::TuiError;

const EVAL_PLUGIN_ID: &str = "bcode.eval";
const EVAL_RUN_PICKER_SURFACE_KIND: &str = "eval-run-picker";
const EVAL_RUN_VIEWER_SURFACE_KIND: &str = "eval-run-viewer";
const DEFAULT_RUNS_ROOT: &str = "target/bcode-evals/runs";

/// Run the eval run picker surface.
///
/// # Errors
///
/// Returns an error when the eval plugin cannot be loaded/opened or terminal I/O fails.
#[allow(clippy::future_not_send)]
pub async fn run_picker<W: Write>(
    terminal: &mut Terminal<&mut W>,
    repo_path: PathBuf,
) -> Result<(), TuiError> {
    Box::pin(run_surface(
        terminal,
        repo_path,
        EVAL_RUN_PICKER_SURFACE_KIND,
        "eval-run-picker".to_string(),
        serde_json::json!({ "runs_root": DEFAULT_RUNS_ROOT }),
    ))
    .await
}

/// Run the eval run viewer surface for an optional run path.
///
/// When `run` is `None`, the picker is opened instead.
///
/// # Errors
///
/// Returns an error when the eval plugin cannot be loaded/opened or terminal I/O fails.
#[allow(clippy::future_not_send)]
pub async fn run_viewer<W: Write>(
    terminal: &mut Terminal<&mut W>,
    repo_path: PathBuf,
    run: Option<PathBuf>,
) -> Result<(), TuiError> {
    let (surface_kind, instance_id, options) = run.map_or_else(
        || {
            (
                EVAL_RUN_PICKER_SURFACE_KIND,
                "eval-run-picker".to_string(),
                serde_json::json!({ "runs_root": DEFAULT_RUNS_ROOT }),
            )
        },
        |run_path| {
            (
                EVAL_RUN_VIEWER_SURFACE_KIND,
                format!("eval-run-viewer:{}", run_path.display()),
                serde_json::json!({ "run_path": run_path, "runs_root": DEFAULT_RUNS_ROOT }),
            )
        },
    );
    Box::pin(run_surface(
        terminal,
        repo_path,
        surface_kind,
        instance_id,
        options,
    ))
    .await
}

#[allow(clippy::future_not_send)]
async fn run_surface<W: Write>(
    terminal: &mut Terminal<&mut W>,
    repo_path: PathBuf,
    surface_kind: &str,
    instance_id: String,
    options: serde_json::Value,
) -> Result<(), TuiError> {
    let runtime = load_eval_tui_runtime()?;
    let surface = crate::plugin_tui::open_plugin_tui_surface(
        &runtime,
        EVAL_PLUGIN_ID,
        surface_kind,
        bcode_plugin_sdk::tui::PluginTuiSurfaceOpenRequest {
            instance_id,
            repo_path: Some(repo_path),
            session_id: None,
            target: None,
            options,
        },
    )
    .await
    .map_err(|error| TuiError::PluginService {
        code: "tui_surface_open_failed".to_string(),
        message: error.to_string(),
    })?;
    let outcome = Box::pin(crate::runtime::run_standalone_plugin_surface(
        terminal,
        EVAL_PLUGIN_ID,
        surface,
    ))
    .await;
    drop(runtime);
    outcome?;
    Ok(())
}

fn load_eval_tui_runtime() -> Result<bcode_plugin::PluginRuntimeHost, TuiError> {
    bcode_plugin::PluginRuntimeHost::load_defaults_with_static_bundled(
        &bcode_plugin::PluginSelection::all_enabled(),
        &crate::static_bundled_plugins(),
    )
    .map_err(|error| TuiError::PluginService {
        code: "plugin_runtime_load_failed".to_string(),
        message: error.to_string(),
    })
}