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