package lib

import (
	"bufio"
	"fmt"
	"os"
	"path/filepath"
	"strings"
)

var translations = make(map[string]map[string]string)

// TranslateInit loads translation files from the given directory
func TranslateInit(langDir string) {
	files, _ := filepath.Glob(filepath.Join(langDir, "*.yaml"))
	for _, f := range files {
		lang := strings.TrimSuffix(filepath.Base(f), ".yaml")
		translations[lang] = make(map[string]string)
		file, _ := os.Open(f)
		if file == nil {
			continue
		}
		scanner := bufio.NewScanner(file)
		for scanner.Scan() {
			line := scanner.Text()
			if idx := strings.Index(line, ": "); idx > 0 {
				translations[lang][line[:idx]] = strings.Trim(line[idx+2:], "\"'")
			}
		}
		file.Close()
	}
}

// T translates a key to the given language, falling back to English
func T(lang, key string) string {
	if t, ok := translations[lang]; ok {
		if s, ok := t[key]; ok {
			return s
		}
	}
	if t, ok := translations["en"]; ok {
		if s, ok := t[key]; ok {
			return s
		}
	}
	return "[" + key + "]"
}

// CategoryTranslate returns the translated category name
func CategoryTranslate(cat int, lang string) string {
	return T(lang, CategoryKey(cat))
}

// SexTranslate returns the translated sex string (ISO 5218)
func SexTranslate(sex int, lang string) string {
	key := fmt.Sprintf("sex_%d", sex)
	return T(lang, key)
}

// DossierLanguage returns the language for a dossier, defaulting to "en"
func DossierLanguage(dossierID string) string {
	if dossierID == "" {
		return "en"
	}
	d, err := DossierGet("", dossierID) // nil ctx = internal operation
	if err != nil || d.Preferences.Language == "" {
		return "en"
	}
	return d.Preferences.Language
}