MainFrame.java

Go to the documentation of this file.

package org.libcsdbg.jtracer;

import java.util.Enumeration;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;

import java.net.URL;
import java.net.Socket;
import java.net.ServerSocket;
import java.net.InetAddress;
import java.net.InetSocketAddress;

import java.nio.channels.FileLock;

import java.beans.PropertyChangeEvent;
import java.beans.PropertyChangeListener;

import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.Insets;
import java.awt.Dimension;
import java.awt.BorderLayout;
import java.awt.GridBagLayout;
import java.awt.GridBagConstraints;
import java.awt.GraphicsEnvironment;
import java.awt.GraphicsDevice;

import javax.swing.JFrame;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.UIManager;

public class MainFrame extends JFrame implements
  Runnable,
  ActionListener,
  PropertyChangeListener
{
  private static final long serialVersionUID = 0x1000;

  private ToolBar tools = null;

  private MenuBar menu = null;

  private LogPane log = null;

  private StatusBar status = null;

  private AboutDialog about = null;

  private FileLock lock = null;

  private Thread server = null;

  private ServerSocket listener = null;

  private SessionManager desktop = null;


  MainFrame()
  {
    super();

    Registry conf = Registry.getCurrent();
    setTitle((String) conf.get("generic", "name"));
    setIconImages(conf.getProjectIcons());

    menu = new MenuBar(this);
    tools = new ToolBar(this);
    log = new LogPane();
    status = new StatusBar("jTracer initialized");
    desktop = new SessionManager(this);

    setJMenuBar(menu);
    add(tools, BorderLayout.NORTH);
    add(status, BorderLayout.SOUTH);

    JPanel nowrap = new JPanel(new BorderLayout());
    nowrap.add(log);
    JScrollPane viewport = new JScrollPane(nowrap);
    viewport.setVerticalScrollBarPolicy(JScrollPane.VERTICAL_SCROLLBAR_ALWAYS);

    GridBagLayout lm = new GridBagLayout();
    GridBagConstraints c = new GridBagConstraints();
    JPanel inset = new JPanel(lm);
    c.weightx = c.weighty = 1;
    c.fill = GridBagConstraints.BOTH;
    c.insets = new Insets(0, 2, 0, 1);
    lm.setConstraints(viewport, c);
    inset.add(viewport);
    add(inset, BorderLayout.CENTER);

    addPropertyChangeListener("isServing", menu);
    addPropertyChangeListener("isServing", tools);
    addPropertyChangeListener("hasClients", menu);
    addPropertyChangeListener("hasClients", tools);
    addWindowListener(desktop);

    firePropertyChange("isServing", null, false);
    firePropertyChange("hasClients", null, false);

    log.appendln("jTracer initialized", "status");
    log.appendln("Ready to start serving", "status");

    setSize((Dimension) conf.get("generic", "size"));
    setLocationRelativeTo(null);
  }


  public void startService()
  {
    if (server != null)
      return;

    log.appendln("Starting server...", "status");
    server = new Thread(this, "server");
    server.setDaemon(true);
    server.start();
  }


  public void stopService()
  {
    if (server == null)
      return;

    log.appendln("Stopping server...", "status");
    Thread thr = server;
    shutdown();
    
    try {
      if (thr.isAlive())
        thr.join();
    }
      
    catch (Throwable t) {
      Registry.debug(t);
    }
    
    log.appendln("Service stopped. All incoming connections dropped", "status");
  }


  public void shutdown()
  {
    try {
      server = null;

      if (listener != null) {
        listener.close();
        listener = null;
      }
    }

    catch (Throwable t) {
      Registry.debug(t);
    }
  }


  public static void main(String[] argv)
  {
    try {
      Registry conf = new Registry();
      conf.setup();

      Highlighter h = new Highlighter();
      MainFrame app = new MainFrame();

      File lockf = conf.getResource("log/.lock");
      FileOutputStream fs = new FileOutputStream(lockf);

      app.lock = fs.getChannel().tryLock();
      if (app.lock == null) {
        String nm = (String) conf.get("generic", "name");
        Alert.error(null, nm + " is already running", true);
      }

      lockf.deleteOnExit();
      UIManager.setLookAndFeel((String) conf.get("generic", "lnf"));
      app.setVisible(true);
    }

    catch (Throwable t) {
      Registry.debug(t);
      System.exit(1);
    }
  }


  public void run()
  {
    try {
      Registry conf = Registry.getCurrent();
      Enumeration keys = conf.getSectionKeys("server");
      if (keys == null)
        throw new Exception("No server configuration found");

      /* Create the listener socket unbound */
      listener = new ServerSocket();
      listener.setReuseAddress(true);

      /* Bind to one or more interfaces */
      InetSocketAddress iface = null;
      while (keys.hasMoreElements()) {
        String key = (String) keys.nextElement();

        iface = (InetSocketAddress) conf.get("server", key);
        listener.bind(iface);

        InetAddress addr = iface.getAddress();
        String msg = "Accepting connections @";
        msg += addr.getHostAddress() + ":" + iface.getPort();

        if (addr.isAnyLocalAddress())
          msg += " (all local)";
        
        else if (addr.isMulticastAddress())
          msg += " (multicast)";
        
        else
          msg += " (" + addr.getCanonicalHostName() + ")";

        log.appendln(msg, "status");
      }

      if (iface == null || !listener.isBound())
        throw new Exception("No server configuration found");

      firePropertyChange("isServing", null, true);
      status.startUptimeTimer();

      /* Listen for incoming connections */
      while (server == Thread.currentThread()) {
        status.setMessage("Waiting for connections", true);

        try {
          Socket peer = listener.accept();

          String addr = peer.getInetAddress().getHostAddress();
          int port = peer.getPort();
          log.appendln("Connected peer @" + addr + ":" + port, "status");

          desktop.register(peer);
        }

        catch (Throwable t) {
          log.appendln(t.toString(), "alert");
          if (!(t instanceof IOException))
            Registry.debug(t);
        }
      }
      
      status.stopUptimeTimer();
      status.setMessage("Service complete", true);
      firePropertyChange("isServing", null, false);
    }

    catch (Throwable t) {
      Registry.debug(t);
      getToolkit().beep();
      shutdown();

      log.appendln("Failed to start server (" + t.toString() + ")", "error");
      status.setMessage("Server failed", false);
    }
  }


  public void actionPerformed(ActionEvent event)
  {
    try {
      String cmd = event.getActionCommand();

      /* Service menu commands */
      if (cmd.equals("Start"))
        startService();

      else if (cmd.equals("Stop"))
        stopService();

      else if (cmd.equals("Restart")) {
        log.appendln("Restarting server (reload configuration)...", "status");
        stopService();
        startService();
      }

      else if (cmd.equals("Clear log"))
        log.clear();

      else if (cmd.equals("Quit")) {
        boolean reply = Alert.prompt(this, "Are you sure you want to quit?");
        if (reply)
          System.exit(0);
      }


      /* View menu commands */
      else if (cmd.equals("Toolbar")) {
        if (menu.getToggleState(0))
          add(tools, BorderLayout.NORTH);
        else
          remove(tools);

        validate();
      }

      else if (cmd.equals("Statusbar")) {
        if (menu.getToggleState(1))
          add(status, BorderLayout.SOUTH);
        else
          remove(status);

        validate();
      }

      else if (cmd.equals("Find"))
        Alert.error(this, "Not implemented yet", false);

      else if (cmd.equals("Preferences"))
        Alert.error(this, "Not implemented yet", false);

      else if (cmd.equals("Always on top")) {
        boolean state = menu.getToggleState(2);
        setAlwaysOnTop(state);
        desktop.setAlwaysOnTop(-1, state);
      }

      else if (cmd.equals("Full screen")) {
        GraphicsEnvironment env;
        env = GraphicsEnvironment.getLocalGraphicsEnvironment();

        GraphicsDevice dev = env.getDefaultScreenDevice();
        if (!dev.isFullScreenSupported())
          return;

        JFrame cur = (JFrame) dev.getFullScreenWindow();
        if (cur == null)
          dev.setFullScreenWindow(this);
        else
          dev.setFullScreenWindow(null);
      }


      /* Client menu commands */
      else if (cmd.equals("Select previous"))
        desktop.shiftSelection(-1);

      else if (cmd.equals("Select next"))
        desktop.shiftSelection(1);

      else if (cmd.equals("Close")) {
        /* This fixes a very tricky bug induced by the OS window manager */
        toFront();
        desktop.disposeCurrent();
      }

      else if (cmd.equals("Cascade"))
        desktop.cascade();

      else if (cmd.equals("Minimize all"))
        desktop.setIconified(true);

      else if (cmd.equals("Restore all"))
        desktop.setIconified(false);

      else if (cmd.equals("Close all")) {
        boolean reply = Alert.prompt(this, "Close all client windows?");
        if (!reply)
          return;

        desktop.disposeAll();
      }

      else if (cmd.startsWith("Select session")) {
        String parts[] = cmd.split("\\s");
        desktop.setCurrent(Integer.valueOf(parts[2]));
      }

      /* Help menu commands */
      else if (
        cmd.equals("Online documentation")  ||
        cmd.equals("Bug tracker")           ||
        cmd.equals("Submit feedback")
      ) {
        Registry conf = Registry.getCurrent();
        String key = "url-" + cmd.toLowerCase().replace(' ', '-');
        conf.browse((URL) conf.get("generic", key));
      }

      else if (cmd.equals("Check for updates"))
        Alert.error(this, "Not implemented yet", false);

      else if (cmd.startsWith("About")) {
        if (about == null)
          about = new AboutDialog(this);

        about.setLocationRelativeTo(this);
        about.setVisible(true);
      }
      
      
      /* Session commands /
      else if (cmd.equals("Remove trace"))
        setStatus(null, true); */
    }

    catch (Throwable t) {
      Registry.debug(t);
    }
  }


  public void propertyChange(PropertyChangeEvent event)
  {
    try {
      String key = event.getPropertyName();
      Object val = event.getNewValue();
      
      if (key.equals("sessionCount")) {
        firePropertyChange("hasClients", null, (Integer) val != 0); 
        menu.listSessions(desktop.getSessions(), desktop.getCurrent());
        status.setIndicators(desktop.getSessionCount(), desktop.getTraceCount());
      }

      else if (key.equals("traceCount")) {
        menu.listSessions(desktop.getSessions(), desktop.getCurrent());
        status.setIndicators(desktop.getSessionCount(), desktop.getTraceCount());
      }
      
      else if (key.equals("currentSession"))
        menu.setSelectedSession((Integer) val);
      
      else if (key.equals("sessionRequest"))
        log.appendln((String) val, "data");
    }

    catch (Throwable t) {
      Registry.debug(t);
    }
  }
}