tijl.dev-core/internal/i18n/i18n.go

package i18n

import (
	"encoding/json"
	"github.com/gofiber/fiber/v2"
	"os"
	"path/filepath"

	log "github.com/sirupsen/logrus"
)

var translations map[string]map[string]string

const DefaultLang = "en"

func Load() {
	translations = make(map[string]map[string]string)

	dir := "locales"
	files, err := filepath.Glob(filepath.Join(dir, "*.json"))
	if err != nil {
		log.Fatalf("Error loading language files: %v", err)
	}

	for _, file := range files {
		lang := filepath.Base(file)
		lang = lang[:len(lang)-len(filepath.Ext(lang))]

		file, err := os.Open(file)
		if err != nil {
			log.Errorf("Error opening translation file %s: %v", file, err)
			continue
		}
		defer file.Close()

		var messages map[string]string
		if err := json.NewDecoder(file).Decode(&messages); err != nil {
			log.Errorf("Error decoding translation file %s: %v", file, err)
			continue
		}
		translations[lang] = messages
	}

	log.Debug("Loaded translations")
}

func Translate(c *fiber.Ctx, key string) string {
	lang := c.Cookies("language")
	if messages, ok := translations[lang]; ok {
		if message, ok := messages[key]; ok {
			return message
		}
	}
	if messages, ok := translations[DefaultLang]; ok {
		if message, ok := messages[key]; ok {
			return message
		}
	}
	return key
}

func GetLanguage(c *fiber.Ctx) string {
	lang := c.Cookies("language")

	if _, ok := translations[lang]; ok {
		return lang
	} else {
		return DefaultLang
	}
}

func GetTranslations(lang string) map[string]string {
	if messages, ok := translations[lang]; ok {
		return messages
	}

	return translations[DefaultLang]
}