/*

 * Licensed to the Apache Software Foundation (ASF) under one or more

 * contributor license agreements.  See the NOTICE file distributed with

 * this work for additional information regarding copyright ownership.

 * The ASF licenses this file to You 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 org.apache.catalina.manager;

import java.io.IOException;

import java.io.PrintWriter;

import java.lang.reflect.Method;

import java.text.MessageFormat;

import java.util.Date;

import java.util.Enumeration;

import java.util.Iterator;

import java.util.Set;

import java.util.Vector;

import javax.management.MBeanServer;

import javax.management.ObjectInstance;

import javax.management.ObjectName;

import javax.servlet.ServletException;

import javax.servlet.http.HttpServletResponse;

import org.apache.catalina.util.RequestUtil;

/**

 * This is a refactoring of the servlet to externalize

 * the output into a simple class. Although we could

 * use XSLT, that is unnecessarily complex.

 *

 * @author Peter Lin

 */

public class StatusTransformer {

    // --------------------------------------------------------- Public Methods

    public static void setContentType(HttpServletResponse response, 

                                      int mode) {

        if (mode == 0){

            response.setContentType("text/html;charset="+Constants.CHARSET);

        } else if (mode == 1){

            response.setContentType("text/xml;charset="+Constants.CHARSET);

        }

    }

    /**

     * Process a GET request for the specified resource.

     *

     * @param request The servlet request we are processing

     * @param response The servlet response we are creating

     *

     * @exception IOException if an input/output error occurs

     * @exception ServletException if a servlet-specified error occurs

     */

    public static void writeHeader(PrintWriter writer, int mode) {

        if (mode == 0){

            // HTML Header Section

            writer.print(Constants.HTML_HEADER_SECTION);

        } else if (mode == 1){

            writer.write(Constants.XML_DECLARATION);

            writer.write

                (Constants.XML_STYLE);

            writer.write("<status>");

        }

    }

    /**

     * Write the header body. XML output doesn't bother

     * to output this stuff, since it's just title.

     * 

     * @param writer The output writer

     * @param args What to write

     * @param mode 0 means write 

     */

    public static void writeBody(PrintWriter writer, Object[] args, int mode) {

        if (mode == 0){

            writer.print(MessageFormat.format

                         (Constants.BODY_HEADER_SECTION, args));

        }

    }

    /**

     * Write the manager webapp information.

     * 

     * @param writer The output writer

     * @param args What to write

     * @param mode 0 means write

     */

    public static void writeManager(PrintWriter writer, Object[] args, 

                                    int mode) {

        if (mode == 0){

            writer.print(MessageFormat.format(Constants.MANAGER_SECTION, args));

        }

    }

    public static void writePageHeading(PrintWriter writer, Object[] args, 

                                        int mode) {

        if (mode == 0){

            writer.print(MessageFormat.format

                         (Constants.SERVER_HEADER_SECTION, args));

        }

    }

    public static void writeServerInfo(PrintWriter writer, Object[] args, 

                                       int mode){

        if (mode == 0){

            writer.print(MessageFormat.format(Constants.SERVER_ROW_SECTION, args));

        }

    }

    /**

     * 

     */

    public static void writeFooter(PrintWriter writer, int mode) {

        if (mode == 0){

            // HTML Tail Section

            writer.print(Constants.HTML_TAIL_SECTION);

        } else if (mode == 1){

            writer.write("</status>");

        }

    }

    /**

     * Write the OS state. Mode 0 will generate HTML.

     * Mode 1 will generate XML.

     */

