/*

 * Copyright (c) 2012, 2018, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

package build.tools.cldrconverter;

import static build.tools.cldrconverter.Bundle.jreTimeZoneNames;

import build.tools.cldrconverter.BundleGenerator.BundleType;

import java.io.File;

import java.io.IOException;

import java.io.UncheckedIOException;

import java.nio.file.*;

import java.text.MessageFormat;

import java.time.*;

import java.util.*;

import java.util.ResourceBundle.Control;

import java.util.logging.Level;

import java.util.logging.Logger;

import java.util.stream.Collectors;

import java.util.stream.IntStream;

import java.util.stream.Stream;

import javax.xml.parsers.SAXParser;

import javax.xml.parsers.SAXParserFactory;

import org.xml.sax.SAXNotRecognizedException;

import org.xml.sax.SAXNotSupportedException;

/**

 * Converts locale data from "Locale Data Markup Language" format to

 * JRE resource bundle format. LDML is the format used by the Common

 * Locale Data Repository maintained by the Unicode Consortium.

 */

public class CLDRConverter {

    static final String LDML_DTD_SYSTEM_ID = "http://www.unicode.org/cldr/dtd/2.0/ldml.dtd";

    static final String SPPL_LDML_DTD_SYSTEM_ID = "http://www.unicode.org/cldr/dtd/2.0/ldmlSupplemental.dtd";

    static final String BCP47_LDML_DTD_SYSTEM_ID = "http://www.unicode.org/cldr/dtd/2.0/ldmlBCP47.dtd";

    private static String CLDR_BASE;

    static String LOCAL_LDML_DTD;

    static String LOCAL_SPPL_LDML_DTD;

    static String LOCAL_BCP47_LDML_DTD;

    private static String SOURCE_FILE_DIR;

    private static String SPPL_SOURCE_FILE;

    private static String SPPL_META_SOURCE_FILE;

    private static String NUMBERING_SOURCE_FILE;

    private static String METAZONES_SOURCE_FILE;

    private static String LIKELYSUBTAGS_SOURCE_FILE;

    private static String TIMEZONE_SOURCE_FILE;

    static String DESTINATION_DIR = "build/gensrc";

    static final String LOCALE_NAME_PREFIX = "locale.displayname.";

    static final String LOCALE_SEPARATOR = LOCALE_NAME_PREFIX + "separator";

    static final String LOCALE_KEYTYPE = LOCALE_NAME_PREFIX + "keytype";

    static final String LOCALE_KEY_PREFIX = LOCALE_NAME_PREFIX + "key.";

    static final String LOCALE_TYPE_PREFIX = LOCALE_NAME_PREFIX + "type.";

    static final String LOCALE_TYPE_PREFIX_CA = LOCALE_TYPE_PREFIX + "ca.";

    static final String CURRENCY_SYMBOL_PREFIX = "currency.symbol.";

    static final String CURRENCY_NAME_PREFIX = "currency.displayname.";

    static final String CALENDAR_NAME_PREFIX = "calendarname.";

    static final String CALENDAR_FIRSTDAY_PREFIX = "firstDay.";

    static final String CALENDAR_MINDAYS_PREFIX = "minDays.";

    static final String TIMEZONE_ID_PREFIX = "timezone.id.";

    static final String EXEMPLAR_CITY_PREFIX = "timezone.excity.";

    static final String ZONE_NAME_PREFIX = "timezone.displayname.";

    static final String METAZONE_ID_PREFIX = "metazone.id.";

    static final String PARENT_LOCALE_PREFIX = "parentLocale.";

    static final String[] EMPTY_ZONE = {"", "", "", "", "", ""};

    private static SupplementDataParseHandler handlerSuppl;

    private static LikelySubtagsParseHandler handlerLikelySubtags;

    static SupplementalMetadataParseHandler handlerSupplMeta;

    static NumberingSystemsParseHandler handlerNumbering;

    static MetaZonesParseHandler handlerMetaZones;

    static TimeZoneParseHandler handlerTimeZone;

    private static BundleGenerator bundleGenerator;

    // java.base module related

    static boolean isBaseModule = false;

    static final Set<Locale> BASE_LOCALES = new HashSet<>();

    // "parentLocales" map

    private static final Map<String, SortedSet<String>> parentLocalesMap = new HashMap<>();

