Installer.java

Go to the documentation of this file.

/*
 * Autopsy Forensic Browser
 *
 * Copyright 2011-2017 Basis Technology Corp.
 * Contact: carrier <at> sleuthkit <dot> org
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.sleuthkit.autopsy.corecomponents;

import java.awt.Font;
import java.awt.GraphicsEnvironment;
import java.awt.Insets;
import java.beans.PropertyChangeEvent;
import java.beans.PropertyChangeListener;
import java.util.Collection;
import java.util.Enumeration;
import java.util.Iterator;
import java.util.Map;
import java.util.TreeMap;
import java.util.logging.Level;
import java.util.stream.Stream;
import javax.swing.BorderFactory;
import javax.swing.ImageIcon;
import javax.swing.SwingUtilities;
import javax.swing.UIManager;
import javax.swing.UIManager.LookAndFeelInfo;
import javax.swing.UnsupportedLookAndFeelException;
import javax.swing.plaf.FontUIResource;
import org.netbeans.spi.sendopts.OptionProcessor;
import org.netbeans.swing.tabcontrol.plaf.DefaultTabbedContainerUI;
import org.openide.modules.ModuleInstall;
import org.openide.util.Lookup;
import org.openide.windows.WindowManager;
import org.sleuthkit.autopsy.casemodule.StartupWindowProvider;
import org.sleuthkit.autopsy.commandlineingest.CommandLineIngestManager;
import org.sleuthkit.autopsy.centralrepository.eventlisteners.CRDefaultSetupAction;
import org.sleuthkit.autopsy.centralrepository.eventlisteners.CentralRepositoryNotificationDialog;
import org.sleuthkit.autopsy.commandlineingest.CommandLineOptionProcessor;
import org.sleuthkit.autopsy.coreutils.Logger;
import org.sleuthkit.autopsy.coreutils.ModuleSettings;

public class Installer extends ModuleInstall {

    private static final long serialVersionUID = 1L;
    private static final Logger logger = Logger.getLogger(Installer.class.getName());
    private static Installer instance;

    public synchronized static Installer getDefault() {
        if (null == instance) {
            instance = new Installer();
        }
        return instance;
    }

    private Installer() {
        super();
    }

    @Override
    public void restored() {
        super.restored();
        Collection<? extends CommandLineOptionProcessor> optionProcessors = Lookup.getDefault().lookupAll(CommandLineOptionProcessor.class);
        CommandLineOptionProcessor processor = null;
        if(!optionProcessors.isEmpty()) {
            Iterator<? extends CommandLineOptionProcessor> iter = optionProcessors.iterator();
             while (iter.hasNext() || processor == null) {
                 processor = iter.next();
             }
        }
        
        
        if (!GraphicsEnvironment.isHeadless()) {
            SwingUtilities.invokeLater(() -> {
                setLookAndFeel();
            });

            UIManager.put("ViewTabDisplayerUI", "org.sleuthkit.autopsy.corecomponents.NoTabsTabDisplayerUI");
            UIManager.put(DefaultTabbedContainerUI.KEY_VIEW_CONTENT_BORDER, BorderFactory.createEmptyBorder());
            UIManager.put("TabbedPane.contentBorderInsets", new Insets(0, 0, 0, 0));
        }
        
        final CommandLineOptionProcessor finalprocessor = processor;
        
        final boolean crFirstTimeSetup = CRDefaultSetupAction.getInstance().setupDefaultCentralRepository();
        
        // When the --nogui flag is supplied invokeWhenUIReady happens a lot sooner
        // than it would when running wiht the gui. Makes sense. That means that 
        // the starupWindowprovider may get kicked off before the command line
        // options have been processed. To solve this issue, check to see if 
        // the command line options have been processed if they haven't, wait for them.
        // Why || instead of &&. If the main window is visible assume that the -nogui
        // switch was not supplied and that things are running normally.
        // Why not just listen to the command processor instead of using the invokeWhen? 
        // If there were no command line options supplied then the process method will never
        // be called. 
        if (GraphicsEnvironment.isHeadless()) {
            if (finalprocessor.getState() == CommandLineOptionProcessor.ProcessState.COMPLETED) {
                new CommandLineIngestManager().start();
            } else {
                finalprocessor.addPropertyChangeListener(new PropertyChangeListener() {
                    @Override
                    public void propertyChange(PropertyChangeEvent evt) {
                        if (evt.getPropertyName().equals(CommandLineOptionProcessor.PROCESSING_COMPLETED)) {
                            new CommandLineIngestManager().start();
                        }
                    }
                });
            }   
        } else {
            WindowManager.getDefault().invokeWhenUIReady(() -> {
                if(WindowManager.getDefault().getMainWindow().isVisible() || finalprocessor == null || finalprocessor.getState() == CommandLineOptionProcessor.ProcessState.COMPLETED) {
                    showStartupWindows(crFirstTimeSetup);
                } else {
                    finalprocessor.addPropertyChangeListener(new PropertyChangeListener(){
                        @Override
                        public void propertyChange(PropertyChangeEvent evt) {
                            if(evt.getPropertyName().equals(CommandLineOptionProcessor.PROCESSING_COMPLETED)) {
                                showStartupWindows(crFirstTimeSetup);
                            }
                        }
                    });
                }
            });
        }
    }
    
    private void showStartupWindows(boolean crFirstTimeSetup) {
        if (crFirstTimeSetup && CentralRepositoryNotificationDialog.shouldDisplay()) {
            CentralRepositoryNotificationDialog.display();
        }
        
        StartupWindowProvider.getInstance().open();
    }

    @Override
    public void uninstalled() {
        super.uninstalled();
    }
    
    private void setLookAndFeel() {
        
        ImageIcon questionIcon = new ImageIcon(getClass().getResource("/org/sleuthkit/autopsy/images/question_32.png"));
        ImageIcon warningIcon = new ImageIcon(getClass().getResource("/org/sleuthkit/autopsy/images/warning_32.png"));
        ImageIcon informationIcon = new ImageIcon(getClass().getResource("/org/sleuthkit/autopsy/images/information_32.png"));
        ImageIcon errorIcon = new ImageIcon(getClass().getResource("/org/sleuthkit/autopsy/images/error_32.png"));
        UIManager.put("OptionPane.errorIcon", errorIcon);
        UIManager.put("OptionPane.warningIcon", warningIcon);
        UIManager.put("OptionPane.questionIcon", questionIcon);
        UIManager.put("OptionPane.informationIcon", informationIcon);

        if (System.getProperty("os.name").toLowerCase().contains("mac")) { //NON-NLS
            setUnixLookAndFeel();
            setModuleSettings("false");
        }else if (System.getProperty("os.name").toLowerCase().contains("nux")){
            setUnixLookAndFeel();
        }
    }

    private void setOSXLookAndFeel() {
        try {
            UIManager.setLookAndFeel(UIManager.getSystemLookAndFeelClassName());
        } catch (ClassNotFoundException | InstantiationException | IllegalAccessException | UnsupportedLookAndFeelException ex) {
            logger.log(Level.WARNING, "Error setting OS-X look-and-feel", ex); //NON-NLS
        }


        // Store the keys that deal with menu items
        final String[] UI_MENU_ITEM_KEYS = new String[]{"MenuBarUI",}; //NON-NLS    
        Map<Object, Object> uiEntries = new TreeMap<>();
        for (String key : UI_MENU_ITEM_KEYS) {
            uiEntries.put(key, UIManager.get(key));
        }

        // Use Metal if available.
        for (LookAndFeelInfo info : UIManager.getInstalledLookAndFeels()) {
            if ("Nimbus".equals(info.getName())) { //NON-NLS
                try {
                    UIManager.setLookAndFeel(info.getClassName());
                } catch (ClassNotFoundException | InstantiationException | IllegalAccessException | UnsupportedLookAndFeelException ex) {
                    logger.log(Level.WARNING, "Error setting OS-X look-and-feel", ex); //NON-NLS
                }
                break;
            }
        }

        // Overwrite the Metal menu item keys to use the Aqua versions.
        uiEntries.entrySet().stream().forEach((entry) -> {
            UIManager.put(entry.getKey(), entry.getValue());
        });
    }
    
    private static void setUIFont (javax.swing.plaf.FontUIResource f){
    java.util.Enumeration<Object> keys = UIManager.getDefaults().keys();
    while (keys.hasMoreElements()) {
      Object key = keys.nextElement();
      Object value = UIManager.getDefaults().get(key);
      if (value instanceof javax.swing.plaf.FontUIResource)
        UIManager.put(key, f);
      }
    } 
    
    private void setModuleSettings(String value) {
        if (ModuleSettings.configExists("timeline")) {
            ModuleSettings.setConfigSetting("timeline", "enable_timeline", value);
        } else {
            ModuleSettings.makeConfigFile("timeline");
            ModuleSettings.setConfigSetting("timeline", "enable_timeline", value);
        }
    }
    
    private void setUnixLookAndFeel(){
        try {
            UIManager.put("swing.boldMetal", Boolean.FALSE);
            UIManager.setLookAndFeel(UIManager.getCrossPlatformLookAndFeelClassName());
            setUIFont (new javax.swing.plaf.FontUIResource("DejaVu Sans Condensed", Font.PLAIN, 11));
            setModuleSettings("true");
        } catch (ClassNotFoundException | InstantiationException | IllegalAccessException | UnsupportedLookAndFeelException ex) {
            logger.log(Level.WARNING, "Error setting crossplatform look-and-feel, setting default look-and-feel",ex);
            setModuleSettings("false");
        }
    }
}