    public static void writeOSState(PrintWriter writer, int mode) {

        long[] result = new long[16];

        boolean ok = false;

        try {

            String methodName = "info";

            Class paramTypes[] = new Class[1];

            paramTypes[0] = result.getClass();

            Object paramValues[] = new Object[1];

            paramValues[0] = result;

            Method method = Class.forName("org.apache.tomcat.jni.OS")

                .getMethod(methodName, paramTypes);

            method.invoke(null, paramValues);

            ok = true;

        } catch (Throwable t) {

            // Ignore

        }

        if (ok) {

            if (mode == 0){

                writer.print("<h1>OS</h1>");

                writer.print("<p>");

                writer.print(" Physical memory: ");

                writer.print(formatSize(new Long(result[0]), true));

                writer.print(" Available memory: ");

                writer.print(formatSize(new Long(result[1]), true));

                writer.print(" Total page file: ");

                writer.print(formatSize(new Long(result[2]), true));

                writer.print(" Free page file: ");

                writer.print(formatSize(new Long(result[3]), true));

                writer.print(" Memory load: ");

                writer.print(new Long(result[6]));

                writer.print("<br>");

                writer.print(" Process kernel time: ");

                writer.print(formatTime(new Long(result[11] / 1000), true));

                writer.print(" Process user time: ");

                writer.print(formatTime(new Long(result[12] / 1000), true));

                writer.print("</p>");

            } else if (mode == 1){

            }

        }

    }

    /**

     * Write the VM state. Mode 0 will generate HTML.

     * Mode 1 will generate XML.

     */

    public static void writeVMState(PrintWriter writer, int mode)

        throws Exception {

        if (mode == 0){

            writer.print("<h1>JVM</h1>");

            writer.print("<p>");

            writer.print(" Free memory: ");

            writer.print(formatSize

                         (new Long(Runtime.getRuntime().freeMemory()), true));

            writer.print(" Total memory: ");

            writer.print(formatSize

                         (new Long(Runtime.getRuntime().totalMemory()), true));

            writer.print(" Max memory: ");

            writer.print(formatSize

                         (new Long(Runtime.getRuntime().maxMemory()), true));

            writer.print("</p>");

        } else if (mode == 1){

            writer.write("<jvm>");

            writer.write("<memory");

            writer.write(" free='" + Runtime.getRuntime().freeMemory() + "'");

            writer.write(" total='" + Runtime.getRuntime().totalMemory() + "'");

            writer.write(" max='" + Runtime.getRuntime().maxMemory() + "'/>");

            writer.write("</jvm>");

        }

    }

    /**

     * Write connector state.

     */

    public static void writeConnectorState(PrintWriter writer, 

                                           ObjectName tpName, String name,

                                           MBeanServer mBeanServer,

                                           Vector globalRequestProcessors,

                                           Vector requestProcessors,

                                           int mode)

