org.asteriskjava.tools.HtmlEventTracer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of asterisk-java Show documentation
The free Java library for Asterisk PBX integration.
The newest version!
package org.asteriskjava.tools;

import org.asteriskjava.live.DefaultAsteriskServer;
import org.asteriskjava.manager.ManagerEventListener;
import org.asteriskjava.manager.event.*;

import java.beans.BeanInfo;
import java.beans.Introspector;
import java.beans.PropertyDescriptor;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * A diagnostic tool that creates an HTML file showing the state changing events
 * received from Asterisk on the Manager API.
 * 
 * The following events are shown:
 * 

 * NewChannel
 * NewState
 * Rename
 * Dial
 * Bridge (Link and Unlink)
 * Hangup
 * 
 * Usage: java org.asteriskjava.tools.HtmlEventTracer host username password
 *
 * @version $Id$
 */
public class HtmlEventTracer implements ManagerEventListener {
    private String filename = "trace.html";
    private PrintWriter writer;
    private final List uniqueIds;
    private final List events;
    private final Map, String> colors;

    public HtmlEventTracer() {
        uniqueIds = new ArrayList<>();
        events = new ArrayList<>();
        colors = new HashMap<>();

        colors.put(NewChannelEvent.class, "#7cd300"); // green
        colors.put(NewStateEvent.class, "#a4b6c8");
        colors.put(NewExtenEvent.class, "#efefef"); // grey
        colors.put(RenameEvent.class, "#ddeeff"); // light blue
        colors.put(DialEvent.class, "#feec30"); // yellow
        colors.put(BridgeEvent.class, "#fff8ae"); // light yellow
        colors.put(HangupEvent.class, "#ff6c17"); // orange

        try {
            writer = new PrintWriter(new OutputStreamWriter(new FileOutputStream(filename), StandardCharsets.UTF_8));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) throws Exception {
        if (args.length != 3) {
            System.err.println("Usage: java org.asteriskjava.tools.HtmlEventTracer host username password");
            System.exit(1);
        }

        final HtmlEventTracer tracer;
        final DefaultAsteriskServer server;

        tracer = new HtmlEventTracer();
        server = new DefaultAsteriskServer(args[0], args[1], args[2]);
        server.initialize();
        server.getManagerConnection().addEventListener(tracer);

        System.err.println("Event tracer successfully started. Press Ctrl-C to write trace file and exit.");

        Runtime.getRuntime().addShutdownHook(new Thread() {
            @Override
            public void run() {
                tracer.write();
                server.shutdown();
            }
        });

        while (true) {
            Thread.sleep(1000);
        }
    }

    public void onManagerEvent(ManagerEvent event) {
        events.add(event);
        System.out.println("> " + event);
        for (String property : new String[]{"uniqueId", "uniqueId1", "uniqueId2", "srcUniqueId", "destUniqueId"}) {
            String uniqueId;

            uniqueId = getProperty(event, property);
            if (uniqueId != null && !uniqueIds.contains(uniqueId)) {
                uniqueIds.add(uniqueId);
            }
        }
    }

    public void write() {
        writer.append("");
        for (String uniqueId : uniqueIds) {
            writer.append("");
        }
        writer.append("");
        writer.println("");

        for (ManagerEvent event : events) {
            boolean print = false;
            StringBuilder line = new StringBuilder();

            line.append("");
            for (String uniqueId : uniqueIds) {
                String text;
                text = getText(uniqueId, event);
                if (text == null) {
                    line.append("");
                } else {
                    String color = getColor(event.getClass());
                    line.append("");
                    print = true;
                }
            }
            line.append("");
            if (print) {
                writer.println(line.toString());
            }
        }
        writer.append(" ");
            writer.append(uniqueId.substring(0, uniqueId.lastIndexOf('.') + 1));
            writer.append("");
            writer.append(uniqueId.substring(uniqueId.lastIndexOf('.') + 1, uniqueId.length()));
            writer.append("
");
            line.append(event.getClass().getSimpleName());
            line.append("
");
            line.append(event.getDateReceived());
            line.append("  ").append(text).append("");
        writer.close();
        System.err.println("Trace file successfully written to " + filename + ".");
    }

    private String getColor(Class clazz) {
        for (Map.Entry, String> entry : colors.entrySet()) {
            if (entry.getKey().isAssignableFrom(clazz)) {
                return entry.getValue();
            }
        }
        return "#ffffff";
    }

    protected String getProperty(Object obj, String property) {
        try {
            BeanInfo beanInfo = Introspector.getBeanInfo(obj.getClass());
            for (PropertyDescriptor propertyDescriptor : beanInfo.getPropertyDescriptors()) {
                if (!propertyDescriptor.getName().equals(property)) {
                    continue;
                }

                final Object o = propertyDescriptor.getReadMethod().invoke(obj);
                return o != null ? o.toString() : null;
            }
        } catch (Exception e) {
            System.err.println("Unable to read property '" + property + "' from object " + obj + ": " + e.getMessage());
            return null;
        }

        return null;
    }

    protected String getText(String uniqueId, ManagerEvent event) {
        String format = null;
        String[] properties = null;

        if (uniqueId.equals(getProperty(event, "uniqueId"))) {
            if (event instanceof NewChannelEvent) {
                format = "%s
%s";
                properties = new String[]{"channel", "state"};
            } else if (event instanceof NewStateEvent) {
                format = "%s
%s";
                properties = new String[]{"channel", "state"};
            } else if (event instanceof NewExtenEvent) {
                format = "%s,%s,%s
%s(%s)";
                properties = new String[]{"context", "extension", "priority", "application", "appData"};
            } else if (event instanceof RenameEvent) {
                format = "old: %s
new: %s";
                properties = new String[]{"oldname", "newname"};
            } else if (event instanceof HoldEvent) {
                format = "%s";
                properties = new String[]{"status"};
            } else if (event instanceof AbstractParkedCallEvent) {
                format = "exten: %s
from: %s";
                properties = new String[]{"exten", "from"};
            } else if (event instanceof HangupEvent) {
                format = "%s
%s (%s)";
                properties = new String[]{"channel", "cause", "causeTxt"};
            } else if (event instanceof AntennaLevelEvent) {
                format = "%s";
                properties = new String[]{"signal"};
            } else if (event instanceof PausedEvent) {
                format = "(%s) %s";
                properties = new String[]{"header", "extension"};
            } else if (event instanceof UnpausedEvent) {
                format = "(%s) %s";
                properties = new String[]{"header", "extension"};
            }
        }

        if (event instanceof BridgeEvent) {
            if (uniqueId.equals(getProperty(event, "uniqueId1"))) {
                format = "%s
%s
%s";
                properties = new String[]{"uniqueId2", "channel2", "bridgeState"};
            } else if (uniqueId.equals(getProperty(event, "uniqueId2"))) {
                format = "%s
%s
%s";
                properties = new String[]{"uniqueId1", "channel1", "bridgeState"};
            }
        }

        if (event instanceof DialEvent) {
            if (uniqueId.equals(getProperty(event, "srcUniqueId"))) {
                format = "To: %s";
                properties = new String[]{"destination"};
            } else if (uniqueId.equals(getProperty(event, "destUniqueId"))) {
                format = "From: %s";
                properties = new String[]{"src"};
            }
        }

        if (format != null && properties != null) {
            String[] args = new String[properties.length];

            for (int i = 0; i < properties.length; i++) {
                String value;

                value = getProperty(event, properties[i]);
                if (value == null) {
                    args[i] = "";
                } else {
                    value = value.replace("<", "<");
                    value = value.replace(">", ">");
                    args[i] = value;
                }
            }

            return String.format(format, (Object[]) args);
        }
        return null;
    }
}
	"); writer.append(uniqueId.substring(0, uniqueId.lastIndexOf('.') + 1)); writer.append(""); writer.append(uniqueId.substring(uniqueId.lastIndexOf('.') + 1, uniqueId.length())); writer.append("
"); line.append(event.getClass().getSimpleName()); line.append(" "); line.append(event.getDateReceived()); line.append("		`").append(text).append("`