net.awired.visuwall.core.application.common.ApplicationHelper.java Source code

Introduction

Here is the source code for net.awired.visuwall.core.application.common.ApplicationHelper.java

Source

/**
 *     Copyright (C) 2010 Julien SMADJA <julien dot smadja at gmail dot com> - Arnaud LEMAIRE <alemaire at norad dot fr>
 *
 *     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 net.awired.visuwall.core.application.common;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.URL;
import java.util.Enumeration;
import java.util.jar.Manifest;
import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import net.awired.visuwall.core.application.enumeration.LogLevelEnum;
import org.slf4j.LoggerFactory;
import ch.qos.logback.classic.Level;
import ch.qos.logback.classic.LoggerContext;
import ch.qos.logback.classic.joran.JoranConfigurator;
import ch.qos.logback.core.joran.spi.JoranException;
import ch.qos.logback.core.util.StatusPrinter;
import com.google.common.io.CharStreams;

public class ApplicationHelper {

    private static final org.slf4j.Logger LOG = LoggerFactory.getLogger(ApplicationHelper.class);

    public static final String LOG_LVL_KEY = "VISUWALL_LOG";
    public static final String HOME_KEY = "VISUWALL_HOME";
    public static final String MANIFEST_VERSION_KEY = "VisuwallVersion";

    public static String findVersion(InputStream manifestIn) {
        // runnnable war
        try {
            Enumeration<URL> manifests = ApplicationHelper.class.getClassLoader()
                    .getResources("META-INF/MANIFEST.MF");
            while (manifests.hasMoreElements()) {
                URL res = manifests.nextElement();
                Manifest manifest = new Manifest(res.openStream());
                String versionManifest = manifest.getMainAttributes().getValue(MANIFEST_VERSION_KEY);
                if (versionManifest != null) {
                    return versionManifest;
                }
            }
        } catch (IOException e) {
        }

        // tomcat like
        try {
            Manifest manifest = new Manifest(manifestIn);
            String versionManifest = manifest.getMainAttributes().getValue(MANIFEST_VERSION_KEY);
            if (versionManifest != null) {
                return versionManifest;
            }
        } catch (IOException e) {
        }
        return "Unknow Version";
    }

    public static Level findLogLvl() {
        String logLvl = System.getProperty(LOG_LVL_KEY);
        LogLevelEnum logLevelEnum = null;
        if (logLvl != null) {
            logLevelEnum = LogLevelEnum.valueOf(logLvl);
        }
        if (logLevelEnum != null) {
            return logLevelEnum.getLevel();
        }
        return null;
    }

    public static String findHomeDir() {

        // TODO redirect to tmp or $TMP or check at start in cli if home is not
        // writable
        // if (home == null) {
        // home = "/tmp";
        // }

        // check JNDI for the home directory first
        try {
            InitialContext iniCtxt = new InitialContext();
            Context env = (Context) iniCtxt.lookup("java:comp/env");
            String value = (String) env.lookup(HOME_KEY);
            if (value != null && value.trim().length() > 0) {
                return value.trim();
            }

            value = (String) iniCtxt.lookup(HOME_KEY);
            if (value != null && value.trim().length() > 0) {
                return value.trim();
            }
        } catch (NamingException e) {
            // ignore
        }

        // finally check the system property
        String sysProp = System.getProperty(HOME_KEY);
        if (sysProp != null) {
            return sysProp.trim();
        }

        // look at the env var next
        try {
            String env = System.getenv(HOME_KEY);
            if (env != null) {
                return env.trim();
            }
        } catch (Throwable _) {
            // when this code runs on JDK1.4, this method fails
            // fall through to the next method
        }

        // if for some reason we can't put it within the webapp, use home
        // directory.
        return System.getProperty("user.home") + "/.visuwall";
    }

    public static void changeLogLvl() {
        Level loglvl = ApplicationHelper.findLogLvl();
        try {
            InputStream logConfStream = ApplicationHelper.class.getResourceAsStream("/visuwall-logback.xml");
            String logConfString = CharStreams.toString(new InputStreamReader(logConfStream));
            if (loglvl != null) {
                // TODO replace with a better regexp to replace all tags like that <logger name="net.awired" level="DEBUG" />
                // TODO or replace by a xml parser or replace by a deep logback communication to set log lvl
                logConfString = logConfString.replace("DEBUG", loglvl.toString());
            }
            LoggerContext lc = (LoggerContext) LoggerFactory.getILoggerFactory();
            try {
                JoranConfigurator configurator = new JoranConfigurator();
                configurator.setContext(lc);
                lc.reset();
                configurator.doConfigure(new ByteArrayInputStream(logConfString.getBytes()));
            } catch (JoranException je) {
                je.printStackTrace();
            }
            StatusPrinter.printInCaseOfErrorsOrWarnings(lc);
        } catch (IOException e) {
            LOG.error("Can not change application log level", e);
        }

        // don't change root lvl as is may put hibernate or jetty in debug
        //        Logger root = (Logger) LoggerFactory.getLogger(Logger.ROOT_LOGGER_NAME);
        //        root.setLevel(loglvl);
    }
}