import java.awt.event.ActionEvent;
import javax.swing.event.TreeSelectionEvent;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.TreePath;
import javax.swing.tree.TreeSelectionModel;
import ch.qos.logback.classic.Level;
import ch.qos.logback.classic.spi.ILoggingEvent;
import ch.qos.logback.core.Appender;

public class LoggerLevelConfigurator extends javax.swing.JPanel implements java.awt.event.ActionListener, javax.swing.event.TreeSelectionListener
{
    private final static org.slf4j.Logger CLASS_LOGGER = org.slf4j.LoggerFactory.getLogger(LoggerLevelConfigurator.class);
	private final static org.slf4j.Logger EXCEPTION_LOGGER =org.slf4j.LoggerFactory.getLogger("ExceptionCatcher");
    private DefaultMutableTreeNode root = null;
    private DefaultTreeModel dtm;
    private final javax.swing.JTree tree;
    private final javax.swing.JPanel lower;
    private javax.swing.JComboBox cb;
    private javax.swing.Action createLoggerAction;
    private ch.qos.logback.classic.LoggerContext context;
    private java.util.Set<Appender<ILoggingEvent> > appenders=new java.util.HashSet();

    public LoggerLevelConfigurator()
    {
        super(new java.awt.BorderLayout());
        buildModel();
        tree = new javax.swing.JTree(dtm);
        add(new javax.swing.JScrollPane(tree));
        lower = new javax.swing.JPanel(new java.awt.BorderLayout());
        cb = new javax.swing.JComboBox(new Object[]
        {Level.OFF, Level.TRACE, Level.DEBUG, Level.INFO, Level.WARN, Level.ERROR, Level.ALL});
        cb.setEnabled(false);
        lower.add(cb);
        add(lower, java.awt.BorderLayout.SOUTH);
        createActions();
        javax.swing.JToolBar tb = new javax.swing.JToolBar();
        tb.setFloatable(false);
        tb.add(createLoggerAction);
        add(tb, java.awt.BorderLayout.NORTH);
        tree.getSelectionModel().setSelectionMode(TreeSelectionModel.SINGLE_TREE_SELECTION);
        cb.addActionListener(this);
        tree.addTreeSelectionListener(this);
    }

    private void createActions()
    {
        createLoggerAction = new javax.swing.AbstractAction("+")//null,de.netsysit.util.ResourceLoader.getIcon("toolbarButtonGraphics/general/Add24.gif"))
        {

            @Override
            public void actionPerformed(ActionEvent e)
            {
                if (tree.isSelectionEmpty())
                {
                    java.awt.Toolkit.getDefaultToolkit().beep();
                    createLoggerAction.setEnabled(false);
                } else
                {
                    DefaultMutableTreeNode node = (DefaultMutableTreeNode) tree.getSelectionPath().getLastPathComponent();
                    ch.qos.logback.classic.Logger log = (ch.qos.logback.classic.Logger) node.getUserObject();
                    java.lang.String name=javax.swing.JOptionPane.showInputDialog("new logger beneath "+log.getName());
                    if(name!=null)
                    {
                        DefaultMutableTreeNode parent=node;
                        java.lang.String[] parts=name.split("\\.");
                        for(int i=0;i<parts.length;++i)
                        {
                            ch.qos.logback.classic.Logger newlogger = context.getLogger(log.getName()+"."+parts[i]);
                            for(Appender<ILoggingEvent> app:appenders)
                            {
                                newlogger.addAppender(app);
                            }
                            DefaultMutableTreeNode newnode=new DefaultMutableTreeNode(newlogger);
                            parent.add(newnode);
                            parent=newnode;
                            log=(ch.qos.logback.classic.Logger) parent.getUserObject();
                        }
                        dtm.reload();
                        TreePath path = new TreePath(parent.getPath());
                        tree.setSelectionPath(path);
                        tree.scrollPathToVisible(path);
                    }
                }
            }
        };
        createLoggerAction.setEnabled(false);
    }

    private void buildModel()
    {
        java.util.HashMap<java.lang.String, DefaultMutableTreeNode> map = new java.util.HashMap();
        context = (ch.qos.logback.classic.LoggerContext) org.slf4j.LoggerFactory.getILoggerFactory();
        for (ch.qos.logback.classic.Logger log : context.getLoggerList())
        {
            java.util.Iterator<Appender<ILoggingEvent> > iter=log.iteratorForAppenders();
            while(iter.hasNext())
            {
                appenders.add(iter.next());
            }
            CLASS_LOGGER.trace(log.getName());
            DefaultMutableTreeNode dmtn = new DefaultMutableTreeNode(log);
            if (log.getName().equals("ROOT"))
                root = dmtn;
            else
                map.put(log.getName(), dmtn);
        }
        for (java.util.Map.Entry<java.lang.String, DefaultMutableTreeNode> entry : map.entrySet())
        {
            if (entry.getKey().indexOf('.') < 0)
            {
                root.add(entry.getValue());
            } else
            {
                java.lang.String parentName = entry.getKey().substring(0, entry.getKey().lastIndexOf('.'));
                CLASS_LOGGER.trace(parentName);
                map.get(parentName).add(entry.getValue());
            }
        }
        dtm = new DefaultTreeModel(root);
        for(Appender<ILoggingEvent> app:appenders)
        {
            CLASS_LOGGER.trace(app.getName());
        }
    }

    @Override
    public void actionPerformed(ActionEvent e)
    {
        if (tree.isSelectionEmpty())
        {
            java.awt.Toolkit.getDefaultToolkit().beep();
            cb.setEnabled(false);
        } else
        {
            DefaultMutableTreeNode node = (DefaultMutableTreeNode) tree.getSelectionPath().getLastPathComponent();
            ch.qos.logback.classic.Logger log = (ch.qos.logback.classic.Logger) node.getUserObject();
            log.setLevel((Level) cb.getSelectedItem());
        }
    }

    @Override
    public void valueChanged(TreeSelectionEvent e)
    {
        if ((e.getPath() != null) && (tree.isSelectionEmpty() == false))
        {
            cb.setEnabled(true);
            createLoggerAction.setEnabled(true);
            Level lvl = null;
            DefaultMutableTreeNode node = (DefaultMutableTreeNode) e.getPath().getLastPathComponent();
            while ((node != null) && (lvl == null))
            {
                ch.qos.logback.classic.Logger log = (ch.qos.logback.classic.Logger) node.getUserObject();
                lvl = log.getLevel();
                node = (DefaultMutableTreeNode) node.getParent();
            }
            CLASS_LOGGER.trace(lvl != null ? lvl.toString() : "null");
            if (lvl == null)
                cb.setEnabled(false);
            else
                cb.setSelectedItem(lvl);
        } else
        {
            cb.setEnabled(false);
            createLoggerAction.setEnabled(false);
        }
    }

    public static void main(java.lang.String[] args)
    {
        javax.swing.JOptionPane.showMessageDialog(null,new LoggerLevelConfigurator());
    }
}