context7_cli/
i18n.rs

1/// Bilingual internationalisation (EN / PT-BR) for user-facing messages.
2///
3/// Language resolution order:
4/// 1. CLI flag `--lang en|pt`
5/// 2. Environment variable `CONTEXT7_LANG`
6/// 3. `sys_locale::get_locale()` — locale starting with `"pt"` → Portuguese
7/// 4. Default: English
8///
9/// Call [`definir_idioma`] once at startup (in `run()`), then call
10/// [`idioma_atual`] or [`t`] anywhere to retrieve localised strings.
11use std::sync::OnceLock;
12
13// ─── IDIOMA ───────────────────────────────────────────────────────────────────
14
15/// Supported display languages.
16#[derive(Debug, Clone, Copy, PartialEq, Eq)]
17pub enum Idioma {
18    /// English output.
19    English,
20    /// Brazilian Portuguese output.
21    Portugues,
22}
23
24/// Global language setting — written once at startup, read-only thereafter.
25static IDIOMA_GLOBAL: OnceLock<Idioma> = OnceLock::new();
26
27/// Returns the currently configured language.
28///
29/// Defaults to [`Idioma::English`] if [`definir_idioma`] has not been called.
30pub fn idioma_atual() -> Idioma {
31    *IDIOMA_GLOBAL.get().unwrap_or(&Idioma::English)
32}
33
34/// Sets the global language. Silently ignored if already set (OnceLock semantics).
35pub fn definir_idioma(idioma: Idioma) {
36    let _ = IDIOMA_GLOBAL.set(idioma);
37}
38
39/// Resolves the language from CLI flag, env var, or system locale.
40///
41/// Resolution order:
42/// 1. `cli_lang` — value of `--lang` flag (if provided)
43/// 2. `CONTEXT7_LANG` environment variable
44/// 3. `sys_locale::get_locale()` — BCP 47 locale (e.g. `"pt-BR"`)
45/// 4. Default: English
46pub fn resolver_idioma(cli_lang: Option<&str>) -> Idioma {
47    // 1. CLI flag
48    if let Some(lang) = cli_lang {
49        return parse_lang_str(lang);
50    }
51
52    // 2. Environment variable
53    if let Ok(env_lang) = std::env::var("CONTEXT7_LANG") {
54        return parse_lang_str(&env_lang);
55    }
56
57    // 3. System locale
58    if let Some(locale) = sys_locale::get_locale() {
59        if locale.to_lowercase().starts_with("pt") {
60            return Idioma::Portugues;
61        }
62    }
63
64    // 4. Default
65    Idioma::English
66}
67
68fn parse_lang_str(s: &str) -> Idioma {
69    match s.to_lowercase().as_str() {
70        "pt" | "pt-br" | "pt_br" | "portugues" | "português" => Idioma::Portugues,
71        _ => Idioma::English,
72    }
73}
74
75// ─── MENSAGEM ─────────────────────────────────────────────────────────────────
76
77/// All user-facing messages indexed by variant.
78///
79/// Each variant maps to a pair of `(English, Portuguese)` strings.
80#[derive(Debug, Clone, Copy)]
81pub enum Mensagem {
82    // Keys subcommand (15 variants)
83    /// "Key added successfully at: {path}"
84    ChaveAdicionada,
85    /// "Key already exists (skipping)."
86    ChaveJaExistia,
87    /// "No key stored."
88    NenhumaChaveArmazenada,
89    /// "Use `context7 keys add <KEY>` to add a key."
90    UsarKeysAdd,
91    /// "{n} key(s) stored:"
92    ContadorChaves,
93    /// "No key stored to remove."
94    NenhumaChaveParaRemover,
95    /// "Index {i} invalid. Use a number between 1 and {n}."
96    IndiceInvalido,
97    /// "Key {masked} removed successfully."
98    ChaveRemovidaSucesso,
99    /// "Operation cancelled."
100    OperacaoCancelada,
101    /// "All keys removed."
102    TodasChavesRemovidas,
103    /// "System does not support XDG directories."
104    SistemaXdgNaoSuportado,
105    /// "{imported}/{total} key(s) imported successfully."
106    ChavesImportadasSucesso,
107    /// "No CONTEXT7_API= key found in: {file}"
108    NenhumaChaveContext7NoArquivo,
109    /// "Are you sure you want to remove ALL keys? [y/N] " / "[s/N] "
110    ConfirmarRemoverTodas,
111    /// Accepted confirmation responses: "y"/"yes" or "s"/"sim"
112    RespostaConfirmacao,
113
114    // Library / Docs (8 variants)
115    /// "No library found."
116    NenhumaBibliotecaEncontrada,
117    /// "Libraries found:"
118    BibliotecasEncontradas,
119    /// "Trust:"
120    ConfiancaScore,
121    /// "No documentation found."
122    NenhumaDocumentacaoEncontrada,
123    /// "Documentation:"
124    TituloDocumentacao,
125    /// "Sources:"
126    TituloFontes,
127    /// "No content available."
128    SemConteudoDisponivel,
129    /// "Searching library: {name}"
130    BuscandoBiblioteca,
131
132    // HTTP / Network errors (8 variants)
133    /// "Network error searching library: {err}"
134    ErroDeRede,
135    /// "Network error fetching documentation: {err}"
136    ErroDeRedeDocs,
137    /// "Failed to deserialise JSON response: {err}"
138    FalhaDesserializar,
139    /// "Rate limit reached (429), waiting for retry…"
140    RateLimitAtingido,
141    /// "Server error ({status}), retrying…"
142    ErrodoServidor,
143    /// "Invalid API key (401/403), trying next…"
144    ChaveApiInvalida,
145    /// "Attempt {n}/{max}"
146    Tentativa,
147    /// "Waiting {ms}ms before retrying…"
148    AguardandoRetry,
149
150    // Config / XDG (7 variants)
151    /// "System does not support XDG — cannot save configuration"
152    ErroCaminhoXdg,
153    /// "Failed to read XDG config at: {path}"
154    FalhaLerConfig,
155    /// "Invalid TOML at: {path}"
156    FalhaTomlInvalido,
157    /// "Failed to write config at: {path}"
158    FalhaEscreverConfig,
159    /// "Failed to create directory: {path}"
160    FalhaCriarDiretorio,
161    /// "No API key configured. Set CONTEXT7_API_KEYS or use `keys add`."
162    NenhumaChaveConfigurada,
163    /// "Failed to serialise configuration to TOML"
164    FalhaSerializarToml,
165
166    // Logging / info (8 variants)
167    /// "Keys loaded from CONTEXT7_API_KEYS environment variable"
168    ChavesCarregadasEnvVar,
169    /// "Keys loaded from XDG configuration"
170    ChavesCarregadasXdg,
171    /// "Failed to read XDG configuration (continuing): {err}"
172    FalhaLerXdgContinuando,
173    /// "Starting context7 with {n} API keys available"
174    IniciandoComChaves,
175    /// "Keys loaded from compile-time CONTEXT7_API_KEYS"
176    ChavesCarregadasCompileTime,
177    /// "Failed to serialise results to JSON"
178    FalhaSerializarJson,
179    /// "Failed to serialise documentation to JSON"
180    FalhaSerializarDocs,
181    /// "Failed to search library '{name}'"
182    FalhaBuscarBiblioteca,
183
184    // Permissions / IO (2 variants)
185    /// "Failed to read metadata of: {path}"
186    FalhaLerMetadados,
187    /// "Failed to set permissions on: {path}"
188    FalhaDefinirPermissoes,
189
190    // API / Docs errors (4 variants)
191    /// "Failed to fetch documentation for: {library_id}"
192    FalhaBuscarDocumentacao,
193    /// "Failed to create HTTP client"
194    FalhaCriarClienteHttp,
195    /// "No documentation available"
196    SemDocumentacaoDisponivel,
197    /// "Library not found. Verify the ID via `context7 library <name>`."
198    BibliotecaNaoEncontradaApi,
199}
200
201impl Mensagem {
202    /// Returns the localised text for this message in the given language.
203    ///
204    /// Prefer this over the global [`t`] function when you need deterministic
205    /// translations without depending on the process-wide language setting
206    /// (useful for tests and library usage).
207    pub fn texto(self, idioma: Idioma) -> &'static str {
208        match idioma {
209            Idioma::English => en(self),
210            Idioma::Portugues => pt(self),
211        }
212    }
213}
214
215/// Returns the localised string for a message in the current language.
216///
217/// For parameterised messages use `format!("{} {}", t(Mensagem::Foo), param)`.
218pub fn t(msg: Mensagem) -> &'static str {
219    match idioma_atual() {
220        Idioma::English => en(msg),
221        Idioma::Portugues => pt(msg),
222    }
223}
224
225fn en(msg: Mensagem) -> &'static str {
226    match msg {
227        // Keys
228        Mensagem::ChaveAdicionada => "Key added successfully at:",
229        Mensagem::ChaveJaExistia => "Key already exists (skipping).",
230        Mensagem::NenhumaChaveArmazenada => "No key stored.",
231        Mensagem::UsarKeysAdd => "Use `context7 keys add <KEY>` to add a key.",
232        Mensagem::ContadorChaves => "key(s) stored:",
233        Mensagem::NenhumaChaveParaRemover => "No key stored to remove.",
234        Mensagem::IndiceInvalido => "Invalid index. Use a number between 1 and",
235        Mensagem::ChaveRemovidaSucesso => "Key removed successfully.",
236        Mensagem::OperacaoCancelada => "Operation cancelled.",
237        Mensagem::TodasChavesRemovidas => "All keys removed.",
238        Mensagem::SistemaXdgNaoSuportado => "System does not support XDG directories.",
239        Mensagem::ChavesImportadasSucesso => "key(s) imported successfully.",
240        Mensagem::NenhumaChaveContext7NoArquivo => "No CONTEXT7_API= key found in:",
241        Mensagem::ConfirmarRemoverTodas => "Are you sure you want to remove ALL keys? [y/N] ",
242        Mensagem::RespostaConfirmacao => "y|yes",
243
244        // Library / Docs
245        Mensagem::NenhumaBibliotecaEncontrada => "No library found.",
246        Mensagem::BibliotecasEncontradas => "Libraries found:",
247        Mensagem::ConfiancaScore => "trust",
248        Mensagem::NenhumaDocumentacaoEncontrada => "No documentation found.",
249        Mensagem::TituloDocumentacao => "Documentation:",
250        Mensagem::TituloFontes => "Sources:",
251        Mensagem::SemConteudoDisponivel => "No content available.",
252        Mensagem::BuscandoBiblioteca => "Searching library:",
253
254        // HTTP / Network
255        Mensagem::ErroDeRede => "Network error searching library:",
256        Mensagem::ErroDeRedeDocs => "Network error fetching documentation:",
257        Mensagem::FalhaDesserializar => "Failed to deserialise JSON response:",
258        Mensagem::RateLimitAtingido => "Rate limit reached (429), waiting for retry…",
259        Mensagem::ErrodoServidor => "Server error, retrying…",
260        Mensagem::ChaveApiInvalida => "Invalid API key (401/403), trying next…",
261        Mensagem::Tentativa => "Attempt",
262        Mensagem::AguardandoRetry => "Waiting before retrying…",
263
264        // Config / XDG
265        Mensagem::ErroCaminhoXdg => "System does not support XDG — cannot save configuration",
266        Mensagem::FalhaLerConfig => "Failed to read XDG config at:",
267        Mensagem::FalhaTomlInvalido => "Invalid TOML at:",
268        Mensagem::FalhaEscreverConfig => "Failed to write config at:",
269        Mensagem::FalhaCriarDiretorio => "Failed to create directory:",
270        Mensagem::NenhumaChaveConfigurada => {
271            "No API key configured. Set CONTEXT7_API_KEYS or use `context7 keys add <KEY>`."
272        }
273        Mensagem::FalhaSerializarToml => "Failed to serialise configuration to TOML",
274
275        // Logging / info
276        Mensagem::ChavesCarregadasEnvVar => {
277            "Keys loaded from CONTEXT7_API_KEYS environment variable"
278        }
279        Mensagem::ChavesCarregadasXdg => "Keys loaded from XDG configuration",
280        Mensagem::FalhaLerXdgContinuando => "Failed to read XDG configuration (continuing):",
281        Mensagem::IniciandoComChaves => "Starting context7 with",
282        Mensagem::ChavesCarregadasCompileTime => "Keys loaded from compile-time CONTEXT7_API_KEYS",
283        Mensagem::FalhaSerializarJson => "Failed to serialise results to JSON",
284        Mensagem::FalhaSerializarDocs => "Failed to serialise documentation to JSON",
285        Mensagem::FalhaBuscarBiblioteca => "Failed to search library",
286
287        // Permissions / IO
288        Mensagem::FalhaLerMetadados => "Failed to read metadata of:",
289        Mensagem::FalhaDefinirPermissoes => "Failed to set permissions on:",
290
291        // API / Docs errors
292        Mensagem::FalhaBuscarDocumentacao => "Failed to fetch documentation for:",
293        Mensagem::FalhaCriarClienteHttp => "Failed to create HTTP client",
294        Mensagem::SemDocumentacaoDisponivel => "No documentation available",
295        Mensagem::BibliotecaNaoEncontradaApi => {
296            "Library not found. Verify the ID via `context7 library <name>`."
297        }
298    }
299}
300
301fn pt(msg: Mensagem) -> &'static str {
302    match msg {
303        // Keys
304        Mensagem::ChaveAdicionada => "Chave adicionada com sucesso em:",
305        Mensagem::ChaveJaExistia => "Chave já existente (ignorando).",
306        Mensagem::NenhumaChaveArmazenada => "Nenhuma chave armazenada.",
307        Mensagem::UsarKeysAdd => "Use `context7 keys add <CHAVE>` para adicionar uma chave.",
308        Mensagem::ContadorChaves => "chave(s) armazenada(s):",
309        Mensagem::NenhumaChaveParaRemover => "Nenhuma chave armazenada para remover.",
310        Mensagem::IndiceInvalido => "Índice inválido. Use um número entre 1 e",
311        Mensagem::ChaveRemovidaSucesso => "Chave removida com sucesso.",
312        Mensagem::OperacaoCancelada => "Operação cancelada.",
313        Mensagem::TodasChavesRemovidas => "Todas as chaves foram removidas.",
314        Mensagem::SistemaXdgNaoSuportado => "Sistema não suporta diretórios XDG.",
315        Mensagem::ChavesImportadasSucesso => "chave(s) importada(s) com sucesso.",
316        Mensagem::NenhumaChaveContext7NoArquivo => "Nenhuma chave CONTEXT7_API= encontrada em:",
317        Mensagem::ConfirmarRemoverTodas => "Tem certeza que deseja remover TODAS as chaves? [s/N] ",
318        Mensagem::RespostaConfirmacao => "s|sim",
319
320        // Library / Docs
321        Mensagem::NenhumaBibliotecaEncontrada => "Nenhuma biblioteca encontrada.",
322        Mensagem::BibliotecasEncontradas => "Bibliotecas encontradas:",
323        Mensagem::ConfiancaScore => "confiança",
324        Mensagem::NenhumaDocumentacaoEncontrada => "Nenhuma documentação encontrada.",
325        Mensagem::TituloDocumentacao => "Documentação:",
326        Mensagem::TituloFontes => "Fontes:",
327        Mensagem::SemConteudoDisponivel => "Sem conteúdo disponível.",
328        Mensagem::BuscandoBiblioteca => "Buscando biblioteca:",
329
330        // HTTP / Network
331        Mensagem::ErroDeRede => "Erro de rede ao buscar biblioteca:",
332        Mensagem::ErroDeRedeDocs => "Erro de rede ao buscar documentação:",
333        Mensagem::FalhaDesserializar => "Falha ao desserializar resposta JSON:",
334        Mensagem::RateLimitAtingido => "Rate limit atingido (429), aguardando retry…",
335        Mensagem::ErrodoServidor => "Erro do servidor, tentando novamente…",
336        Mensagem::ChaveApiInvalida => "Chave de API inválida (401/403), tentando próxima…",
337        Mensagem::Tentativa => "Tentativa",
338        Mensagem::AguardandoRetry => "Aguardando antes de tentar novamente…",
339
340        // Config / XDG
341        Mensagem::ErroCaminhoXdg => {
342            "Sistema não suporta diretórios XDG — impossível salvar configuração"
343        }
344        Mensagem::FalhaLerConfig => "Falha ao ler configuração XDG em:",
345        Mensagem::FalhaTomlInvalido => "TOML inválido em:",
346        Mensagem::FalhaEscreverConfig => "Falha ao escrever config em:",
347        Mensagem::FalhaCriarDiretorio => "Falha ao criar diretório:",
348        Mensagem::NenhumaChaveConfigurada => {
349            "Nenhuma chave de API encontrada. Configure CONTEXT7_API_KEYS ou use `context7 keys add <CHAVE>`."
350        }
351        Mensagem::FalhaSerializarToml => "Falha ao serializar configuração para TOML",
352
353        // Logging / info
354        Mensagem::ChavesCarregadasEnvVar => {
355            "Chaves carregadas via variável de ambiente CONTEXT7_API_KEYS"
356        }
357        Mensagem::ChavesCarregadasXdg => "Chaves carregadas via configuração XDG",
358        Mensagem::FalhaLerXdgContinuando => "Falha ao ler configuração XDG (continuando):",
359        Mensagem::IniciandoComChaves => "Iniciando context7 com",
360        Mensagem::ChavesCarregadasCompileTime => {
361            "Chaves carregadas via compile-time CONTEXT7_API_KEYS"
362        }
363        Mensagem::FalhaSerializarJson => "Falha ao serializar resultados para JSON",
364        Mensagem::FalhaSerializarDocs => "Falha ao serializar documentação para JSON",
365        Mensagem::FalhaBuscarBiblioteca => "Falha ao buscar biblioteca",
366
367        // Permissions / IO
368        Mensagem::FalhaLerMetadados => "Falha ao ler metadados de:",
369        Mensagem::FalhaDefinirPermissoes => "Falha ao definir permissões em:",
370
371        // API / Docs errors
372        Mensagem::FalhaBuscarDocumentacao => "Falha ao buscar documentação para:",
373        Mensagem::FalhaCriarClienteHttp => "Falha ao criar cliente HTTP",
374        Mensagem::SemDocumentacaoDisponivel => "Nenhuma documentação disponível",
375        Mensagem::BibliotecaNaoEncontradaApi => {
376            "Biblioteca não encontrada. Verifique o ID via `context7 library <nome>`."
377        }
378    }
379}
380
381// ─── TESTES ───────────────────────────────────────────────────────────────────
382
383#[cfg(test)]
384mod testes {
385    use super::*;
386
387    #[test]
388    fn testa_idioma_atual_padrao_e_english() {
389        // Se OnceLock ainda não foi setado neste processo de teste, deve ser English
390        // (pode ter sido setado por outro teste — verificamos apenas que não panics)
391        let _ = idioma_atual();
392    }
393
394    #[test]
395    fn testa_resolver_idioma_cli_flag_pt() {
396        assert_eq!(resolver_idioma(Some("pt")), Idioma::Portugues);
397        assert_eq!(resolver_idioma(Some("pt-BR")), Idioma::Portugues);
398        assert_eq!(resolver_idioma(Some("PT_BR")), Idioma::Portugues);
399    }
400
401    #[test]
402    fn testa_resolver_idioma_cli_flag_en() {
403        assert_eq!(resolver_idioma(Some("en")), Idioma::English);
404        assert_eq!(resolver_idioma(Some("en-US")), Idioma::English);
405    }
406
407    #[test]
408    fn testa_resolver_idioma_sem_flag_nem_env_retorna_english_ou_pt() {
409        // Sem flag e sem env, deve retornar English ou Portugues (dependendo do sistema)
410        let idioma = resolver_idioma(None);
411        assert!(idioma == Idioma::English || idioma == Idioma::Portugues);
412    }
413
414    #[test]
415    fn testa_t_mensagem_nenhuma_chave_en() {
416        let msg_en = en(Mensagem::NenhumaChaveArmazenada);
417        assert!(!msg_en.is_empty());
418        assert!(
419            msg_en.to_lowercase().contains("no") || msg_en.to_lowercase().contains("key"),
420            "EN deve conter 'no' ou 'key', obteve: {}",
421            msg_en
422        );
423    }
424
425    #[test]
426    fn testa_t_mensagem_nenhuma_chave_pt() {
427        let msg_pt = pt(Mensagem::NenhumaChaveArmazenada);
428        assert!(!msg_pt.is_empty());
429        assert!(
430            msg_pt.to_lowercase().contains("nenhuma") || msg_pt.to_lowercase().contains("chave"),
431            "PT deve conter 'nenhuma' ou 'chave', obteve: {}",
432            msg_pt
433        );
434    }
435
436    #[test]
437    fn testa_confirmacao_resposta_en_contem_y() {
438        let confirmacao = en(Mensagem::RespostaConfirmacao);
439        assert!(
440            confirmacao.contains('y'),
441            "EN: resposta confirmação deve conter 'y'"
442        );
443    }
444
445    #[test]
446    fn testa_confirmacao_resposta_pt_contem_s() {
447        let confirmacao = pt(Mensagem::RespostaConfirmacao);
448        assert!(
449            confirmacao.contains('s'),
450            "PT: resposta confirmação deve conter 's'"
451        );
452    }
453
454    #[test]
455    fn testa_todas_variantes_en_nao_sao_vazias() {
456        let variantes = [
457            Mensagem::ChaveAdicionada,
458            Mensagem::ChaveJaExistia,
459            Mensagem::NenhumaChaveArmazenada,
460            Mensagem::UsarKeysAdd,
461            Mensagem::ContadorChaves,
462            Mensagem::NenhumaChaveParaRemover,
463            Mensagem::IndiceInvalido,
464            Mensagem::ChaveRemovidaSucesso,
465            Mensagem::OperacaoCancelada,
466            Mensagem::TodasChavesRemovidas,
467            Mensagem::SistemaXdgNaoSuportado,
468            Mensagem::ChavesImportadasSucesso,
469            Mensagem::NenhumaChaveContext7NoArquivo,
470            Mensagem::ConfirmarRemoverTodas,
471            Mensagem::RespostaConfirmacao,
472            Mensagem::NenhumaBibliotecaEncontrada,
473            Mensagem::BibliotecasEncontradas,
474            Mensagem::ConfiancaScore,
475            Mensagem::NenhumaDocumentacaoEncontrada,
476            Mensagem::TituloDocumentacao,
477            Mensagem::TituloFontes,
478            Mensagem::SemConteudoDisponivel,
479            Mensagem::BuscandoBiblioteca,
480            Mensagem::ErroDeRede,
481            Mensagem::ErroDeRedeDocs,
482            Mensagem::FalhaDesserializar,
483            Mensagem::RateLimitAtingido,
484            Mensagem::ErrodoServidor,
485            Mensagem::ChaveApiInvalida,
486            Mensagem::Tentativa,
487            Mensagem::AguardandoRetry,
488            Mensagem::ErroCaminhoXdg,
489            Mensagem::FalhaLerConfig,
490            Mensagem::FalhaTomlInvalido,
491            Mensagem::FalhaEscreverConfig,
492            Mensagem::FalhaCriarDiretorio,
493            Mensagem::NenhumaChaveConfigurada,
494            Mensagem::FalhaSerializarToml,
495            Mensagem::ChavesCarregadasEnvVar,
496            Mensagem::ChavesCarregadasXdg,
497            Mensagem::FalhaLerXdgContinuando,
498            Mensagem::IniciandoComChaves,
499            Mensagem::ChavesCarregadasCompileTime,
500            Mensagem::FalhaSerializarJson,
501            Mensagem::FalhaSerializarDocs,
502            Mensagem::FalhaBuscarBiblioteca,
503            Mensagem::FalhaLerMetadados,
504            Mensagem::FalhaDefinirPermissoes,
505            Mensagem::FalhaBuscarDocumentacao,
506            Mensagem::FalhaCriarClienteHttp,
507            Mensagem::SemDocumentacaoDisponivel,
508            Mensagem::BibliotecaNaoEncontradaApi,
509        ];
510
511        for v in &variantes {
512            let msg_en = en(*v);
513            let msg_pt = pt(*v);
514            assert!(
515                !msg_en.is_empty(),
516                "EN mensagem vazia para variante {:?}",
517                v
518            );
519            assert!(
520                !msg_pt.is_empty(),
521                "PT mensagem vazia para variante {:?}",
522                v
523            );
524        }
525    }
526}
context7_cli/i18n.rs

context7_cli/
i18n.rs
