1 /*
   2  * Copyright (c) 2012, 2018, 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                         String valStr = (String)value;
 215                         if (type == BundleType.TIMEZONE &&
 216                             !key.startsWith(CLDRConverter.EXEMPLAR_CITY_PREFIX) ||
 217                             valStr.startsWith(META_VALUE_PREFIX)) {
 218                             out.printf("            { \"%s\", %s },\n", key, CLDRConverter.saveConvert(valStr, useJava));
 219                         } else {
 220                             out.printf("            { \"%s\", \"%s\" },\n", key, CLDRConverter.saveConvert(valStr, useJava));
 221                         }
 222                     } else if (value instanceof String[]) {
 223                         String[] values = (String[]) value;
 224                         out.println("            { \"" + key + "\",\n                new String[] {");
 225                         for (String s : values) {
 226                             out.println("                    \"" + CLDRConverter.saveConvert(s, useJava) + "\",");
 227                         }
 228                         out.println("                }\n            },");
 229                     } else {
 230                         throw new RuntimeException("unknown value type: " + value.getClass().getName());
 231                     }
 232                 } else {
 233                     out.println(key + "=" + CLDRConverter.saveConvert((String) map.get(key), useJava));
 234                 }
 235             }
 236             if (useJava) {
 237                 out.println("        };\n        return data;\n    }\n}");
 238             }
 239         }
 240     }
 241 
 242     @Override
 243     public void generateMetaInfo(Map<String, SortedSet<String>> metaInfo) throws IOException {
 244         String dirName = CLDRConverter.DESTINATION_DIR + File.separator + "sun" + File.separator + "util" +
 245             File.separator +
 246             (CLDRConverter.isBaseModule ? "cldr" + File.separator + File.separator :
 247                       "resources" + File.separator + "cldr" + File.separator + "provider" + File.separator);
 248         File dir = new File(dirName);
 249         if (!dir.exists()) {
 250             dir.mkdirs();
 251         }
 252         String className =
 253             (CLDRConverter.isBaseModule ? "CLDRBaseLocaleDataMetaInfo" : "CLDRLocaleDataMetaInfo");
 254         File file = new File(dir, className + ".java");
 255         if (!file.exists()) {
 256             file.createNewFile();
 257         }
 258         CLDRConverter.info("Generating file " + file);
 259 
 260         try (PrintWriter out = new PrintWriter(file, "us-ascii")) {
 261             out.printf(CopyrightHeaders.getOpenJDKCopyright());
 262 
 263             out.printf((CLDRConverter.isBaseModule ? "package sun.util.cldr;\n\n" :
 264                                   "package sun.util.resources.cldr.provider;\n\n")
 265                       + "import java.util.HashMap;\n"
 266                       + "import java.util.Locale;\n"
 267                       + "import java.util.Map;\n"
 268                       + "import sun.util.locale.provider.LocaleDataMetaInfo;\n"
 269                       + "import sun.util.locale.provider.LocaleProviderAdapter;\n\n");
 270             out.printf("public class %s implements LocaleDataMetaInfo {\n", className);
 271             out.printf("    private static final Map<String, String> resourceNameToLocales = new HashMap<>();\n" +
 272                        (CLDRConverter.isBaseModule ?
 273                        "    private static final Map<Locale, String[]> parentLocalesMap = new HashMap<>();\n\n" :
 274                        "\n") +
 275                        "    static {\n");
 276 
 277             for (String key : metaInfo.keySet()) {
 278                 if (key.startsWith(CLDRConverter.PARENT_LOCALE_PREFIX)) {
 279                     String parentTag = key.substring(CLDRConverter.PARENT_LOCALE_PREFIX.length());
 280                     if ("root".equals(parentTag)) {
 281                         out.printf("        parentLocalesMap.put(Locale.ROOT,\n");
 282                     } else {
 283                         out.printf("        parentLocalesMap.put(Locale.forLanguageTag(\"%s\"),\n",
 284                                    parentTag);
 285                     }
 286                     String[] children = toLocaleList(metaInfo.get(key), true).split(" ");
 287                     Arrays.sort(children);
 288                     out.printf("             new String[] {\n" +
 289                                "                 ");
 290                     int count = 0;
 291                     for (int i = 0; i < children.length; i++) {
 292                         String child = children[i];
 293                         out.printf("\"%s\", ", child);
 294                         count += child.length() + 4;
 295                         if (i != children.length - 1 && count > 64) {
 296                             out.printf("\n                 ");
 297                             count = 0;
 298                         }
 299                     }
 300                     out.printf("\n             });\n");
 301                 } else {
 302                     if ("AvailableLocales".equals(key)) {
 303                         out.printf("        resourceNameToLocales.put(\"%s\",\n", key);
 304                         out.printf("              \"%s\");\n", toLocaleList(metaInfo.get(key), false));
 305                     }
 306                 }
 307             }
 308 
 309             out.printf("    }\n\n");
 310 
 311             // end of static initializer block.
 312 
 313             // Canonical TZ names for delayed initialization
 314             if (CLDRConverter.isBaseModule) {
 315                 out.printf("    private static class TZCanonicalIDMapHolder {\n");
 316                 out.printf("        static final Map<String, String> tzCanonicalIDMap = new HashMap<>(600);\n");
 317                 out.printf("        static {\n");
 318                 CLDRConverter.handlerTimeZone.getData().entrySet().stream()
 319                     .forEach(e -> {
 320                         String[] ids = ((String)e.getValue()).split("\\s");
 321                         out.printf("            tzCanonicalIDMap.put(\"%s\", \"%s\");\n", e.getKey(),
 322                                 ids[0]);
 323                         for (int i = 1; i < ids.length; i++) {
 324                             out.printf("            tzCanonicalIDMap.put(\"%s\", \"%s\");\n", ids[i],
 325                                 ids[0]);
 326                         }
 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> tzCanonicalIDs() {\n" +
 344                            "        return TZCanonicalIDMapHolder.tzCanonicalIDMap;\n" +
 345                            "    }\n\n");
 346                 out.printf("    public Map<Locale, String[]> parentLocales() {\n" +
 347                            "        return parentLocalesMap;\n" +
 348                            "    }\n}");
 349             } else {
 350                 out.printf("}");
 351             }
 352         }
 353     }
 354 
 355     private static final Locale.Builder LOCALE_BUILDER = new Locale.Builder();
 356     private static boolean isBaseLocale(String localeID) {
 357         localeID = localeID.replaceAll("-", "_");
 358         // ignore script here
 359         Locale locale = LOCALE_BUILDER
 360                             .clear()
 361                             .setLanguage(CLDRConverter.getLanguageCode(localeID))
 362                             .setRegion(CLDRConverter.getRegionCode(localeID))
 363                             .build();
 364         return CLDRConverter.BASE_LOCALES.contains(locale);
 365     }
 366 
 367     private static String toLocaleList(SortedSet<String> set, boolean all) {
 368         StringBuilder sb = new StringBuilder(set.size() * 6);
 369         for (String id : set) {
 370             if (!"root".equals(id)) {
 371                 if (!all && CLDRConverter.isBaseModule ^ isBaseLocale(id)) {
 372                     continue;
 373                 }
 374                 sb.append(' ');
 375                 sb.append(id);
 376             }
 377         }
 378         return sb.toString();
 379     }
 380 }