org.tinygroup.weixin.impl.WeiXinManagerDefault Maven / Gradle / Ivy

Go to download
/**
 * Copyright (c) 2012-2016, www.tinygroup.org ([email protected]).
 *
 *  Licensed under the GPL, Version 3.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.gnu.org/licenses/gpl.html
 *
 *  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 org.tinygroup.weixin.impl;

import org.tinygroup.template.TemplateContext;
import org.tinygroup.template.TemplateException;
import org.tinygroup.template.TemplateRender;
import org.tinygroup.template.impl.TemplateContextDefault;
import org.tinygroup.template.impl.TemplateRenderDefault;
import org.tinygroup.weixin.WeiXinContext;
import org.tinygroup.weixin.WeiXinManager;
import org.tinygroup.weixin.common.UrlConfig;
import org.tinygroup.weixin.common.UrlConfigs;
import org.tinygroup.weixin.exception.WeiXinException;

import java.util.HashMap;
import java.util.Map;

public class WeiXinManagerDefault implements WeiXinManager {

    private Map urlConfigMap = new HashMap();
    private TemplateRender templateRender = new TemplateRenderDefault();

    public UrlConfig getUrl(String key) {
        return urlConfigMap.get(key);
    }

    public void addUrlConfig(UrlConfig urlConfig) {
        urlConfigMap.put(urlConfig.getKey(), urlConfig);
    }

    public void addUrlConfigs(UrlConfigs urlConfigs) {
        for (UrlConfig urlConfig : urlConfigs.getUrlConfigs()) {
            addUrlConfig(urlConfig);
        }
    }

    public void removeUrlConfig(UrlConfig urlConfig) {
        urlConfigMap.remove(urlConfig.getKey());
    }

    public void removeUrlConfigs(UrlConfigs urlConfigs) {
        for (UrlConfig urlConfig : urlConfigs.getUrlConfigs()) {
            removeUrlConfig(urlConfig);
        }
    }

    public String renderUrl(String key, WeiXinContext context) {
        UrlConfig urlConfig = getUrl(key);
        TemplateContext templateContext = new TemplateContextDefault();
        templateContext.setParent(context);

        Object input = context.getInput();
        // 默认策略：将对象以key值为名称放到上下文环境
        templateContext.put(urlConfig.getKey(), input);

        String result;
        try {
            result = templateRender.renderTemplateContent(urlConfig.getUrl(), templateContext);
        } catch (TemplateException e) {
            throw new WeiXinException(e);
        } finally {
            templateContext.setParent(null);
        }

        return result;

    }
}