package webutility
  
  import (
  	"encoding/json"

  	"errors"

  	"io/ioutil"
  )

  type Dictionary struct {

  	locales       map[string]map[string]string
  	supported     []string
  	defaultLocale string

  }
  
  func NewDictionary() Dictionary {
  	return Dictionary{
  		locales: map[string]map[string]string{},
  	}
  }

  func (d *Dictionary) AddLocale(loc, filePath string) error {

  	file, err := ioutil.ReadFile(filePath)
  	if err != nil {
  		return err
  	}
  
  	var data interface{}
  	err = json.Unmarshal(file, &data)
  	if err != nil {
  		return err
  	}
  
  	l := map[string]string{}
  	for k, v := range data.(map[string]interface{}) {
  		l[k] = v.(string)
  	}

  	d.locales[loc] = l

  	d.supported = append(d.supported, loc)

  
  	return nil
  }

  func (d *Dictionary) Translate(loc, key string) string {
  	return d.locales[loc][key]
  }

  func (d *Dictionary) HasLocale(loc string) bool {
  	for _, v := range d.supported {
  		if v == loc {

  			return true
  		}
  	}
  	return false

  }

  
  func (d *Dictionary) SetDefaultLocale(loc string) error {
  	if !d.HasLocale(loc) {
  		return errors.New("dictionary does not contain translations for " + loc)
  	}
  	d.defaultLocale = loc
  	return nil
  }
  
  func (d *Dictionary) GetDefaultLocale() string {
  	return d.defaultLocale
  }