        throws Exception {

        if (mode == 0) {

            writer.print("<h1>");

            writer.print(name);

            writer.print("</h1>");

            writer.print("<p>");

            writer.print(" Max threads: ");

            writer.print(mBeanServer.getAttribute(tpName, "maxThreads"));

            writer.print(" Current thread count: ");

            writer.print(mBeanServer.getAttribute(tpName, "currentThreadCount"));

            writer.print(" Current thread busy: ");

            writer.print(mBeanServer.getAttribute(tpName, "currentThreadsBusy"));

            try {

                Object value = mBeanServer.getAttribute(tpName, "keepAliveCount");

                writer.print(" Keeped alive sockets count: ");

                writer.print(value);

            } catch (Exception e) {

                // Ignore

            }

            writer.print("<br>");

            ObjectName grpName = null;

            Enumeration enumeration = globalRequestProcessors.elements();

            while (enumeration.hasMoreElements()) {

                ObjectName objectName = (ObjectName) enumeration.nextElement();

                if (name.equals(objectName.getKeyProperty("name"))) {

                    grpName = objectName;

                }

            }

            if (grpName == null) {

                return;

            }

            writer.print(" Max processing time: ");

            writer.print(formatTime(mBeanServer.getAttribute

                                    (grpName, "maxTime"), false));

            writer.print(" Processing time: ");

            writer.print(formatTime(mBeanServer.getAttribute

                                    (grpName, "processingTime"), true));

            writer.print(" Request count: ");

            writer.print(mBeanServer.getAttribute(grpName, "requestCount"));

            writer.print(" Error count: ");

            writer.print(mBeanServer.getAttribute(grpName, "errorCount"));

            writer.print(" Bytes received: ");

            writer.print(formatSize(mBeanServer.getAttribute

                                    (grpName, "bytesReceived"), true));

            writer.print(" Bytes sent: ");

            writer.print(formatSize(mBeanServer.getAttribute

                                    (grpName, "bytesSent"), true));

            writer.print("</p>");

            writer.print("<table border=\"0\"><tr><th>Stage</th><th>Time</th><th>B Sent</th><th>B Recv</th><th>Client</th><th>VHost</th><th>Request</th></tr>");

            enumeration = requestProcessors.elements();

            while (enumeration.hasMoreElements()) {

                ObjectName objectName = (ObjectName) enumeration.nextElement();

                if (name.equals(objectName.getKeyProperty("worker"))) {

                    writer.print("<tr>");

                    writeProcessorState(writer, objectName, mBeanServer, mode);

                    writer.print("</tr>");

                }

            }

            writer.print("</table>");

            writer.print("<p>");

            writer.print("P: Parse and prepare request S: Service F: Finishing R: Ready K: Keepalive");

            writer.print("</p>");

        } else if (mode == 1){

            writer.write("<connector name='" + name + "'>");

            writer.write("<threadInfo ");

            writer.write(" maxThreads=\"" + mBeanServer.getAttribute(tpName, "maxThreads") + "\"");

            writer.write(" currentThreadCount=\"" + mBeanServer.getAttribute(tpName, "currentThreadCount") + "\"");

            writer.write(" currentThreadsBusy=\"" + mBeanServer.getAttribute(tpName, "currentThreadsBusy") + "\"");

            writer.write(" />");

            ObjectName grpName = null;

            Enumeration enumeration = globalRequestProcessors.elements();

            while (enumeration.hasMoreElements()) {

                ObjectName objectName = (ObjectName) enumeration.nextElement();

                if (name.equals(objectName.getKeyProperty("name"))) {

                    grpName = objectName;

                }

            }

            if (grpName != null) {

                writer.write("<requestInfo ");

                writer.write(" maxTime=\"" + mBeanServer.getAttribute(grpName, "maxTime") + "\"");

                writer.write(" processingTime=\"" + mBeanServer.getAttribute(grpName, "processingTime") + "\"");

                writer.write(" requestCount=\"" + mBeanServer.getAttribute(grpName, "requestCount") + "\"");

                writer.write(" errorCount=\"" + mBeanServer.getAttribute(grpName, "errorCount") + "\"");

                writer.write(" bytesReceived=\"" + mBeanServer.getAttribute(grpName, "bytesReceived") + "\"");

                writer.write(" bytesSent=\"" + mBeanServer.getAttribute(grpName, "bytesSent") + "\"");

                writer.write(" />");

                writer.write("<workers>");

                enumeration = requestProcessors.elements();

                while (enumeration.hasMoreElements()) {

                    ObjectName objectName = (ObjectName) enumeration.nextElement();

                    if (name.equals(objectName.getKeyProperty("worker"))) {

                        writeProcessorState(writer, objectName, mBeanServer, mode);

                    }

                }

                writer.write("</workers>");

            }

            writer.write("</connector>");

        }

    }

    /**

     * Write processor state.

     */

    protected static void writeProcessorState(PrintWriter writer, 

                                              ObjectName pName,

                                              MBeanServer mBeanServer, 

                                              int mode)

