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

use cod_cli::milestone::edit::EditMilestoneArgs;
use cod_client::CodebergClient;
use cod_endpoints::endpoint_generator::EndpointGenerator;
use cod_render::spinner::spin_until_ready;
use cod_render::ui::fuzzy_select_with_key;
use cod_render::ui::multi_fuzzy_select_with_key;
use cod_types::api::edit_options::edit_milestone_option::EditMilestoneOption;
use cod_types::api::milestone::Milestone;
use cod_types::api::state_type::StateType;
use strum::IntoEnumIterator;
use strum::{Display, EnumIter};

use crate::text_manipulation::edit_prompt_for;
use crate::text_manipulation::input_prompt_for;
use crate::text_manipulation::select_prompt_for;

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

pub async fn edit_milestone(
    _args: EditMilestoneArgs,
    client: &CodebergClient,
) -> anyhow::Result<()> {
    let milestones_list = spin_until_ready(client.get_repo_milestones(None, None)).await?;

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

    let selected_milestone =
        fuzzy_select_with_key(milestones_list, select_prompt_for("milestone"))?
            .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_milestone_options = create_update_data(selected_update_fields, &selected_milestone)?;

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

    let api_endpoint = EndpointGenerator::repo_update_milestone(selected_milestone.id)?;

    let updated_milestone: Milestone = client
        .patch_body(api_endpoint, edit_milestone_options)
        .await?;

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

    Ok(())
}

fn create_update_data(
    selected_update_fields: Vec<EditableFields>,
    selected_milestone: &Milestone,
) -> anyhow::Result<EditMilestoneOption> {
    use EditableFields::*;

    let mut edit_milestone_options = EditMilestoneOption::from_milestone(selected_milestone);

    if selected_update_fields.contains(&Description) {
        let new_description =
            inquire::Editor::new(edit_prompt_for("a milestone description").as_str())
                .with_predefined_text(
                    selected_milestone
                        .description
                        .as_deref()
                        .unwrap_or_default(),
                )
                .prompt()?;
        edit_milestone_options.description.replace(new_description);
    }

    if selected_update_fields.contains(&State) {
        let new_state = fuzzy_select_with_key(
            StateType::available_for_choosing().to_vec(),
            select_prompt_for("state"),
        )?;
        edit_milestone_options
            .state
            .replace(new_state.unwrap_or(selected_milestone.state));
    }

    if selected_update_fields.contains(&Title) {
        let new_title =
            inquire::Text::new(input_prompt_for("Choose a new milestone title").as_str())
                .with_default(selected_milestone.title.as_str())
                .prompt()?;
        edit_milestone_options.title.replace(new_title);
    }

    Ok(edit_milestone_options)
}