aeinstaller/gui/MainWin.java

package net.oni2.aeinstaller.gui;

import java.awt.BorderLayout;
import java.awt.Desktop;
import java.awt.GridLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ItemEvent;
import java.awt.event.ItemListener;
import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.Date;
import java.util.HashMap;
import java.util.HashSet;
import java.util.ResourceBundle;
import java.util.TreeMap;
import java.util.TreeSet;
import java.util.Vector;

import javax.swing.AbstractAction;
import javax.swing.Icon;
import javax.swing.ImageIcon;
import javax.swing.JButton;
import javax.swing.JCheckBox;
import javax.swing.JComboBox;
import javax.swing.JFileChooser;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JMenu;
import javax.swing.JMenuItem;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JRadioButton;
import javax.swing.JSplitPane;
import javax.swing.SwingUtilities;
import javax.swing.ToolTipManager;
import javax.swing.filechooser.FileFilter;

import net.oni2.aeinstaller.AEInstaller2;
import net.oni2.aeinstaller.backend.AppExecution;
import net.oni2.aeinstaller.backend.Paths;
import net.oni2.aeinstaller.backend.Settings;
import net.oni2.aeinstaller.backend.Settings.Platform;
import net.oni2.aeinstaller.backend.SizeFormatter;
import net.oni2.aeinstaller.backend.depot.DepotManager;
import net.oni2.aeinstaller.backend.mods.Mod;
import net.oni2.aeinstaller.backend.mods.ModManager;
import net.oni2.aeinstaller.backend.mods.Type;
import net.oni2.aeinstaller.backend.mods.download.ModDownloader;
import net.oni2.aeinstaller.backend.mods.download.ModDownloader.State;
import net.oni2.aeinstaller.backend.mods.download.ModDownloaderListener;
import net.oni2.aeinstaller.backend.oni.InstallProgressListener;
import net.oni2.aeinstaller.backend.oni.Installer;
import net.oni2.aeinstaller.backend.oni.OniSplit;
import net.oni2.aeinstaller.gui.about.AboutDialog;
import net.oni2.aeinstaller.gui.downloadwindow.Downloader;
import net.oni2.aeinstaller.gui.modtable.DownloadSizeListener;
import net.oni2.aeinstaller.gui.modtable.ModSelectionListener;
import net.oni2.aeinstaller.gui.modtable.ModTable;
import net.oni2.aeinstaller.gui.settings.SettingsDialog;
import net.oni2.aeinstaller.gui.toolmanager.ToolManager;

import org.javabuilders.BuildResult;
import org.javabuilders.annotations.DoInBackground;
import org.javabuilders.event.BackgroundEvent;
import org.javabuilders.swing.SwingJavaBuilder;
import org.simplericity.macify.eawt.ApplicationEvent;
import org.simplericity.macify.eawt.ApplicationListener;

/**
 * @author Christian Illy
 */