        throws Exception {

        Integer stageValue = 

            (Integer) mBeanServer.getAttribute(pName, "stage");

        int stage = stageValue.intValue();

        boolean fullStatus = true;

        boolean showRequest = true;

        String stageStr = null;

        switch (stage) {

        case (1/*org.apache.coyote.Constants.STAGE_PARSE*/):

            stageStr = "P";

            fullStatus = false;

            break;

        case (2/*org.apache.coyote.Constants.STAGE_PREPARE*/):

            stageStr = "P";

            fullStatus = false;

            break;

        case (3/*org.apache.coyote.Constants.STAGE_SERVICE*/):

            stageStr = "S";

            break;

        case (4/*org.apache.coyote.Constants.STAGE_ENDINPUT*/):

            stageStr = "F";

            break;

        case (5/*org.apache.coyote.Constants.STAGE_ENDOUTPUT*/):

            stageStr = "F";

            break;

        case (7/*org.apache.coyote.Constants.STAGE_ENDED*/):

            stageStr = "R";

            fullStatus = false;

            break;

        case (6/*org.apache.coyote.Constants.STAGE_KEEPALIVE*/):

            stageStr = "K";

            fullStatus = true;

            showRequest = false;

            break;

        case (0/*org.apache.coyote.Constants.STAGE_NEW*/):

            stageStr = "R";

            fullStatus = false;

            break;

        default:

            // Unknown stage

            stageStr = "?";

            fullStatus = false;

        }

        if (mode == 0) {

            writer.write("<td><strong>");

            writer.write(stageStr);

            writer.write("</strong></td>");

            if (fullStatus) {

                writer.write("<td>");

                writer.print(formatTime(mBeanServer.getAttribute

                                        (pName, "requestProcessingTime"), false));

                writer.write("</td>");

                writer.write("<td>");

                if (showRequest) {

                    writer.print(formatSize(mBeanServer.getAttribute

                                            (pName, "requestBytesSent"), false));

                } else {

                    writer.write("?");

                }

                writer.write("</td>");

                writer.write("<td>");

                if (showRequest) {

                    writer.print(formatSize(mBeanServer.getAttribute

                                            (pName, "requestBytesReceived"), 

                                            false));

                } else {

                    writer.write("?");

                }

                writer.write("</td>");

                writer.write("<td>");

                writer.print(filter(mBeanServer.getAttribute

                                    (pName, "remoteAddr")));

                writer.write("</td>");

                writer.write("<td nowrap>");

                writer.write(filter(mBeanServer.getAttribute

                                    (pName, "virtualHost")));

                writer.write("</td>");

                writer.write("<td nowrap>");

                if (showRequest) {

                    writer.write(filter(mBeanServer.getAttribute

                                        (pName, "method")));

                    writer.write(" ");

                    writer.write(filter(mBeanServer.getAttribute

                                        (pName, "currentUri")));

                    String queryString = (String) mBeanServer.getAttribute

                        (pName, "currentQueryString");

                    if ((queryString != null) && (!queryString.equals(""))) {

                        writer.write("?");

                        writer.print(RequestUtil.filter(queryString));

                    }

                    writer.write(" ");

                    writer.write(filter(mBeanServer.getAttribute

                                        (pName, "protocol")));

                } else {

                    writer.write("?");

                }

                writer.write("</td>");

            } else {

                writer.write("<td>?</td><td>?</td><td>?</td><td>?</td><td>?</td><td>?</td>");

            }

        } else if (mode == 1){

            writer.write("<worker ");

            writer.write(" stage=\"" + stageStr + "\"");

            if (fullStatus) {

                writer.write(" requestProcessingTime=\"" 

                             + mBeanServer.getAttribute

                             (pName, "requestProcessingTime") + "\"");

                writer.write(" requestBytesSent=\"");

                if (showRequest) {

                    writer.write("" + mBeanServer.getAttribute

                                 (pName, "requestBytesSent"));

                } else {

                    writer.write("0");

                }

                writer.write("\"");

                writer.write(" requestBytesReceived=\"");

                if (showRequest) {

                    writer.write("" + mBeanServer.getAttribute

                                 (pName, "requestBytesReceived"));

                } else {

                    writer.write("0");

                }

                writer.write("\"");

                writer.write(" remoteAddr=\"" 

                             + filter(mBeanServer.getAttribute

                                      (pName, "remoteAddr")) + "\"");

                writer.write(" virtualHost=\"" 

                             + filter(mBeanServer.getAttribute

                                      (pName, "virtualHost")) + "\"");

                if (showRequest) {

                    writer.write(" method=\"" 

                                 + filter(mBeanServer.getAttribute

                                          (pName, "method")) + "\"");

                    writer.write(" currentUri=\"" 

                                 + filter(mBeanServer.getAttribute

                                          (pName, "currentUri")) + "\"");

                    String queryString = (String) mBeanServer.getAttribute

                        (pName, "currentQueryString");

                    if ((queryString != null) && (!queryString.equals(""))) {

                        writer.write(" currentQueryString=\"" 

                                     + RequestUtil.filter(queryString) + "\"");

                    } else {

                        writer.write(" currentQueryString=\"?\"");

                    }

                    writer.write(" protocol=\"" 

                                 + filter(mBeanServer.getAttribute

                                          (pName, "protocol")) + "\"");

                } else {

                    writer.write(" method=\"?\"");

                    writer.write(" currentUri=\"?\"");

                    writer.write(" currentQueryString=\"?\"");

                    writer.write(" protocol=\"?\"");

                }