ICULocaleService.java

// © 2016 and later: Unicode, Inc. and others.
// License & terms of use: http://www.unicode.org/copyright.html
/*
 *******************************************************************************
 * Copyright (C) 2001-2016, International Business Machines Corporation and
 * others. All Rights Reserved.
 *******************************************************************************
 */
package com.ibm.icu.impl;

import com.ibm.icu.util.ULocale;
import java.util.Collections;
import java.util.Locale;
import java.util.Map;
import java.util.Set;

public class ICULocaleService extends ICUService {
    private ULocale fallbackLocale;
    private String fallbackLocaleName;

    /** Construct an ICULocaleService. */
    public ICULocaleService() {}

    /** Construct an ICULocaleService with a name (useful for debugging). */
    public ICULocaleService(String name) {
        super(name);
    }

    /**
     * Convenience override for callers using locales. This calls get(ULocale, int, ULocale[]) with
     * KIND_ANY for kind and null for actualReturn.
     */
    public Object get(ULocale locale) {
        return get(locale, LocaleKey.KIND_ANY, null);
    }

    /**
     * Convenience override for callers using locales. This calls get(ULocale, int, ULocale[]) with
     * a null actualReturn.
     */
    public Object get(ULocale locale, int kind) {
        return get(locale, kind, null);
    }

    /**
     * Convenience override for callers using locales. This calls get(ULocale, int, ULocale[]) with
     * KIND_ANY for kind.
     */
    public Object get(ULocale locale, ULocale[] actualReturn) {
        return get(locale, LocaleKey.KIND_ANY, actualReturn);
    }

    /**
     * Convenience override for callers using locales. This uses createKey(ULocale.toString(), kind)
     * to create a key, calls getKey, and then if actualReturn is not null, returns the actualResult
     * from getKey (stripping any prefix) into a ULocale.
     */
    public Object get(ULocale locale, int kind, ULocale[] actualReturn) {
        Key key = createKey(locale, kind);
        if (actualReturn == null) {
            return getKey(key);
        }

        String[] temp = new String[1];
        Object result = getKey(key, temp);
        if (result != null) {
            int n = temp[0].indexOf("/");
            if (n >= 0) {
                temp[0] = temp[0].substring(n + 1);
            }
            actualReturn[0] = new ULocale(temp[0]);
        }
        return result;
    }

    /**
     * Convenience override for callers using locales. This calls registerObject(Object, ULocale,
     * int kind, boolean visible) passing KIND_ANY for the kind, and true for the visibility.
     */
    public Factory registerObject(Object obj, ULocale locale) {
        return registerObject(obj, locale, LocaleKey.KIND_ANY, true);
    }

    /**
     * Convenience override for callers using locales. This calls registerObject(Object, ULocale,
     * int kind, boolean visible) passing KIND_ANY for the kind.
     */
    public Factory registerObject(Object obj, ULocale locale, boolean visible) {
        return registerObject(obj, locale, LocaleKey.KIND_ANY, visible);
    }

    /**
     * Convenience function for callers using locales. This calls registerObject(Object, ULocale,
     * int kind, boolean visible) passing true for the visibility.
     */
    public Factory registerObject(Object obj, ULocale locale, int kind) {
        return registerObject(obj, locale, kind, true);
    }

    /**
     * Convenience function for callers using locales. This instantiates a SimpleLocaleKeyFactory,
     * and registers the factory.
     */
    public Factory registerObject(Object obj, ULocale locale, int kind, boolean visible) {
        Factory factory = new SimpleLocaleKeyFactory(obj, locale, kind, visible);
        return registerFactory(factory);
    }

    /**
     * Convenience method for callers using locales. This returns the standard Locale list, built
     * from the Set of visible ids.
     */
    public Locale[] getAvailableLocales() {
        // TODO make this wrap getAvailableULocales later
        Set<String> visIDs = getVisibleIDs();
        Locale[] locales = new Locale[visIDs.size()];
        int n = 0;
        for (String id : visIDs) {
            Locale loc = LocaleUtility.getLocaleFromName(id);
            locales[n++] = loc;
        }
        return locales;
    }

    /**
     * Convenience method for callers using locales. This returns the standard ULocale list, built
     * from the Set of visible ids.
     */
    public ULocale[] getAvailableULocales() {
        Set<String> visIDs = getVisibleIDs();
        ULocale[] locales = new ULocale[visIDs.size()];
        int n = 0;
        for (String id : visIDs) {
            locales[n++] = new ULocale(id);
        }
        return locales;
    }

