shuttle_common/
tables.rs

1use chrono::{DateTime, Local, SecondsFormat};
2use comfy_table::{
3    presets::{NOTHING, UTF8_BORDERS_ONLY},
4    Attribute, Cell, Color, ContentArrangement, Table,
5};
6
7use crate::{
8    models::{
9        certificate::CertificateResponse,
10        deployment::DeploymentResponse,
11        project::ProjectResponse,
12        resource::{ResourceResponse, ResourceType},
13    },
14    secrets::SecretStore,
15    DatabaseInfo,
16};
17
18pub fn get_certificates_table(certs: &[CertificateResponse], raw: bool) -> String {
19    let mut table = Table::new();
20    table
21        .load_preset(if raw { NOTHING } else { UTF8_BORDERS_ONLY })
22        .set_content_arrangement(ContentArrangement::Disabled)
23        .set_header(vec!["Certificate ID", "Subject", "Expires"]);
24
25    for cert in certs {
26        table.add_row(vec![
27            Cell::new(&cert.id).add_attribute(Attribute::Bold),
28            Cell::new(&cert.subject),
29            Cell::new(&cert.not_after),
30        ]);
31    }
32
33    table.to_string()
34}
35
36pub fn deployments_table(deployments: &[DeploymentResponse], raw: bool) -> String {
37    let mut table = Table::new();
38    table
39        .load_preset(if raw { NOTHING } else { UTF8_BORDERS_ONLY })
40        .set_content_arrangement(ContentArrangement::Disabled)
41        .set_header(vec!["Deployment ID", "Status", "Date", "Git revision"]);
42
43    for deploy in deployments.iter() {
44        let datetime: DateTime<Local> = DateTime::from(deploy.created_at);
45        table.add_row(vec![
46            Cell::new(&deploy.id).add_attribute(Attribute::Bold),
47            Cell::new(&deploy.state).fg(deploy.state.get_color_comfy_table()),
48            Cell::new(datetime.to_rfc3339_opts(SecondsFormat::Secs, false)),
49            Cell::new(
50                deploy
51                    .build_meta
52                    .as_ref()
53                    .map(ToString::to_string)
54                    .unwrap_or_default(),
55            ),
56        ]);
57    }
58
59    table.to_string()
60}
61
62pub fn get_projects_table(projects: &[ProjectResponse], raw: bool) -> String {
63    let mut table = Table::new();
64    table
65        .load_preset(if raw { NOTHING } else { UTF8_BORDERS_ONLY })
66        .set_content_arrangement(ContentArrangement::Disabled)
67        .set_header(vec!["Project ID", "Project Name", "Deployment Status"]);
68
69    for project in projects {
70        let state = project
71            .deployment_state
72            .as_ref()
73            .map(|s| s.to_string())
74            .unwrap_or_default();
75        let color = project
76            .deployment_state
77            .as_ref()
78            .map(|s| s.get_color_comfy_table())
79            .unwrap_or(Color::White);
80        table.add_row(vec![
81            Cell::new(&project.id).add_attribute(Attribute::Bold),
82            Cell::new(&project.name),
83            Cell::new(state).fg(color),
84        ]);
85    }
86
87    table.to_string()
88}
89
90pub fn get_resource_tables(
91    resources: &[ResourceResponse],
92    service_name: &str,
93    raw: bool,
94    show_secrets: bool,
95) -> String {
96    if resources.is_empty() {
97        return "No resources are linked to this service\n".to_string();
98    }
99    let mut output = Vec::new();
100    output.push(get_secrets_table(
101        &resources
102            .iter()
103            .filter(|r| matches!(r.r#type, ResourceType::Secrets))
104            .map(Clone::clone)
105            .collect::<Vec<_>>(),
106        service_name,
107        raw,
108    ));
109    output.push(get_databases_table(
110        &resources
111            .iter()
112            .filter(|r| {
113                matches!(
114                    r.r#type,
115                    ResourceType::DatabaseSharedPostgres
116                        | ResourceType::DatabaseAwsRdsMariaDB
117                        | ResourceType::DatabaseAwsRdsMySql
118                        | ResourceType::DatabaseAwsRdsPostgres
119                )
120            })
121            .map(Clone::clone)
122            .collect::<Vec<_>>(),
123        service_name,
124        raw,
125        show_secrets,
126    ));
127    output.join("\n")
128}
129
130fn get_databases_table(
131    databases: &[ResourceResponse],
132    service_name: &str,
133    raw: bool,
134    show_secrets: bool,
135) -> String {
136    if databases.is_empty() {
137        return String::new();
138    }
139
140    let mut table = Table::new();
141    table
142        .load_preset(if raw { NOTHING } else { UTF8_BORDERS_ONLY })
143        .set_content_arrangement(ContentArrangement::Disabled)
144        .set_header(vec!["Type", "Connection string"]);
145
146    for database in databases {
147        let connection_string = serde_json::from_value::<DatabaseInfo>(database.output.clone())
148            .expect("resource data to be a valid database")
149            .connection_string(show_secrets);
150
151        table.add_row(vec![database.r#type.to_string(), connection_string]);
152    }
153
154    let show_secret_hint = if databases.is_empty() || show_secrets {
155        ""
156    } else {
157        "Hint: you can show the secrets of these resources using `shuttle resource list --show-secrets`\n"
158    };
159
160    format!("These databases are linked to {service_name}\n{table}\n{show_secret_hint}")
161}
162
163fn get_secrets_table(secrets: &[ResourceResponse], service_name: &str, raw: bool) -> String {
164    let Some(secrets) = secrets.first() else {
165        return String::new();
166    };
167    let secrets = serde_json::from_value::<SecretStore>(secrets.output.clone()).unwrap();
168    if secrets.secrets.is_empty() {
169        return String::new();
170    }
171
172    let mut table = Table::new();
173    table
174        .load_preset(if raw { NOTHING } else { UTF8_BORDERS_ONLY })
175        .set_content_arrangement(ContentArrangement::Disabled)
176        .set_header(vec!["Key"]);
177
178    for key in secrets.secrets.keys() {
179        table.add_row(vec![key]);
180    }
181
182    format!("These secrets can be accessed by {service_name}\n{table}")
183}
shuttle_common/tables.rs

shuttle_common/
tables.rs
