codeberg_cli/actions/pull_request/
list.rs

1use forgejo_api::structs::{
2    PullRequest, RepoListPullRequestsQuery, RepoListPullRequestsQueryState,
3};
4use itertools::Itertools;
5
6use crate::actions::GeneralArgs;
7use crate::render::option::{option_debug_display, option_display};
8use crate::render::spinner::spin_until_ready;
9
10use crate::types::api::pr_query_sort::{self, PullRequestsQuerySort};
11use crate::types::context::BergContext;
12
13use crate::types::api::state_type::ViewStateType;
14use crate::types::git::OwnerRepo;
15use clap::Parser;
16
17/// List pull requests
18#[derive(Parser, Debug, Clone, Copy)]
19pub struct ListPullRequestArgs {
20    /// Number of pull requests to be displayed
21    #[arg(short, long, value_name = "N", default_value_t = 5)]
22    pub count: u32,
23
24    /// Sort using a certain criteria
25    #[arg(long, default_value_t = pr_query_sort::PullRequestsQuerySort::Recentupdate)]
26    pub sort: PullRequestsQuerySort,
27
28    /// Filter pull requests with the chosen state
29    #[arg(short, long, default_value_t = ViewStateType::All)]
30    pub state: ViewStateType,
31}
32
33impl ListPullRequestArgs {
34    pub async fn run(self, general_args: GeneralArgs) -> anyhow::Result<()> {
35        let _ = general_args;
36        let ctx = BergContext::new(self, general_args).await?;
37
38        let state = match self.state {
39            ViewStateType::All => RepoListPullRequestsQueryState::All,
40            ViewStateType::Closed => RepoListPullRequestsQueryState::Closed,
41            ViewStateType::Open => RepoListPullRequestsQueryState::Open,
42        };
43        let query = RepoListPullRequestsQuery {
44            state: Some(state),
45            sort: Some(self.sort.into()),
46            limit: Some(self.count),
47            ..Default::default()
48        };
49
50        let OwnerRepo { repo, owner } = ctx.owner_repo()?;
51        let pull_requests_list = spin_until_ready(ctx.client.repo_list_pull_requests(
52            owner.as_str(),
53            repo.as_str(),
54            query,
55        ))
56        .await?;
57
58        present_pull_requests_list(&ctx, pull_requests_list);
59
60        Ok(())
61    }
62}
63
64fn present_pull_requests_list(
65    ctx: &BergContext<ListPullRequestArgs>,
66    pull_requests: Vec<PullRequest>,
67) {
68    let pull_requests_empty = pull_requests.is_empty();
69
70    let mut table = ctx.make_table().add_table((!pull_requests_empty).then(|| {
71        let mut inner_table = ctx.make_table();
72
73        inner_table
74            .set_header(vec!["Number", "Status", "Name", "Labels"])
75            .add_rows(pull_requests.iter().map(|issue| {
76                let PullRequest {
77                    title,
78                    number,
79                    labels,
80                    state,
81                    ..
82                } = issue;
83                let labels = option_display(&labels.as_ref().map(|labels| {
84                    if labels.is_empty() {
85                        String::from("x")
86                    } else {
87                        labels
88                            .iter()
89                            .filter_map(|label| label.name.as_ref())
90                            .join(",")
91                    }
92                }));
93                vec![
94                    option_display(number),
95                    option_debug_display(state),
96                    option_display(title),
97                    labels,
98                ]
99            }));
100
101        inner_table
102    }));
103
104    table.set_header(vec![format!(
105        "Pull Requests {}",
106        pull_requests_empty
107            .then_some(" (empty)")
108            .unwrap_or_default()
109    )]);
110
111    println!("{table}", table = table.show());
112}