com.codequicker.quick.templates.state.EngineContext.java Source code

Introduction

Here is the source code for com.codequicker.quick.templates.state.EngineContext.java

Source

/*
 * Copyright 2016 Rajesh Putta
       
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.codequicker.quick.templates.state;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.util.HashMap;
import java.util.Map;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;

import org.xml.sax.SAXException;

import com.codequicker.quick.templates.exceptions.TemplateRuntimeException;
import com.codequicker.quick.templates.utils.TemplateUtil;
import com.google.gson.JsonElement;
import com.google.gson.JsonParser;

/*
* @author Rajesh Putta
*/
public class EngineContext {

    private Map<String, Object> context = new HashMap<String, Object>();

    private Map<String, Object> variableContext = new HashMap<String, Object>();

    private JsonElement jsonObject = null;

    private Object xmlRoot = null;

    public void set(String key, Object value) {
        if (key == null || key.trim().equals(""))
            throw new TemplateRuntimeException("key cannot be null or empty...");

        this.context.put(key, value);
    }

    public void setXml(String key, String payload) {
        if (TemplateUtil.isNullOrEmpty(payload))
            return;

        DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
        //        factory.setNamespaceAware(true);
        DocumentBuilder builder;

        try {
            builder = factory.newDocumentBuilder();
            Object xmlRoot = builder.parse(new ByteArrayInputStream(payload.getBytes("UTF8"))).getDocumentElement();

            context.put(key, xmlRoot);

        } catch (ParserConfigurationException e) {
            e.printStackTrace();
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        } catch (SAXException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public void setXmlRoot(Object xmlRoot) {
        this.xmlRoot = xmlRoot;
    }

    public Object getXmlRoot() {
        return xmlRoot;
    }

    public void setJson(String key, String payload) {
        if (TemplateUtil.isNullOrEmpty(payload) || TemplateUtil.isNullOrEmpty(key))
            return;

        JsonParser jsonParser = new JsonParser();
        JsonElement jsonObject = jsonParser.parse(payload).getAsJsonObject();

        context.put(key, jsonObject);
    }

    public void setJsonObject(JsonElement jsonObject) {
        this.jsonObject = jsonObject;
    }

    public JsonElement getJsonObject() {
        return jsonObject;
    }

    @SuppressWarnings({ "rawtypes", "unchecked" })
    public void setAll(Map other) {
        this.context.putAll(other);
    }

    public Object get(String key) {
        return this.context.get(key);
    }

    public void setVariable(String variable, Object reference) {
        this.variableContext.put(variable, reference);
    }

    public Object getVariable(String variable) {
        return this.variableContext.get(variable);
    }

    public void removeVariable(String variable) {
        this.variableContext.remove(variable);
    }

    public Map<String, Object> getVariableContext() {
        return variableContext;
    }

    public Map<String, Object> getContext() {
        return context;
    }

    @Override
    public int hashCode() {
        return this.context.hashCode();
    }

    @Override
    public String toString() {
        return this.context.toString();
    }

}