com.clickntap.smart.SmartConfServlet Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Stripecube Show documentation
Stripecube is an open source Java framework for Web Applications
There is a newer version: 1.30
package com.clickntap.smart;

import com.clickntap.utils.ConstUtils;
import org.apache.commons.io.FileUtils;
import org.apache.log4j.PropertyConfigurator;

import javax.servlet.*;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.Enumeration;
import java.util.Properties;

public class SmartConfServlet implements Servlet {

    public void init(ServletConfig conf) throws ServletException {
        try {
            File root = new File(conf.getServletContext().getRealPath(""));
            File webInf = new File(conf.getServletContext().getRealPath("WEB-INF"));
            File dir = new File(webInf.getCanonicalPath() + getPathSeparator() + "conf");
            File out = new File(dir.getCanonicalPath() + getPathSeparator() + "auto");
            out.mkdirs();
            Properties envProperties = new Properties();
            String envFile = conf.getInitParameter("smartEnv");
            if (envFile != null) {
                if (root.getAbsolutePath().contains("stg")) {
                    envFile = envFile.replace("-env", "-stg-env");
                }
                if (root.getAbsolutePath().contains("pre")) {
                    envFile = envFile.replace("-env", "-pre-env");
                }
                if (root.getAbsolutePath().contains("demo")) {
                    envFile = envFile.replace("-env", "-demo-env");
                }
                if (root.getAbsolutePath().contains("dev")) {
                    envFile = envFile.replace("-env", "-dev-env");
                }
                File file = new File(envFile);
                if (!file.exists())
                    file = new File(root.getCanonicalPath() + ConstUtils.SLASH + envFile);
                if (!file.exists())
                    file = new File(ConstUtils.SLASH + envFile);
                if (file.exists()) {
                    FileInputStream in = new FileInputStream(file);
                    envProperties.load(in);
                    in.close();
                    file = new File(file.getCanonicalPath().replace("-env", "-log4j"));
                    if (file.exists()) {
                        PropertyConfigurator.configure(file.getCanonicalPath().replace("-env", "-log4j"));
                    } else {
                        System.out.println("LOG4J NOT FOUND: " + file.getCanonicalPath().replace("-env", "-log4j"));
                    }
                } else {
                    System.out.println("ENV NOT FOUND: " + file.getCanonicalPath());
                }
            }
            FileUtils.cleanDirectory(out);
            if (dir.exists() && dir.isDirectory()) {
                for (File file : dir.listFiles()) {
                    if (file.isFile()) {
                        String data = FileUtils.readFileToString(file, ConstUtils.UTF_8);
                        Enumeration enu = envProperties.keys();
                        while (enu.hasMoreElements()) {
                            String key = enu.nextElement().toString();
                            data = data.replaceAll("\\[" + key + "\\]", envProperties.getProperty(key));
                        }
                        data = data.replaceAll("\\[WEB-INF\\]", webInf.getCanonicalPath().replace("\\", "/"));
                        data = data.replaceAll("\\[ROOT\\]", root.getCanonicalPath().replace("\\", "/"));
                        out.mkdirs();
                        FileUtils.writeStringToFile(new File(out.getAbsolutePath() + getPathSeparator() + file.getName()), data, ConstUtils.UTF_8);
                    }
                }
            }
        } catch (Exception e) {
            throw new ServletException(e);
        }
    }

    private String getPathSeparator() {
        return System.getProperty("file.separator");
    }

    public void destroy() {
    }

    public ServletConfig getServletConfig() {
        return null;
    }

    public String getServletInfo() {
        return null;
    }

    public void service(ServletRequest arg0, ServletResponse arg1) throws ServletException, IOException {

    }

}