Version 4.0.0: java.awt dependencies move
[nikiroo-utils.git] / src / be / nikiroo / utils / resources / Bundle.java
index 2039d578e3c581a98efabad7369dbd0c879ed626..3c448efacf8f3bce0ef62341f0d6368e54d64e05 100644 (file)
@@ -11,60 +11,100 @@ import java.io.Reader;
 import java.io.Writer;
 import java.lang.reflect.Field;
 import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
 import java.util.Locale;
+import java.util.Map;
 import java.util.MissingResourceException;
 import java.util.PropertyResourceBundle;
 import java.util.ResourceBundle;
 
 /**
- * This class encapsulate a {@link ResourceBundle} in UTF-8. It only allows to
+ * This class encapsulate a {@link ResourceBundle} in UTF-8. It allows to
  * retrieve values associated to an enumeration, and allows some additional
  * methods.
- * 
- * @author niki
+ * <p>
+ * It also sports a writable change map, and you can save back the
+ * {@link Bundle} to file with {@link Bundle#updateFile(String)}.
  * 
  * @param <E>
  *            the enum to use to get values out of this class
+ * 
+ * @author niki
  */
+
 public class Bundle<E extends Enum<E>> {
+       /** The type of E. */
        protected Class<E> type;
-       protected Enum<?> name;
-       private ResourceBundle map;
+       /**
+        * The {@link Enum} associated to this {@link Bundle} (all the keys used in
+        * this {@link Bundle} will be of this type).
+        */
+       protected Enum<?> keyType;
+
+       private TransBundle<E> descriptionBundle;
+
+       /** R/O map */
+       private Map<String, String> map;
+       /** R/W map */
+       private Map<String, String> changeMap;
 
        /**
         * Create a new {@link Bundles} of the given name.
         * 
         * @param type
         *            a runtime instance of the class of E
-        * 
         * @param name
         *            the name of the {@link Bundles}
+        * @param descriptionBundle
+        *            the description {@link TransBundle}, that is, a
+        *            {@link TransBundle} dedicated to the description of the values
+        *            of the given {@link Bundle} (can be NULL)
         */
-       protected Bundle(Class<E> type, Enum<?> name) {
+       protected Bundle(Class<E> type, Enum<?> name,
+                       TransBundle<E> descriptionBundle) {
                this.type = type;
-               this.name = name;
-               setBundle(name, Locale.getDefault());
+               this.keyType = name;
+               this.descriptionBundle = descriptionBundle;
+
+               this.map = new HashMap<String, String>();
+               this.changeMap = new HashMap<String, String>();
+               setBundle(name, Locale.getDefault(), false);
        }
 
        /**
         * Return the value associated to the given id as a {@link String}.
         * 
-        * @param mame
+        * @param id
         *            the id of the value to get
         * 
         * @return the associated value, or NULL if not found (not present in the
         *         resource file)
         */
        public String getString(E id) {
-               return getStringX(id, null);
+               return getString(id.name());
+       }
+
+       /**
+        * Set the value associated to the given id as a {@link String}.
+        * 
+        * @param id
+        *            the id of the value to get
+        * @param value
+        *            the value
+        * 
+        */
+       public void setString(E id, String value) {
+               setString(id.name(), value);
        }
 
        /**
         * Return the value associated to the given id as a {@link String} suffixed
         * with the runtime value "_suffix" (that is, "_" and suffix).
+        * <p>
+        * Will only accept suffixes that form an existing id.
         * 
-        * @param mame
+        * @param id
         *            the id of the value to get
         * @param suffix
         *            the runtime suffix
@@ -76,17 +116,45 @@ public class Bundle<E extends Enum<E>> {
                String key = id.name()
                                + (suffix == null ? "" : "_" + suffix.toUpperCase());
 
-               if (containsKey(key)) {
-                       return getString(key).trim();
+               try {
+                       id = Enum.valueOf(type, key);
+                       return getString(id);
+               } catch (IllegalArgumentException e) {
+
                }
 
                return null;
        }
 
+       /**
+        * Set the value associated to the given id as a {@link String} suffixed
+        * with the runtime value "_suffix" (that is, "_" and suffix).
+        * <p>
+        * Will only accept suffixes that form an existing id.
+        * 
+        * @param id
+        *            the id of the value to get
+        * @param suffix
+        *            the runtime suffix
+        * @param value
+        *            the value
+        */
+       public void setStringX(E id, String suffix, String value) {
+               String key = id.name()
+                               + (suffix == null ? "" : "_" + suffix.toUpperCase());
+
+               try {
+                       id = Enum.valueOf(type, key);
+                       setString(id, value);
+               } catch (IllegalArgumentException e) {
+
+               }
+       }
+
        /**
         * Return the value associated to the given id as a {@link Boolean}.
         * 
-        * @param mame
+        * @param id
         *            the id of the value to get
         * 
         * @return the associated value
@@ -107,9 +175,9 @@ public class Bundle<E extends Enum<E>> {
        }
 
        /**
-        * Return the value associated to the given id as a {@link boolean}.
+        * Return the value associated to the given id as a {@link Boolean}.
         * 
-        * @param mame
+        * @param id
         *            the id of the value to get
         * @param def
         *            the default value when it is not present in the config file or
@@ -128,7 +196,7 @@ public class Bundle<E extends Enum<E>> {
        /**
         * Return the value associated to the given id as an {@link Integer}.
         * 
-        * @param mame
+        * @param id
         *            the id of the value to get
         * 
         * @return the associated value
@@ -143,9 +211,9 @@ public class Bundle<E extends Enum<E>> {
        }
 
        /**
-        * Return the value associated to the given id as a {@link int}.
+        * Return the value associated to the given id as an int.
         * 
-        * @param mame
+        * @param id
         *            the id of the value to get
         * @param def
         *            the default value when it is not present in the config file or
@@ -164,18 +232,176 @@ public class Bundle<E extends Enum<E>> {
        /**
         * Return the value associated to the given id as a {@link Character}.
         * 
-        * @param mame
+        * @param id
         *            the id of the value to get
         * 
         * @return the associated value
         */
-       public char getChar(E id) {
+       public Character getCharacter(E id) {
                String s = getString(id).trim();
                if (s.length() > 0) {
                        return s.charAt(0);
                }
 
-               return ' ';
+               return null;
+       }
+
+       /**
+        * Return the value associated to the given id as a {@link Character}.
+        * 
+        * @param id
+        *            the id of the value to get
+        * @param def
+        *            the default value when it is not present in the config file or
+        *            if it is not a char value
+        * 
+        * @return the associated value
+        */
+       public char getCharacter(E id, char def) {
+               String s = getString(id).trim();
+               if (s.length() > 0) {
+                       return s.charAt(0);
+               }
+
+               return def;
+       }
+
+       /**
+        * Return the value associated to the given id as a colour if it is found
+        * and can be parsed.
+        * <p>
+        * The returned value is an ARGB value.
+        * 
+        * @param id
+        *            the id of the value to get
+        * 
+        * @return the associated value
+        */
+       public Integer getColor(E id) {
+               Integer rep = null;
+
+               String bg = getString(id).trim();
+
+               int r = 0, g = 0, b = 0, a = -1;
+               if (bg.startsWith("#") && (bg.length() == 7 || bg.length() == 9)) {
+                       try {
+                               r = Integer.parseInt(bg.substring(1, 3), 16);
+                               g = Integer.parseInt(bg.substring(3, 5), 16);
+                               b = Integer.parseInt(bg.substring(5, 7), 16);
+                               if (bg.length() == 9) {
+                                       a = Integer.parseInt(bg.substring(7, 9), 16);
+                               } else {
+                                       a = 255;
+                               }
+
+                       } catch (NumberFormatException e) {
+                               // no changes
+                       }
+               }
+
+               // Try by name if still not found
+               if (a == -1) {
+                       if ("black".equalsIgnoreCase(bg)) {
+                               a = 255;
+                               r = 0;
+                               g = 0;
+                               b = 0;
+                       } else if ("white".equalsIgnoreCase(bg)) {
+                               a = 255;
+                               r = 255;
+                               g = 255;
+                               b = 255;
+                       } else if ("red".equalsIgnoreCase(bg)) {
+                               a = 255;
+                               r = 255;
+                               g = 0;
+                               b = 0;
+                       } else if ("green".equalsIgnoreCase(bg)) {
+                               a = 255;
+                               r = 0;
+                               g = 255;
+                               b = 0;
+                       } else if ("blue".equalsIgnoreCase(bg)) {
+                               a = 255;
+                               r = 0;
+                               g = 0;
+                               b = 255;
+                       } else if ("grey".equalsIgnoreCase(bg)
+                                       || "gray".equalsIgnoreCase(bg)) {
+                               a = 255;
+                               r = 128;
+                               g = 128;
+                               b = 128;
+                       } else if ("cyan".equalsIgnoreCase(bg)) {
+                               a = 255;
+                               r = 0;
+                               g = 255;
+                               b = 255;
+                       } else if ("magenta".equalsIgnoreCase(bg)) {
+                               a = 255;
+                               r = 255;
+                               g = 0;
+                               b = 255;
+                       } else if ("yellow".equalsIgnoreCase(bg)) {
+                               a = 255;
+                               r = 255;
+                               g = 255;
+                               b = 0;
+                       }
+               }
+
+               if (a != -1) {
+                       rep = ((a & 0xFF) << 24) //
+                                       | ((r & 0xFF) << 16) //
+                                       | ((g & 0xFF) << 8) //
+                                       | ((b & 0xFF) << 0);
+               }
+
+               return rep;
+       }
+
+       /**
+        * Set the value associated to the given id as a colour.
+        * <p>
+        * The value is an BGRA value.
+        * 
+        * @param id
+        *            the id of the value to set
+        * @param color
+        *            the new colour
+        */
+       public void setColor(E id, Integer color) {
+               int a = (color >> 24) & 0xFF;
+               int r = (color >> 16) & 0xFF;
+               int g = (color >> 8) & 0xFF;
+               int b = (color >> 0) & 0xFF;
+
+               String rs = Integer.toString(r, 16);
+               String gs = Integer.toString(g, 16);
+               String bs = Integer.toString(b, 16);
+               String as = "";
+               if (a < 255) {
+                       as = Integer.toString(a, 16);
+               }
+
+               setString(id, "#" + rs + gs + bs + as);
+       }
+
+       /**
+        * Create/update the .properties file.
+        * <p>
+        * Will use the most likely candidate as base if the file does not already
+        * exists and this resource is translatable (for instance, "en_US" will use
+        * "en" as a base if the resource is a translation file).
+        * <p>
+        * Will update the files in {@link Bundles#getDirectory()}; it <b>MUST</b>
+        * be set.
+        * 
+        * @throws IOException
+        *             in case of IO errors
+        */
+       public void updateFile() throws IOException {
+               updateFile(Bundles.getDirectory());
        }
 
        /**
@@ -186,7 +412,8 @@ public class Bundle<E extends Enum<E>> {
         * "en" as a base if the resource is a translation file).
         * 
         * @param path
-        *            the path where the .properties files are
+        *            the path where the .properties files are, <b>MUST NOT</b> be
+        *            NULL
         * 
         * @throws IOException
         *             in case of IO errors
@@ -204,7 +431,7 @@ public class Bundle<E extends Enum<E>> {
                for (Field field : type.getDeclaredFields()) {
                        Meta meta = field.getAnnotation(Meta.class);
                        if (meta != null) {
-                               E id = E.valueOf(type, field.getName());
+                               E id = Enum.valueOf(type, field.getName());
                                String info = getMetaInfo(meta);
 
                                if (info != null) {
@@ -219,11 +446,27 @@ public class Bundle<E extends Enum<E>> {
                writer.close();
        }
 
+       /**
+        * The description {@link TransBundle}, that is, a {@link TransBundle}
+        * dedicated to the description of the values of the given {@link Bundle}
+        * (can be NULL).
+        * 
+        * @return the description {@link TransBundle}
+        */
+       public TransBundle<E> getDescriptionBundle() {
+               return descriptionBundle;
+       }
+
        /**
         * Reload the {@link Bundle} data files.
+        * 
+        * @param resetToDefault
+        *            reset to the default configuration (do not look into the
+        *            possible user configuration files, only take the original
+        *            configuration)
         */
-       public void reload() {
-               setBundle(name, null);
+       public void reload(boolean resetToDefault) {
+               setBundle(keyType, Locale.getDefault(), resetToDefault);
        }
 
        /**
@@ -235,12 +478,7 @@ public class Bundle<E extends Enum<E>> {
         * @return true if it does
         */
        protected boolean containsKey(String key) {
-               try {
-                       map.getObject(key);
-                       return true;
-               } catch (MissingResourceException e) {
-                       return false;
-               }
+               return changeMap.containsKey(key) || map.containsKey(key);
        }
 
        /**
@@ -253,13 +491,30 @@ public class Bundle<E extends Enum<E>> {
         * @return the value, or NULL
         */
        protected String getString(String key) {
-               if (containsKey(key)) {
-                       return map.getString(key);
+               if (changeMap.containsKey(key)) {
+                       return changeMap.get(key);
+               }
+
+               if (map.containsKey(key)) {
+                       return map.get(key);
                }
 
                return null;
        }
 
+       /**
+        * Set the value for this key, in the change map (it is kept in memory, not
+        * yet on disk).
+        * 
+        * @param key
+        *            the key
+        * @param value
+        *            the associated value
+        */
+       protected void setString(String key, String value) {
+               changeMap.put(key, value == null ? null : value.trim());
+       }
+
        /**
         * Return formated, display-able information from the {@link Meta} field
         * given. Each line will always starts with a "#" character.
@@ -270,44 +525,45 @@ public class Bundle<E extends Enum<E>> {
         * @return the information to display or NULL if none
         */
        protected String getMetaInfo(Meta meta) {
-               String what = meta.what();
-               String where = meta.where();
-               String format = meta.format();
+               String desc = meta.description();
+               boolean group = meta.group();
+               Meta.Format format = meta.format();
+               String[] list = meta.list();
+               boolean nullable = meta.nullable();
                String info = meta.info();
+               boolean array = meta.array();
 
-               int opt = what.length() + where.length() + format.length();
-               if (opt + info.length() == 0)
+               // Default, empty values -> NULL
+               if (desc.length() + list.length + info.length() == 0 && !group
+                               && nullable && format == Meta.Format.STRING) {
                        return null;
+               }
 
                StringBuilder builder = new StringBuilder();
-               builder.append("# ");
-
-               if (opt > 0) {
-                       builder.append("(");
-                       if (what.length() > 0) {
-                               builder.append("WHAT: " + what);
-                               if (where.length() + format.length() > 0)
-                                       builder.append(", ");
-                       }
-
-                       if (where.length() > 0) {
-                               builder.append("WHERE: " + where);
-                               if (format.length() > 0)
-                                       builder.append(", ");
-                       }
+               builder.append("# ").append(desc);
+               if (desc.length() > 20) {
+                       builder.append("\n#");
+               }
 
-                       if (format.length() > 0) {
-                               builder.append("FORMAT: " + format);
+               if (group) {
+                       builder.append("This item is used as a group, its content is not expected to be used.");
+               } else {
+                       builder.append(" (FORMAT: ").append(format)
+                                       .append(nullable ? "" : " (required)");
+                       builder.append(") ").append(info);
+
+                       if (list.length > 0) {
+                               builder.append("\n# ALLOWED VALUES:");
+                               for (String value : list) {
+                                       builder.append(" \"").append(value).append("\"");
+                               }
                        }
 
-                       builder.append(")");
-                       if (info.length() > 0) {
-                               builder.append("\n# ");
+                       if (array) {
+                               builder.append("\n# (This item accept a list of comma-separated values)");
                        }
                }
 
-               builder.append(info);
-
                return builder.toString();
        }
 
@@ -317,7 +573,7 @@ public class Bundle<E extends Enum<E>> {
         * @return the name
         */
        protected String getBundleDisplayName() {
-               return name.toString();
+               return keyType.toString();
        }
 
        /**
@@ -374,7 +630,7 @@ public class Bundle<E extends Enum<E>> {
                        value = "";
                }
 
-               String[] lines = value.replaceAll("\t", "\\t").split("\n");
+               String[] lines = value.replaceAll("\t", "\\\\\\t").split("\n");
                for (int i = 0; i < lines.length; i++) {
                        writer.write(lines[i]);
                        if (i < lines.length - 1) {
@@ -391,42 +647,108 @@ public class Bundle<E extends Enum<E>> {
         *            the path where the .properties files are
         * 
         * @return the source {@link File}
-        * 
-        * @throws IOException
-        *             in case of IO errors
         */
        protected File getUpdateFile(String path) {
-               return new File(path, name.name() + ".properties");
+               return new File(path, keyType.name() + ".properties");
        }
 
        /**
-        * Change the currently used bundle.
+        * Change the currently used bundle, and reset all changes.
         * 
         * @param name
         *            the name of the bundle to load
         * @param locale
         *            the {@link Locale} to use
+        * @param resetToDefault
+        *            reset to the default configuration (do not look into the
+        *            possible user configuration files, only take the original
+        *            configuration)
         */
-       protected void setBundle(Enum<?> name, Locale locale) {
-               map = null;
+       protected void setBundle(Enum<?> name, Locale locale, boolean resetToDefault) {
+               changeMap.clear();
                String dir = Bundles.getDirectory();
 
-               if (dir != null) {
+               boolean found = false;
+               if (!resetToDefault && dir != null) {
                        try {
                                File file = getPropertyFile(dir, name.name(), locale);
                                if (file != null) {
                                        Reader reader = new InputStreamReader(new FileInputStream(
                                                        file), "UTF8");
-                                       map = new PropertyResourceBundle(reader);
+                                       resetMap(new PropertyResourceBundle(reader));
+                                       found = true;
                                }
                        } catch (IOException e) {
                                e.printStackTrace();
                        }
                }
 
-               if (map == null) {
-                       map = ResourceBundle.getBundle(type.getPackage().getName() + "."
-                                       + name.name(), locale, new FixedResourceBundleControl());
+               if (!found) {
+                       String bname = type.getPackage().getName() + "." + name.name();
+                       try {
+                               resetMap(ResourceBundle
+                                               .getBundle(bname, locale, type.getClassLoader(),
+                                                               new FixedResourceBundleControl()));
+                       } catch (Exception e) {
+                               // We have no bundle for this Bundle
+                               System.err.println("No bundle found for: " + bname);
+                               resetMap(null);
+                       }
+               }
+       }
+
+       /**
+        * Reset the backing map to the content of the given bundle, or empty if
+        * bundle is NULL.
+        * 
+        * @param bundle
+        *            the bundle to copy
+        */
+       protected void resetMap(ResourceBundle bundle) {
+               this.map.clear();
+
+               if (bundle != null) {
+                       for (E field : type.getEnumConstants()) {
+                               try {
+                                       String value = bundle.getString(field.name());
+                                       this.map.put(field.name(),
+                                                       value == null ? null : value.trim());
+                               } catch (MissingResourceException e) {
+                               }
+                       }
+               }
+       }
+
+       /**
+        * Take a snapshot of the changes in memory in this {@link Bundle} made by
+        * the "set" methods ( {@link Bundle#setString(Enum, String)}...) at the
+        * current time.
+        * 
+        * @return a snapshot to use with {@link Bundle#restoreSnapshot(Object)}
+        */
+       public Object takeSnapshot() {
+               return new HashMap<String, String>(changeMap);
+       }
+
+       /**
+        * Restore a snapshot taken with {@link Bundle}, or reset the current
+        * changes if the snapshot is NULL.
+        * 
+        * @param snap
+        *            the snapshot or NULL
+        */
+       @SuppressWarnings("unchecked")
+       public void restoreSnapshot(Object snap) {
+               if (snap == null) {
+                       changeMap.clear();
+               } else {
+                       if (snap instanceof Map) {
+                               changeMap = (Map<String, String>) snap;
+                       } else {
+                               throw new RuntimeException(
+                                               "Restoring changes in a Bundle must be done on a changes snapshot, "
+                                                               + "or NULL to discard current changes");
+                       }
                }
        }
 
@@ -434,9 +756,9 @@ public class Bundle<E extends Enum<E>> {
         * Return the resource file that is closer to the {@link Locale}.
         * 
         * @param dir
-        *            the dirctory to look into
+        *            the directory to look into
         * @param name
-        *            the file basename (without <tt>.properties</tt>)
+        *            the file base name (without <tt>.properties</tt>)
         * @param locale
         *            the {@link Locale}
         * 
@@ -464,9 +786,9 @@ public class Bundle<E extends Enum<E>> {
                        file = new File(dir, name + loc + ".properties");
                        if (file.exists()) {
                                break;
-                       } else {
-                               file = null;
                        }
+
+                       file = null;
                }
 
                return file;