/*
 * Copyright 2014, The Sporting Exchange Limited
 *
 * 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.betfair.cougar.core.impl.jmx;

import com.betfair.cougar.core.api.jmx.JMXHttpParser;
import com.betfair.cougar.core.api.jmx.JMXHttpParserReader;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.sun.jdmk.comm.AuthInfo;
import org.springframework.beans.factory.InitializingBean;

import javax.management.MBeanServer;
import javax.management.ObjectName;
import java.util.logging.Level;

public class HtmlAdaptor implements InitializingBean, JMXHttpParserReader {
	private static final Logger LOGGER = LoggerFactory.getLogger(HtmlAdaptor.class);
    private static final String PARSER_NAME = "CoUGAR.internal:name=HtmlAdaptorParser";

    private final TlsHtmlAdaptorServer htmlAdaptor;
	private String name = "CoUGAR:name=HtmlAdaptor";
    private String username;
	private String password;
	private boolean httpExport = true;
	private MBeanServer mBeanServer;
	private HtmlAdaptorParser htmlParser;

    public HtmlAdaptor(final TlsHtmlAdaptorServer htmlAdaptor) {
        this.htmlAdaptor = htmlAdaptor;
    }

    public void afterPropertiesSet() throws Exception {
		if (httpExport) {
			ObjectName adaptorObjectName = new ObjectName(name);
			if(mBeanServer.isRegistered(adaptorObjectName)){
				mBeanServer.unregisterMBean(adaptorObjectName);
			}
	        mBeanServer.registerMBean(htmlAdaptor, adaptorObjectName);
			if (username != null && username.length() > 0 &&
			        password != null && password.length() > 0) {
			    LOGGER.info("JMX HTML interface password protected");
	    		htmlAdaptor.addUserAuthenticationInfo(new AuthInfo(username, password));
			} else {
		          LOGGER.info("JMX HTML interface password NOT protected");
			}

			ObjectName parserObjectName = new ObjectName(PARSER_NAME);
			if(mBeanServer.isRegistered(parserObjectName)){
				mBeanServer.unregisterMBean(parserObjectName);
			}
			htmlParser = new HtmlAdaptorParser(mBeanServer);
			addCustomParser(new ThreadDumper());
			mBeanServer.registerMBean(htmlParser, parserObjectName);
			htmlAdaptor.setParser(parserObjectName);

            String protocol = htmlAdaptor.isTlsEnabled() ? "https" : "http";
			LOGGER.info("Starting JMX HTML interface on "+protocol+"://localhost:%d/", htmlAdaptor.getPort());
			htmlAdaptor.start();

	        // Create a shutdown hook to close the jmx port cleanly
	        Runtime.getRuntime().addShutdownHook(new Thread("JMX HTTP Shutdown Hook") {

				@Override
				public void run() {
					LOGGER.info("Gracefully shutting down JMX http server");
					try {
						htmlAdaptor.stop();
					} catch (Exception e) {
						LOGGER.warn("Failed to shutdown JMX http server", e);
					}
				}});

		} else {
			LOGGER.info("JMX HTML interface not started");
		}
	}

	public void setName(String name) {
		this.name = name;
	}

	public void setMBeanServer(MBeanServer mBeanServer) {
		this.mBeanServer = mBeanServer;
	}

    public void setUsername(String username) {
        this.username = username;
    }

    public void setPassword(String password) {
        this.password = password;
    }

	public void setHttpExport(boolean httpExport) {
		this.httpExport = httpExport;
	}

    @Override
    public void addCustomParser(JMXHttpParser parser) {
    	if (httpExport) {
			LOGGER.info("JMX HTML custom parser added for path {}", parser.getPath());
    		htmlParser.addCustomParser(parser);
    	} else {
			LOGGER.info("No JMX HTML interface - parser ignored");

    	}
    }
}