001/* 002 * (C) Copyright 2006-2011 Nuxeo SA (http://nuxeo.com/) and others. 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 * 016 * Contributors: 017 * bstefanescu 018 * 019 * $Id$ 020 */ 021 022package org.nuxeo.ecm.platform.rendering.fm.extensions; 023 024import java.util.List; 025 026import org.nuxeo.ecm.core.api.IdRef; 027import org.nuxeo.ecm.core.api.PathRef; 028 029import freemarker.template.SimpleScalar; 030import freemarker.template.TemplateMethodModelEx; 031import freemarker.template.TemplateModelException; 032 033/** 034 * @author <a href="mailto:bs@nuxeo.com">Bogdan Stefanescu</a> 035 */ 036public class DocRefMethod implements TemplateMethodModelEx { 037 038 @Override 039 public Object exec(@SuppressWarnings("rawtypes") List arguments) throws TemplateModelException { 040 if (arguments.size() != 1) { 041 throw new TemplateModelException("Invalid number of arguments for docRef(id) method"); 042 } 043 String value = null; 044 SimpleScalar scalar = (SimpleScalar) arguments.get(0); 045 if (scalar != null) { 046 value = scalar.getAsString(); 047 } else { 048 throw new TemplateModelException("the argument is not defined"); 049 } 050 051 if (value.startsWith("/")) { 052 return new PathRef(value); 053 } else { 054 return new IdRef(value); 055 } 056 } 057 058}