001    /**
002     * Copyright (c) 2000-2013 Liferay, Inc. All rights reserved.
003     *
004     * This library is free software; you can redistribute it and/or modify it under
005     * the terms of the GNU Lesser General Public License as published by the Free
006     * Software Foundation; either version 2.1 of the License, or (at your option)
007     * any later version.
008     *
009     * This library is distributed in the hope that it will be useful, but WITHOUT
010     * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
011     * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
012     * details.
013     */
014    
015    package com.liferay.portal.xml.xpath;
016    
017    import com.liferay.portal.kernel.cache.Lifecycle;
018    import com.liferay.portal.kernel.cache.ThreadLocalCache;
019    import com.liferay.portal.kernel.cache.ThreadLocalCacheManager;
020    
021    import java.util.HashMap;
022    import java.util.List;
023    import java.util.Map;
024    import java.util.regex.Matcher;
025    import java.util.regex.Pattern;
026    
027    import org.jaxen.Context;
028    import org.jaxen.Function;
029    import org.jaxen.FunctionCallException;
030    import org.jaxen.Navigator;
031    import org.jaxen.function.StringFunction;
032    
033    /**
034     * @author Raymond Aug??
035     */
036    public class MatchesFunction implements Function {
037    
038            public static Boolean evaluate(
039                    Object stringArgument, Object regexArgument, Navigator navigator) {
040    
041                    String string = StringFunction.evaluate(stringArgument, navigator);
042                    String regex = StringFunction.evaluate(regexArgument, navigator);
043    
044                    return evaluate(string, regex, 0, navigator);
045            }
046    
047            public static Boolean evaluate(
048                    Object stringArg, Object regexArgument, Object flagsArgument,
049                    Navigator navigator) {
050    
051                    String string = StringFunction.evaluate(stringArg, navigator);
052                    String regex = StringFunction.evaluate(regexArgument, navigator);
053                    String flagsString = StringFunction.evaluate(flagsArgument, navigator);
054    
055                    int flags = 0;
056    
057                    for (int i = 0; i < flagsString.length(); i++) {
058                            if (flagsString.charAt(i) == 'i') {
059                                    flags &= Pattern.CASE_INSENSITIVE;
060                            }
061                            else if (flagsString.charAt(i) == 'm') {
062                                    flags &= Pattern.MULTILINE;
063                            }
064                            else if (flagsString.charAt(i) == 's') {
065                                    flags &= Pattern.DOTALL;
066                            }
067                            else if (flagsString.charAt(i) == 'x') {
068                                    flags &= Pattern.COMMENTS;
069                            }
070                    }
071    
072                    return evaluate(string, regex, flags, navigator);
073            }
074    
075            public static Boolean evaluate(
076                    String string, String regex, int flags, Navigator navigator) {
077    
078                    ThreadLocalCache<Map<String, Pattern>> threadLocalPatterns =
079                            ThreadLocalCacheManager.getThreadLocalCache(
080                                    Lifecycle.ETERNAL, MatchesFunction.class.getName());
081    
082                    Map<String, Pattern> patterns = threadLocalPatterns.get(
083                            _THREAD_LOCAL_PATTERNS_KEY);
084    
085                    if (patterns == null) {
086                            patterns = new HashMap<String, Pattern>();
087                    }
088    
089                    Pattern pattern = patterns.get(regex);
090    
091                    if (pattern != null) {
092                            Matcher matcher = pattern.matcher(string);
093    
094                            return matcher.find();
095                    }
096    
097                    pattern = Pattern.compile(regex, flags);
098    
099                    patterns.put(regex, pattern);
100    
101                    threadLocalPatterns.put(_THREAD_LOCAL_PATTERNS_KEY, patterns);
102    
103                    Matcher matcher = pattern.matcher(string);
104    
105                    return matcher.find();
106            }
107    
108            @Override
109            @SuppressWarnings("rawtypes")
110            public Object call(Context context, List arguments)
111                    throws FunctionCallException {
112    
113                    if (arguments.size() == 2) {
114                            return evaluate(
115                                    arguments.get(0), arguments.get(1), context.getNavigator());
116                    }
117                    else if ((arguments.size() == 3) && (arguments.get(2) != null)) {
118                            return evaluate(
119                                    arguments.get(0), arguments.get(1), arguments.get(2),
120                                    context.getNavigator());
121                    }
122    
123                    throw new FunctionCallException(
124                            "matches() requires two or three arguments");
125            }
126    
127            private static final String _THREAD_LOCAL_PATTERNS_KEY = "matches";
128    
129    }