    001// Copyright 2006-2013 The Apache Software Foundation
    002//
    003// Licensed under the Apache License, Version 2.0 (the "License");
    004// you may not use this file except in compliance with the License.
    005// You may obtain a copy of the License at
    006//
    007// http://www.apache.org/licenses/LICENSE-2.0
    008//
    009// Unless required by applicable law or agreed to in writing, software
    010// distributed under the License is distributed on an "AS IS" BASIS,
    011// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    012// See the License for the specific language governing permissions and
    013// limitations under the License.
    014
    015package org.apache.tapestry5.internal.services;
    016
    017import java.util.Arrays;
    018import java.util.List;
    019import java.util.Locale;
    020import java.util.concurrent.Callable;
    021
    022import org.apache.tapestry5.SymbolConstants;
    023import org.apache.tapestry5.ioc.Messages;
    024import org.apache.tapestry5.ioc.Resource;
    025import org.apache.tapestry5.ioc.annotations.PostInjection;
    026import org.apache.tapestry5.ioc.annotations.Symbol;
    027import org.apache.tapestry5.ioc.internal.util.TapestryException;
    028import org.apache.tapestry5.ioc.internal.util.URLChangeTracker;
    029import org.apache.tapestry5.ioc.services.ClasspathURLConverter;
    030import org.apache.tapestry5.ioc.services.ThreadLocale;
    031import org.apache.tapestry5.model.ComponentModel;
    032import org.apache.tapestry5.services.InvalidationEventHub;
    033import org.apache.tapestry5.services.UpdateListener;
    034import org.apache.tapestry5.services.messages.ComponentMessagesSource;
    035import org.apache.tapestry5.services.messages.PropertiesFileParser;
    036import org.apache.tapestry5.services.pageload.ComponentRequestSelectorAnalyzer;
    037import org.apache.tapestry5.services.pageload.ComponentResourceLocator;
    038import org.apache.tapestry5.services.pageload.ComponentResourceSelector;
    039
    040public class ComponentMessagesSourceImpl implements ComponentMessagesSource, UpdateListener
    041{
    042    private final MessagesSourceImpl messagesSource;
    043
    044    private final MessagesBundle appCatalogBundle;
    045
    046    private final ComponentRequestSelectorAnalyzer componentRequestSelectorAnalyzer;
    047
    048    private final ThreadLocale threadLocale;
    049
    050    private class ComponentModelBundle implements MessagesBundle
    051    {
    052        private final ComponentModel model;
    053
    054        public ComponentModelBundle(ComponentModel model)
    055        {
    056            this.model = model;
    057        }
    058
    059        public Resource getBaseResource()
    060        {
    061            return model.getBaseResource();
    062        }
    063
    064        public Object getId()
    065        {
    066            return model.getComponentClassName();
    067        }
    068
    069        public MessagesBundle getParent()
    070        {
    071            ComponentModel parentModel = model.getParentModel();
    072
    073            if (parentModel == null)
    074                return appCatalogBundle;
    075
    076            return new ComponentModelBundle(parentModel);
    077        }
    078    }
    079
    080    public ComponentMessagesSourceImpl(@Symbol(SymbolConstants.PRODUCTION_MODE)
    081                                       boolean productionMode, List<Resource> appCatalogResources, PropertiesFileParser parser,
    082                                       ComponentResourceLocator resourceLocator, ClasspathURLConverter classpathURLConverter,
    083                                       ComponentRequestSelectorAnalyzer componentRequestSelectorAnalyzer,
    084                                       ThreadLocale threadLocale)
    085    {
    086        this(productionMode, appCatalogResources, resourceLocator, parser, new URLChangeTracker(classpathURLConverter), componentRequestSelectorAnalyzer, threadLocale);
    087    }
    088
    089    ComponentMessagesSourceImpl(boolean productionMode, Resource appCatalogResource,
    090                                ComponentResourceLocator resourceLocator, PropertiesFileParser parser,
    091                                URLChangeTracker tracker, ComponentRequestSelectorAnalyzer componentRequestSelectorAnalyzer,
    092                                ThreadLocale threadLocale)
    093    {
    094        this(productionMode, Arrays.asList(appCatalogResource), resourceLocator, parser, tracker, componentRequestSelectorAnalyzer, threadLocale);
    095    }
    096
    097    ComponentMessagesSourceImpl(boolean productionMode, List<Resource> appCatalogResources,
    098                                ComponentResourceLocator resourceLocator, PropertiesFileParser parser,
    099                                URLChangeTracker tracker, ComponentRequestSelectorAnalyzer componentRequestSelectorAnalyzer,
    100                                ThreadLocale threadLocale)
    101    {
    102        messagesSource = new MessagesSourceImpl(productionMode, productionMode ? null : tracker, resourceLocator,
    103                parser);
    104
    105        appCatalogBundle = createAppCatalogBundle(appCatalogResources);
    106        this.componentRequestSelectorAnalyzer = componentRequestSelectorAnalyzer;
    107        this.threadLocale = threadLocale;
    108    }
    109
    110    @PostInjection
    111    public void setupReload(ReloadHelper reloadHelper)
    112    {
    113        reloadHelper.addReloadCallback(new Runnable()
    114        {
    115            public void run()
    116            {
    117                messagesSource.invalidate();
    118            }
    119        });
    120    }
    121
    122    public void checkForUpdates()
    123    {
    124        messagesSource.checkForUpdates();
    125    }
    126
    127    public Messages getMessages(ComponentModel componentModel, Locale locale)
    128    {
    129       return getMessagesWithForcedLocale(() -> getMessages(componentModel, componentRequestSelectorAnalyzer.buildSelectorForRequest()), locale);
    130    }
    131
    132    private Messages getMessagesWithForcedLocale(Callable<Messages> callable, Locale locale)
    133    {
    134        final Locale original = threadLocale.getLocale();
    135        try
    136        {
    137           threadLocale.setLocale(locale);
    138           return callable.call();
    139        } catch (Exception e) {
    140            throw new TapestryException(e.getMessage(), e);
    141        }
    142        finally {
    143            threadLocale.setLocale(original);
    144        }
    145    }
    146
    147    public Messages getMessages(ComponentModel componentModel, ComponentResourceSelector selector)
    148    {
    149        final MessagesBundle bundle = new ComponentModelBundle(componentModel);
    150        return messagesSource.getMessages(bundle, selector);
    151    }
    152
    153    public Messages getApplicationCatalog(Locale locale)
    154    {
    155        return getMessagesWithForcedLocale(() -> messagesSource.getMessages(appCatalogBundle, componentRequestSelectorAnalyzer.buildSelectorForRequest()), locale);
    156    }
    157
    158    private MessagesBundle createAppCatalogBundle(List<Resource> resources)
    159    {
    160        MessagesBundle current = null;
    161
    162        for (Resource r : resources)
    163        {
    164            current = createMessagesBundle(r, current);
    165        }
    166
    167        return current;
    168    }
    169
    170    private MessagesBundle createMessagesBundle(final Resource resource, final MessagesBundle parent)
    171    {
    172        return new MessagesBundle()
    173        {
    174            public Resource getBaseResource()
    175            {
    176                return resource;
    177            }
    178
    179            public Object getId()
    180            {
    181                return resource.getPath();
    182            }
    183
    184            public MessagesBundle getParent()
    185            {
    186                return parent;
    187            }
    188        };
    189    }
    190
    191    public InvalidationEventHub getInvalidationEventHub()
    192    {
    193        return messagesSource;
    194    }
    195}