1 package be
.nikiroo
.utils
.resources
;
3 import java
.io
.BufferedWriter
;
5 import java
.io
.FileInputStream
;
6 import java
.io
.FileOutputStream
;
7 import java
.io
.IOException
;
8 import java
.io
.InputStreamReader
;
9 import java
.io
.OutputStreamWriter
;
10 import java
.io
.Reader
;
11 import java
.io
.Writer
;
12 import java
.lang
.reflect
.Field
;
13 import java
.util
.ArrayList
;
14 import java
.util
.HashMap
;
15 import java
.util
.List
;
16 import java
.util
.Locale
;
18 import java
.util
.MissingResourceException
;
19 import java
.util
.PropertyResourceBundle
;
20 import java
.util
.ResourceBundle
;
23 * This class encapsulate a {@link ResourceBundle} in UTF-8. It allows to
24 * retrieve values associated to an enumeration, and allows some additional
27 * It also sports a writable change map, and you can save back the
28 * {@link Bundle} to file with {@link Bundle#updateFile(String)}.
31 * the enum to use to get values out of this class
36 public class Bundle
<E
extends Enum
<E
>> {
38 protected Class
<E
> type
;
40 * The {@link Enum} associated to this {@link Bundle} (all the keys used in
41 * this {@link Bundle} will be of this type).
43 protected Enum
<?
> keyType
;
45 private TransBundle
<E
> descriptionBundle
;
48 private Map
<String
, String
> map
;
50 private Map
<String
, String
> changeMap
;
53 * Create a new {@link Bundles} of the given name.
56 * a runtime instance of the class of E
58 * the name of the {@link Bundles}
59 * @param descriptionBundle
60 * the description {@link TransBundle}, that is, a
61 * {@link TransBundle} dedicated to the description of the values
62 * of the given {@link Bundle} (can be NULL)
64 protected Bundle(Class
<E
> type
, Enum
<?
> name
,
65 TransBundle
<E
> descriptionBundle
) {
68 this.descriptionBundle
= descriptionBundle
;
70 this.map
= new HashMap
<String
, String
>();
71 this.changeMap
= new HashMap
<String
, String
>();
72 setBundle(name
, Locale
.getDefault(), false);
76 * Return the value associated to the given id as a {@link String}.
79 * the id of the value to get
81 * @return the associated value, or NULL if not found (not present in the
84 public String
getString(E id
) {
85 return getString(id
.name());
89 * Set the value associated to the given id as a {@link String}.
92 * the id of the value to get
97 public void setString(E id
, String value
) {
98 setString(id
.name(), value
);
102 * Return the value associated to the given id as a {@link String} suffixed
103 * with the runtime value "_suffix" (that is, "_" and suffix).
105 * Will only accept suffixes that form an existing id.
108 * the id of the value to get
112 * @return the associated value, or NULL if not found (not present in the
115 public String
getStringX(E id
, String suffix
) {
116 String key
= id
.name()
117 + (suffix
== null ?
"" : "_" + suffix
.toUpperCase());
120 id
= Enum
.valueOf(type
, key
);
121 return getString(id
);
122 } catch (IllegalArgumentException e
) {
130 * Set the value associated to the given id as a {@link String} suffixed
131 * with the runtime value "_suffix" (that is, "_" and suffix).
133 * Will only accept suffixes that form an existing id.
136 * the id of the value to get
142 public void setStringX(E id
, String suffix
, String value
) {
143 String key
= id
.name()
144 + (suffix
== null ?
"" : "_" + suffix
.toUpperCase());
147 id
= Enum
.valueOf(type
, key
);
148 setString(id
, value
);
149 } catch (IllegalArgumentException e
) {
155 * Return the value associated to the given id as a {@link Boolean}.
158 * the id of the value to get
160 * @return the associated value
162 public Boolean
getBoolean(E id
) {
163 String str
= getString(id
);
164 if (str
!= null && str
.length() > 0) {
165 if (str
.equalsIgnoreCase("true") || str
.equalsIgnoreCase("on")
166 || str
.equalsIgnoreCase("yes"))
168 if (str
.equalsIgnoreCase("false") || str
.equalsIgnoreCase("off")
169 || str
.equalsIgnoreCase("no"))
178 * Return the value associated to the given id as a {@link Boolean}.
181 * the id of the value to get
183 * the default value when it is not present in the config file or
184 * if it is not a boolean value
186 * @return the associated value
188 public boolean getBoolean(E id
, boolean def
) {
189 Boolean b
= getBoolean(id
);
197 * Return the value associated to the given id as an {@link Integer}.
200 * the id of the value to get
202 * @return the associated value
204 public Integer
getInteger(E id
) {
206 return Integer
.parseInt(getString(id
));
207 } catch (Exception e
) {
214 * Return the value associated to the given id as an int.
217 * the id of the value to get
219 * the default value when it is not present in the config file or
220 * if it is not a int value
222 * @return the associated value
224 public int getInteger(E id
, int def
) {
225 Integer i
= getInteger(id
);
233 * Return the value associated to the given id as a {@link Character}.
236 * the id of the value to get
238 * @return the associated value
240 public Character
getCharacter(E id
) {
241 String s
= getString(id
).trim();
242 if (s
.length() > 0) {
250 * Return the value associated to the given id as a {@link Character}.
253 * the id of the value to get
255 * the default value when it is not present in the config file or
256 * if it is not a char value
258 * @return the associated value
260 public char getCharacter(E id
, char def
) {
261 String s
= getString(id
);
262 if (s
!= null && s
.length() > 0) {
263 return s
.trim().charAt(0);
270 * Return the value associated to the given id as a colour if it is found
273 * The returned value is an ARGB value.
276 * the id of the value to get
278 * @return the associated value
280 public Integer
getColor(E id
) {
283 String bg
= getString(id
).trim();
285 int r
= 0, g
= 0, b
= 0, a
= -1;
286 if (bg
.startsWith("#") && (bg
.length() == 7 || bg
.length() == 9)) {
288 r
= Integer
.parseInt(bg
.substring(1, 3), 16);
289 g
= Integer
.parseInt(bg
.substring(3, 5), 16);
290 b
= Integer
.parseInt(bg
.substring(5, 7), 16);
291 if (bg
.length() == 9) {
292 a
= Integer
.parseInt(bg
.substring(7, 9), 16);
297 } catch (NumberFormatException e
) {
302 // Try by name if still not found
304 if ("black".equalsIgnoreCase(bg
)) {
309 } else if ("white".equalsIgnoreCase(bg
)) {
314 } else if ("red".equalsIgnoreCase(bg
)) {
319 } else if ("green".equalsIgnoreCase(bg
)) {
324 } else if ("blue".equalsIgnoreCase(bg
)) {
329 } else if ("grey".equalsIgnoreCase(bg
)
330 || "gray".equalsIgnoreCase(bg
)) {
335 } else if ("cyan".equalsIgnoreCase(bg
)) {
340 } else if ("magenta".equalsIgnoreCase(bg
)) {
345 } else if ("yellow".equalsIgnoreCase(bg
)) {
354 rep
= ((a
& 0xFF) << 24) //
355 | ((r
& 0xFF) << 16) //
356 | ((g
& 0xFF) << 8) //
364 * Set the value associated to the given id as a colour.
366 * The value is an BGRA value.
369 * the id of the value to set
373 public void setColor(E id
, Integer color
) {
374 int a
= (color
>> 24) & 0xFF;
375 int r
= (color
>> 16) & 0xFF;
376 int g
= (color
>> 8) & 0xFF;
377 int b
= (color
>> 0) & 0xFF;
379 String rs
= Integer
.toString(r
, 16);
380 String gs
= Integer
.toString(g
, 16);
381 String bs
= Integer
.toString(b
, 16);
384 as
= Integer
.toString(a
, 16);
387 setString(id
, "#" + rs
+ gs
+ bs
+ as
);
391 * Create/update the .properties file.
393 * Will use the most likely candidate as base if the file does not already
394 * exists and this resource is translatable (for instance, "en_US" will use
395 * "en" as a base if the resource is a translation file).
397 * Will update the files in {@link Bundles#getDirectory()}; it <b>MUST</b>
400 * @throws IOException
401 * in case of IO errors
403 public void updateFile() throws IOException
{
404 updateFile(Bundles
.getDirectory());
408 * Create/update the .properties file.
410 * Will use the most likely candidate as base if the file does not already
411 * exists and this resource is translatable (for instance, "en_US" will use
412 * "en" as a base if the resource is a translation file).
415 * the path where the .properties files are, <b>MUST NOT</b> be
418 * @throws IOException
419 * in case of IO errors
421 public void updateFile(String path
) throws IOException
{
422 File file
= getUpdateFile(path
);
424 BufferedWriter writer
= new BufferedWriter(new OutputStreamWriter(
425 new FileOutputStream(file
), "UTF-8"));
431 for (Field field
: type
.getDeclaredFields()) {
432 Meta meta
= field
.getAnnotation(Meta
.class);
434 E id
= Enum
.valueOf(type
, field
.getName());
435 String info
= getMetaInfo(meta
);
442 writeValue(writer
, id
);
450 * Delete the .properties file.
452 * Will use the most likely candidate as base if the file does not already
453 * exists and this resource is translatable (for instance, "en_US" will use
454 * "en" as a base if the resource is a translation file).
456 * Will delete the files in {@link Bundles#getDirectory()}; it <b>MUST</b>
459 * @return TRUE if the file was deleted
461 public boolean deleteFile() {
462 return deleteFile(Bundles
.getDirectory());
466 * Delete the .properties file.
468 * Will use the most likely candidate as base if the file does not already
469 * exists and this resource is translatable (for instance, "en_US" will use
470 * "en" as a base if the resource is a translation file).
473 * the path where the .properties files are, <b>MUST NOT</b> be
476 * @return TRUE if the file was deleted
478 public boolean deleteFile(String path
) {
479 File file
= getUpdateFile(path
);
480 return file
.delete();
484 * The description {@link TransBundle}, that is, a {@link TransBundle}
485 * dedicated to the description of the values of the given {@link Bundle}
488 * @return the description {@link TransBundle}
490 public TransBundle
<E
> getDescriptionBundle() {
491 return descriptionBundle
;
495 * Reload the {@link Bundle} data files.
497 * @param resetToDefault
498 * reset to the default configuration (do not look into the
499 * possible user configuration files, only take the original
502 public void reload(boolean resetToDefault
) {
503 setBundle(keyType
, Locale
.getDefault(), resetToDefault
);
507 * Check if the internal map contains the given key.
510 * the key to check for
512 * @return true if it does
514 protected boolean containsKey(String key
) {
515 return changeMap
.containsKey(key
) || map
.containsKey(key
);
519 * Get the value for the given key if it exists in the internal map, or NULL
523 * the key to check for
525 * @return the value, or NULL
527 protected String
getString(String key
) {
528 if (changeMap
.containsKey(key
)) {
529 return changeMap
.get(key
);
532 if (map
.containsKey(key
)) {
540 * Set the value for this key, in the change map (it is kept in memory, not
546 * the associated value
548 protected void setString(String key
, String value
) {
549 changeMap
.put(key
, value
== null ?
null : value
.trim());
553 * Return formated, display-able information from the {@link Meta} field
554 * given. Each line will always starts with a "#" character.
557 * the {@link Meta} field
559 * @return the information to display or NULL if none
561 protected String
getMetaInfo(Meta meta
) {
562 String desc
= meta
.description();
563 boolean group
= meta
.group();
564 Meta
.Format format
= meta
.format();
565 String
[] list
= meta
.list();
566 boolean nullable
= meta
.nullable();
567 String def
= meta
.def();
568 String info
= meta
.info();
569 boolean array
= meta
.array();
571 // Default, empty values -> NULL
572 if (desc
.length() + list
.length
+ info
.length() + def
.length() == 0
573 && !group
&& nullable
&& format
== Meta
.Format
.STRING
) {
577 StringBuilder builder
= new StringBuilder();
578 builder
.append("# ").append(desc
);
579 if (desc
.length() > 20) {
580 builder
.append("\n#");
584 builder
.append("This item is used as a group, its content is not expected to be used.");
586 builder
.append(" (FORMAT: ").append(format
)
587 .append(nullable ?
"" : " (required)");
588 builder
.append(") ").append(info
);
590 if (list
.length
> 0) {
591 builder
.append("\n# ALLOWED VALUES:");
592 for (String value
: list
) {
593 builder
.append(" \"").append(value
).append("\"");
598 builder
.append("\n# (This item accept a list of comma-separated values)");
602 return builder
.toString();
606 * The display name used in the <tt>.properties file</tt>.
610 protected String
getBundleDisplayName() {
611 return keyType
.toString();
615 * Write the header found in the configuration <tt>.properties</tt> file of
616 * this {@link Bundles}.
619 * the {@link Writer} to write the header in
621 * @throws IOException
622 * in case of IO error
624 protected void writeHeader(Writer writer
) throws IOException
{
625 writer
.write("# " + getBundleDisplayName() + "\n");
630 * Write the given id to the config file, i.e., "MY_ID = my_curent_value"
631 * followed by a new line
634 * the {@link Writer} to write into
638 * @throws IOException
639 * in case of IO error
641 protected void writeValue(Writer writer
, E id
) throws IOException
{
642 writeValue(writer
, id
.name(), getString(id
));
646 * Write the given data to the config file, i.e., "MY_ID = my_curent_value"
647 * followed by a new line
650 * the {@link Writer} to write into
656 * @throws IOException
657 * in case of IO error
659 protected void writeValue(Writer writer
, String id
, String value
)
668 String
[] lines
= value
.replaceAll("\t", "\\\\\\t").split("\n");
669 for (int i
= 0; i
< lines
.length
; i
++) {
670 writer
.write(lines
[i
]);
671 if (i
< lines
.length
- 1) {
672 writer
.write("\\n\\");
679 * Return the source file for this {@link Bundles} from the given path.
682 * the path where the .properties files are
684 * @return the source {@link File}
686 protected File
getUpdateFile(String path
) {
687 return new File(path
, keyType
.name() + ".properties");
691 * Change the currently used bundle, and reset all changes.
694 * the name of the bundle to load
696 * the {@link Locale} to use
697 * @param resetToDefault
698 * reset to the default configuration (do not look into the
699 * possible user configuration files, only take the original
702 protected void setBundle(Enum
<?
> name
, Locale locale
, boolean resetToDefault
) {
704 String dir
= Bundles
.getDirectory();
705 String bname
= type
.getPackage().getName() + "." + name
.name();
707 boolean found
= false;
708 if (!resetToDefault
&& dir
!= null) {
709 // Look into Bundles.getDirectory() for .properties files
711 File file
= getPropertyFile(dir
, name
.name(), locale
);
713 Reader reader
= new InputStreamReader(new FileInputStream(
715 resetMap(new PropertyResourceBundle(reader
));
718 } catch (IOException e
) {
724 // Look into the package itself for resources
726 resetMap(ResourceBundle
727 .getBundle(bname
, locale
, type
.getClassLoader(),
728 new FixedResourceBundleControl()));
730 } catch (MissingResourceException e
) {
731 } catch (Exception e
) {
737 // We have no bundle for this Bundle
738 System
.err
.println("No bundle found for: " + bname
);
744 * Reset the backing map to the content of the given bundle, or with default
745 * valiues if bundle is NULL.
750 protected void resetMap(ResourceBundle bundle
) {
752 for (Field field
: type
.getDeclaredFields()) {
754 Meta meta
= field
.getAnnotation(Meta
.class);
756 E id
= Enum
.valueOf(type
, field
.getName());
759 if (bundle
!= null) {
760 value
= bundle
.getString(id
.name());
765 this.map
.put(id
.name(), value
== null ?
null : value
.trim());
767 } catch (MissingResourceException e
) {
773 * Take a snapshot of the changes in memory in this {@link Bundle} made by
774 * the "set" methods ( {@link Bundle#setString(Enum, String)}...) at the
777 * @return a snapshot to use with {@link Bundle#restoreSnapshot(Object)}
779 public Object
takeSnapshot() {
780 return new HashMap
<String
, String
>(changeMap
);
784 * Restore a snapshot taken with {@link Bundle}, or reset the current
785 * changes if the snapshot is NULL.
788 * the snapshot or NULL
790 @SuppressWarnings("unchecked")
791 public void restoreSnapshot(Object snap
) {
795 if (snap
instanceof Map
) {
796 changeMap
= (Map
<String
, String
>) snap
;
798 throw new RuntimeException(
799 "Restoring changes in a Bundle must be done on a changes snapshot, "
800 + "or NULL to discard current changes");
806 * Return the resource file that is closer to the {@link Locale}.
809 * the directory to look into
811 * the file base name (without <tt>.properties</tt>)
815 * @return the closest match or NULL if none
817 private File
getPropertyFile(String dir
, String name
, Locale locale
) {
818 List
<String
> locales
= new ArrayList
<String
>();
819 if (locale
!= null) {
820 String country
= locale
.getCountry() == null ?
"" : locale
822 String language
= locale
.getLanguage() == null ?
"" : locale
824 if (!language
.isEmpty() && !country
.isEmpty()) {
825 locales
.add("_" + language
+ "-" + country
);
827 if (!language
.isEmpty()) {
828 locales
.add("_" + language
);
835 for (String loc
: locales
) {
836 file
= new File(dir
, name
+ loc
+ ".properties");