    /**
     * A subclass of Key that implements a locale fallback mechanism. The first locale to search for
     * is the locale provided by the client, and the fallback locale to search for is the current
     * default locale. If a prefix is present, the currentDescriptor includes it before the locale
     * proper, separated by "/". This is the default key instantiated by ICULocaleService.
     *
     * <p>Canonicalization adjusts the locale string so that the section before the first understore
     * is in lower case, and the rest is in upper case, with no trailing underscores.
     */
    public static class LocaleKey extends ICUService.Key {
        private int kind;
        private int varstart;
        private String primaryID;
        private String fallbackID;
        private String currentID;

        public static final int KIND_ANY = -1;

        /** Create a LocaleKey with canonical primary and fallback IDs. */
        public static LocaleKey createWithCanonicalFallback(
                String primaryID, String canonicalFallbackID) {
            return createWithCanonicalFallback(primaryID, canonicalFallbackID, KIND_ANY);
        }

        /** Create a LocaleKey with canonical primary and fallback IDs. */
        public static LocaleKey createWithCanonicalFallback(
                String primaryID, String canonicalFallbackID, int kind) {
            if (primaryID == null) {
                return null;
            }
            String canonicalPrimaryID = ULocale.getName(primaryID);
            return new LocaleKey(primaryID, canonicalPrimaryID, canonicalFallbackID, kind);
        }

        /** Create a LocaleKey with canonical primary and fallback IDs. */
        public static LocaleKey createWithCanonical(
                ULocale locale, String canonicalFallbackID, int kind) {
            if (locale == null) {
                return null;
            }
            String canonicalPrimaryID = locale.getName();
            return new LocaleKey(canonicalPrimaryID, canonicalPrimaryID, canonicalFallbackID, kind);
        }

        /**
         * PrimaryID is the user's requested locale string, canonicalPrimaryID is this string in
         * canonical form, fallbackID is the current default locale's string in canonical form.
         */
        protected LocaleKey(
                String primaryID, String canonicalPrimaryID, String canonicalFallbackID, int kind) {
            super(primaryID);
            this.kind = kind;

            if (canonicalPrimaryID == null || canonicalPrimaryID.equalsIgnoreCase("root")) {
                this.primaryID = "";
                this.fallbackID = null;
            } else {
                int idx = canonicalPrimaryID.indexOf('@');
                if (idx == 4 && canonicalPrimaryID.regionMatches(true, 0, "root", 0, 4)) {
                    this.primaryID = canonicalPrimaryID.substring(4);
                    this.varstart = 0;
                    this.fallbackID = null;
                } else {
                    this.primaryID = canonicalPrimaryID;
                    this.varstart = idx;

                    if (canonicalFallbackID == null || this.primaryID.equals(canonicalFallbackID)) {
                        this.fallbackID = "";
                    } else {
                        this.fallbackID = canonicalFallbackID;
                    }
                }
            }

            this.currentID =
                    varstart == -1 ? this.primaryID : this.primaryID.substring(0, varstart);
        }

        /** Return the prefix associated with the kind, or null if the kind is KIND_ANY. */
        public String prefix() {
            return kind == KIND_ANY ? null : Integer.toString(kind());
        }

        /** Return the kind code associated with this key. */
        public int kind() {
            return kind;
        }

        /** Return the (canonical) original ID. */
        @Override
        public String canonicalID() {
            return primaryID;
        }

        /** Return the (canonical) current ID, or null if no current id. */
        @Override
        public String currentID() {
            return currentID;
        }

        /**
         * Return the (canonical) current descriptor, or null if no current id. Includes the
         * keywords, whereas the ID does not include keywords.
         */
        @Override
        public String currentDescriptor() {
            String result = currentID();
            if (result != null) {
                StringBuilder buf =
                        new StringBuilder(); // default capacity 16 is usually good enough
                if (kind != KIND_ANY) {
                    buf.append(prefix());
                }
                buf.append('/');
                buf.append(result);
                if (varstart != -1) {
                    buf.append(primaryID.substring(varstart, primaryID.length()));
                }
                result = buf.toString();
            }
            return result;
        }

        /** Convenience method to return the locale corresponding to the (canonical) original ID. */
        public ULocale canonicalLocale() {
            return new ULocale(primaryID);
        }

