1 /*
   2  * Copyright (c) 2012, 2017, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.  Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 
  26 package build.tools.cldrconverter;
  27 
  28 import java.io.File;
  29 import java.io.IOException;
  30 import java.io.PrintWriter;
  31 import java.util.Arrays;
  32 import java.util.Formatter;
  33 import java.util.HashSet;
  34 import java.util.HashMap;
  35 import java.util.LinkedHashMap;
  36 import java.util.Map;
  37 import java.util.Locale;
  38 import java.util.Objects;
  39 import java.util.Set;
  40 import java.util.SortedSet;
  41 
  42 class ResourceBundleGenerator implements BundleGenerator {
  43     // preferred timezones - keeping compatibility with JDK1.1 3 letter abbreviations
  44     private static final String[] preferredTZIDs = {
  45         "America/Los_Angeles",
  46         "America/Denver",
  47         "America/Phoenix",
  48         "America/Chicago",
  49         "America/New_York",
  50         "America/Indianapolis",
  51         "Pacific/Honolulu",
  52         "America/Anchorage",
  53         "America/Halifax",
  54         "America/Sitka",
  55         "America/St_Johns",
  56         "Europe/Paris",
  57         // Although CLDR does not support abbreviated zones, handle "GMT" as a
  58         // special case here, as it is specified in the javadoc.
  59         "GMT",
  60         "Africa/Casablanca",
  61         "Asia/Jerusalem",
  62         "Asia/Tokyo",
  63         "Europe/Bucharest",
  64         "Asia/Shanghai",
  65         "UTC",
  66     };
  67 
  68     // For duplicated values
  69     private static final String META_VALUE_PREFIX = "metaValue_";
  70 
  71     @Override
  72     public void generateBundle(String packageName, String baseName, String localeID, boolean useJava,
  73                                Map<String, ?> map, BundleType type) throws IOException {
  74         String suffix = useJava ? ".java" : ".properties";
  75         String dirName = CLDRConverter.DESTINATION_DIR + File.separator + "sun" + File.separator
  76                 + packageName + File.separator + "resources" + File.separator + "cldr";
  77         packageName = packageName + ".resources.cldr";
  78 
  79         if (CLDRConverter.isBaseModule ^ isBaseLocale(localeID)) {
  80             return;
  81         }
  82 
  83         // Assume that non-base resources go into jdk.localedata
  84         if (!CLDRConverter.isBaseModule) {
  85             dirName = dirName + File.separator + "ext";
  86             packageName = packageName + ".ext";
  87         }
  88 
  89         File dir = new File(dirName);
  90         if (!dir.exists()) {
  91             dir.mkdirs();
  92         }
  93         File file = new File(dir, baseName + ("root".equals(localeID) ? "" : "_" + localeID) + suffix);
  94         if (!file.exists()) {
  95             file.createNewFile();
  96         }
  97         CLDRConverter.info("\tWriting file " + file);
  98 
  99         String encoding;
 100         if (useJava) {
 101             if (CLDRConverter.USE_UTF8) {
 102                 encoding = "utf-8";
 103             } else {
 104                 encoding = "us-ascii";
 105             }
 106         } else {
 107             encoding = "iso-8859-1";
 108         }
 109 
 110         Formatter fmt = null;
 111         if (type == BundleType.TIMEZONE) {
 112             fmt = new Formatter();
 113             Set<String> metaKeys = new HashSet<>();
 114             for (String key : map.keySet()) {
 115                 if (key.startsWith(CLDRConverter.METAZONE_ID_PREFIX)) {
 116                     String meta = key.substring(CLDRConverter.METAZONE_ID_PREFIX.length());
 117                     String[] value;
 118                     value = (String[]) map.get(key);
 119                     fmt.format("        final String[] %s = new String[] {\n", meta);
 120                     for (String s : value) {
 121                         fmt.format("               \"%s\",\n", CLDRConverter.saveConvert(s, useJava));
 122                     }
 123                     fmt.format("            };\n");
 124                     metaKeys.add(key);
 125                 }
 126             }
 127             for (String key : metaKeys) {
 128                 map.remove(key);
 129             }
 130 
 131             // Make it preferred ordered
 132             LinkedHashMap<String, Object> newMap = new LinkedHashMap<>();
 133             for (String preferred : preferredTZIDs) {
 134                 if (map.containsKey(preferred)) {
 135                     newMap.put(preferred, map.remove(preferred));
 136                 } else if (("GMT".equals(preferred) || "UTC".equals(preferred)) &&
 137                            metaKeys.contains(CLDRConverter.METAZONE_ID_PREFIX+preferred)) {
 138                     newMap.put(preferred, preferred);
 139                 }
 140             }
 141             newMap.putAll(map);
 142             map = newMap;
 143         } else {
 144             // generic reduction of duplicated values
 145             Map<String, Object> newMap = null;
 146             for (String key : map.keySet()) {
 147                 Object val = map.get(key);
 148                 String metaVal = null;
 149 
 150                 for (Map.Entry<String, ?> entry : map.entrySet()) {
 151                     String k = entry.getKey();
 152                     if (!k.equals(key) &&
 153                         Objects.deepEquals(val, entry.getValue()) &&
 154                         !(Objects.nonNull(newMap) && newMap.containsKey(k))) {
 155                         if (Objects.isNull(newMap)) {
 156                             newMap = new HashMap<>();
 157                             fmt = new Formatter();
 158                         }
 159 
 160                         if (Objects.isNull(metaVal)) {
 161                             metaVal = META_VALUE_PREFIX + key.replaceAll("\\.", "_");
 162 
 163                             if (val instanceof String[]) {
 164                                 fmt.format("        final String[] %s = new String[] {\n", metaVal);
 165                                 for (String s : (String[])val) {
 166                                     fmt.format("               \"%s\",\n", CLDRConverter.saveConvert(s, useJava));
 167                                 }
 168                                 fmt.format("            };\n");
 169                             } else {
 170                                 fmt.format("        final String %s = \"%s\";\n", metaVal, CLDRConverter.saveConvert((String)val, useJava));
 171                             }
 172                         }
 173 
 174                         newMap.put(k, metaVal);
 175                     }
 176                 }
 177 
 178                 if (Objects.nonNull(metaVal)) {
 179                     newMap.put(key, metaVal);
 180                 }
 181             }
 182 
 183             if (Objects.nonNull(newMap)) {
 184                 for (String key : map.keySet()) {
 185                     newMap.putIfAbsent(key, map.get(key));
 186                 }
 187                 map = newMap;
 188             }
 189         }
 190 
 191         try (PrintWriter out = new PrintWriter(file, encoding)) {
 192             // Output copyright headers
 193             out.println(CopyrightHeaders.getOpenJDKCopyright());
 194             out.println(CopyrightHeaders.getUnicodeCopyright());
 195 
 196             if (useJava) {
 197                 out.println("package sun." + packageName + ";\n");
 198                 out.printf("import %s;\n\n", type.getPathName());
 199                 out.printf("public class %s%s extends %s {\n", baseName, "root".equals(localeID) ? "" : "_" + localeID, type.getClassName());
 200 
 201                 out.println("    @Override\n" +
 202                             "    protected final Object[][] getContents() {");
 203                 if (fmt != null) {
 204                     out.print(fmt.toString());
 205                 }
 206                 out.println("        final Object[][] data = new Object[][] {");
 207             }
 208             for (String key : map.keySet()) {
 209                 if (useJava) {
 210                     Object value = map.get(key);
 211                     if (value == null) {
 212                         CLDRConverter.warning("null value for " + key);
 213                     } else if (value instanceof String) {
 214                         if (type == BundleType.TIMEZONE ||
 215                             ((String)value).startsWith(META_VALUE_PREFIX)) {
 216                             out.printf("            { \"%s\", %s },\n", key, CLDRConverter.saveConvert((String) value, useJava));
 217                         } else {
 218                             out.printf("            { \"%s\", \"%s\" },\n", key, CLDRConverter.saveConvert((String) value, useJava));
 219                         }
 220                     } else if (value instanceof String[]) {
 221                         String[] values = (String[]) value;
 222                         out.println("            { \"" + key + "\",\n                new String[] {");
 223                         for (String s : values) {
 224                             out.println("                    \"" + CLDRConverter.saveConvert(s, useJava) + "\",");
 225                         }
 226                         out.println("                }\n            },");
 227                     } else {
 228                         throw new RuntimeException("unknown value type: " + value.getClass().getName());
 229                     }
 230                 } else {
 231                     out.println(key + "=" + CLDRConverter.saveConvert((String) map.get(key), useJava));
 232                 }
 233             }
 234             if (useJava) {
 235                 out.println("        };\n        return data;\n    }\n}");
 236             }
 237         }
 238     }
 239 
 240     @Override
 241     public void generateMetaInfo(Map<String, SortedSet<String>> metaInfo) throws IOException {
 242         String dirName = CLDRConverter.DESTINATION_DIR + File.separator + "sun" + File.separator + "util" +
 243             File.separator +
 244             (CLDRConverter.isBaseModule ? "cldr" + File.separator + File.separator :
 245                       "resources" + File.separator + "cldr" + File.separator + "provider" + File.separator);
 246         File dir = new File(dirName);
 247         if (!dir.exists()) {
 248             dir.mkdirs();
 249         }
 250         String className =
 251             (CLDRConverter.isBaseModule ? "CLDRBaseLocaleDataMetaInfo" : "CLDRLocaleDataMetaInfo");
 252         File file = new File(dir, className + ".java");
 253         if (!file.exists()) {
 254             file.createNewFile();
 255         }
 256         CLDRConverter.info("Generating file " + file);
 257 
 258         try (PrintWriter out = new PrintWriter(file, "us-ascii")) {
 259             out.printf(CopyrightHeaders.getOpenJDKCopyright());
 260 
 261             out.printf((CLDRConverter.isBaseModule ? "package sun.util.cldr;\n\n" :
 262                                   "package sun.util.resources.cldr.provider;\n\n")
 263                       + "import java.util.HashMap;\n"
 264                       + "import java.util.Locale;\n"
 265                       + "import java.util.Map;\n"
 266                       + "import sun.util.locale.provider.LocaleDataMetaInfo;\n"
 267                       + "import sun.util.locale.provider.LocaleProviderAdapter;\n\n");
 268             out.printf("public class %s implements LocaleDataMetaInfo {\n", className);
 269             out.printf("    private static final Map<String, String> resourceNameToLocales = new HashMap<>();\n" +
 270                        (CLDRConverter.isBaseModule ?
 271                        "    private static final Map<Locale, String[]> parentLocalesMap = new HashMap<>();\n" +
 272                        "    private static final Map<String, String> languageAliasMap = new HashMap<>();\n\n" :
 273                        "\n") +
 274                        "    static {\n");
 275 
 276             for (String key : metaInfo.keySet()) {
 277                 if (key.startsWith(CLDRConverter.PARENT_LOCALE_PREFIX)) {
 278                     String parentTag = key.substring(CLDRConverter.PARENT_LOCALE_PREFIX.length());
 279                     if ("root".equals(parentTag)) {
 280                         out.printf("        parentLocalesMap.put(Locale.ROOT,\n");
 281                     } else {
 282                         out.printf("        parentLocalesMap.put(Locale.forLanguageTag(\"%s\"),\n",
 283                                    parentTag);
 284                     }
 285                     String[] children = toLocaleList(metaInfo.get(key), true).split(" ");
 286                     Arrays.sort(children);
 287                     out.printf("             new String[] {\n" +
 288                                "                 ");
 289                     int count = 0;
 290                     for (int i = 0; i < children.length; i++) {
 291                         String child = children[i];
 292                         out.printf("\"%s\", ", child);
 293                         count += child.length() + 4;
 294                         if (i != children.length - 1 && count > 64) {
 295                             out.printf("\n                 ");
 296                             count = 0;
 297                         }
 298                     }
 299                     out.printf("\n             });\n");
 300                 } else {
 301                     if ("AvailableLocales".equals(key)) {
 302                         out.printf("        resourceNameToLocales.put(\"%s\",\n", key);
 303                         out.printf("              \"%s\");\n", toLocaleList(applyLanguageAliases(metaInfo.get(key)), false));
 304                     }
 305                 }
 306             }
 307             // for languageAliasMap
 308             if (CLDRConverter.isBaseModule) {
 309                 CLDRConverter.handlerSupplMeta.getLanguageAliasData().forEach((key, value) -> {
 310                     out.printf("                languageAliasMap.put(\"%s\", \"%s\");\n", key, value);
 311                 });
 312             }
 313 
 314             out.printf("    }\n\n");
 315 
 316             // end of static initializer block.
 317 
 318             // Short TZ names for delayed initialization
 319             if (CLDRConverter.isBaseModule) {
 320                 out.printf("    private static class TZShortIDMapHolder {\n");
 321                 out.printf("        static final Map<String, String> tzShortIDMap = new HashMap<>();\n");
 322                 out.printf("        static {\n");
 323                 CLDRConverter.handlerTimeZone.getData().entrySet().stream()
 324                     .forEach(e -> {
 325                         out.printf("            tzShortIDMap.put(\"%s\", \"%s\");\n", e.getKey(),
 326                                 ((String)e.getValue()));
 327                     });
 328                 out.printf("        }\n    }\n\n");
 329             }
 330 
 331             out.printf("    @Override\n" +
 332                         "    public LocaleProviderAdapter.Type getType() {\n" +
 333                         "        return LocaleProviderAdapter.Type.CLDR;\n" +
 334                         "    }\n\n");
 335 
 336             out.printf("    @Override\n" +
 337                         "    public String availableLanguageTags(String category) {\n" +
 338                         "        return resourceNameToLocales.getOrDefault(category, \"\");\n" +
 339                         "    }\n\n");
 340 
 341             if (CLDRConverter.isBaseModule) {
 342                 out.printf("    @Override\n" +
 343                            "    public Map<String, String> getLanguageAliasMap() {\n" +
 344                            "        return languageAliasMap;\n" +
 345                            "    }\n\n");     
 346                 out.printf("    @Override\n" +
 347                            "    public Map<String, String> tzShortIDs() {\n" +
 348                            "        return TZShortIDMapHolder.tzShortIDMap;\n" +
 349                            "    }\n\n");
 350                 out.printf("    public Map<Locale, String[]> parentLocales() {\n" +
 351                            "        return parentLocalesMap;\n" +
 352                            "    }\n}");
 353             } else {
 354                 out.printf("}");
 355             }
 356         }
 357     }
 358 
 359     private static final Locale.Builder LOCALE_BUILDER = new Locale.Builder();
 360     private static boolean isBaseLocale(String localeID) {
 361         localeID = localeID.replaceAll("-", "_");
 362         // ignore script here
 363         Locale locale = LOCALE_BUILDER
 364                             .clear()
 365                             .setLanguage(CLDRConverter.getLanguageCode(localeID))
 366                             .setRegion(CLDRConverter.getRegionCode(localeID))
 367                             .build();
 368         return CLDRConverter.BASE_LOCALES.contains(locale);
 369     }
 370 
 371     private static String toLocaleList(SortedSet<String> set, boolean all) {
 372         StringBuilder sb = new StringBuilder(set.size() * 6);
 373         for (String id : set) {
 374             if (!"root".equals(id)) {
 375                 if (!all && CLDRConverter.isBaseModule ^ isBaseLocale(id)) {
 376                     continue;
 377                 }
 378                 sb.append(' ');
 379                 sb.append(id);
 380             }
 381         }
 382         return sb.toString();
 383     }
 384     
 385     private static SortedSet<String> applyLanguageAliases(SortedSet<String> tags) {
 386         CLDRConverter.handlerSupplMeta.getLanguageAliasData().forEach((key, value) -> {
 387             if (tags.remove(key)) {
 388                 tags.add(value);
 389             }
 390         });
 391         return tags;
 392     }
 393 }