package util

import (
	"io/fs"
	"path/filepath"
)

const defaultLocale = "en"

type Locale map[string]string

func LoadLocale(assetFS fs.FS, locale string) (Locale, error) {
	localeMap := make(Locale)
	return localeMap, ReadYAML(assetFS, filepath.Join("locale", locale+".yaml"), &localeMap)
}

func LoadDefaultLocale(assetFS fs.FS) (Locale, error) {
	return LoadLocale(assetFS, defaultLocale)
}

func (t Locale) Get(key string) string {
	if t[key] == "" {
		return key
	}
	return t[key]
}