        /** Convenience method to return the ulocale corresponding to the (canonical) currentID. */
        public ULocale currentLocale() {
            if (varstart == -1) {
                return new ULocale(currentID);
            } else {
                return new ULocale(currentID + primaryID.substring(varstart));
            }
        }

        /**
         * If the key has a fallback, modify the key and return true, otherwise return false.
         *
         * <p>First falls back through the primary ID, then through the fallbackID. The final
         * fallback is "" (root) unless the primary id was "" (root), in which case there is no
         * fallback.
         */
        @Override
        public boolean fallback() {
            int x = currentID.lastIndexOf('_');
            if (x != -1) {
                while (--x >= 0 && currentID.charAt(x) == '_') { // handle zh__PINYIN
                }
                currentID = currentID.substring(0, x + 1);
                return true;
            }
            if (fallbackID != null) {
                currentID = fallbackID;
                if (fallbackID.length() == 0) {
                    fallbackID = null;
                } else {
                    fallbackID = "";
                }
                return true;
            }
            currentID = null;
            return false;
        }

        /**
         * If a key created from id would eventually fallback to match the canonical ID of this key,
         * return true.
         */
        @Override
        public boolean isFallbackOf(String id) {
            return LocaleUtility.isFallbackOf(canonicalID(), id);
        }
    }

    /** A subclass of Factory that uses LocaleKeys. If 'visible' the factory reports its IDs. */
    public abstract static class LocaleKeyFactory implements Factory {
        protected final String name;
        protected final boolean visible;

        public static final boolean VISIBLE = true;
        public static final boolean INVISIBLE = false;

        /** Constructor used by subclasses. */
        protected LocaleKeyFactory(boolean visible) {
            this.visible = visible;
            this.name = null;
        }

        /** Constructor used by subclasses. */
        protected LocaleKeyFactory(boolean visible, String name) {
            this.visible = visible;
            this.name = name;
        }

        /**
         * Implement superclass abstract method. This checks the currentID of the key against the
         * supported IDs, and passes the canonicalLocale and kind off to handleCreate (which
         * subclasses must implement).
         */
        @Override
        public Object create(Key key, ICUService service) {
            if (handlesKey(key)) {
                LocaleKey lkey = (LocaleKey) key;
                int kind = lkey.kind();

                ULocale uloc = lkey.currentLocale();
                return handleCreate(uloc, kind, service);
            } else {
                // System.out.println("factory: " + this + " did not support id: " +
                // key.currentID());
                // System.out.println("supported ids: " + getSupportedIDs());
            }
            return null;
        }

        protected boolean handlesKey(Key key) {
            if (key != null) {
                String id = key.currentID();
                Set<String> supported = getSupportedIDs();
                return supported.contains(id);
            }
            return false;
        }

        /** Override of superclass method. */
        @Override
        public void updateVisibleIDs(Map<String, Factory> result) {
            Set<String> cache = getSupportedIDs();
            for (String id : cache) {
                if (visible) {
                    result.put(id, this);
                } else {
                    result.remove(id);
                }
            }
        }

        /** Return a localized name for the locale represented by id. */
        @Override
        public String getDisplayName(String id, ULocale locale) {
            // assume if the user called this on us, we must have handled some fallback of this id
            //          if (isSupportedID(id)) {
            if (locale == null) {
                return id;
            }
            ULocale loc = new ULocale(id);
            return loc.getDisplayName(locale);
            //              }
            //          return null;
        }

        /// CLOVER:OFF
        /**
         * Utility method used by create(Key, ICUService). Subclasses can implement this instead of
         * create.
         */
        protected Object handleCreate(ULocale loc, int kind, ICUService service) {
            return null;
        }

        /// CLOVER:ON

        /** Return true if this id is one the factory supports (visible or otherwise). */
        protected boolean isSupportedID(String id) {
            return getSupportedIDs().contains(id);
        }

        /**
         * Return the set of ids that this factory supports (visible or otherwise). This can be
         * called often and might need to be cached if it is expensive to create.
         */
        protected Set<String> getSupportedIDs() {
            return Collections.emptySet();
        }

        /** For debugging. */
        @Override
        public String toString() {
            StringBuilder buf = new StringBuilder(super.toString());
            if (name != null) {
                buf.append(", name: ");
                buf.append(name);
            }
            buf.append(", visible: ");
            buf.append(visible);
            return buf.toString();
        }
    }

