MainFrame.java

Go to the documentation of this file.

package org.libcsdbg.jtracer.installer;

/*
import java.util.Hashtable;
import java.util.Enumeration;
import java.net.URL;
import java.awt.Container;
import java.awt.EventQueue;
import javax.swing.ImageIcon;
import javax.swing.SwingUtilities;
*/

import java.util.Vector;
import java.util.zip.ZipEntry;
import java.util.jar.JarInputStream;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.FileWriter;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;

import java.net.URL;

import java.nio.channels.FileLock;
import java.nio.file.Files;

import java.awt.event.KeyEvent;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.Font;
import java.awt.Color;
import java.awt.Dimension;
import java.awt.Insets;
import java.awt.BorderLayout;
import java.awt.GridBagLayout;
import java.awt.GridBagConstraints;

import javax.swing.event.HyperlinkEvent;
import javax.swing.event.HyperlinkListener;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JEditorPane;
import javax.swing.JScrollPane;
import javax.swing.JCheckBox;
import javax.swing.JProgressBar;
import javax.swing.JFileChooser;
import javax.swing.UIManager;
import javax.swing.WindowConstants;

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

  private JLabel caption = null;

  private JLabel progress = null;

  private JPanel viewport = null;

  private Button previous = null, next = null, select = null;

  private TextField home = null, prefix = null;

  private JProgressBar setupBar = null;

  private AboutDialog about = null;

  private Vector<JPanel> screens = null;

  private int current = 0;

  private FileLock lock = null;

  private Thread installer = null;


  MainFrame()
  {
    super();
    Registry conf = Registry.getCurrent();
    screens = new Vector<JPanel>();
  
    String pkg = (String) conf.get("installed", "package");
    setTitle("Installation of " + pkg);
    setIconImages(conf.getProjectIcons());

    add(createCaption(), BorderLayout.NORTH);
    add(createControls(), BorderLayout.SOUTH);
    try {
      createScreen(0);
      createScreen(1);
      createScreen(2);
    }
    
    catch (Throwable t) {
      Registry.debug(t);
      dispose();
      return;
    }

    viewport = new JPanel(new BorderLayout());
    viewport.add(screens.get(0), BorderLayout.CENTER);
    add(viewport, BorderLayout.CENTER);

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


  private JPanel createCaption()
  {
    Registry conf = Registry.getCurrent();
    GridBagLayout lm = new GridBagLayout();
    GridBagConstraints c = new GridBagConstraints();
    JPanel retval = new JPanel(lm);

    String pkg = (String) conf.get("installed", "package");
    caption = new JLabel("Installing " + pkg);
    caption.setFont((Font) conf.get("caption", "font"));
    caption.setForeground((Color) conf.get("caption", "fgcolor"));

    c.gridx = c.gridy = 0;
    c.weightx = 1.0;
    c.insets = new Insets(12, 12, 0, 12);
    c.fill = GridBagConstraints.HORIZONTAL;
    c.anchor = GridBagConstraints.WEST;
    lm.setConstraints(caption, c);
    retval.add(caption);
    
    return retval;
  }


  private JPanel createControls()
  {
    GridBagLayout lm = new GridBagLayout();
    GridBagConstraints c = new GridBagConstraints();
    JPanel retval = new JPanel(lm);

    Button info = new Button("About", this);
    info.setMnemonic(KeyEvent.VK_A);

    Button cancel = new Button("Cancel", this);
    cancel.setMnemonic(KeyEvent.VK_C);

    next = new Button("Next", this);
    next.setMnemonic(KeyEvent.VK_N);

    previous = new Button("Previous", this);
    previous.setMnemonic(KeyEvent.VK_P);
    previous.setEnabled(false);

    c.gridx = c.gridy = 0;
    c.insets = new Insets(12, 12, 6, 12);
    lm.setConstraints(info, c);
    retval.add(info);
    
    c.gridx++;
    c.weightx = 1.0;
    c.insets.left = 0;
    c.anchor = GridBagConstraints.WEST;
    lm.setConstraints(cancel, c);
    retval.add(cancel);

    c.gridx++;
    c.weightx = 0;
    c.anchor = GridBagConstraints.CENTER;
    lm.setConstraints(previous, c);
    retval.add(previous);

    c.gridx++;
    c.insets.left = 0;
    lm.setConstraints(next, c);
    retval.add(next);

    return retval;
  }


  private void createScreen(int i) throws IOException
  {
    Registry conf = Registry.getCurrent();
    GridBagLayout lm = new GridBagLayout();
    GridBagConstraints c = new GridBagConstraints();
    JPanel scr = new JPanel(lm);

    Font fnt = (Font) conf.get("component", "font");
    Color fg = (Color) conf.get("component", "fgcolor");

    switch (i) {
      case 0:
      case 1:
        File page = null;
        if (i == 0)
          page = conf.getResource("var/package.html");  
        else
          page = conf.getResource("var/license.html");  

        JEditorPane pager = new JEditorPane(page.toURI().toURL());
        pager.putClientProperty(JEditorPane.W3C_LENGTH_UNITS, true);
        pager.putClientProperty(JEditorPane.HONOR_DISPLAY_PROPERTIES, true);
        pager.addHyperlinkListener(this);
        pager.setEditable(false);
        pager.setMargin(new Insets(0, 0, 0, 0));
        JScrollPane viewport = new JScrollPane(pager);


        c.gridx = c.gridy = 0;
        c.weightx = c.weighty = 1;
        c.insets = new Insets(12, 12, 0, 12);
        c.fill = GridBagConstraints.BOTH;
        lm.setConstraints(viewport, c);
        scr.add(viewport);
        break;
      
      case 2:
        JLabel l = new JLabel("User data in: ");
        l.setFont(fnt);
        l.setForeground(fg);
        c.gridx = c.gridy = 0;
        c.insets = new Insets(12, 12, 0, 0);
        c.anchor = GridBagConstraints.EAST;
        lm.setConstraints(l, c);
        scr.add(l);

        l = new JLabel("Install in: ");
        l.setFont(fnt);
        l.setForeground(fg);
        c.gridy++;
        lm.setConstraints(l, c);
        scr.add(l);

        String pkg = (String) conf.get("installed", "name");
        File f = new File(Registry.getHomeDirectory(), "." + pkg);
        if (Registry.isLinux())
          home = new TextField(f.getCanonicalPath());
        else
          home = new TextField("C:\\Program Files\\" + pkg + "\\");

        home.setFont(fnt);
        home.setForeground(fg);
        home.setEnabled(false);
        c.gridy = 0;
        c.gridx++;
        c.weightx = 1;
        c.insets.left = 2;
        //c.ipady = 7;
        c.anchor = GridBagConstraints.CENTER;
        c.fill = GridBagConstraints.HORIZONTAL;
        lm.setConstraints(home, c);
        scr.add(home);

        if (Registry.isLinux())
          prefix = new TextField("/usr/local/");
        else
          prefix = new TextField("C:\\Program Files\\" + pkg + "\\");

        prefix.setFont(fnt);
        prefix.setForeground(fg);
        c.gridy++;
        lm.setConstraints(prefix, c);
        scr.add(prefix);

        select = new Button("Select", this);
        select.setMnemonic(KeyEvent.VK_S);
        c.gridx++;
        c.weightx = 0;
        c.insets.right = 12;
        //c.ipady = 0;
        c.fill = GridBagConstraints.NONE;
        lm.setConstraints(select, c);
        scr.add(select);

        JCheckBox storeApi = new JCheckBox("Install all documentation", true);
        storeApi.setFont(fnt);
        storeApi.setForeground(fg);
        c.gridx = 1;
        c.gridy++;
        c.anchor = GridBagConstraints.WEST;
        lm.setConstraints(storeApi, c);
        scr.add(storeApi);

        JCheckBox themes = new JCheckBox("Install all secondary themes", true);
        themes.setFont(fnt);
        themes.setForeground(fg);
        c.gridy++;
        lm.setConstraints(themes, c);
        scr.add(themes);

        JCheckBox appIcon = new JCheckBox("Install desktop icon", true);
        appIcon.setFont(fnt);
        appIcon.setForeground(fg);
        c.gridy++;
        lm.setConstraints(appIcon, c);
        scr.add(appIcon);

        progress = new JLabel("Installation progress: Unpacking");
        progress.setFont(fnt);
        progress.setForeground(fg);
        progress.setVisible(false);
        c.gridx = 0;
        c.gridy++;
        c.gridwidth = 3;
        c.insets = new Insets(24, 12, 0, 12);
        c.anchor = GridBagConstraints.WEST;
        lm.setConstraints(progress, c);
        scr.add(progress);

        fnt = (Font) conf.get("progressbar", "font");
        fg = (Color) conf.get("progressbar", "fgcolor");
        setupBar = new JProgressBar(0, 100);
        setupBar.setStringPainted(true);
        setupBar.setFont(fnt);
        setupBar.setForeground(fg);
        setupBar.setVisible(false);
        c.gridy++;
        c.weightx = 1;
        c.insets.top = 12;
        c.fill = GridBagConstraints.HORIZONTAL;
        lm.setConstraints(setupBar, c);
        scr.add(setupBar);

        l = new JLabel();
        c.gridy++;
        c.weighty = 1;
        c.fill = GridBagConstraints.BOTH;
        lm.setConstraints(l, c);
        scr.add(l);
        break;
    }

    screens.add(scr);
  }


  private void showScreen()
  {
    System.out.println("Current stage: " + current);
    Registry conf = Registry.getCurrent();

    viewport.removeAll();
    viewport.add(screens.get(current), BorderLayout.CENTER);
    viewport.repaint();
    viewport.validate();

    switch (current) {
      case 0:
        String pkg = (String) conf.get("installed", "package");
        caption.setText("Installing " + pkg);

        previous.setEnabled(false);
        break;
      
      case 1:
        String license = (String) conf.get("installed", "license");
        caption.setText(license);

        previous.setEnabled(true);
        next.setText("Next");
        next.setActionCommand("Next");
        next.setMnemonic(KeyEvent.VK_N);          
        break;
      
      case 2:
        caption.setText("Select options");

        next.setText("Install");
        next.setActionCommand("Install");
        next.setMnemonic(KeyEvent.VK_I);
    }
  }


  public static void extractJar(File jar, File dir, JProgressBar bar) throws IOException
  {
    if (dir == null) {
      String path = System.getProperty("java.io.tmpdir");
      dir = new File(path, ".jTracer.installer");
    }

    try {
      boolean autodel = false;
      dir.mkdir();
      if (jar == null) {
        autodel = true;
        dir.deleteOnExit();
        jar = new File(System.getProperty("java.class.path"));
      }

      JarInputStream in = new JarInputStream(new FileInputStream(jar));
      ZipEntry entry;

      float bytes = 0, sz = jar.length();
      while ((entry = in.getNextEntry()) != null) {
        File from = new File(entry.getName());
        File to = new File(dir, entry.getName());

        if (entry.isDirectory()) {
          String parts[] = entry.getName().split("/"); //File.separator);
          File subdir = new File(dir, "");

          for (int i = 0; i < parts.length; i++) {
            subdir = new File(subdir, parts[i]);
            System.out.println(parts[i]);
            subdir.mkdir();

            bytes += to.length();
            if (bar != null)
              bar.setValue((int) ((bytes / sz) * 100));
            
            if (autodel)
              subdir.deleteOnExit();
          }

          System.out.println(subdir.getAbsolutePath());
          continue;
        }

        System.out.println(from.getAbsolutePath() + " -> " + to.getAbsolutePath());
        /*String parent = from.getParent();
        if (parent != null) {
          String parts[] = parent.split(File.separator);
          File subdir = new File(dir, "");

          for (int i = 0; i < parts.length; i++) {
            subdir = new File(subdir, parts[i]);
            //System.out.println(subdir.getAbsolutePath());
            subdir.mkdir();
            if (autodel)
              subdir.deleteOnExit();
          }
        }*/

        to.createNewFile();
        if (autodel)
          to.deleteOnExit();

        /*if (bar != null)
          bar.setMaximum((int) to.length());*/

        DataOutputStream out = new DataOutputStream(new FileOutputStream(to));
        while (in.available() == 1) {
          //byte[] data = new byte[1];
          int data = in.read();
          if (data == -1)
            break;
          out.writeByte(data);

          bytes++;
          if (bar != null)
            bar.setValue((int) ((bytes / sz) * 100));
        }
    
        out.close();
      }

      if (!Registry.isLinux() && bar != null)
        bar.setValue(100);
      in.close();
    }

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


  public static void main(String[] argv)
  {
    try {
      if (Registry.isLinux())
        System.out.println("Linux");
      else
        System.out.println("Windows");

      Registry conf = new Registry();
      conf.setup();

      extractJar(null, null, null);
      MainFrame app = new MainFrame();
      File lockf = conf.getResource("var/.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()
  {
    //while (installer == Thread.currentThread()) {
      try {
        Registry conf = Registry.getCurrent();
        File pkg = conf.getResource("config/package.jar");
        File dest = new File(home.getText());
        dest.mkdirs();

        extractJar(pkg, dest, setupBar);
        progress.setText("Installation progress: Configuring");
        File exec = new File(dest, "jTracer-1.03.jar");
        exec.setExecutable(true, false);

        if (Registry.isLinux()) {
          File theme = new File(dest, "res/theme/default");
          File link = new File(dest, "res/theme/current");
          Files.createSymbolicLink(link.toPath(), theme.toPath());
        
          File launcher = new File(prefix.getText() + "/bin/jTracer");
          launcher.createNewFile();
          launcher.setExecutable(true, false);
          FileWriter out = new FileWriter(launcher);
          out.write("#!/bin/bash\n\n");
          out.write("java -jar " + exec.getAbsolutePath() + " &\n");
          out.close();
        }

        progress.setText("Installation progress: Complete");
        System.out.println("finished");
      }

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

      next.setEnabled(true);
      next.setText("Finish");
      next.setActionCommand("Finish");
      next.setMnemonic(KeyEvent.VK_F);
    //}
  }


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

      if (cmd.equals("Cancel")) {
        Registry conf = Registry.getCurrent();
  
        String pkg = (String) conf.get("installed", "package");
        String msg = "Really cancel the installation of " + pkg + "?";

        if (Alert.prompt(this, msg))
          System.exit(0);
      }

      else if (cmd.equals("Previous")) {
        if (current == 0)
          return;
        
        current--;
        showScreen();
      }
      
      else if (cmd.equals("Next")) {
        if (current == screens.size() - 1)
          return;

        current++;
        showScreen();
      }

      else if (cmd.equals("Select")) {
        JFileChooser chooser = new JFileChooser();
        chooser.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);

        int retval = chooser.showDialog(this, "Select");
        if (retval == JFileChooser.APPROVE_OPTION)
          prefix.setText(chooser.getSelectedFile().getPath());
      }

      else if (cmd.equals("Install")) {
        progress.setVisible(true);
        setupBar.setVisible(true);
        
        next.setEnabled(false);
        previous.setEnabled(false);
        select.setEnabled(false);
        prefix.setEnabled(false);

        setupBar.setValue(0);       
        installer = new Thread(this);
        installer.start();
      }
      
      else if (cmd.equals("Finish"))
        System.exit(0);
      
      else if (cmd.startsWith("About")) {
        if (about == null)
          about = new AboutDialog(this);
        
        about.setLocationRelativeTo(this);
        about.setVisible(true);
      }
    }

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


  public void hyperlinkUpdate(HyperlinkEvent event)
  {
    try {
      if (event.getEventType() != HyperlinkEvent.EventType.ACTIVATED)
        return;

      URL url = event.getURL();
      String proto = url.getProtocol();
      Registry conf = Registry.getCurrent();

      if (proto.equals("file")) {
        if (about != null)
          about.setVisible(false);
      }

      else if (proto.equals("mailto"))
        conf.mail(url);

      else if (proto.equals("http"))
        conf.browse(url);
    }

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