1use crate::mascaramento::mascarar;
8use crate::ssh::SaidaExecucao;
9use crate::vps::modelo::VpsRegistro;
10use secrecy::ExposeSecret;
11use serde_json::json;
12use std::io::{self, Write};
13use std::sync::atomic::{AtomicBool, Ordering};
14
15static QUIET: AtomicBool = AtomicBool::new(false);
17
18static JSON_ERROS: AtomicBool = AtomicBool::new(false);
20
21pub fn definir_quiet(quiet: bool) {
23 QUIET.store(quiet, Ordering::SeqCst);
24}
25
26pub fn definir_json_erros(json: bool) {
28 JSON_ERROS.store(json, Ordering::SeqCst);
29}
30
31#[must_use]
33pub fn esta_quiet() -> bool {
34 QUIET.load(Ordering::SeqCst)
35}
36
37#[must_use]
39pub fn quer_json_erros() -> bool {
40 JSON_ERROS.load(Ordering::SeqCst)
41}
42
43pub fn escrever_linha(conteudo: &str) -> io::Result<()> {
51 let stdout = io::stdout();
52 let mut handle = stdout.lock();
53 handle.write_all(conteudo.as_bytes())?;
54 handle.write_all(b"\n")?;
55 handle.flush()?;
56 Ok(())
57}
58
59pub fn imprimir_sucesso(mensagem: &str) {
61 if esta_quiet() {
62 return;
63 }
64 println!("{mensagem}");
65}
66
67pub fn imprimir_banner_humano(mensagem: &str) {
71 if esta_quiet() || quer_json_erros() {
72 return;
73 }
74 if !std::io::IsTerminal::is_terminal(&std::io::stdout()) {
75 return;
76 }
77 if std::env::var_os("SSH_CLI_FORCE_TEXT").is_none() {
78 }
80 println!("{mensagem}");
81}
82
83pub fn imprimir_erro(mensagem: &str) {
85 eprintln!("{mensagem}");
86}
87
88pub fn imprimir_erro_envelope(
90 exit_code: i32,
91 message: &str,
92 remote_exit_code: Option<i32>,
93) -> io::Result<()> {
94 let mut v = json!({
95 "exit_code": exit_code,
96 "message": message,
97 });
98 if let Some(r) = remote_exit_code {
99 v["remote_exit_code"] = json!(r);
100 }
101 let s = serde_json::to_string(&v).unwrap_or_else(|_| {
102 format!(r#"{{"exit_code":{exit_code},"message":"erro de serialização"}}"#)
103 });
104 let mut err = io::stderr().lock();
105 err.write_all(s.as_bytes())?;
106 err.write_all(b"\n")?;
107 err.flush()?;
108 Ok(())
109}
110
111pub fn imprimir_json_value(v: &serde_json::Value) -> io::Result<()> {
113 let s = serde_json::to_string_pretty(v).map_err(io::Error::other)?;
114 escrever_linha(&s)
115}
116
117#[allow(clippy::too_many_arguments)]
119pub fn imprimir_doctor_texto(
120 camada: &str,
121 config_path: &str,
122 existe: bool,
123 perms: &str,
124 schema_version: u32,
125 hosts: usize,
126 known_hosts: &str,
127 active_file: &str,
128 secrets_at_rest: &str,
129 secrets_key_source: &str,
130 secrets_key_file: &str,
131 plaintext_opt_out: bool,
132) {
133 if esta_quiet() {
134 return;
135 }
136 println!("Camada vencedora: {camada}");
137 println!("Config path: {config_path}");
138 println!("Existe: {existe}");
139 println!("Permissões: {perms}");
140 println!("Schema: {schema_version}");
141 println!("Hosts: {hosts}");
142 println!("known_hosts: {known_hosts}");
143 println!("active file: {active_file}");
144 println!("Secrets at-rest: {secrets_at_rest} (key source: {secrets_key_source})");
145 println!("Secrets key file: {secrets_key_file}");
146 println!(
147 "Plaintext opt-out: {}",
148 if plaintext_opt_out { "yes" } else { "no" }
149 );
150 println!("Telemetria: desabilitada");
151}
152
153pub fn imprimir_lista_texto(registros: &[VpsRegistro]) {
155 if esta_quiet() {
156 return;
157 }
158 if registros.is_empty() {
159 println!(
160 "{}",
161 crate::i18n::t(crate::i18n::Mensagem::VpsRegistroVazio)
162 );
163 return;
164 }
165
166 println!(
167 "{:<20} {:<30} {:<6} {:<15} {:<20}",
168 "NOME", "HOST", "PORTA", "USUÁRIO", "SENHA"
169 );
170 for r in registros {
171 println!(
172 "{:<20} {:<30} {:<6} {:<15} {:<20}",
173 r.nome,
174 r.host,
175 r.porta,
176 r.usuario,
177 mascarar(r.senha.expose_secret())
178 );
179 }
180}
181
182pub fn imprimir_lista_json(registros: &[VpsRegistro]) {
184 let lista: Vec<_> = registros.iter().map(registro_para_json_mascarado).collect();
185 match serde_json::to_string_pretty(&lista) {
186 Ok(s) => println!("{s}"),
187 Err(erro) => eprintln!("erro ao serializar JSON: {erro}"),
188 }
189}
190
191pub fn imprimir_detalhes_texto(r: &VpsRegistro) {
193 if esta_quiet() {
194 return;
195 }
196 println!("Nome: {}", r.nome);
197 println!("Host: {}", r.host);
198 println!("Porta: {}", r.porta);
199 println!("Usuário: {}", r.usuario);
200 println!(
202 "Senha: {}",
203 if r.senha.expose_secret().is_empty() {
204 "(não definida)".to_string()
205 } else {
206 mascarar(r.senha.expose_secret())
207 }
208 );
209 println!(
210 "Key path: {}",
211 r.key_path.as_deref().unwrap_or("(não definida)")
212 );
213 println!(
214 "Senha sudo: {}",
215 r.senha_sudo
216 .as_ref()
217 .map_or_else(|| "(não definida)".into(), |s| mascarar(s.expose_secret()))
218 );
219 println!(
220 "Senha su: {}",
221 r.senha_su
222 .as_ref()
223 .map_or_else(|| "(não definida)".into(), |s| mascarar(s.expose_secret()))
224 );
225 println!("Timeout (ms): {}", r.timeout_ms);
226 println!("Max cmd chars: {}", r.max_command_chars);
227 println!("Max out chars: {}", r.max_output_chars);
228 println!("Disable sudo: {}", r.disable_sudo);
229 println!("Schema version: {}", r.schema_version);
230 println!("Adicionado em: {}", r.adicionado_em);
231}
232
233pub fn imprimir_detalhes_json(r: &VpsRegistro) {
235 let v = registro_para_json_mascarado(r);
236 match serde_json::to_string_pretty(&v) {
237 Ok(s) => println!("{s}"),
238 Err(erro) => eprintln!("erro ao serializar JSON: {erro}"),
239 }
240}
241
242fn registro_para_json_mascarado(r: &VpsRegistro) -> serde_json::Value {
243 let password = if r.senha.expose_secret().is_empty() {
245 json!(null)
246 } else {
247 json!(mascarar(r.senha.expose_secret()))
248 };
249 json!({
250 "name": r.nome,
251 "host": r.host,
252 "port": r.porta,
253 "user": r.usuario,
254 "password": password,
255 "key_path": r.key_path,
256 "key_passphrase": r.key_passphrase.as_ref().map(|s| mascarar(s.expose_secret())),
257 "sudo_password": r.senha_sudo.as_ref().map(|s| mascarar(s.expose_secret())),
258 "su_password": r.senha_su.as_ref().map(|s| mascarar(s.expose_secret())),
259 "timeout_ms": r.timeout_ms,
260 "max_command_chars": r.max_command_chars,
261 "max_output_chars": r.max_output_chars,
262 "disable_sudo": r.disable_sudo,
263 "schema_version": r.schema_version,
264 "added_at": r.adicionado_em,
265 })
266}
267
268pub fn imprimir_saida_execucao(saida: &SaidaExecucao) {
279 println!("--- stdout ---");
280 if saida.stdout.is_empty() {
281 println!("(vazio)");
282 } else {
283 println!("{}", saida.stdout);
284 }
285 println!("--- stderr ---");
286 if saida.stderr.is_empty() {
287 println!("(vazio)");
288 } else {
289 println!("{}", saida.stderr);
290 }
291 let code_str = saida
292 .exit_code
293 .map(|c| c.to_string())
294 .unwrap_or_else(|| "N/A".to_string());
295 println!("--- exit code: {} ({}ms) ---", code_str, saida.duracao_ms);
296 if saida.truncado_stdout {
297 println!("(stdout foi truncado)");
298 }
299 if saida.truncado_stderr {
300 println!("(stderr foi truncado)");
301 }
302}
303
304pub fn imprimir_saida_execucao_json(saida: &SaidaExecucao) {
306 let v = json!({
307 "stdout": saida.stdout,
308 "stderr": saida.stderr,
309 "exit_code": saida.exit_code,
310 "truncated_stdout": saida.truncado_stdout,
311 "truncated_stderr": saida.truncado_stderr,
312 "duration_ms": saida.duracao_ms,
313 });
314 match serde_json::to_string_pretty(&v) {
315 Ok(s) => println!("{s}"),
316 Err(e) => eprintln!("erro ao serializar JSON: {e}"),
317 }
318}
319
320pub fn imprimir_health_check(nome: &str, latencia_ms: u64) {
322 if esta_quiet() {
323 return;
324 }
325 println!(
326 "{}",
327 crate::i18n::t(crate::i18n::Mensagem::HealthCheckOk {
328 nome: nome.to_string(),
329 })
330 );
331 println!(" latência: {latencia_ms}ms");
332}
333
334pub fn imprimir_health_check_json(nome: &str, latencia_ms: u64) {
336 let v = json!({
337 "name": nome,
338 "status": "ok",
339 "latency_ms": latencia_ms,
340 });
341 match serde_json::to_string_pretty(&v) {
342 Ok(s) => println!("{s}"),
343 Err(e) => eprintln!("erro ao serializar JSON: {e}"),
344 }
345}
346
347pub fn imprimir_transferencia_json(
349 direction: &str,
350 vps: &str,
351 local: &str,
352 remote: &str,
353 bytes: u64,
354 duration_ms: u64,
355) {
356 let v = json!({
357 "ok": true,
358 "direction": direction,
359 "vps": vps,
360 "local": local,
361 "remote": remote,
362 "bytes": bytes,
363 "duration_ms": duration_ms,
364 });
365 match serde_json::to_string_pretty(&v) {
366 Ok(s) => {
367 let _ = escrever_linha(&s);
368 }
369 Err(e) => eprintln!("erro ao serializar JSON: {e}"),
370 }
371}
372
373pub fn imprimir_tunnel_listening_json(
375 vps: &str,
376 local_port: u16,
377 remote_host: &str,
378 remote_port: u16,
379 timeout_ms: u64,
380) {
381 let v = json!({
382 "ok": true,
383 "event": "tunnel_listening",
384 "vps": vps,
385 "local_port": local_port,
386 "remote_host": remote_host,
387 "remote_port": remote_port,
388 "timeout_ms": timeout_ms,
389 });
390 match serde_json::to_string_pretty(&v) {
391 Ok(s) => {
392 let _ = escrever_linha(&s);
393 }
394 Err(e) => eprintln!("erro ao serializar JSON: {e}"),
395 }
396}
397
398#[cfg(test)]
399mod testes {
400 use super::*;
401 use crate::ssh::SaidaExecucao;
402 use crate::vps::modelo::VpsRegistro;
403 use secrecy::SecretString;
404
405 fn registro_teste() -> VpsRegistro {
406 VpsRegistro::novo(
407 "vps-teste".into(),
408 "1.2.3.4".into(),
409 22,
410 "root".into(),
411 SecretString::from("senha-super-secreta".to_string()),
412 None,
413 None,
414 Some(5000),
415 Some(1000),
416 Some(1000),
417 Some(SecretString::from("sudo-password-longa-aqui".to_string())),
418 None,
419 false,
420 )
421 }
422
423 #[test]
424 fn registro_para_json_mascarado_contem_campos_obrigatorios() {
425 let r = registro_teste();
426 let json = registro_para_json_mascarado(&r);
427 assert_eq!(json["name"], "vps-teste");
428 assert_eq!(json["host"], "1.2.3.4");
429 assert_eq!(json["port"], 22);
430 assert_eq!(json["user"], "root");
431 assert_eq!(json["password"].as_str().unwrap(), "***");
432 assert_eq!(json["sudo_password"].as_str().unwrap(), "***");
433 assert!(json["su_password"].is_null());
434 assert_eq!(json["timeout_ms"], 5000);
435 assert_eq!(json["max_command_chars"], 1000);
436 assert_eq!(json["max_output_chars"], 1000);
437 assert_eq!(json["schema_version"], 2);
438 }
439
440 #[test]
441 fn registro_para_json_mascarado_senha_sudo_nula_quando_nao_definida() {
442 let mut r = registro_teste();
443 r.senha_sudo = None;
444 let json = registro_para_json_mascarado(&r);
445 assert!(json["sudo_password"].is_null());
446 }
447
448 #[test]
449 fn registro_para_json_mascarado_su_password_presente() {
450 let mut r = registro_teste();
451 r.senha_su = Some(SecretString::from("senha-su-muito-longa-aqui".to_string()));
452 let json = registro_para_json_mascarado(&r);
453 assert_eq!(json["su_password"].as_str().unwrap(), "***");
454 }
455
456 #[test]
457 fn registro_para_json_mascarado_password_null_quando_vazio() {
458 let mut r = registro_teste();
459 r.senha = SecretString::from(String::new());
460 let json = registro_para_json_mascarado(&r);
461 assert!(json["password"].is_null());
462 }
463
464 #[test]
465 fn escribir_linha_ok() {
466 let resultado = escrever_linha("teste de escrita");
467 assert!(resultado.is_ok());
468 }
469
470 #[test]
471 fn escribir_linha_com_caracteres_especiais() {
472 let resultado = escrever_linha("linha com \t tab e \"aspas\"");
473 assert!(resultado.is_ok());
474 }
475
476 #[test]
477 fn salida_execucao_completa_formatada() {
478 let saida = SaidaExecucao {
479 stdout: "output do comando".to_string(),
480 stderr: "erro do comando".to_string(),
481 exit_code: Some(0),
482 truncado_stdout: false,
483 truncado_stderr: false,
484 duracao_ms: 150,
485 };
486 let resultado = escrever_linha(&format!(
487 "stdout: {}, stderr: {}, exit: {:?}",
488 saida.stdout, saida.stderr, saida.exit_code
489 ));
490 assert!(resultado.is_ok());
491 }
492
493 #[test]
494 fn salida_execucao_sem_exit_code() {
495 let saida = SaidaExecucao {
496 stdout: "".to_string(),
497 stderr: "".to_string(),
498 exit_code: None,
499 truncado_stdout: false,
500 truncado_stderr: false,
501 duracao_ms: 0,
502 };
503 let code_str = saida
504 .exit_code
505 .map(|c| c.to_string())
506 .unwrap_or_else(|| "N/A".to_string());
507 assert_eq!(code_str, "N/A");
508 }
509
510 #[test]
511 fn vps_registro_debug_nao_expoe_senha() {
512 let r = registro_teste();
513 let json = registro_para_json_mascarado(&r);
514 let json_str = serde_json::to_string(&json).unwrap();
515 assert!(!json_str.contains("senha-super-secreta"));
516 assert!(!json_str.contains("sudo-password-longa-aqui"));
517 }
518
519 #[test]
520 fn salida_execucao_truncada_mostra_aviso() {
521 let saida = SaidaExecucao {
522 stdout: "output".to_string(),
523 stderr: "erro".to_string(),
524 exit_code: Some(1),
525 truncado_stdout: true,
526 truncado_stderr: true,
527 duracao_ms: 100,
528 };
529 assert!(saida.truncado_stdout);
530 assert!(saida.truncado_stderr);
531 }
532
533 #[test]
534 fn salida_execucao_com_exit_code_numerico() {
535 let saida = SaidaExecucao {
536 stdout: "".to_string(),
537 stderr: "".to_string(),
538 exit_code: Some(127),
539 truncado_stdout: false,
540 truncado_stderr: false,
541 duracao_ms: 0,
542 };
543 let code_str = saida
544 .exit_code
545 .map(|c| c.to_string())
546 .unwrap_or_else(|| "N/A".to_string());
547 assert_eq!(code_str, "127");
548 }
549
550 #[test]
551 fn escribir_linha_string_vazia() {
552 let resultado = escrever_linha("");
553 assert!(resultado.is_ok());
554 }
555
556 #[test]
557 fn escribir_linha_com_unicode_brasileiro() {
558 let resultado = escrever_linha("ação você está Itaú");
559 assert!(resultado.is_ok());
560 }
561
562 #[test]
563 fn escribir_linha_com_emojis() {
564 let resultado = escrever_linha("texto com 🚀 e 🔐");
565 assert!(resultado.is_ok());
566 }
567
568 #[test]
569 fn escribir_linha_com_newlines() {
570 let resultado = escrever_linha("linha1\nlinha2\nlinha3");
571 assert!(resultado.is_ok());
572 }
573
574 #[test]
575 fn escribir_linha_longo_texto() {
576 let texto_longo = "a".repeat(10000);
577 let resultado = escrever_linha(&texto_longo);
578 assert!(resultado.is_ok());
579 }
580
581 #[test]
582 fn registro_para_json_mascarado_com_senha_curta_mascara_com_asteriscos() {
583 let mut r = registro_teste();
584 r.senha = SecretString::from("curta".to_string());
585 let json = registro_para_json_mascarado(&r);
586 let senha_str = json["password"].as_str().unwrap();
587 assert_eq!(senha_str, "***");
588 }
589
590 #[test]
591 fn registro_para_json_mascarado_com_sudo_e_su_definidos() {
592 let mut r = registro_teste();
593 r.senha_sudo = Some(SecretString::from("sudo-pass-longa-aqui".to_string()));
594 r.senha_su = Some(SecretString::from("su-pass-longa-aqui".to_string()));
595 let json = registro_para_json_mascarado(&r);
596 assert!(!json["sudo_password"].is_null());
597 assert!(!json["su_password"].is_null());
598 assert_eq!(json["sudo_password"].as_str().unwrap(), "***");
599 assert_eq!(json["su_password"].as_str().unwrap(), "***");
600 }
601
602 #[test]
603 fn saida_execucao_formatacao_completa() {
604 let saida = SaidaExecucao {
605 stdout: "comando executado".to_string(),
606 stderr: "aviso harmless".to_string(),
607 exit_code: Some(0),
608 truncado_stdout: false,
609 truncado_stderr: false,
610 duracao_ms: 1000,
611 };
612 assert_eq!(saida.stdout, "comando executado");
613 assert_eq!(saida.stderr, "aviso harmless");
614 assert_eq!(saida.exit_code, Some(0));
615 assert_eq!(saida.duracao_ms, 1000);
616 assert!(!saida.truncado_stdout);
617 assert!(!saida.truncado_stderr);
618 }
619
620 #[test]
621 fn saida_execucao_sem_stderr() {
622 let saida = SaidaExecucao {
623 stdout: "ok".to_string(),
624 stderr: String::new(),
625 exit_code: Some(0),
626 truncado_stdout: false,
627 truncado_stderr: false,
628 duracao_ms: 50,
629 };
630 assert!(saida.stderr.is_empty());
631 }
632
633 #[test]
634 fn saida_execucao_com_sinal_em_vez_de_exit_code() {
635 let saida = SaidaExecucao {
636 stdout: String::new(),
637 stderr: "signal received".to_string(),
638 exit_code: None,
639 truncado_stdout: false,
640 truncado_stderr: false,
641 duracao_ms: 5000,
642 };
643 assert!(saida.exit_code.is_none());
644 }
645
646 #[test]
647 fn saida_execucao_json_contem_campos_obrigatorios() {
648 let saida = SaidaExecucao {
649 stdout: "output".to_string(),
650 stderr: "erro".to_string(),
651 exit_code: Some(0),
652 truncado_stdout: false,
653 truncado_stderr: false,
654 duracao_ms: 100,
655 };
656 imprimir_saida_execucao_json(&saida);
657 }
658}