Skip to main content

codeberg_cli/actions/pull_request/
list.rs

1use forgejo_api::structs::{
2    PullRequest, RepoListPullRequestsQuery, RepoListPullRequestsQueryState,
3};
4use itertools::Itertools;
5use miette::IntoDiagnostic;
6
7use crate::actions::GlobalArgs;
8use crate::render::json::JsonToStdout;
9use crate::render::option::{option_debug_display, option_display};
10use crate::render::spinner::spin_until_ready;
11
12use crate::types::api::pr_query_sort::{self, PullRequestsQuerySort};
13use crate::types::context::BergContext;
14
15use crate::types::api::state_type::ViewStateType;
16use crate::types::git::OwnerRepo;
17use crate::types::output::OutputMode;
18use clap::Parser;
19
20/// List pull requests
21#[derive(Parser, Debug, Clone)]
22pub struct ListPullRequestArgs {
23    /// Sort using a certain criteria
24    #[arg(long, default_value_t = pr_query_sort::PullRequestsQuerySort::Recentupdate)]
25    pub sort: PullRequestsQuerySort,
26
27    /// Filter pull requests with the chosen state
28    #[arg(short, long, default_value_t = ViewStateType::All)]
29    pub state: ViewStateType,
30}
31
32impl ListPullRequestArgs {
33    pub async fn run(self, global_args: GlobalArgs) -> miette::Result<()> {
34        let state = match self.state {
35            ViewStateType::All => RepoListPullRequestsQueryState::All,
36            ViewStateType::Closed => RepoListPullRequestsQueryState::Closed,
37            ViewStateType::Open => RepoListPullRequestsQueryState::Open,
38        };
39        let query = RepoListPullRequestsQuery {
40            state: Some(state),
41            sort: Some(self.sort.into()),
42            ..Default::default()
43        };
44
45        let ctx = BergContext::new(self, global_args).await?;
46
47        let OwnerRepo { repo, owner } = ctx.owner_repo()?;
48        let (_, pull_requests_list) = spin_until_ready(
49            ctx.client
50                .repo_list_pull_requests(owner.as_str(), repo.as_str(), query)
51                .send(),
52        )
53        .await
54        .into_diagnostic()?;
55
56        match ctx.global_args.output_mode {
57            OutputMode::Pretty => {
58                present_pull_requests_list(&ctx, pull_requests_list);
59            }
60            OutputMode::Json => pull_requests_list.print_json()?,
61        }
62
63        Ok(())
64    }
65}
66
67fn present_pull_requests_list(
68    ctx: &BergContext<ListPullRequestArgs>,
69    pull_requests: Vec<PullRequest>,
70) {
71    let pull_requests_empty = pull_requests.is_empty();
72
73    let table = ctx
74        .make_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    let header = format!(
103        "Pull Requests {}",
104        if pull_requests_empty {
105            " (empty)"
106        } else {
107            Default::default()
108        }
109    );
110
111    let underscore = "=".repeat(header.len());
112    let out = [header, underscore, table.show()].join("\n");
113
114    println!("{out}");
115}