Skip to main content

sqlite_graphrag/
i18n.rs

1//! Bilingual human-readable message layer.
2//!
3//! The CLI uses `--lang en|pt` (global flag) or `SQLITE_GRAPHRAG_LANG` (env var) to choose
4//! the language of stderr progress messages. JSON stdout is deterministic and identical
5//! across languages — only strings intended for humans pass through this module.
6//!
7//! Detection (highest to lowest priority):
8//! 1. Explicit `--lang` flag
9//! 2. Env var `SQLITE_GRAPHRAG_LANG`
10//! 3. OS locale (`LANG`, `LC_ALL`) with `pt` prefix
11//! 4. Fallback `English`
12
13use std::sync::OnceLock;
14
15#[derive(Copy, Clone, Debug, PartialEq, Eq, clap::ValueEnum)]
16pub enum Language {
17    #[value(name = "en", aliases = ["english", "EN"])]
18    English,
19    #[value(name = "pt", aliases = ["portugues", "portuguese", "pt-BR", "pt-br", "PT"])]
20    Portuguese,
21}
22
23impl Language {
24    /// Parses a command-line string into a `Language` without relying on clap.
25    /// Accepts the same aliases defined in `#[value(...)]`: "en", "pt", etc.
26    pub fn from_str_opt(s: &str) -> Option<Self> {
27        match s.to_lowercase().as_str() {
28            "en" | "english" => Some(Language::English),
29            "pt" | "pt-br" | "portugues" | "portuguese" => Some(Language::Portuguese),
30            _ => None,
31        }
32    }
33
34    pub fn from_env_or_locale() -> Self {
35        // Priority 1: explicit SQLITE_GRAPHRAG_LANG env var (highest precedence).
36        // Empty string treated as unset per POSIX convention.
37        if let Ok(v) = std::env::var("SQLITE_GRAPHRAG_LANG") {
38            if !v.is_empty() {
39                let lower = v.to_lowercase();
40                if lower.starts_with("pt") {
41                    return Language::Portuguese;
42                }
43                if lower.starts_with("en") {
44                    return Language::English;
45                }
46                tracing::warn!(target: "i18n",
47                    value = %v,
48                    "SQLITE_GRAPHRAG_LANG value not recognized, falling back to locale detection"
49                );
50            }
51        }
52        // Priority 2: POSIX locale precedence LC_ALL > LC_MESSAGES > LANG.
53        // We read these via std::env (not via sys_locale) because:
54        // (a) `sys_locale::get_locale()` calls into native OS APIs (CFLocaleCopyCurrent
55        //     on macOS, GetUserDefaultLocaleName on Windows) which cache the
56        //     system locale and IGNORE env vars set at runtime by tests;
57        // (b) POSIX specifies LC_ALL > LC_MESSAGES > LANG ordering and an
58        //     unrecognised LC_ALL value must stop iteration (fall back to
59        //     English default).
60        for var in ["LC_ALL", "LC_MESSAGES", "LANG"] {
61            if let Ok(v) = std::env::var(var) {
62                if v.is_empty() {
63                    continue;
64                }
65                let lower = v.to_lowercase();
66                if lower.starts_with("pt") {
67                    return Language::Portuguese;
68                }
69                if lower.starts_with("en") {
70                    return Language::English;
71                }
72                // Unrecognised value in a higher-precedence variable stops
73                // iteration per POSIX.1-2017 §8.2.
74                if var == "LC_ALL" {
75                    return Language::English;
76                }
77            }
78        }
79        // Priority 3: cross-platform locale detection via native OS APIs.
80        // Only reached when no POSIX env var is set.
81        if let Some(locale) = sys_locale::get_locale() {
82            let lower = locale.to_lowercase();
83            if lower.starts_with("pt") {
84                return Language::Portuguese;
85            }
86            if lower.starts_with("en") {
87                return Language::English;
88            }
89        }
90        Language::English
91    }
92}
93
94static GLOBAL_LANGUAGE: OnceLock<Language> = OnceLock::new();
95
96/// Initializes the global language. Subsequent calls are silently ignored
97/// (OnceLock semantics) — guaranteeing thread-safety and determinism.
98///
99/// v1.0.36 (L6): early-return when already initialized so the env-fallback
100/// resolver (`from_env_or_locale`) does not run a second time. Without this
101/// guard, calling `init(None)` after `current()` already populated the
102/// OnceLock causes `from_env_or_locale` to fire its `tracing::warn!` twice
103/// for unrecognized `SQLITE_GRAPHRAG_LANG` values.
104pub fn init(explicit: Option<Language>) {
105    if GLOBAL_LANGUAGE.get().is_some() {
106        return;
107    }
108    let resolved = explicit.unwrap_or_else(Language::from_env_or_locale);
109    let _ = GLOBAL_LANGUAGE.set(resolved);
110}
111
112/// Returns the active language, or fallback English if `init` was never called.
113pub fn current() -> Language {
114    *GLOBAL_LANGUAGE.get_or_init(Language::from_env_or_locale)
115}
116
117/// Translates a bilingual message by selecting the active variant.
118///
119/// v1.0.36 (M4): inputs are constrained to `&'static str` so the function
120/// can return one of them directly without `Box::leak`. The previous
121/// implementation leaked one allocation per call which accumulated in
122/// long-running pipelines; this version is allocation-free. All in-tree
123/// callers already pass string literals, which are `&'static str`.
124pub fn tr(en: &'static str, pt: &'static str) -> &'static str {
125    match current() {
126        Language::English => en,
127        Language::Portuguese => pt,
128    }
129}
130
131/// Progress message emitted after pruning relationships.
132///
133/// English-only: this string is emitted to stderr as a progress notice and
134/// does not vary by language because the prune-relations command targets
135/// agent-first pipelines where deterministic output matters.
136pub fn relations_pruned(count: usize, relation: &str, namespace: &str) -> String {
137    format!("pruned {count} '{relation}' relationships in namespace '{namespace}'")
138}
139
140/// Progress message for dry-run preview of prune-relations.
141///
142/// English-only: emitted to stderr as a progress notice.
143pub fn prune_dry_run(count: usize, relation: &str) -> String {
144    format!("dry run: {count} '{relation}' relationships would be removed")
145}
146
147/// Warning message when --yes is not passed for destructive prune-relations.
148///
149/// English-only: emitted to stderr as a progress notice.
150pub fn prune_requires_yes() -> String {
151    "destructive operation requires --yes flag; use --dry-run to preview".to_string()
152}
153
154/// Localized prefix for error messages displayed to the end user.
155pub fn error_prefix() -> &'static str {
156    match current() {
157        Language::English => "Error",
158        Language::Portuguese => "Erro",
159    }
160}
161
162/// Error messages for `AppError` variants — always English.
163///
164/// These strings end up inside `AppError` inner fields and may appear in
165/// deterministic JSON stdout (e.g. ingest NDJSON). Portuguese translations
166/// for stderr live in `pub mod app_error_pt` and are applied by
167/// `localized_message_for(Language::Portuguese)`.
168pub mod errors_msg {
169    pub fn memory_not_found(nome: &str, namespace: &str) -> String {
170        format!("memory '{nome}' not found in namespace '{namespace}'")
171    }
172
173    pub fn memory_or_entity_not_found(name: &str, namespace: &str) -> String {
174        format!("memory or entity '{name}' not found in namespace '{namespace}'")
175    }
176
177    pub fn database_not_found(path: &str) -> String {
178        format!("database not found at {path}. Run 'sqlite-graphrag init' first.")
179    }
180
181    pub fn entity_not_found(nome: &str, namespace: &str) -> String {
182        format!("entity \"{nome}\" does not exist in namespace \"{namespace}\"")
183    }
184
185    pub fn relationship_not_found(de: &str, rel: &str, para: &str, namespace: &str) -> String {
186        format!(
187            "relationship \"{de}\" --[{rel}]--> \"{para}\" does not exist in namespace \"{namespace}\""
188        )
189    }
190
191    pub fn duplicate_memory(nome: &str, namespace: &str) -> String {
192        format!(
193            "memory '{nome}' already exists in namespace '{namespace}'. Use --force-merge to update."
194        )
195    }
196
197    pub fn duplicate_memory_soft_deleted(name: &str, namespace: &str) -> String {
198        format!(
199            "memory '{name}' exists but is soft-deleted in namespace '{namespace}'; \
200             use --force-merge to restore and update, or `restore` to revive it"
201        )
202    }
203
204    pub fn optimistic_lock_conflict(expected: i64, current_ts: i64) -> String {
205        format!(
206            "optimistic lock conflict: expected updated_at={expected}, but current is {current_ts}"
207        )
208    }
209
210    pub fn version_not_found(versao: i64, nome: &str) -> String {
211        format!("version {versao} not found for memory '{nome}'")
212    }
213
214    pub fn no_recall_results(max_distance: f32, query: &str, namespace: &str) -> String {
215        format!(
216            "no results within --max-distance {max_distance} for query '{query}' in namespace '{namespace}'"
217        )
218    }
219
220    pub fn soft_deleted_memory_not_found(nome: &str, namespace: &str) -> String {
221        format!("soft-deleted memory '{nome}' not found in namespace '{namespace}'")
222    }
223
224    pub fn concurrent_process_conflict() -> String {
225        "optimistic lock conflict: memory was modified by another process".to_string()
226    }
227
228    pub fn entity_limit_exceeded(max: usize) -> String {
229        format!("entities exceed limit of {max}")
230    }
231
232    pub fn relationship_limit_exceeded(max: usize) -> String {
233        format!("relationships exceed limit of {max}")
234    }
235}
236
237/// Localized validation messages for memory fields.
238pub mod validation {
239    use super::current;
240    use crate::i18n::Language;
241
242    pub fn name_length(max: usize) -> String {
243        match current() {
244            Language::English => format!("name must be 1-{max} chars"),
245            Language::Portuguese => format!("nome deve ter entre 1 e {max} caracteres"),
246        }
247    }
248
249    pub fn reserved_name() -> String {
250        match current() {
251            Language::English => {
252                "names and namespaces starting with __ are reserved for internal use".to_string()
253            }
254            Language::Portuguese => {
255                "nomes e namespaces iniciados com __ são reservados para uso interno".to_string()
256            }
257        }
258    }
259
260    pub fn name_kebab(nome: &str) -> String {
261        match current() {
262            Language::English => format!(
263                "name must be kebab-case slug (lowercase letters, digits, hyphens): '{nome}'"
264            ),
265            Language::Portuguese => {
266                format!("nome deve estar em kebab-case (minúsculas, dígitos, hífens): '{nome}'")
267            }
268        }
269    }
270
271    pub fn description_exceeds(max: usize) -> String {
272        match current() {
273            Language::English => format!("description must be <= {max} chars"),
274            Language::Portuguese => format!("descrição deve ter no máximo {max} caracteres"),
275        }
276    }
277
278    pub fn body_exceeds(max: usize) -> String {
279        match current() {
280            Language::English => format!("body exceeds {max} bytes"),
281            Language::Portuguese => format!("corpo excede {max} bytes"),
282        }
283    }
284
285    pub fn new_name_length(max: usize) -> String {
286        match current() {
287            Language::English => format!("new-name must be 1-{max} chars"),
288            Language::Portuguese => format!("novo nome deve ter entre 1 e {max} caracteres"),
289        }
290    }
291
292    pub fn new_name_kebab(nome: &str) -> String {
293        match current() {
294            Language::English => format!(
295                "new-name must be kebab-case slug (lowercase letters, digits, hyphens): '{nome}'"
296            ),
297            Language::Portuguese => format!(
298                "novo nome deve estar em kebab-case (minúsculas, dígitos, hífens): '{nome}'"
299            ),
300        }
301    }
302
303    pub fn namespace_length() -> String {
304        match current() {
305            Language::English => "namespace must be 1-80 chars".to_string(),
306            Language::Portuguese => "namespace deve ter entre 1 e 80 caracteres".to_string(),
307        }
308    }
309
310    pub fn namespace_format() -> String {
311        match current() {
312            Language::English => "namespace must be alphanumeric + hyphens/underscores".to_string(),
313            Language::Portuguese => {
314                "namespace deve ser alfanumérico com hífens/sublinhados".to_string()
315            }
316        }
317    }
318
319    pub fn path_traversal(p: &str) -> String {
320        match current() {
321            Language::English => format!("path traversal rejected: {p}"),
322            Language::Portuguese => format!("traversal de caminho rejeitado: {p}"),
323        }
324    }
325
326    pub fn invalid_tz(v: &str) -> String {
327        match current() {
328            Language::English => format!(
329                "SQLITE_GRAPHRAG_DISPLAY_TZ invalid: '{v}'; use an IANA name like 'America/Sao_Paulo'"
330            ),
331            Language::Portuguese => format!(
332                "SQLITE_GRAPHRAG_DISPLAY_TZ inválido: '{v}'; use um nome IANA como 'America/Sao_Paulo'"
333            ),
334        }
335    }
336
337    pub fn empty_query() -> String {
338        match current() {
339            Language::English => "query cannot be empty".to_string(),
340            Language::Portuguese => "a consulta não pode estar vazia".to_string(),
341        }
342    }
343
344    pub fn empty_body() -> String {
345        match current() {
346            Language::English => "body cannot be empty: provide --body, --body-file, or --body-stdin with content, or supply a graph via --entities-file/--graph-stdin".to_string(),
347            Language::Portuguese => "o corpo não pode estar vazio: forneça --body, --body-file ou --body-stdin com conteúdo, ou um grafo via --entities-file/--graph-stdin".to_string(),
348        }
349    }
350
351    pub fn invalid_namespace_config(path: &str, err: &str) -> String {
352        match current() {
353            Language::English => {
354                format!("invalid project namespace config '{path}': {err}")
355            }
356            Language::Portuguese => {
357                format!("configuração de namespace de projeto inválida '{path}': {err}")
358            }
359        }
360    }
361
362    pub fn invalid_projects_mapping(path: &str, err: &str) -> String {
363        match current() {
364            Language::English => format!("invalid projects mapping '{path}': {err}"),
365            Language::Portuguese => format!("mapeamento de projetos inválido '{path}': {err}"),
366        }
367    }
368
369    pub fn self_referential_link() -> String {
370        match current() {
371            Language::English => "--from and --to must be different entities — self-referential relationships are not supported".to_string(),
372            Language::Portuguese => "--from e --to devem ser entidades diferentes — relacionamentos auto-referenciais não são suportados".to_string(),
373        }
374    }
375
376    pub fn invalid_link_weight(weight: f64) -> String {
377        match current() {
378            Language::English => {
379                format!("--weight: must be between 0.0 and 1.0 (actual: {weight})")
380            }
381            Language::Portuguese => {
382                format!("--weight: deve estar entre 0.0 e 1.0 (atual: {weight})")
383            }
384        }
385    }
386
387    pub fn sync_destination_equals_source() -> String {
388        match current() {
389            Language::English => {
390                "destination path must differ from the source database path".to_string()
391            }
392            Language::Portuguese => {
393                "caminho de destino deve ser diferente do caminho do banco de dados fonte"
394                    .to_string()
395            }
396        }
397    }
398
399    /// Portuguese translations for `AppError` Display messages.
400    ///
401    /// Each helper mirrors a single `AppError` variant's `#[error(...)]` text in
402    /// Portuguese, keeping the language barrier confined to this module. The
403    /// English source of truth lives in `src/errors.rs` via `thiserror`.
404    pub mod app_error_pt {
405        pub fn validation(msg: &str) -> String {
406            format!("erro de validação: {msg}")
407        }
408
409        pub fn duplicate(msg: &str) -> String {
410            let translated = msg
411                .replace("already exists in namespace", "já existe no namespace")
412                .replace(
413                    "exists but is soft-deleted in namespace",
414                    "existe mas está excluída temporariamente no namespace",
415                )
416                .replace(
417                    "Use --force-merge to update.",
418                    "Use --force-merge para atualizar.",
419                )
420                .replace(
421                    "use --force-merge to restore and update, or `restore` to revive it",
422                    "use --force-merge para restaurar e atualizar, ou `restore` para revivê-la",
423                )
424                .replace("memory", "memória");
425            format!("duplicata detectada: {translated}")
426        }
427
428        pub fn conflict(msg: &str) -> String {
429            let translated = msg
430                .replace("optimistic lock conflict", "conflito de lock otimista")
431                .replace("but current is", "mas atual é")
432                .replace(
433                    "was modified by another process",
434                    "foi modificada por outro processo",
435                );
436            format!("conflito: {translated}")
437        }
438
439        pub fn not_found(msg: &str) -> String {
440            // G55 T3: add replacements for the read.rs format produced by the
441            // T1 fix: `memory not found: name='X' in namespace 'Y'`.
442            // The existing chain did not catch ` in namespace '` when broken
443            // by the name label, leaving a bilingual hybrid. New patterns
444            // must run BEFORE the catch-all `memory` → `memória` to avoid
445            // being shadowed.
446            let translated = msg
447                .replace("memory not found:", "memória não encontrada:")
448                .replace("not found in namespace", "não encontrada no namespace")
449                .replace("not found for memory", "não encontrada para memória")
450                .replace("does not exist in namespace", "não existe no namespace")
451                .replace("memory or entity", "memória ou entidade")
452                .replace("name='", "nome='")
453                .replace("memory", "memória")
454                .replace("entity", "entidade")
455                .replace(" in namespace '", " no namespace '")
456                .replace("version", "versão")
457                .replace("soft-deleted", "excluída temporariamente");
458            format!("não encontrado: {translated}")
459        }
460
461        // G55 S2 (v1.0.80): structured variant helpers. They synthesize the
462        // canonical English message and feed it through the `not_found`
463        // replace-chain so the pt-BR translation stays in one place.
464        pub fn memory_not_found(name: &str, namespace: &str) -> String {
465            not_found(&format!(
466                "memory not found: name='{name}' in namespace '{namespace}'"
467            ))
468        }
469
470        pub fn memory_not_found_by_id(id: i64) -> String {
471            not_found(&format!("memory not found: id={id}"))
472        }
473
474        pub fn namespace_error(msg: &str) -> String {
475            format!("namespace não resolvido: {msg}")
476        }
477
478        pub fn limit_exceeded(msg: &str) -> String {
479            let translated = msg
480                .replace("exceeds limit of", "excede limite de")
481                .replace("body exceeds", "corpo excede")
482                .replace("entities exceed limit", "entidades excedem limite")
483                .replace(
484                    "relationships exceed limit",
485                    "relacionamentos excedem limite",
486                );
487            format!("limite excedido: {translated}")
488        }
489
490        pub fn database(err: &str) -> String {
491            format!("erro de banco de dados: {err}")
492        }
493
494        pub fn embedding(msg: &str) -> String {
495            format!("erro de embedding: {msg}")
496        }
497
498        pub fn vec_extension(msg: &str) -> String {
499            format!("extensão sqlite-vec falhou: {msg}")
500        }
501
502        pub fn provider_error(code: &str, message: &str) -> String {
503            format!("erro do provedor (código {code}): {message}")
504        }
505
506        pub fn db_busy(msg: &str) -> String {
507            format!("banco ocupado: {msg}")
508        }
509
510        pub fn batch_partial_failure(total: usize, failed: usize) -> String {
511            format!("falha parcial em batch: {failed} de {total} itens falharam")
512        }
513
514        pub fn io(err: &str) -> String {
515            format!("erro de I/O: {err}")
516        }
517
518        pub fn internal(err: &str) -> String {
519            format!("erro interno: {err}")
520        }
521
522        pub fn json(err: &str) -> String {
523            format!("erro de JSON: {err}")
524        }
525
526        pub fn lock_busy(msg: &str) -> String {
527            format!("lock ocupado: {msg}")
528        }
529
530        pub fn all_slots_full(max: usize, waited_secs: u64) -> String {
531            format!(
532                "todos os {max} slots de concorrência ocupados após aguardar {waited_secs}s \
533                 (exit 75); use --max-concurrency ou aguarde outras invocações terminarem"
534            )
535        }
536
537        pub fn job_singleton_locked(job_type: &str, namespace: &str) -> String {
538            format!(
539                "job {job_type} para o namespace '{namespace}' já está em execução (exit 75); \
540                 aguarde a conclusão ou passe --wait-job-singleton <SEGUNDOS>"
541            )
542        }
543
544        pub fn embedding_singleton_locked(namespace: &str) -> String {
545            format!(
546                "singleton de embedding para o namespace '{namespace}' já está retido (exit 75); \
547                 outra CLI está chamando o LLM neste banco; passe --wait-embed-singleton <SEGUNDOS> para aguardar"
548            )
549        }
550
551        pub fn low_memory(available_mb: u64, required_mb: u64) -> String {
552            format!(
553                "memória disponível ({available_mb}MB) abaixo do mínimo requerido ({required_mb}MB) \
554                 para carregar o modelo; aborte outras cargas ou use --skip-memory-guard (exit 77)"
555            )
556        }
557
558        pub fn shutdown(signal: &str) -> String {
559            format!(
560                "sinal de desligamento recebido: {signal}; operação cancelada pelo usuário (exit 19)"
561            )
562        }
563
564        pub fn preflight_failed(detail: &str) -> String {
565            format!(
566                "validação pré-execução falhou (exit 16): {detail}; corrija a condição e tente novamente (definir SQLITE_GRAPHRAG_SKIP_PREFLIGHT=1 desabilita esta validação em emergências)"
567            )
568        }
569
570        pub fn binary_not_found(name: &str) -> String {
571            format!("binário não encontrado: {name} — instale e adicione ao PATH")
572        }
573
574        pub fn rate_limited(detail: &str) -> String {
575            format!("taxa de requisição excedida: {detail}")
576        }
577
578        pub fn timeout(operation: &str, secs: u64) -> String {
579            format!("timeout após {secs}s: {operation}")
580        }
581    }
582
583    /// Portuguese translations for runtime startup messages emitted from `main.rs`.
584    ///
585    /// These mirror the English text supplied alongside each call to
586    /// `output::emit_progress_i18n` / `output::emit_error_i18n`, keeping the PT
587    /// strings confined to this module per the language policy.
588    pub mod runtime_pt {
589        pub fn embedding_heavy_must_measure_ram() -> String {
590            "comando intensivo em embedding precisa medir RAM disponível".to_string()
591        }
592
593        pub fn heavy_command_detected(available_mb: u64, safe_concurrency: usize) -> String {
594            format!(
595                "Comando pesado detectado; memória disponível: {available_mb} MB; \
596                 concorrência segura: {safe_concurrency}"
597            )
598        }
599
600        pub fn reducing_concurrency(
601            requested_concurrency: usize,
602            effective_concurrency: usize,
603        ) -> String {
604            format!(
605                "Reduzindo a concorrência solicitada de {requested_concurrency} para \
606                 {effective_concurrency} para evitar oversubscription de memória"
607            )
608        }
609
610        pub fn initializing_embedding_model() -> &'static str {
611            "Inicializando modelo de embedding (pode baixar na primeira execução)..."
612        }
613
614        pub fn embedding_chunks_serially(count: usize) -> String {
615            format!("Embedando {count} chunks serialmente para manter memória limitada...")
616        }
617
618        pub fn remember_step_input_validated(available_mb: u64) -> String {
619            format!("Etapa remember: entrada validada; memória disponível {available_mb} MB")
620        }
621
622        pub fn remember_step_chunking_completed(
623            total_passage_tokens: usize,
624            model_max_length: usize,
625            chunks_count: usize,
626            rss_mb: u64,
627        ) -> String {
628            format!(
629                "Etapa remember: tokenizer contou {total_passage_tokens} tokens de passagem \
630                 (máximo do modelo {model_max_length}); chunking gerou {chunks_count} chunks; \
631                 RSS do processo {rss_mb} MB"
632            )
633        }
634
635        pub fn remember_step_embeddings_completed(rss_mb: u64) -> String {
636            format!("Etapa remember: embeddings dos chunks concluídos; RSS do processo {rss_mb} MB")
637        }
638
639        pub fn restore_recomputing_embedding() -> &'static str {
640            "Recalculando embedding da memória restaurada..."
641        }
642
643        pub fn edit_recomputing_embedding() -> &'static str {
644            "Recalculando embedding da memória editada..."
645        }
646    }
647}
648
649#[cfg(test)]
650mod tests {
651    use super::*;
652    use serial_test::serial;
653
654    #[test]
655    #[serial]
656    fn fallback_english_when_env_absent() {
657        std::env::remove_var("SQLITE_GRAPHRAG_LANG");
658        std::env::set_var("LC_ALL", "C");
659        std::env::set_var("LANG", "C");
660        assert_eq!(Language::from_env_or_locale(), Language::English);
661        std::env::remove_var("LC_ALL");
662        std::env::remove_var("LANG");
663    }
664
665    #[test]
666    #[serial]
667    fn env_pt_selects_portuguese() {
668        std::env::remove_var("LC_ALL");
669        std::env::remove_var("LANG");
670        std::env::set_var("SQLITE_GRAPHRAG_LANG", "pt");
671        assert_eq!(Language::from_env_or_locale(), Language::Portuguese);
672        std::env::remove_var("SQLITE_GRAPHRAG_LANG");
673    }
674
675    #[test]
676    #[serial]
677    fn env_pt_br_selects_portuguese() {
678        std::env::remove_var("LC_ALL");
679        std::env::remove_var("LANG");
680        std::env::set_var("SQLITE_GRAPHRAG_LANG", "pt-BR");
681        assert_eq!(Language::from_env_or_locale(), Language::Portuguese);
682        std::env::remove_var("SQLITE_GRAPHRAG_LANG");
683    }
684
685    #[test]
686    #[serial]
687    fn locale_ptbr_utf8_selects_portuguese() {
688        std::env::remove_var("SQLITE_GRAPHRAG_LANG");
689        std::env::set_var("LC_ALL", "pt_BR.UTF-8");
690        assert_eq!(Language::from_env_or_locale(), Language::Portuguese);
691        std::env::remove_var("LC_ALL");
692    }
693
694    #[test]
695    #[serial]
696    fn posix_precedence_lc_all_overrides_lang() {
697        std::env::remove_var("SQLITE_GRAPHRAG_LANG");
698        std::env::remove_var("LC_MESSAGES");
699        std::env::set_var("LC_ALL", "en_US.UTF-8");
700        std::env::set_var("LANG", "pt_BR.UTF-8");
701        assert_eq!(
702            Language::from_env_or_locale(),
703            Language::English,
704            "LC_ALL=en_US must override LANG=pt_BR per POSIX"
705        );
706        std::env::remove_var("LC_ALL");
707        std::env::remove_var("LANG");
708    }
709
710    #[test]
711    #[serial]
712    fn posix_precedence_lc_all_unrecognized_stops_iteration() {
713        std::env::remove_var("SQLITE_GRAPHRAG_LANG");
714        std::env::remove_var("LC_MESSAGES");
715        std::env::set_var("LC_ALL", "ja_JP.UTF-8");
716        std::env::set_var("LANG", "pt_BR.UTF-8");
717        assert_eq!(
718            Language::from_env_or_locale(),
719            Language::English,
720            "LC_ALL=ja_JP set must stop iteration; falls back to English default"
721        );
722        std::env::remove_var("LC_ALL");
723        std::env::remove_var("LANG");
724    }
725
726    #[test]
727    #[serial]
728    fn lang_pt_selects_portuguese_when_lc_all_unset() {
729        std::env::remove_var("SQLITE_GRAPHRAG_LANG");
730        std::env::remove_var("LC_ALL");
731        std::env::remove_var("LC_MESSAGES");
732        std::env::set_var("LANG", "pt_BR.UTF-8");
733        assert_eq!(Language::from_env_or_locale(), Language::Portuguese);
734        std::env::remove_var("LANG");
735    }
736
737    mod validation_tests {
738        use super::*;
739
740        #[test]
741        fn name_length_en() {
742            let msg = match Language::English {
743                Language::English => format!("name must be 1-{} chars", 80),
744                Language::Portuguese => format!("nome deve ter entre 1 e {} caracteres", 80),
745            };
746            assert!(msg.contains("name must be 1-80 chars"), "obtido: {msg}");
747        }
748
749        #[test]
750        fn name_length_pt() {
751            let msg = match Language::Portuguese {
752                Language::English => format!("name must be 1-{} chars", 80),
753                Language::Portuguese => format!("nome deve ter entre 1 e {} caracteres", 80),
754            };
755            assert!(
756                msg.contains("nome deve ter entre 1 e 80 caracteres"),
757                "obtido: {msg}"
758            );
759        }
760
761        #[test]
762        fn name_kebab_en() {
763            let nome = "Invalid_Name";
764            let msg = match Language::English {
765                Language::English => format!(
766                    "name must be kebab-case slug (lowercase letters, digits, hyphens): '{nome}'"
767                ),
768                Language::Portuguese => {
769                    format!("nome deve estar em kebab-case (minúsculas, dígitos, hífens): '{nome}'")
770                }
771            };
772            assert!(msg.contains("kebab-case slug"), "obtido: {msg}");
773            assert!(msg.contains("Invalid_Name"), "obtido: {msg}");
774        }
775
776        #[test]
777        fn name_kebab_pt() {
778            let nome = "Invalid_Name";
779            let msg = match Language::Portuguese {
780                Language::English => format!(
781                    "name must be kebab-case slug (lowercase letters, digits, hyphens): '{nome}'"
782                ),
783                Language::Portuguese => {
784                    format!("nome deve estar em kebab-case (minúsculas, dígitos, hífens): '{nome}'")
785                }
786            };
787            assert!(msg.contains("kebab-case"), "obtido: {msg}");
788            assert!(msg.contains("minúsculas"), "obtido: {msg}");
789            assert!(msg.contains("Invalid_Name"), "obtido: {msg}");
790        }
791
792        #[test]
793        fn description_exceeds_en() {
794            let msg = match Language::English {
795                Language::English => format!("description must be <= {} chars", 500),
796                Language::Portuguese => format!("descrição deve ter no máximo {} caracteres", 500),
797            };
798            assert!(msg.contains("description must be <= 500"), "obtido: {msg}");
799        }
800
801        #[test]
802        fn description_exceeds_pt() {
803            let msg = match Language::Portuguese {
804                Language::English => format!("description must be <= {} chars", 500),
805                Language::Portuguese => format!("descrição deve ter no máximo {} caracteres", 500),
806            };
807            assert!(
808                msg.contains("descrição deve ter no máximo 500"),
809                "obtido: {msg}"
810            );
811        }
812
813        #[test]
814        fn body_exceeds_en() {
815            let limite = crate::constants::MAX_MEMORY_BODY_LEN;
816            let msg = match Language::English {
817                Language::English => format!("body exceeds {limite} bytes"),
818                Language::Portuguese => format!("corpo excede {limite} bytes"),
819            };
820            assert!(msg.contains("body exceeds 512000"), "obtido: {msg}");
821        }
822
823        #[test]
824        fn body_exceeds_pt() {
825            let limite = crate::constants::MAX_MEMORY_BODY_LEN;
826            let msg = match Language::Portuguese {
827                Language::English => format!("body exceeds {limite} bytes"),
828                Language::Portuguese => format!("corpo excede {limite} bytes"),
829            };
830            assert!(msg.contains("corpo excede 512000"), "obtido: {msg}");
831        }
832
833        #[test]
834        fn new_name_length_en() {
835            let msg = match Language::English {
836                Language::English => format!("new-name must be 1-{} chars", 80),
837                Language::Portuguese => format!("novo nome deve ter entre 1 e {} caracteres", 80),
838            };
839            assert!(msg.contains("new-name must be 1-80"), "obtido: {msg}");
840        }
841
842        #[test]
843        fn new_name_length_pt() {
844            let msg = match Language::Portuguese {
845                Language::English => format!("new-name must be 1-{} chars", 80),
846                Language::Portuguese => format!("novo nome deve ter entre 1 e {} caracteres", 80),
847            };
848            assert!(
849                msg.contains("novo nome deve ter entre 1 e 80"),
850                "obtido: {msg}"
851            );
852        }
853
854        #[test]
855        fn new_name_kebab_en() {
856            let nome = "Bad Name";
857            let msg = match Language::English {
858                Language::English => format!(
859                    "new-name must be kebab-case slug (lowercase letters, digits, hyphens): '{nome}'"
860                ),
861                Language::Portuguese => format!(
862                    "novo nome deve estar em kebab-case (minúsculas, dígitos, hífens): '{nome}'"
863                ),
864            };
865            assert!(msg.contains("new-name must be kebab-case"), "obtido: {msg}");
866        }
867
868        #[test]
869        fn new_name_kebab_pt() {
870            let nome = "Bad Name";
871            let msg = match Language::Portuguese {
872                Language::English => format!(
873                    "new-name must be kebab-case slug (lowercase letters, digits, hyphens): '{nome}'"
874                ),
875                Language::Portuguese => format!(
876                    "novo nome deve estar em kebab-case (minúsculas, dígitos, hífens): '{nome}'"
877                ),
878            };
879            assert!(
880                msg.contains("novo nome deve estar em kebab-case"),
881                "obtido: {msg}"
882            );
883        }
884
885        #[test]
886        fn reserved_name_en() {
887            let msg = match Language::English {
888                Language::English => {
889                    "names and namespaces starting with __ are reserved for internal use"
890                        .to_string()
891                }
892                Language::Portuguese => {
893                    "nomes e namespaces iniciados com __ são reservados para uso interno"
894                        .to_string()
895                }
896            };
897            assert!(msg.contains("reserved for internal use"), "obtido: {msg}");
898        }
899
900        #[test]
901        fn reserved_name_pt() {
902            let msg = match Language::Portuguese {
903                Language::English => {
904                    "names and namespaces starting with __ are reserved for internal use"
905                        .to_string()
906                }
907                Language::Portuguese => {
908                    "nomes e namespaces iniciados com __ são reservados para uso interno"
909                        .to_string()
910                }
911            };
912            assert!(msg.contains("reservados para uso interno"), "obtido: {msg}");
913        }
914    }
915
916    mod app_error_pt_translation_tests {
917        use crate::errors::AppError;
918
919        #[test]
920        fn localized_message_pt_not_found_fully_translated() {
921            let err =
922                AppError::NotFound("memory 'test-mem' not found in namespace 'global'".into());
923            let pt = err.localized_message_for(crate::i18n::Language::Portuguese);
924            assert!(
925                pt.contains("memória"),
926                "PT must translate 'memory' to 'memória': {pt}"
927            );
928            assert!(
929                pt.contains("não encontrada no namespace"),
930                "PT must translate full phrase: {pt}"
931            );
932            assert!(
933                !pt.contains("not found in namespace"),
934                "PT must not contain English phrase: {pt}"
935            );
936        }
937
938        #[test]
939        fn localized_message_pt_duplicate_fully_translated() {
940            let err = AppError::Duplicate(
941                "memory 'x' already exists in namespace 'global'. Use --force-merge to update."
942                    .into(),
943            );
944            let pt = err.localized_message_for(crate::i18n::Language::Portuguese);
945            assert!(pt.contains("memória"), "PT must translate 'memory': {pt}");
946            assert!(
947                pt.contains("já existe no namespace"),
948                "PT must translate 'already exists': {pt}"
949            );
950        }
951    }
952}