| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| |
| package freemarker.core; |
| |
| import java.util.ArrayList; |
| import java.util.HashMap; |
| import java.util.LinkedHashMap; |
| import java.util.ListIterator; |
| |
| import freemarker.template.SimpleSequence; |
| import freemarker.template.TemplateCollectionModel; |
| import freemarker.template.TemplateException; |
| import freemarker.template.TemplateHashModelEx; |
| import freemarker.template.TemplateModel; |
| import freemarker.template._TemplateAPI; |
| |
| final class HashLiteral extends Expression { |
| |
| private final ArrayList keys, values; |
| private final int size; |
| |
| HashLiteral(ArrayList/*<Expression>*/ keys, ArrayList/*<Expression>*/ values) { |
| this.keys = keys; |
| this.values = values; |
| this.size = keys.size(); |
| keys.trimToSize(); |
| values.trimToSize(); |
| } |
| |
| @Override |
| TemplateModel _eval(Environment env) throws TemplateException { |
| return new SequenceHash(env); |
| } |
| |
| @Override |
| public String getCanonicalForm() { |
| StringBuilder buf = new StringBuilder("{"); |
| for (int i = 0; i < size; i++) { |
| Expression key = (Expression) keys.get(i); |
| Expression value = (Expression) values.get(i); |
| buf.append(key.getCanonicalForm()); |
| buf.append(": "); |
| buf.append(value.getCanonicalForm()); |
| if (i != size - 1) { |
| buf.append(", "); |
| } |
| } |
| buf.append("}"); |
| return buf.toString(); |
| } |
| |
| @Override |
| String getNodeTypeSymbol() { |
| return "{...}"; |
| } |
| |
| @Override |
| boolean isLiteral() { |
| if (constantValue != null) { |
| return true; |
| } |
| for (int i = 0; i < size; i++) { |
| Expression key = (Expression) keys.get(i); |
| Expression value = (Expression) values.get(i); |
| if (!key.isLiteral() || !value.isLiteral()) { |
| return false; |
| } |
| } |
| return true; |
| } |
| |
| |
| @Override |
| protected Expression deepCloneWithIdentifierReplaced_inner( |
| String replacedIdentifier, Expression replacement, ReplacemenetState replacementState) { |
| ArrayList clonedKeys = (ArrayList) keys.clone(); |
| for (ListIterator iter = clonedKeys.listIterator(); iter.hasNext(); ) { |
| iter.set(((Expression) iter.next()).deepCloneWithIdentifierReplaced( |
| replacedIdentifier, replacement, replacementState)); |
| } |
| ArrayList clonedValues = (ArrayList) values.clone(); |
| for (ListIterator iter = clonedValues.listIterator(); iter.hasNext(); ) { |
| iter.set(((Expression) iter.next()).deepCloneWithIdentifierReplaced( |
| replacedIdentifier, replacement, replacementState)); |
| } |
| return new HashLiteral(clonedKeys, clonedValues); |
| } |
| |
| private class SequenceHash implements TemplateHashModelEx { |
| |
| private HashMap map; // maps keys to integer offset |
| private TemplateCollectionModel keyCollection, valueCollection; // ordered lists of keys and values |
| |
| SequenceHash(Environment env) throws TemplateException { |
| if (_TemplateAPI.getTemplateLanguageVersionAsInt(HashLiteral.this) >= _TemplateAPI.VERSION_INT_2_3_21) { |
| map = new LinkedHashMap(); |
| for (int i = 0; i < size; i++) { |
| Expression keyExp = (Expression) keys.get(i); |
| Expression valExp = (Expression) values.get(i); |
| String key = keyExp.evalAndCoerceToPlainText(env); |
| TemplateModel value = valExp.eval(env); |
| if (env == null || !env.isClassicCompatible()) { |
| valExp.assertNonNull(value, env); |
| } |
| map.put(key, value); |
| } |
| } else { |
| // Legacy hash literal, where repeated keys were kept when doing ?values or ?keys, yet overwritten when |
| // doing hash[key]. |
| map = new HashMap(); |
| ArrayList keyList = new ArrayList(size); |
| ArrayList valueList = new ArrayList(size); |
| for (int i = 0; i < size; i++) { |
| Expression keyExp = (Expression) keys.get(i); |
| Expression valExp = (Expression) values.get(i); |
| String key = keyExp.evalAndCoerceToPlainText(env); |
| TemplateModel value = valExp.eval(env); |
| if (env == null || !env.isClassicCompatible()) { |
| valExp.assertNonNull(value, env); |
| } |
| map.put(key, value); |
| keyList.add(key); |
| valueList.add(value); |
| } |
| keyCollection = new CollectionAndSequence(new SimpleSequence(keyList)); |
| valueCollection = new CollectionAndSequence(new SimpleSequence(valueList)); |
| } |
| } |
| |
| public int size() { |
| return size; |
| } |
| |
| public TemplateCollectionModel keys() { |
| if (keyCollection == null) { |
| // This can only happen when IcI >= 2.3.21, an the map is a LinkedHashMap. |
| keyCollection = new CollectionAndSequence(new SimpleSequence(map.keySet())); |
| } |
| return keyCollection; |
| } |
| |
| public TemplateCollectionModel values() { |
| if (valueCollection == null) { |
| // This can only happen when IcI >= 2.3.21, an the map is a LinkedHashMap. |
| valueCollection = new CollectionAndSequence(new SimpleSequence(map.values())); |
| } |
| return valueCollection; |
| } |
| |
| public TemplateModel get(String key) { |
| return (TemplateModel) map.get(key); |
| } |
| |
| public boolean isEmpty() { |
| return size == 0; |
| } |
| |
| @Override |
| public String toString() { |
| return getCanonicalForm(); |
| } |
| |
| } |
| |
| @Override |
| int getParameterCount() { |
| return size * 2; |
| } |
| |
| @Override |
| Object getParameterValue(int idx) { |
| checkIndex(idx); |
| return idx % 2 == 0 ? keys.get(idx / 2) : values.get(idx / 2); |
| } |
| |
| @Override |
| ParameterRole getParameterRole(int idx) { |
| checkIndex(idx); |
| return idx % 2 == 0 ? ParameterRole.ITEM_KEY : ParameterRole.ITEM_VALUE; |
| } |
| |
| private void checkIndex(int idx) { |
| if (idx >= size * 2) { |
| throw new IndexOutOfBoundsException(); |
| } |
| } |
| |
| } |