    private static final ResourceBundle.Control defCon =

        ResourceBundle.Control.getControl(ResourceBundle.Control.FORMAT_DEFAULT);

    private static final String[] AVAILABLE_TZIDS = TimeZone.getAvailableIDs();

    private static String zoneNameTempFile;

    private static String tzDataDir;

    private static final Map<String, String> canonicalTZMap = new HashMap<>();

    static enum DraftType {

        UNCONFIRMED,

        PROVISIONAL,

        CONTRIBUTED,

        APPROVED;

        private static final Map<String, DraftType> map = new HashMap<>();

        static {

            for (DraftType dt : values()) {

                map.put(dt.getKeyword(), dt);

            }

        }

        static private DraftType defaultType = CONTRIBUTED;

        private final String keyword;

        private DraftType() {

            keyword = this.name().toLowerCase(Locale.ROOT);

        }

        static DraftType forKeyword(String keyword) {

            return map.get(keyword);

        }

        static DraftType getDefault() {

            return defaultType;

        }

        static void setDefault(String keyword) {

            defaultType = Objects.requireNonNull(forKeyword(keyword));

        }

        String getKeyword() {

            return keyword;

        }

    }

    static boolean USE_UTF8 = false;

    private static boolean verbose;

    private CLDRConverter() {

       // no instantiation

    }

    @SuppressWarnings("AssignmentToForLoopParameter")

    public static void main(String[] args) throws Exception {

        if (args.length != 0) {

            String currentArg = null;

            try {

                for (int i = 0; i < args.length; i++) {

                    currentArg = args[i];

                    switch (currentArg) {

                    case "-draft":

                        String draftDataType = args[++i];

                        try {

                            DraftType.setDefault(draftDataType);

                        } catch (NullPointerException e) {

                            severe("Error: incorrect draft value: %s%n", draftDataType);

                            System.exit(1);

                        }

                        info("Using the specified data type: %s%n", draftDataType);

                        break;

                    case "-base":

                        // base directory for input files

                        CLDR_BASE = args[++i];

                        if (!CLDR_BASE.endsWith("/")) {

                            CLDR_BASE += "/";

                        }

                        break;

                    case "-baselocales":

                        // base locales

                        setupBaseLocales(args[++i]);

                        break;

                    case "-basemodule":

                        // indicates java.base module resource generation

                        isBaseModule = true;

                        break;

                    case "-o":

                        // output directory

                        DESTINATION_DIR = args[++i];

                        break;

                    case "-utf8":

                        USE_UTF8 = true;

                        break;

                    case "-verbose":

                        verbose = true;

                        break;

                    case "-zntempfile":

                        zoneNameTempFile = args[++i];

                        break;

                    case "-tzdatadir":

                        tzDataDir = args[++i];

                        break;

                    case "-help":

                        usage();

                        System.exit(0);

                        break;

                    default:

                        throw new RuntimeException();

                    }

                }

            } catch (RuntimeException e) {

                severe("unknown or imcomplete arg(s): " + currentArg);

                usage();

                System.exit(1);

            }

        }

        // Set up path names

        LOCAL_LDML_DTD = CLDR_BASE + "/dtd/ldml.dtd";

        LOCAL_SPPL_LDML_DTD = CLDR_BASE + "/dtd/ldmlSupplemental.dtd";

        LOCAL_BCP47_LDML_DTD = CLDR_BASE + "/dtd/ldmlBCP47.dtd";

        SOURCE_FILE_DIR = CLDR_BASE + "/main";

        SPPL_SOURCE_FILE = CLDR_BASE + "/supplemental/supplementalData.xml";

        LIKELYSUBTAGS_SOURCE_FILE = CLDR_BASE + "/supplemental/likelySubtags.xml";

        NUMBERING_SOURCE_FILE = CLDR_BASE + "/supplemental/numberingSystems.xml";

        METAZONES_SOURCE_FILE = CLDR_BASE + "/supplemental/metaZones.xml";

        TIMEZONE_SOURCE_FILE = CLDR_BASE + "/bcp47/timezone.xml";

        SPPL_META_SOURCE_FILE = CLDR_BASE + "/supplemental/supplementalMetadata.xml";

        if (BASE_LOCALES.isEmpty()) {

            setupBaseLocales("en-US");

        }

        bundleGenerator = new ResourceBundleGenerator();

        // Parse data independent of locales

        parseSupplemental();

        parseBCP47();

        List<Bundle> bundles = readBundleList();

        convertBundles(bundles);

        if (isBaseModule) {

            generateZoneName();

        }

    }