public class MainWin extends JFrame implements ApplicationListener,
                DownloadSizeListener, ModSelectionListener {
        private static final long serialVersionUID = -4027395051382659650L;

        private ResourceBundle bundle = ResourceBundle
                        .getBundle("net.oni2.aeinstaller.localization."
                                        + getClass().getSimpleName());
        @SuppressWarnings("unused")
        private BuildResult result = SwingJavaBuilder.build(this, bundle);

        private JMenu mainMenu;
        private JMenu toolsMenu;
        private TreeSet<JMenuItem> toolsMenuItems = new TreeSet<JMenuItem>();

        private JSplitPane contents;

        private JComboBox cmbModTypes;
        private JRadioButton radAll;
        private JRadioButton radOnline;
        private JRadioButton radLocal;
        private ModTable tblMods;
        private JLabel lblDownloadSizeVal;

        private JLabel lblSubmitterVal;
        private JLabel lblCreatorVal;
        private JLabel lblTypesVal;
        private JLabel lblPlatformVal;
        private JLabel lblPackageNumberVal;
        private HTMLLinkLabel lblDescriptionVal;

        private JButton btnInstall;

        private TreeSet<Mod> execUpdates = null;

        private enum EInstallResult {
                DONE,
                OFFLINE,
                INCOMPATIBLE
        };

        private EInstallResult installDone = EInstallResult.DONE;

        /**
         * Constructor of main window.
         */
        public MainWin() {
                this.setTitle(SwingJavaBuilder.getConfig().getResource("appname")
                                + " - v"
                                + SwingJavaBuilder.getConfig().getResource("appversion"));

                contents.setDividerLocation(400);
                contents.setResizeWeight(0.4);

                if (Settings.getPlatform() == Platform.MACOS) {
                        mainMenu.setVisible(false);
                }

                ToolTipManager.sharedInstance().setInitialDelay(250);

                getRootPane().setDefaultButton(btnInstall);
                lblDownloadSizeVal.setText(SizeFormatter.format(0, 2));
                radAll.setSelected(true);

                tblMods.addModSelectionListener(this);
                tblMods.addDownloadSizeListener(this);
        }

        private void initModTypeBox() {
                cmbModTypes.removeAllItems();

                TreeMap<String, Type> types = new TreeMap<String, Type>();
                for (Type t : ModManager.getInstance().getTypesWithContent()) {
                        types.put(t.getName(), t);
                }
                cmbModTypes.addItem("-All-");
                for (Type t : types.values()) {
                        cmbModTypes.addItem(t);
                }
                cmbModTypes.setSelectedIndex(0);
        }

        private void exit() {
                dispose();
                System.exit(0);
        }

        private void saveLocalData() {
                Settings.getInstance().serializeToFile();
                DepotManager.getInstance().saveToFile(Settings.getDepotCacheFilename());
        }

        @DoInBackground(progressMessage = "updateDepot.title", cancelable = false, indeterminateProgress = false)
        private void execDepotUpdate(final BackgroundEvent evt) {
                if (!Settings.getInstance().isOfflineMode()) {
                        long start = new Date().getTime();

                        try {
                                DepotManager.getInstance().updateInformation(false);
                        } catch (Exception e) {
                                e.printStackTrace();
                        }

                        System.out.println("Took: " + (new Date().getTime() - start)
                                        + " msec");
                }

                ModManager.getInstance().init();
                tblMods.reloadData();
                initModTypeBox();

                tblMods.setVisible(true);
        }

        @SuppressWarnings("unused")
        private void checkUpdates(Object evtSource) {
                if ((evtSource != this)
                                || Settings.getInstance().get("notifyupdates", true)) {
                        if (Settings.getInstance().isOfflineMode()) {
                                if (evtSource != this) {
                                        JOptionPane.showMessageDialog(this,
                                                        bundle.getString("offlineMode.text"),
                                                        bundle.getString("offlineMode.title"),
                                                        JOptionPane.WARNING_MESSAGE);
                                }
                        } else {
                                TreeSet<Mod> mods = ModManager.getInstance().getUpdatableMods();
                                TreeSet<Mod> tools = ModManager.getInstance()
                                                .getUpdatableTools();
                                int size = 0;
                                JPanel panPackages = new JPanel(new GridLayout(0, 1));
                                execUpdates = new TreeSet<Mod>();
                                execUpdates.addAll(mods);
                                execUpdates.addAll(tools);
                                for (final Mod m : mods) {
                                        size += m.getZipSize();
                                        JCheckBox check = new JCheckBox("Mod: " + m.getName());
                                        check.setSelected(true);
                                        check.addItemListener(new ItemListener() {
                                                @Override
                                                public void itemStateChanged(ItemEvent e) {
                                                        if (e.getStateChange() == ItemEvent.SELECTED)
                                                                execUpdates.add(m);
                                                        else
                                                                execUpdates.remove(m);
                                                }
                                        });
                                        panPackages.add(check);
                                }
                                for (final Mod m : tools) {
                                        size += m.getZipSize();
                                        JCheckBox check = new JCheckBox("Tool: " + m.getName());
                                        check.setSelected(true);
                                        panPackages.add(check);
                                }
                                if (size > 0) {
                                        // Build info dialog content
                                        JPanel packages = new JPanel(new BorderLayout(0, 7));
                                        JLabel lblIntro = new JLabel("<html>"
                                                        + bundle.getString("updatesAvailable.text")
                                                        + "</html>");
                                        JLabel lblSize = new JLabel("<html>"
                                                        + String.format(bundle
                                                                        .getString("updatesAvailableSize.text"),
                                                                        SizeFormatter.format(size, 3)) + "</html>");
                                        packages.add(lblIntro, BorderLayout.NORTH);
                                        packages.add(panPackages, BorderLayout.CENTER);
                                        packages.add(lblSize, BorderLayout.SOUTH);

                                        JPanel pan = new JPanel(new BorderLayout(0, 25));
                                        pan.add(packages, BorderLayout.CENTER);
                                        JCheckBox checkFutureUpdates = new JCheckBox(
                                                        bundle.getString("checkOnStartup.text"));
                                        checkFutureUpdates.setSelected(Settings.getInstance().get(
                                                        "notifyupdates", true));
                                        checkFutureUpdates.addItemListener(new ItemListener() {
                                                @Override
                                                public void itemStateChanged(ItemEvent evt) {
                                                        Settings.getInstance().put("notifyupdates",
                                                                        evt.getStateChange() == ItemEvent.SELECTED);
                                                }
                                        });
                                        pan.add(checkFutureUpdates, BorderLayout.SOUTH);

                                        // Show dialog
                                        int res = JOptionPane.showConfirmDialog(this, pan,
                                                        bundle.getString("updatesAvailable.title"),
                                                        JOptionPane.YES_NO_OPTION,
                                                        JOptionPane.QUESTION_MESSAGE);
                                        if (res == JOptionPane.NO_OPTION) {
                                                execUpdates = null;
                                        }
                                }
                        }
                }
        }

        @SuppressWarnings("unused")
        private void doUpdate() {
                if (execUpdates != null && execUpdates.size() > 0) {
                        Downloader dl = new Downloader(execUpdates);
                        try {
                                dl.setVisible(true);
                                if (dl.isFinished()) {
                                        TreeSet<Integer> installed = Installer.getInstalledTools();
                                        TreeSet<Mod> tools = new TreeSet<Mod>();
                                        for (Mod m : execUpdates)
                                                if (m.isTool()
                                                                && installed.contains(m.getPackageNumber()))
                                                        tools.add(m);
                                        if (tools.size() > 0) {
                                                Installer.installTools(tools);
                                        }
                                }
                        } finally {
                                dl.dispose();
                        }
                }
                execUpdates = null;
        }

        @SuppressWarnings("unused")
        private void focus() {
                SwingUtilities.invokeLater(new Runnable() {

                        @Override
                        public void run() {
                                toFront();
                                repaint();
                        }
                });

        }

        private void showSettings() {
                new SettingsDialog().setVisible(true);
        }

        private void showAbout() {
                new AboutDialog().setVisible(true);
        }

        private JFileChooser getConfigOpenSaveDialog(boolean save) {
                JFileChooser fc = new JFileChooser();
                fc.setCurrentDirectory(Paths.getEditionBasePath());
                if (save)
                        fc.setDialogType(JFileChooser.SAVE_DIALOG);
                else
                        fc.setDialogType(JFileChooser.OPEN_DIALOG);
                fc.setFileSelectionMode(JFileChooser.FILES_ONLY);
                fc.setFileFilter(new FileFilter() {
                        @Override
                        public String getDescription() {
                                return "XML files";
                        }

                        @Override
                        public boolean accept(File arg0) {
                                return (arg0.isDirectory())
                                                || (arg0.getName().toLowerCase().endsWith(".xml"));
                        }
                });
                fc.setMultiSelectionEnabled(false);
                return fc;
        }

        @SuppressWarnings("unused")
        private void loadConfig() {
                JFileChooser fc = getConfigOpenSaveDialog(false);
                int res = fc.showOpenDialog(this);
                if (res == JFileChooser.APPROVE_OPTION) {
                        if (fc.getSelectedFile().exists())
                                tblMods.reloadSelection(fc.getSelectedFile());
                }
        }

        @SuppressWarnings("unused")
        private void saveConfig() {
                JFileChooser fc = getConfigOpenSaveDialog(true);
                int res = fc.showSaveDialog(this);
                if (res == JFileChooser.APPROVE_OPTION) {
                        File f = fc.getSelectedFile();
                        if (!f.getName().endsWith(".xml"))
                                f = new File(f.getParentFile(), f.getName() + ".xml");
                        ModManager.getInstance().saveModSelection(f,
                                        tblMods.getSelectedMods());
                }
        }

        @DoInBackground(progressMessage = "initializingEdition.title", cancelable = false, indeterminateProgress = false)
        private void reglobalize(final BackgroundEvent evt) {
                Installer.initializeEdition(new InstallProgressListener() {
                        @Override
                        public void installProgressUpdate(int done, int total, String step) {
                                evt.setProgressEnd(total);
                                evt.setProgressValue(done);
                                evt.setProgressMessage(step);
                        }
                });
        }

        @SuppressWarnings("unused")
        private void tools() {
                new ToolManager().setVisible(true);
        }

        @SuppressWarnings("unused")
        private void refreshToolsMenu() {
                for (JMenuItem i : toolsMenuItems) {
                        toolsMenu.remove(i);
                }
                toolsMenuItems.clear();
                for (Mod m : ModManager.getInstance().getInstalledTools()) {
                        if (m.getExeFile() != null && m.getExeFile().exists()) {
                                JMenuItem item = new JMenuItem();
                                final Vector<String> params = new Vector<String>();
                                params.add(m.getExeFile().getPath());
                                final File wd = m.getWorkingDir();
                                Icon ico = null;
                                if (m.getIconFile() != null && m.getIconFile().exists()) {
                                        ico = new ImageIcon(m.getIconFile().getPath());
                                } else {
                                        URL icon = AEInstaller2.class
                                                        .getResource("images/transparent.png");
                                        ico = new ImageIcon(icon);
                                }
                                item.setAction(new AbstractAction(m.getName(), ico) {
                                        private static final long serialVersionUID = 1L;

                                        @Override
                                        public void actionPerformed(ActionEvent e) {
                                                AppExecution.execute(params, wd);
                                        }
                                });
                                toolsMenuItems.add(item);
                                toolsMenu.add(item);
                        }
                }
        }

        @SuppressWarnings("unused")
        private void revertSelection() {
                tblMods.revertSelection();
        }

        @DoInBackground(progressMessage = "mandatoryFiles.title", cancelable = false, indeterminateProgress = false)
        private void checkMandatoryFiles(final BackgroundEvent evt) {
                if (!Settings.getInstance().isOfflineMode()) {
                        TreeSet<Mod> mand = new TreeSet<Mod>();
                        for (Mod m : ModManager.getInstance().getMandatoryTools()) {
                                if (m.isNewerAvailable()) {
                                        mand.add(m);
                                }
                        }
                        for (Mod m : ModManager.getInstance().getMandatoryMods()) {
                                if (m.isNewerAvailable()) {
                                        mand.add(m);
                                }
                        }
                        if (mand.size() > 0) {
                                ModDownloader m = new ModDownloader(mand,
                                                new ModDownloaderListener() {
                                                        @Override
                                                        public void updateStatus(ModDownloader source,
                                                                        State state, int filesDown, int filesTotal,
                                                                        int bytesDown, int bytesTotal,
                                                                        int duration, int remaining, int speed) {
                                                                evt.setProgressEnd(filesTotal);
                                                                evt.setProgressValue(filesDown);
                                                        }
                                                });
                                while (!m.isFinished()) {
                                        try {
                                                Thread.sleep(10);
                                        } catch (InterruptedException e) {
                                                e.printStackTrace();
                                        }
                                }
                        }
                        evt.setProgressMessage(bundle
                                        .getString("mandatoryToolsInstall.title"));
                        Installer
                                        .installTools(ModManager.getInstance().getMandatoryTools());
                }
        }

        @DoInBackground(progressMessage = "installing.title", cancelable = false, indeterminateProgress = false)
        private void install(final BackgroundEvent evt) {
                TreeSet<Mod> mods = new TreeSet<Mod>();
                mods.addAll(ModManager.getInstance().getMandatoryMods());
                mods.addAll(tblMods.getSelectedMods());

                boolean instReady = false;
                installDone = EInstallResult.DONE;

                while (!instReady) {
                        TreeSet<Mod> toDownload = new TreeSet<Mod>();
                        for (Mod m : mods) {
                                if (!m.isLocalAvailable())
                                        toDownload.add(m);
                        }
                        if (Settings.getInstance().isOfflineMode()) {
                                installDone = EInstallResult.OFFLINE;
                                break;
                        }
                        if (toDownload.size() > 0) {
                                Downloader dl = new Downloader(toDownload);
                                try {
                                        dl.setVisible(true);
                                        if (!dl.isFinished())
                                                break;
                                } finally {
                                        dl.dispose();
                                }
                        }
                        HashMap<Mod, HashSet<Mod>> dependencies = ModManager.getInstance()
                                        .checkDependencies(mods);
                        if (dependencies.size() > 0) {
                                System.out.println("Unmet dependencies: "
                                                + dependencies.toString());
                                for (Mod m : dependencies.keySet()) {
                                        for (Mod mDep : dependencies.get(m))
                                                mods.add(mDep);
                                }
                        } else {
                                HashMap<Mod, HashSet<Mod>> conflicts = ModManager.getInstance()
                                                .checkIncompabitilites(mods);
                                if (conflicts.size() > 0) {
                                        installDone = EInstallResult.INCOMPATIBLE;
                                        System.err.println("Incompatible mods: "
                                                        + conflicts.toString());
                                        break;
                                } else {
                                        instReady = true;
                                }
                        }
                }

                if (instReady) {
                        TreeSet<Mod> actuallyMods = new TreeSet<Mod>();
                        TreeSet<Mod> actuallyTools = new TreeSet<Mod>();

                        for (Mod m : mods) {
                                if (m.isTool())
                                        actuallyTools.add(m);
                                else
                                        actuallyMods.add(m);
                        }

                        if (actuallyTools.size() > 0) {
                                Installer.installTools(actuallyTools);
                        }

                        Installer.install(actuallyMods, new InstallProgressListener() {
                                @Override
                                public void installProgressUpdate(int done, int total,
                                                String step) {
                                        evt.setProgressEnd(total);
                                        evt.setProgressValue(done);
                                        evt.setProgressMessage(step);
                                }
                        });
                        installDone = EInstallResult.DONE;
                }
        }

        @SuppressWarnings("unused")
        private void installDone() {
                switch (installDone) {
                        case DONE:
                                JOptionPane.showMessageDialog(this,
                                                bundle.getString("installDone.text"),
                                                bundle.getString("installDone.title"),
                                                JOptionPane.INFORMATION_MESSAGE);
                                break;
                        case OFFLINE:
                                JOptionPane.showMessageDialog(this,
                                                bundle.getString("offlineMode.text"),
                                                bundle.getString("offlineMode.title"),
                                                JOptionPane.WARNING_MESSAGE);
                                break;
                        case INCOMPATIBLE:
                                break;
                }
        }

        @Override
        public void modSelectionChanged(ModTable source, Mod m) {
                lblSubmitterVal.setText("");
                lblCreatorVal.setText("");
                lblDescriptionVal.setText("");
                lblTypesVal.setText("");
                lblPlatformVal.setText("");
                lblPackageNumberVal.setText("");
                if (m != null) {
                        lblSubmitterVal.setText(m.getName());
                        lblCreatorVal.setText(m.getCreator());
                        lblDescriptionVal.setText(m.getDescription());

                        String types = "";
                        for (Type t : m.getTypes()) {
                                if (types.length() > 0)
                                        types += ", ";
                                types += t.getName();
                        }
                        lblTypesVal.setText(types);
                        lblPlatformVal.setText(m.getPlatform().toString());
                        lblPackageNumberVal.setText(m.getPackageNumberString());
                }
        }

        private void updateTableFilter() {
                Object o = cmbModTypes.getSelectedItem();
                Type t = null;
                if (o instanceof Type)
                        t = (Type) o;
                int downloadState = 0;
                if (radOnline.isSelected())
                        downloadState = 1;
                if (radLocal.isSelected())
                        downloadState = 2;
                tblMods.setFilter(t, downloadState);
        }

        @SuppressWarnings("unused")
        private void modTypeSelection() {
                updateTableFilter();
        }

        @SuppressWarnings("unused")
        private void showTypeSelection() {
                updateTableFilter();
        }

        @Override
        public void downloadSizeChanged(int newSize) {
                lblDownloadSizeVal.setText(SizeFormatter.format(newSize, 2));
        }

        @SuppressWarnings("unused")
        private void checkInitialize() {
                if (!Installer.isEditionInitialized()) {
                        if (!OniSplit.isOniSplitInstalled()) {
                                JOptionPane.showMessageDialog(this,
                                                bundle.getString("noOniSplit.text"),
                                                bundle.getString("noOniSplit.title"),
                                                JOptionPane.ERROR_MESSAGE);
                                exit();
                        } else {
                                int res = JOptionPane
                                                .showConfirmDialog(this,
                                                                bundle.getString("askInitialize.text"),
                                                                bundle.getString("askInitialize.title"),
                                                                JOptionPane.YES_NO_OPTION,
                                                                JOptionPane.QUESTION_MESSAGE);
                                if (res == JOptionPane.NO_OPTION) {
                                        saveLocalData();
                                        exit();
                                }
                        }
                }
        }

        @DoInBackground(progressMessage = "initializingEdition.title", cancelable = false, indeterminateProgress = false)
        private void initialize(final BackgroundEvent evt) {
                if (!Installer.isEditionInitialized()) {
                        Installer.initializeEdition(new InstallProgressListener() {
                                @Override
                                public void installProgressUpdate(int done, int total,
                                                String step) {
                                        evt.setProgressEnd(total);
                                        evt.setProgressValue(done);
                                        evt.setProgressMessage(step);
                                }
                        });
                }
        }

        private Vector<String> getBasicOniLaunchParams() {
                Vector<String> params = new Vector<String>();
                File exe = null;
                switch (Settings.getPlatform()) {
                        case WIN:
                                exe = new File(Paths.getEditionBasePath(), "Oni.exe");
                                if (exe.exists())
                                        params.add(exe.getPath());
                                break;
                        case MACOS:
                                exe = new File(Paths.getEditionBasePath(),
                                                "Oni.app/Contents/MacOS/Oni");
                                if (exe.exists())
                                        params.add(exe.getPath());
                                break;
                        case LINUX:
                                String wine = Settings.getWinePath();
                                exe = new File(Paths.getEditionBasePath(), "Oni.exe");
                                if (exe.exists()) {
                                        if (wine != null) {
                                                params.add(wine);
                                                params.add(exe.getPath());
                                        }
                                }
                                break;
                        default:
                }
                if (params.size() > 0) {
                        params.add("-debugfiles");
                }
                return params;
        }

        @SuppressWarnings("unused")
        private void oniFull() {
                Vector<String> params = getBasicOniLaunchParams();
                if (params.size() > 0) {
                        AppExecution.execute(params, Paths.getEditionBasePath());
                }
        }

        @SuppressWarnings("unused")
        private void oniWin() {
                Vector<String> params = getBasicOniLaunchParams();
                if (params.size() > 0) {
                        params.add("-noswitch");
                        AppExecution.execute(params, Paths.getEditionBasePath());
                }
        }

        @SuppressWarnings("unused")
        private void openEditionFolder() {
                try {
                        Desktop.getDesktop().open(Paths.getEditionBasePath());
                } catch (IOException e) {
                        e.printStackTrace();
                }
        }

        @Override
        public void handleAbout(ApplicationEvent event) {
                event.setHandled(true);
                showAbout();
        }

        @Override
        public void handleOpenApplication(ApplicationEvent event) {
        }

        @Override
        public void handleOpenFile(ApplicationEvent event) {
        }

        @Override
        public void handlePreferences(ApplicationEvent event) {
                showSettings();
        }

        @Override
        public void handlePrintFile(ApplicationEvent event) {
        }

        @Override
        public void handleQuit(ApplicationEvent event) {
                event.setHandled(true);
                saveLocalData();
                exit();
        }

        @Override
        public void handleReOpenApplication(ApplicationEvent event) {
        }

}
Revision:	634
Committed:	Sat Jan 19 11:58:17 2013 UTC (12 years, 8 months ago) by alloc
Content type:	text/x-java
Original Path:	*AE/installer2/src/net/oni2/aeinstaller/gui/MainWin.java*
File size:	21490 byte(s)
Log Message:	AEI2 0.92b:\n- Using zipped Depot cache