summaryrefslogtreecommitdiff
path: root/src/systems/cmd/processer.rs
blob: 2b66eef887571e966e10ea206f3d8b6bfbfa8a55 (plain)
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
use just_progress::progress;
use log::{error, info, warn};
use rust_i18n::t;

use crate::systems::cmd::_commands::{jv_cmd_nodes, jv_cmd_process_node};
use crate::systems::cmd::cmd_system::JVCommandContext;
use crate::systems::cmd::errors::CmdProcessError;
use crate::systems::render::renderer::JVRenderResult;

pub async fn jv_cmd_process(
    args: &[String],
    ctx: JVCommandContext,
    renderer_override: String,
) -> Result<JVRenderResult, CmdProcessError> {
    let result = process(args, ctx, renderer_override).await;

    // Regardless of the result, the progress bar output should be terminated after the command finishes running.
    // Otherwise, the program will block on the progress bar output.
    progress::clear_all();
    progress::close();
    result
}

async fn process(
    args: &[String],
    ctx: JVCommandContext,
    renderer_override: String,
) -> Result<JVRenderResult, CmdProcessError> {
    info!("{}", t!("verbose.cmd_process_start"));

    let nodes = jv_cmd_nodes();

    // Why add a space?
    // Add a space at the end of the command string for precise command prefix matching.
    // For example: when the input command is "bananas", if there are two commands "banana" and "bananas",
    // without a space it might incorrectly match "banana" (because "bananas".starts_with("banana") is true).
    // After adding a space, "bananas " will not match "banana ", thus avoiding ambiguity caused by overlapping prefixes.
    let command = format!("{} ", args.join(" "));

    // Find all nodes that match the command prefix
    let matching_nodes: Vec<&String> = nodes
        .iter()
        // Also add a space to the node string to ensure consistent matching logic
        .filter(|node| command.starts_with(&format!("{} ", node)))
        .collect();

    match matching_nodes.len() {
        0 => {
            // No matching node found
            error!("{}", t!("verbose.cmd_match_no_node"));
            Err(CmdProcessError::NoMatchingCommand)
        }
        1 => {
            let matched_prefix = matching_nodes[0];

            // DEBUG
            info!(
                "{}",
                t!("verbose.cmd_match_matched_single", node = matched_prefix)
            );

            let prefix_len = matched_prefix.split_whitespace().count();
            let trimmed_args: Vec<String> = args.iter().skip(prefix_len).cloned().collect();
            return jv_cmd_process_node(matched_prefix, trimmed_args, ctx, renderer_override).await;
        }
        _ => {
            // Multiple matching nodes found
            // Find the node with the longest length (most specific match)
            let matched_prefix = matching_nodes.iter().max_by_key(|node| node.len()).unwrap();

            // DEBUG
            let nodes_str = matching_nodes
                .iter()
                .map(|s| s.as_str())
                .collect::<Vec<_>>()
                .join(", ");
            warn!(
                "{}",
                t!(
                    "verbose.cmd_match_matched_multi",
                    nodes = nodes_str,
                    node = matched_prefix
                )
            );

            let prefix_len = matched_prefix.split_whitespace().count();
            let trimmed_args: Vec<String> = args.iter().skip(prefix_len).cloned().collect();
            return jv_cmd_process_node(matched_prefix, trimmed_args, ctx, renderer_override).await;
        }
    }
}