package net.oni2.aeinstaller.updater.gui; import java.awt.BorderLayout; import java.awt.Dimension; import java.awt.event.ActionEvent; import java.awt.event.ActionListener; import java.awt.event.WindowAdapter; import java.awt.event.WindowEvent; import java.io.File; import java.io.FileNotFoundException; import java.util.List; import javax.swing.JButton; import javax.swing.JFrame; import javax.swing.JLabel; import javax.swing.JOptionPane; import javax.swing.JProgressBar; import javax.swing.SwingWorker; import net.oni2.aeinstaller.updater.backend.Paths; import net.oni2.platformtools.applicationinvoker.ApplicationInvoker; import net.oni2.platformtools.applicationinvoker.EExeType; import net.oni2.platformtools.applicationinvoker.ERuntimeNotInstalledException; import net.oni2.svnaccess.SVN; import net.oni2.svnaccess.SVNUpdateListener; /** * @author Christian Illy */ public class MainWin extends JFrame { private static final long serialVersionUID = -3653187495409881426L; JLabel step = new JLabel("Preparing"); JProgressBar bar = new JProgressBar(0, 1); JButton closeBtn = new JButton("Close and launch AEI"); /** * Constructor of main window. */ public MainWin() { super("AEInstaller2 self updater"); System.out.println("AEI2 updater version 1.4"); setLayout(new BorderLayout(2, 4)); bar.setPreferredSize(new Dimension(250, 16)); closeBtn.setEnabled(false); add(bar, BorderLayout.CENTER); add(step, BorderLayout.NORTH); add(closeBtn, BorderLayout.SOUTH); setResizable(false); setDefaultCloseOperation(DO_NOTHING_ON_CLOSE); pack(); setLocationRelativeTo(null); closeBtn.addActionListener(new ActionListener() { @Override public void actionPerformed(ActionEvent e) { exit(); } }); addWindowListener(new WindowAdapter() { @Override public void windowClosing(WindowEvent e) { super.windowClosing(e); if (closeBtn.isEnabled()) { exit(); } } }); Updater upd = new Updater(); upd.execute(); } private void exit() { File aei = new File(new File(Paths.getInstallerPath(), "bin"), "AEInstaller2.jar"); if (aei.exists()) { try { ApplicationInvoker .execute(EExeType.JAR, null, aei, null, false); } catch (FileNotFoundException e) { // TODO Auto-generated catch block e.printStackTrace(); } catch (ERuntimeNotInstalledException e) { // TODO Auto-generated catch block e.printStackTrace(); } } System.out.println("Closing AEI updater"); dispose(); } class Status { public Status(int done, int total) { this.done = done; this.total = total; } /** * Steps done */ public int done; /** * Steps in total to do */ public int total; } class Updater extends SwingWorker { protected void setStep(String text) { step.setText(text); System.out.println(text); } @Override protected Status doInBackground() throws Exception { setStep("Waiting for AEI to close"); int i = 0; while (!checkWritable() && i < 20) { i++; Thread.sleep(500); } if (i >= 20) { JOptionPane .showMessageDialog( null, "Could not update because the main file of AEI was locked.\nPerhaps you are still running an instance of AEI?", "Could not update!", JOptionPane.ERROR_MESSAGE); System.out.println("Could not update because the main file of AEI was locked."); System.exit(1); return null; } setStep("Updating"); SVN svn = new SVN(); try { boolean showError = false; File wcDir = new File(Paths.getPrefsPath(), "bin"); System.out.println("AEI WC path: " + wcDir.getAbsolutePath()); int x = svn.checkSVN("http://svn.aei.oni2.net", wcDir); switch (x) { case -2: // No repos connection System.out.println("Error: No repository connection"); showError = true; break; case 0: // Repos up to date System.out.println("AEI up to date"); break; case -1:// no WC yet case 1:// Update available case 2:// missing files switch (x) { case -1: System.out.println("No working copy created so far"); break; case 1: System.out.println("Updating to HEAD"); break; case 2: System.out.println("Updating for missing files"); break; } showError = !svn.updateWC("http://svn.aei.oni2.net", wcDir, new SVNUpdateListener() { public void statusUpdate(int done, int total) { publish(new Status(done, total)); } }); if (showError) { System.out.println("Error: Updating failed!"); } break; } if (showError) { JOptionPane .showMessageDialog( null, "Perhaps you don't have a internet connection right now or\nyou have (not) entered (wrong) proxy data?", "Updating AEI failed!", JOptionPane.ERROR_MESSAGE); System.exit(1); } } catch (Exception e) { e.printStackTrace(); } return null; } private boolean checkWritable() { File aei = new File(new File(Paths.getInstallerPath(), "bin"), "AEInstaller2.jar"); File temp = new File(new File(Paths.getInstallerPath(), "bin"), "temp.jar"); if (!aei.exists()) return true; if (aei.renameTo(temp)) { temp.renameTo(aei); return true; } return false; } @Override protected void process(List chunks) { super.process(chunks); if (chunks.size() > 0) { Status s = chunks.get(chunks.size() - 1); bar.setValue(s.done); bar.setMaximum(s.total); } } @Override protected void done() { super.done(); step.setText("AEI is up to date"); bar.setValue(1); bar.setMaximum(1); closeBtn.setEnabled(true); } } }