    private static void usage() {

        errout("Usage: java CLDRConverter [options]%n"

                + "\t-help          output this usage message and exit%n"

                + "\t-verbose       output information%n"

                + "\t-draft [contributed | approved | provisional | unconfirmed]%n"

                + "\t\t       draft level for using data (default: contributed)%n"

                + "\t-base dir      base directory for CLDR input files%n"

                + "\t-basemodule    generates bundles that go into java.base module%n"

                + "\t-baselocales loc(,loc)*      locales that go into the base module%n"

                + "\t-o dir         output directory (default: ./build/gensrc)%n"

                + "\t-zntempfile    template file for java.time.format.ZoneName.java%n"

                + "\t-tzdatadir     tzdata directory for java.time.format.ZoneName.java%n"

                + "\t-utf8          use UTF-8 rather than \\uxxxx (for debug)%n");

    }

    static void info(String fmt, Object... args) {

        if (verbose) {

            System.out.printf(fmt, args);

        }

    }

    static void info(String msg) {

        if (verbose) {

            System.out.println(msg);

        }

    }

    static void warning(String fmt, Object... args) {

        System.err.print("Warning: ");

        System.err.printf(fmt, args);

    }

    static void warning(String msg) {

        System.err.print("Warning: ");

        errout(msg);

    }

    static void severe(String fmt, Object... args) {

        System.err.print("Error: ");

        System.err.printf(fmt, args);

    }

    static void severe(String msg) {

        System.err.print("Error: ");

        errout(msg);

    }

    private static void errout(String msg) {

        if (msg.contains("%n")) {

            System.err.printf(msg);

        } else {

            System.err.println(msg);

        }

    }

    /**

     * Configure the parser to allow access to DTDs on the file system.

     */

    private static void enableFileAccess(SAXParser parser) throws SAXNotSupportedException {

        try {

            parser.setProperty("http://javax.xml.XMLConstants/property/accessExternalDTD", "file");

        } catch (SAXNotRecognizedException ignore) {

            // property requires >= JAXP 1.5

        }

    }

    private static List<Bundle> readBundleList() throws Exception {

        List<Bundle> retList = new ArrayList<>();

        Path path = FileSystems.getDefault().getPath(SOURCE_FILE_DIR);

        try (DirectoryStream<Path> dirStr = Files.newDirectoryStream(path)) {

            for (Path entry : dirStr) {

                String fileName = entry.getFileName().toString();

                if (fileName.endsWith(".xml")) {

                    String id = fileName.substring(0, fileName.indexOf('.'));

                    Locale cldrLoc = Locale.forLanguageTag(toLanguageTag(id));

                    StringBuilder sb = getCandLocales(cldrLoc);

                    if (sb.indexOf("root") == -1) {

                        sb.append("root");

                    }

                    Bundle b = new Bundle(id, sb.toString(), null, null);

                    // Insert the bundle for root at the top so that it will get

                    // processed first.

                    if ("root".equals(id)) {

                        retList.add(0, b);

                    } else {

                        retList.add(b);

                    }

                }

            }

        }

        return retList;

    }

    private static final Map<String, Map<String, Object>> cldrBundles = new HashMap<>();

    private static Map<String, SortedSet<String>> metaInfo = new HashMap<>();

    static {

        // For generating information on supported locales.

        metaInfo.put("AvailableLocales", new TreeSet<>());

    }

    static Map<String, Object> getCLDRBundle(String id) throws Exception {

        Map<String, Object> bundle = cldrBundles.get(id);

        if (bundle != null) {

            return bundle;

        }

        File file = new File(SOURCE_FILE_DIR + File.separator + id + ".xml");

        if (!file.exists()) {

            // Skip if the file doesn't exist.

            return Collections.emptyMap();

        }

        info("..... main directory .....");

        LDMLParseHandler handler = new LDMLParseHandler(id);

        parseLDMLFile(file, handler);

        bundle = handler.getData();

        cldrBundles.put(id, bundle);

        if (id.equals("root")) {

            // Calendar data (firstDayOfWeek & minDaysInFirstWeek)

            bundle = handlerSuppl.getData("root");

            if (bundle != null) {

                //merge two maps into one map

                Map<String, Object> temp = cldrBundles.remove(id);

                bundle.putAll(temp);

                cldrBundles.put(id, bundle);

            }

        }

        return bundle;

    }

