codeberg_cli/actions/pull_request/
list.rs

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