package com.trendmicro.grid.acl.ds.trivial;
   
   import com.trendmicro.grid.acl.l0.datatypes.Category;
   
   import javax.xml.bind.annotation.*;
   import java.util.HashMap;
   import java.util.List;
   import java.util.Locale;
   import java.util.Map;
  
  /**
   * Implements an I18N and L10N aware definition of a category.
   *
   * @author juergen_kellerer, 2011-01-19
   * @version 1.0
   */
  @XmlType(namespace = I18NAwareDefinition.NS)
  @XmlAccessorType(XmlAccessType.FIELD)
  @XmlRootElement(name = "category-definition", namespace = I18NAwareDefinition.NS)
  public class CategoryDefinition extends I18NAwareDefinition {
  
  	@XmlElement(name = "category", required = true)
  	private Category category;
  
  	@XmlElement(name = "displayName")
  	@XmlElementWrapper(name = "translations")
  	private List<DisplayName> displayNames;
  
  	private transient Map<Locale, Category> localizedCategories;
  
  	/**
  	 * Constructs an empty display name (used by JAXB only).
  	 */
  	public CategoryDefinition() {
  	}
  
  	/**
  	 * Constructs a category definition.
  	 *
  	 * @param category		 The category that is defined by this element.
  	 * @param displayNames	 The localized display names.
  	 * @param availableLocales The locales that this definition should be made available to.
  	 */
  	public CategoryDefinition(Category category, List<DisplayName> displayNames, List<Locale> availableLocales) {
  		super(availableLocales);
  		this.category = category;
  		this.displayNames = displayNames;
  	}
  
  	/**
  	 * Returns the embedded category.
  	 *
  	 * @return the embedded category.
  	 */
  	public Category getCategory() {
  		return category;
  	}
  
  	@Override
  	public String getName() {
  		return category.getName();
  	}
  
  	/**
  	 * Returns a map of all available localized variants of this category.
  	 *
  	 * @return a map of all available localized variants of this category.
  	 */
  	public Map<Locale, Category> getLocalizedCategories() {
  		if (localizedCategories == null) {
  			Map<Locale, Category> lc = new HashMap<Locale, Category>();
  			for (DisplayName name : displayNames) {
  				Category c = category.clone().localize(name.getValue(), name.getLocale());
  				lc.put(name.getLocale(), c);
  				lc.put(new Locale(name.getLocale().getLanguage()), c);
  			}
  			localizedCategories = lc;
  		}
  
  		return localizedCategories;
  	}
  
  	/**
  	 * Returns a localized version of the category or the default category if no localized version is available.
  	 *
  	 * @param locale the locale to use for looking-up a localized category version.
  	 * @return a localized version of the category or the default category if no localized version is available.
  	 */
  	public Category getLocalizedCategory(Locale locale) {
  		Map<Locale, Category> categories = getLocalizedCategories();
  
  		Category category = categories.get(locale);
  		if (category == null && !locale.getCountry().isEmpty())
  			category = categories.get(new Locale(locale.getLanguage()));
  
  		return category == null ? this.category : category;
  	}
  
  	@Override
 	public String toString() {
 		return "CategoryDefinition{" +
 			"category=" + category +
 			", displayNames=" + displayNames +
 			'}';
 	}
 
 	@Override
 	public boolean equals(Object o) {
 		if (this == o) return true;
 		if (!(o instanceof CategoryDefinition)) return false;
 
 		CategoryDefinition that = (CategoryDefinition) o;
 
 		if (!category.equals(that.category)) return false;
 		return !(displayNames != null ? !displayNames.equals(that.displayNames) : that.displayNames != null);
 	}
 
 	@Override
 	public int hashCode() {
 		int result = category.hashCode();
 		result = 31 * result + (displayNames != null ? displayNames.hashCode() : 0);
 		return result;
 	}
 }