    // Parsers for data in "supplemental" directory

    //

    private static void parseSupplemental() throws Exception {

        // Parse SupplementalData file and store the information in the HashMap

        // Calendar information such as firstDay and minDay are stored in

        // supplementalData.xml as of CLDR1.4. Individual territory is listed

        // with its ISO 3166 country code while default is listed using UNM49

        // region and composition numerical code (001 for World.)

        //

        // SupplementalData file also provides the "parent" locales which

        // are othrwise not to be fallen back. Process them here as well.

        //

        handlerSuppl = new SupplementDataParseHandler();

        parseLDMLFile(new File(SPPL_SOURCE_FILE), handlerSuppl);

        Map<String, Object> parentData = handlerSuppl.getData("root");

        parentData.keySet().stream()

                .filter(key -> key.startsWith(PARENT_LOCALE_PREFIX))

                .forEach(key -> {

                parentLocalesMap.put(key, new TreeSet(

                    Arrays.asList(((String)parentData.get(key)).split(" "))));

            });

        // Parse numberingSystems to get digit zero character information.

        handlerNumbering = new NumberingSystemsParseHandler();

        parseLDMLFile(new File(NUMBERING_SOURCE_FILE), handlerNumbering);

        // Parse metaZones to create mappings between Olson tzids and CLDR meta zone names

        handlerMetaZones = new MetaZonesParseHandler();

        parseLDMLFile(new File(METAZONES_SOURCE_FILE), handlerMetaZones);

        // Parse likelySubtags

        handlerLikelySubtags = new LikelySubtagsParseHandler();

        parseLDMLFile(new File(LIKELYSUBTAGS_SOURCE_FILE), handlerLikelySubtags);

        // Parse supplementalMetadata

        // Currently interested in deprecated time zone ids and language aliases.

        handlerSupplMeta = new SupplementalMetadataParseHandler();

        parseLDMLFile(new File(SPPL_META_SOURCE_FILE), handlerSupplMeta);

    }

    // Parsers for data in "bcp47" directory

    //

    private static void parseBCP47() throws Exception {

        // Parse timezone

        handlerTimeZone = new TimeZoneParseHandler();

        parseLDMLFile(new File(TIMEZONE_SOURCE_FILE), handlerTimeZone);

        // canonical tz name map

        // alias -> primary

        handlerTimeZone.getData().forEach((k, v) -> {

            String[] ids = ((String)v).split("\\s");

            for (int i = 1; i < ids.length; i++) {

                canonicalTZMap.put(ids[i], ids[0]);

            }

        });

    }

    private static void parseLDMLFile(File srcfile, AbstractLDMLHandler handler) throws Exception {

        info("..... Parsing " + srcfile.getName() + " .....");

        SAXParserFactory pf = SAXParserFactory.newInstance();

        pf.setValidating(true);

        SAXParser parser = pf.newSAXParser();

        enableFileAccess(parser);

        parser.parse(srcfile, handler);

    }

    private static StringBuilder getCandLocales(Locale cldrLoc) {

        List<Locale> candList = getCandidateLocales(cldrLoc);

        StringBuilder sb = new StringBuilder();

        for (Locale loc : candList) {

            if (!loc.equals(Locale.ROOT)) {

                sb.append(toLocaleName(loc.toLanguageTag()));

                sb.append(",");

            }

        }

        return sb;

    }

    private static List<Locale> getCandidateLocales(Locale cldrLoc) {

        List<Locale> candList = new ArrayList<>();

        candList = applyParentLocales("", defCon.getCandidateLocales("",  cldrLoc));

        return candList;

    }

    private static void convertBundles(List<Bundle> bundles) throws Exception {

        // parent locales map. The mappings are put in base metaInfo file

        // for now.

        if (isBaseModule) {

            metaInfo.putAll(parentLocalesMap);

        }

        for (Bundle bundle : bundles) {

            // Get the target map, which contains all the data that should be

            // visible for the bundle's locale

            Map<String, Object> targetMap = bundle.getTargetMap();