com.deepoove.poi.render.RenderAPI Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2014-2015 the original author or authors.
 *
 * Licensed 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 com.deepoove.poi.render;

import java.beans.PropertyDescriptor;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.apache.poi.POIXMLException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.deepoove.poi.XWPFTemplate;
import com.deepoove.poi.data.TextRenderData;
import com.deepoove.poi.exception.RenderException;
import com.deepoove.poi.policy.RenderPolicy;
import com.deepoove.poi.template.ElementTemplate;
import com.deepoove.poi.template.run.TextRunTemplate;

/**
 * @author Sayi
 * @version
 * @since 0.0.3
 */
public class RenderAPI {

	private static final Logger logger = LoggerFactory.getLogger(RenderAPI.class);

	public static void debug(XWPFTemplate template, Map datas) {
		List all = template.getElementTemplates();
		Set tagtKeys = new HashSet();

		if (all == null || all.isEmpty()) {
			if (null == datas || datas.isEmpty()) {
				logger.debug("no template gramer find or no render data find");
				return;
			}
		}

		logger.debug("template tag size is :" + (null == all ? 0 : all.size()));
		for (ElementTemplate ele : all) {
			logger.debug("parse the tag:" + ele.getTagName());
			tagtKeys.add(ele.getTagName());
		}

		Set keySet = datas.keySet();
		HashSet copySet = new HashSet(keySet);

		copySet.removeAll(tagtKeys);
		Iterator iterator = copySet.iterator();
		while (iterator.hasNext()) {
			String key = iterator.next();
			logger.warn("cannot find the gramer tag in template:" + key);
		}
	}

	public static void selfRender(XWPFTemplate template) {
		if (null == template) throw new POIXMLException("template is null,should be setted first.");
		List elementTemplates = template.getElementTemplates();
		if (null == elementTemplates || elementTemplates.isEmpty()) return;
		for (ElementTemplate runTemplate : elementTemplates) {
			logger.debug("tag-name:" + runTemplate.getTagName());
			logger.debug(runTemplate.getClass().toString());
			RenderPolicy policy = template.getPolicy(TextRunTemplate.class);
			policy.render(runTemplate, new TextRenderData(runTemplate.getSource()), template);

		}
	}

	public static void render(XWPFTemplate template, Map datas) {
		if (null == template) throw new POIXMLException("template is null,should be setted first.");
		List elementTemplates = template.getElementTemplates();
		if (null == elementTemplates || elementTemplates.isEmpty() || null == datas
				|| datas.isEmpty())
			return;
		for (ElementTemplate runTemplate : elementTemplates) {
			logger.debug("tag-name:" + runTemplate.getTagName());
			logger.debug(runTemplate.getClass().toString());
			RenderPolicy policy = null == template.getPolicy(runTemplate.getTagName())
					? template.getPolicy(runTemplate.getClass())
					: template.getPolicy(runTemplate.getTagName());
			if (null == policy) throw new RenderException(
					"cannot find render policy: [" + runTemplate.getTagName() + "]");
			policy.render(runTemplate, datas.get(runTemplate.getTagName()), template);

		}
	}

	public static void render(XWPFTemplate template, Object dataSrouce) {
		render(template, convert2Map(dataSrouce));
	}

	private static Map convert2Map(Object dataSrouce) {
		Map ret = new HashMap();
		try {
			Class clazz = dataSrouce.getClass();
			while (clazz != Object.class) {
				Field[] fields = clazz.getDeclaredFields();
				for (Field f : fields) {
					PropertyDescriptor pd = new PropertyDescriptor(f.getName(),
							dataSrouce.getClass());
					Method readMethod = pd.getReadMethod();
					Object value = readMethod.invoke(dataSrouce);
					ret.put(f.getName(), value);
				}
				clazz = clazz.getSuperclass();
			}
		} catch (Exception e) {
			logger.error("Convert datasource failed.", e);
			throw new RenderException("Convert datasource failed.");
		}
		return ret;
	}

}