TemplateMaker.java

/**
 * Axelor Business Solutions
 *
 * Copyright (C) 2016 Axelor (<http://axelor.com>).
 *
 * This program is free software: you can redistribute it and/or  modify
 * it under the terms of the GNU Affero General Public License, version 3,
 * as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.axelor.tool.template;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.Date;
import java.util.Locale;
import java.util.Map;

import org.joda.time.LocalDate;
import org.joda.time.LocalDateTime;
import org.joda.time.LocalTime;
import org.stringtemplate.v4.AttributeRenderer;
import org.stringtemplate.v4.DateRenderer;
import org.stringtemplate.v4.Interpreter;
import org.stringtemplate.v4.ModelAdaptor;
import org.stringtemplate.v4.NumberRenderer;
import org.stringtemplate.v4.ST;
import org.stringtemplate.v4.STGroup;
import org.stringtemplate.v4.StringRenderer;
import org.stringtemplate.v4.misc.ObjectModelAdaptor;

import com.axelor.apps.tool.exception.IExceptionMessage;
import com.axelor.auth.AuthUtils;
import com.axelor.db.Model;
import com.axelor.db.mapper.Mapper;
import com.axelor.db.mapper.Property;
import com.axelor.i18n.I18n;
import com.axelor.inject.Beans;
import com.axelor.meta.db.MetaSelectItem;
import com.axelor.meta.db.repo.MetaSelectItemRepository;
import com.axelor.rpc.Resource;
import com.beust.jcommander.internal.Maps;
import com.google.common.base.Charsets;
import com.google.common.base.Preconditions;
import com.google.common.base.Strings;
import com.google.common.io.Files;

public class TemplateMaker {
	
	private Map<String, Object> context;
	private Map<String, Object> localContext;
	
	private String template;
	private STGroup stGroup;
	private Locale locale;
	
	public TemplateMaker(Locale locale, char delimiterStartChar, char delimiterStopChar) {
		this.locale = locale;
		this.stGroup = new STGroup(delimiterStartChar,delimiterStopChar);
		//Custom renderer
		this.stGroup.registerModelAdaptor(Model.class, new ModelFormatRenderer());
		this.stGroup.registerRenderer(LocalDate.class, new LocalDateRenderer());
		this.stGroup.registerRenderer(LocalDateTime.class, new LocalDateTimeRenderer());
		this.stGroup.registerRenderer(LocalTime.class, new LocalTimeRenderer());
		//Default renderer provide by ST
		this.stGroup.registerRenderer(String.class, new StringRenderer());
		this.stGroup.registerRenderer(Number.class, new NumberRenderer());
		this.stGroup.registerRenderer(Date.class, new DateRenderer());
	}
	
	public void setContext(Model model) {
		this.setContext(model, null, null);
	}
	
	public void setContext(Model model, String nameInContext) {
		this.setContext(model, null, nameInContext);
	}
	
	public void setContext(Model model, Map<String, Object> map) {
		this.setContext(model, map, null);
	}
	
	public void setContext(Model model, Map<String, Object> map, String nameInContext) {
		Preconditions.checkNotNull(model);
		this.context = makeContext(nameInContext, model, map);
	}
	
	private Map<String, Object> makeContext(String nameInContext, Model model, Map<String, Object> map) {
		Map<String, Object> _map = Maps.newHashMap();
		
		if(nameInContext != null) {
			_map.put(nameInContext, model);
		}
		else {
			_map.putAll(Resource.toMap(model));
		}
		
		if(map != null) {
			_map.putAll(map);
		}
		
		return _map;
	}
	
	public void setTemplate(String text) {
		this.template = text;
	}
	
	public void setTemplate(File file) throws FileNotFoundException {
		if (!file.isFile()) {
			throw new FileNotFoundException(I18n.get(IExceptionMessage.TEMPLATE_MAKER_1)+": " + file.getName());
		}
		
		String text;
		try {
			text = Files.toString(file, Charsets.UTF_8);
		} catch (IOException e) {
			throw new IllegalArgumentException(e);
		}
		
		this.setTemplate(text);
	}
	
	public void addInContext(String key, Object value) {
		if(localContext == null) {
			localContext = Maps.newHashMap();
		}
		localContext.put(key, value);
	}
	
	public void addInContext(Map<String, Object> map) {
		if(localContext == null) {
			localContext = Maps.newHashMap();
		}
		localContext.putAll(map);
	}

	public Class<?> getBeanClass(Model model) {
		return model.getClass();
	}
	
	public String make() {
		if (Strings.isNullOrEmpty(this.template)) {
			throw new IllegalArgumentException(I18n.get(IExceptionMessage.TEMPLATE_MAKER_2));
		}
		
		ST st = new ST(stGroup, template);
		
		Map<String, Object> _map = Maps.newHashMap();
		if(localContext != null && !localContext.isEmpty()) {
			_map.putAll(localContext);
		}
		_map.putAll(context);
		
		//Internal context
		_map.put("__user__", AuthUtils.getUser());
		_map.put("__date__", new LocalDate());
		_map.put("__time__", new LocalTime());
		_map.put("__datetime__", new LocalDateTime());
		
		for (String key : _map.keySet()) {
			st.add(key, _map.get(key));
		}
		
		return _make(st);
	}
	
	private String _make(ST st) {
		return st.render(locale);
	}
	
	class ModelFormatRenderer implements ModelAdaptor {
		
		private Property getProperty(Class<?> beanClass, String name) {
			return Mapper.of(beanClass).getProperty(name);
		}
		
		private String getSelectionValue(Property prop, Object o, Object value) {
			if(value == null) {
				return "";
			}
			MetaSelectItem item = Beans.get(MetaSelectItemRepository.class).all()
					.filter("self.select.name = ?1 AND self.value = ?2",
							prop.getSelection(), value).fetchOne();
			
			if (item != null) {
				return item.getTitle();
			}
			return value == null ? "" : value.toString();
		}

		@Override
		public Object getProperty(Interpreter interp, ST self, Object o, Object property, String propertyName) {
			Property prop = this.getProperty(o.getClass(), (String) property);
			ModelAdaptor adap = self.groupThatCreatedThisInstance.getModelAdaptor(ObjectModelAdaptor.class);
			
			if (prop == null || Strings.isNullOrEmpty(prop.getSelection())) {
				return adap.getProperty(interp, self, o, property, propertyName);
			}

			Object value = adap.getProperty(interp, self, o, property, propertyName);
			return getSelectionValue(prop, o, value);
		}
	}
	
	class LocalDateRenderer implements AttributeRenderer {

		@Override
		public String toString(Object o, String formatString, Locale locale) {
	        if ( formatString==null ) return o.toString();
	        LocalDate ld = (LocalDate) o;
	        return ld.toString(formatString);
		}
	}
	
	class LocalDateTimeRenderer implements AttributeRenderer {

		@Override
		public String toString(Object o, String formatString, Locale locale) {
	        if ( formatString==null ) return o.toString();
	        LocalDateTime ld = (LocalDateTime) o;
	        return ld.toString(formatString);
		}
	}
	
	class LocalTimeRenderer implements AttributeRenderer {

		@Override
		public String toString(Object o, String formatString, Locale locale) {
	        if ( formatString==null ) return o.toString();
	        LocalTime ld = (LocalTime) o;
	        return ld.toString(formatString);
		}
	}
}