com.wavemaker.runtime.WMAppContext.java Source code

Introduction

Here is the source code for com.wavemaker.runtime.WMAppContext.java

Source

/*
 *  Copyright (C) 2012-2013 CloudJee, Inc. All rights reserved.
 *
 *  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.wavemaker.runtime;

import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Properties;

import javax.servlet.ServletContext;
import javax.servlet.ServletContextEvent;

import org.apache.commons.io.IOUtils;
import org.cloudfoundry.runtime.env.CloudEnvironment;
import org.json.JSONObject;
import org.springframework.core.io.Resource;
import org.springframework.web.context.support.ServletContextResource;

import com.wavemaker.common.CommonConstants;
import com.wavemaker.common.WMRuntimeException;
import com.wavemaker.runtime.data.util.DataServiceConstants;

/**
 * This singleton class is to store any properties in the scope of the application context.
 * 
 * @author Seung Lee
 * @author Jeremy Grelle
 */
public class WMAppContext {

    private static WMAppContext instance;

    private int defaultTenantID = DataServiceConstants.DEFAULT_TENANT_ID;

    private String tenantFieldName = DataServiceConstants.DEFAULT_TENANT_FIELD;

    private String tenantColumnName = "";

    private HashMap<String, ProjectInfo> hm;

    private ServletContext context;

    private String appName;

    private JSONObject typesObj;

    private final CloudEnvironment cloudEnvironment = new CloudEnvironment();

    private WMAppContext(ServletContextEvent event) {
        this.context = event.getServletContext();
        this.appName = this.context.getServletContextName();
        if (this.appName == null) {
            this.appName = "Project Name";
        }

        // In Studio, the tenant field and def tenant ID is injected by ProjectManager when a project opens
        if (!this.appName.equals(DataServiceConstants.WAVEMAKER_STUDIO)) {
            // Store types.js contents in memory
            try {
                Resource typesResource = new ServletContextResource(this.context, "/types.js");
                String s = IOUtils.toString(typesResource.getInputStream());
                this.typesObj = new JSONObject(s.substring(11));
            } catch (Exception e) {
                e.printStackTrace();
                return;
            }

            // Set up multi-tenant info
            Resource appPropsResource = null;
            try {
                appPropsResource = new ServletContextResource(this.context,
                        "/WEB-INF/" + CommonConstants.APP_PROPERTY_FILE);
            } catch (WMRuntimeException re) {
                return;
            } catch (Exception e) {
                e.printStackTrace();
                return;
            }

            if (!appPropsResource.exists()) {
                return;
            }

            Properties props;

            try {
                props = new Properties();
                InputStream is = appPropsResource.getInputStream();
                props.load(is);
                is.close();
            } catch (IOException ioe) {
                ioe.printStackTrace();
                return;
            }

            this.tenantFieldName = props.getProperty(DataServiceConstants.TENANT_FIELD_PROPERTY_NAME);
            this.tenantColumnName = props.getProperty(DataServiceConstants.TENANT_COLUMN_PROPERTY_NAME);
            this.defaultTenantID = Integer
                    .parseInt(props.getProperty(DataServiceConstants.DEFAULT_TENANT_ID_PROPERTY_NAME));
        }
    }

    public static synchronized WMAppContext getInstance(ServletContextEvent event) {
        if (instance == null) {
            instance = new WMAppContext(event);
        }

        return instance;
    }

    public static synchronized WMAppContext getInstance() {
        return instance;
    }

    @Override
    public Object clone() throws CloneNotSupportedException {
        throw new CloneNotSupportedException();
    }

    public int getDefaultTenantID() {
        if (this.appName.equals(DataServiceConstants.WAVEMAKER_STUDIO)) {
            String p = (String) RuntimeAccess.getInstance().getSession()
                    .getAttribute(DataServiceConstants.CURRENT_PROJECT_NAME);
            if (this.hm == null) {
                return DataServiceConstants.DEFAULT_TENANT_ID;
            } else {
                ProjectInfo pi = this.hm.get(p);
                if (pi == null) {
                    return DataServiceConstants.DEFAULT_TENANT_ID;
                } else {
                    return pi.defTenantId;
                }
            }
        } else {
            return this.defaultTenantID;
        }
    }

    public String getTenantFieldName() {
        if (this.appName.equals(DataServiceConstants.WAVEMAKER_STUDIO)) {
            String p = (String) RuntimeAccess.getInstance().getSession()
                    .getAttribute(DataServiceConstants.CURRENT_PROJECT_NAME);
            if (this.hm == null) {
                return DataServiceConstants.DEFAULT_TENANT_FIELD;
            } else {
                ProjectInfo pi = this.hm.get(p);
                if (pi == null) {
                    return DataServiceConstants.DEFAULT_TENANT_FIELD;
                } else {
                    return pi.tenantFldName;
                }
            }
        } else {
            return this.tenantFieldName;
        }
    }

    public String getTenantColumnName() {
        if (this.appName.equals(DataServiceConstants.WAVEMAKER_STUDIO)) {
            String p = (String) RuntimeAccess.getInstance().getSession()
                    .getAttribute(DataServiceConstants.CURRENT_PROJECT_NAME);
            if (this.hm == null) {
                return "";
            } else {
                ProjectInfo pi = this.hm.get(p);
                if (pi == null) {
                    return "";
                } else {
                    return pi.tenantColName;
                }
            }
        } else {
            return this.tenantColumnName;
        }
    }

    public void setTenantInfoForProj(String projName, String val1, int val2, String val3) {
        if (this.hm == null) {
            this.hm = new HashMap<String, ProjectInfo>();
        }
        ProjectInfo pi = new ProjectInfo(val1, val2, val3);
        this.hm.put(projName, pi);
    }

    public boolean isMultiTenant() {
        String tf;
        boolean multiTenancy;

        if (this.appName.equals(DataServiceConstants.WAVEMAKER_STUDIO)) {
            String p = (String) RuntimeAccess.getInstance().getSession()
                    .getAttribute(DataServiceConstants.CURRENT_PROJECT_NAME);

            if (this.hm == null) {
                tf = DataServiceConstants.DEFAULT_TENANT_FIELD;
            } else {
                ProjectInfo pi = this.hm.get(p);
                if (pi == null) {
                    tf = DataServiceConstants.DEFAULT_TENANT_FIELD;
                } else {
                    tf = pi.tenantFldName;
                }
            }
        } else {
            tf = this.tenantFieldName;
        }

        if (tf.equalsIgnoreCase(DataServiceConstants.DEFAULT_TENANT_FIELD) || tf.length() == 0 || tf == null) {
            multiTenancy = false;
        } else {
            multiTenancy = true;
        }

        return multiTenancy;
    }

    public boolean isCloudFoundry() {
        return this.cloudEnvironment.getInstanceInfo() != null;
    }

    public CloudEnvironment getCloudEnvironment() {
        return this.cloudEnvironment;
    }

    public String getAppName() {
        return this.appName;
    }

    public String getAppContextRoot() {
        return this.context.getRealPath("");
    }

    public JSONObject getTypesObject() {
        return this.typesObj;
    }

    public void setTypesObject(JSONObject val) {
        this.typesObj = val;
    }

    class ProjectInfo {

        String tenantFldName;

        int defTenantId;

        String tenantColName;

        private ProjectInfo(String val1, int val2, String val3) {
            this.tenantFldName = val1;
            this.defTenantId = val2;
            this.tenantColName = val3;
        }
    }
}