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

use crate::cli::label::edit::EditLabelArgs;
use crate::client::BergClient;
use crate::endpoints::endpoint_generator::EndpointGenerator;
use crate::render::spinner::spin_until_ready;
use crate::render::ui::{fuzzy_select_with_key, multi_fuzzy_select_with_key};
use crate::types::api::edit_options::edit_label_option::EditLabelOption;
use crate::types::api::label::Label;
use strum::{Display, EnumIter, IntoEnumIterator};

use crate::actions::text_manipulation::{edit_prompt_for, input_prompt_for, select_prompt_for};

#[derive(Display, EnumIter, PartialEq, Eq)]
enum EditableFields {
    Name,
    Description,
    Color,
}

pub async fn edit_label(_args: EditLabelArgs, client: &BergClient) -> anyhow::Result<()> {
    let labels_list = spin_until_ready(client.get_repo_labels(None)).await?;

    if labels_list.is_empty() {
        println!("No labels found in this repository");
    }

    let selected_label =
        fuzzy_select_with_key(labels_list, select_prompt_for("label")).and_then(|maybe_label| {
            maybe_label.ok_or_else(|| anyhow::anyhow!("Nothing selected. Aborting."))
        })?;

    let selected_update_fields = multi_fuzzy_select_with_key(
        EditableFields::iter().collect::<Vec<_>>(),
        select_prompt_for("options"),
        |_| false,
    )?;

    let edit_label_options = create_update_data(selected_update_fields, &selected_label)?;

    tracing::debug!("{edit_label_options:?}");

    let api_endpoint = EndpointGenerator::repo_labels_with_id(selected_label.id)?;

    let updated_label: Label = client.patch_body(api_endpoint, edit_label_options).await?;

    tracing::debug!("{updated_label:?}");

    Ok(())
}
fn create_update_data(
    selected_update_fields: Vec<EditableFields>,
    selected_label: &Label,
) -> anyhow::Result<EditLabelOption> {
    use EditableFields::*;

    let mut edit_label_options = EditLabelOption::from_label(selected_label);

    if selected_update_fields.contains(&Name) {
        let new_title = inquire::Text::new(input_prompt_for("Choose a new label name").as_str())
            .with_default(selected_label.name.as_str())
            .prompt()?;
        edit_label_options.name.replace(new_title);
    }

    if selected_update_fields.contains(&Color) {
        let new_title = inquire::Text::new(input_prompt_for("Choose a new label color").as_str())
            .with_help_message("(format: #xxxxxx)")
            .with_default(selected_label.color.as_str())
            .prompt()?;
        edit_label_options.color.replace(new_title);
    }

    if selected_update_fields.contains(&Description) {
        let new_description = inquire::Editor::new(edit_prompt_for("a label description").as_str())
            .with_predefined_text(selected_label.description.as_str())
            .prompt()?;
        edit_label_options.description.replace(new_description);
    }

    Ok(edit_label_options)
}