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.portlet.dynamicdatamapping.template;
016    
017    import com.liferay.portal.kernel.io.unsync.UnsyncStringWriter;
018    import com.liferay.portal.kernel.template.Template;
019    import com.liferay.portal.kernel.template.TemplateConstants;
020    import com.liferay.portal.kernel.template.TemplateManagerUtil;
021    import com.liferay.portal.kernel.template.TemplateResource;
022    import com.liferay.portal.kernel.template.TemplateVariableCodeHandler;
023    import com.liferay.portal.kernel.template.TemplateVariableDefinition;
024    import com.liferay.portal.kernel.template.URLTemplateResource;
025    import com.liferay.portal.kernel.util.StringBundler;
026    import com.liferay.portal.kernel.util.StringPool;
027    import com.liferay.portal.kernel.util.StringUtil;
028    
029    import java.io.Writer;
030    
031    import java.net.URL;
032    
033    /**
034     * @author Marcellus Tavares
035     */
036    public class DDMTemplateVariableCodeHandler
037            implements TemplateVariableCodeHandler {
038    
039            public DDMTemplateVariableCodeHandler(String templatePath) {
040                    _templatePath = templatePath;
041            }
042    
043            @Override
044            public String[] generate(
045                            TemplateVariableDefinition templateVariableDefinition,
046                            String language)
047                    throws Exception {
048    
049                    String resourceName = getResourceName(
050                            templateVariableDefinition.getDataType());
051    
052                    Template template = getTemplate(resourceName);
053    
054                    String content = getTemplateContent(
055                            template, templateVariableDefinition, language);
056    
057                    String[] lines = getContentLines(content);
058    
059                    String[] dataContentArray = getDataContentArray(lines);
060    
061                    if (!templateVariableDefinition.isRepeatable()) {
062                            return dataContentArray;
063                    }
064    
065                    return handleRepeatableField(
066                            templateVariableDefinition, language, dataContentArray);
067            }
068    
069            protected String[] getContentLines(String content) {
070                    String[] lines = StringUtil.splitLines(content);
071    
072                    for (int i = 0; i < lines.length; i++) {
073                            lines[i] = StringUtil.trim(lines[i]);
074                    }
075    
076                    return lines;
077            }
078    
079            protected String[] getDataContentArray(String[] lines) {
080                    String[] dataContentArray = new String[] {
081                            StringPool.BLANK, StringPool.BLANK, StringPool.BLANK};
082    
083                    for (int i = 0; i < lines.length; i++) {
084                            dataContentArray[i] = lines[i];
085                    }
086    
087                    return dataContentArray;
088            }
089    
090            protected String getResourceName(String dataType) {
091                    if (isCommonResource(dataType)) {
092                            dataType = "common";
093                    }
094    
095                    StringBundler sb = new StringBundler(3);
096    
097                    sb.append(_templatePath);
098                    sb.append(dataType);
099                    sb.append(".ftl");
100    
101                    return sb.toString();
102            }
103    
104            protected Template getTemplate(String resource) throws Exception {
105                    TemplateResource templateResource = getTemplateResource(resource);
106    
107                    return TemplateManagerUtil.getTemplate(
108                            TemplateConstants.LANG_TYPE_FTL, templateResource, false);
109            }
110    
111            protected String getTemplateContent(
112                            Template template,
113                            TemplateVariableDefinition templateVariableDefinition,
114                            String language)
115                    throws Exception {
116    
117                    prepareTemplate(template, templateVariableDefinition, language);
118    
119                    Writer writer = new UnsyncStringWriter();
120    
121                    template.processTemplate(writer);
122    
123                    return writer.toString();
124            }
125    
126            protected TemplateResource getTemplateResource(String resource) {
127                    Class<?> clazz = getClass();
128    
129                    ClassLoader classLoader = clazz.getClassLoader();
130    
131                    URL url = classLoader.getResource(resource);
132    
133                    return new URLTemplateResource(resource, url);
134            }
135    
136            protected String[] handleRepeatableField(
137                            TemplateVariableDefinition templateVariableDefinition,
138                            String language, String[] dataContentArray)
139                    throws Exception {
140    
141                    Template template = getTemplate(_templatePath + "repeatable.ftl");
142    
143                    String content = getTemplateContent(
144                            template, templateVariableDefinition, language);
145    
146                    String[] lines = getContentLines(content);
147    
148                    String tempDataContent0 = dataContentArray[0];
149    
150                    dataContentArray[0] =
151                            lines[0] + StringPool.NEW_LINE + StringPool.TAB + lines[1];
152                    dataContentArray[1] =
153                            tempDataContent0 + dataContentArray[1] + dataContentArray[2];
154                    dataContentArray[2] = lines[2] + StringPool.NEW_LINE + lines[3];
155    
156                    return dataContentArray;
157            }
158    
159            protected boolean isCommonResource(String dataType) {
160                    if (dataType.equals("boolean") || dataType.equals("date") ||
161                            dataType.equals("document-library") || dataType.equals("image") ||
162                            dataType.equals("link-to-page")) {
163    
164                            return false;
165                    }
166    
167                    return true;
168            }
169    
170            protected void prepareTemplate(
171                    Template template,
172                    TemplateVariableDefinition templateVariableDefinition,
173                    String language) {
174    
175                    template.put("dataType", templateVariableDefinition.getDataType());
176                    template.put("help", templateVariableDefinition.getHelp());
177                    template.put("label", templateVariableDefinition.getLabel());
178                    template.put("language", language);
179                    template.put("name", templateVariableDefinition.getName());
180                    template.put("repeatable", templateVariableDefinition.isRepeatable());
181            }
182    
183            private String _templatePath;
184    
185    }