    /** A LocaleKeyFactory that just returns a single object for a kind/locale. */
    public static class SimpleLocaleKeyFactory extends LocaleKeyFactory {
        private final Object obj;
        private final String id;
        private final int kind;

        // TODO: remove when we no longer need this
        public SimpleLocaleKeyFactory(Object obj, ULocale locale, int kind, boolean visible) {
            this(obj, locale, kind, visible, null);
        }

        public SimpleLocaleKeyFactory(
                Object obj, ULocale locale, int kind, boolean visible, String name) {
            super(visible, name);

            this.obj = obj;
            this.id = locale.getBaseName();
            this.kind = kind;
        }

        /** Returns the service object if kind/locale match. Service is not used. */
        @Override
        public Object create(Key key, ICUService service) {
            if (!(key instanceof LocaleKey)) {
                return null;
            }

            LocaleKey lkey = (LocaleKey) key;
            if (kind != LocaleKey.KIND_ANY && kind != lkey.kind()) {
                return null;
            }
            if (!id.equals(lkey.currentID())) {
                return null;
            }

            return obj;
        }

        @Override
        protected boolean isSupportedID(String idToCheck) {
            return this.id.equals(idToCheck);
        }

        @Override
        public void updateVisibleIDs(Map<String, Factory> result) {
            if (visible) {
                result.put(id, this);
            } else {
                result.remove(id);
            }
        }

        @Override
        public String toString() {
            StringBuilder buf = new StringBuilder(super.toString());
            buf.append(", id: ");
            buf.append(id);
            buf.append(", kind: ");
            buf.append(kind);
            return buf.toString();
        }
    }

    /**
     * A LocaleKeyFactory that creates a service based on the ICU locale data. This is a base class
     * for most ICU factories. Subclasses instantiate it with a constructor that takes a bundle
     * name, which determines the supported IDs. Subclasses then override handleCreate to create the
     * actual service object. The default implementation returns a resource bundle.
     */
    public static class ICUResourceBundleFactory extends LocaleKeyFactory {
        protected final String bundleName;

        /** Convenience constructor that uses the main ICU bundle name. */
        public ICUResourceBundleFactory() {
            this(ICUData.ICU_BASE_NAME);
        }

        /** A service factory based on ICU resource data in resources with the given name. */
        public ICUResourceBundleFactory(String bundleName) {
            super(true);

            this.bundleName = bundleName;
        }

        /** Return the supported IDs. This is the set of all locale names for the bundleName. */
        @Override
        protected Set<String> getSupportedIDs() {
            return ICUResourceBundle.getFullLocaleNameSet(bundleName, loader());
        }

        /** Override of superclass method. */
        @Override
        public void updateVisibleIDs(Map<String, Factory> result) {
            Set<String> visibleIDs =
                    ICUResourceBundle.getAvailableLocaleNameSet(
                            bundleName, loader()); // only visible ids
            for (String id : visibleIDs) {
                result.put(id, this);
            }
        }

        /**
         * Create the service. The default implementation returns the resource bundle for the
         * locale, ignoring kind, and service.
         */
        @Override
        protected Object handleCreate(ULocale loc, int kind, ICUService service) {
            return ICUResourceBundle.getBundleInstance(bundleName, loc, loader());
        }

        protected ClassLoader loader() {
            return ClassLoaderUtil.getClassLoader(getClass());
        }

        @Override
        public String toString() {
            return super.toString() + ", bundle: " + bundleName;
        }
    }

    /**
     * Return the name of the current fallback locale. If it has changed since this was last
     * accessed, the service cache is cleared.
     */
    public String validateFallbackLocale() {
        ULocale loc = ULocale.getDefault();
        if (loc != fallbackLocale) {
            synchronized (this) {
                if (loc != fallbackLocale) {
                    fallbackLocaleName = loc.getBaseName();
                    clearServiceCache();
                    fallbackLocale = loc;
                }
            }
        }
        return fallbackLocaleName;
    }

    @Override
    public Key createKey(String id) {
        return LocaleKey.createWithCanonicalFallback(id, validateFallbackLocale());
    }

    public Key createKey(String id, int kind) {
        return LocaleKey.createWithCanonicalFallback(id, validateFallbackLocale(), kind);
    }

    public Key createKey(ULocale l, int kind) {
        return LocaleKey.createWithCanonical(l, validateFallbackLocale(), kind);
    }
}