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

context7_cli/
i18n.rs
