ImportWizardAction.java

package nl.fontys.sofa.limo.view.wizard.importer;

import java.awt.Component;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.List;
import javax.swing.JComponent;
import nl.fontys.sofa.limo.api.dao.DAO;
import nl.fontys.sofa.limo.api.dao.ProcedureCategoryDAO;
import nl.fontys.sofa.limo.api.service.provider.EventService;
import nl.fontys.sofa.limo.api.service.provider.HubService;
import nl.fontys.sofa.limo.api.service.provider.HubTypeService;
import nl.fontys.sofa.limo.api.service.provider.LegTypeService;
import nl.fontys.sofa.limo.api.service.provider.ProcedureService;
import nl.fontys.sofa.limo.domain.BaseEntity;
import nl.fontys.sofa.limo.domain.component.event.Event;
import nl.fontys.sofa.limo.domain.component.hub.Hub;
import nl.fontys.sofa.limo.domain.component.procedure.Procedure;
import nl.fontys.sofa.limo.domain.component.procedure.ProcedureCategory;
import nl.fontys.sofa.limo.domain.component.type.HubType;
import nl.fontys.sofa.limo.domain.component.type.LegType;
import nl.fontys.sofa.limo.view.StartupOptionProcessor;
import nl.fontys.sofa.limo.view.util.LIMOResourceBundle;
import nl.fontys.sofa.limo.view.wizard.export.ExportWizardAction;
import org.openide.DialogDisplayer;
import org.openide.WizardDescriptor;
import org.openide.awt.ActionID;
import org.openide.awt.ActionReference;
import org.openide.awt.ActionReferences;
import org.openide.awt.ActionRegistration;
import org.openide.util.Lookup;

/**
 * @author Matthias Brück
 */
@ActionID(category = "Master Data", id = "nl.fontys.sofa.limo.view.wizard.importer.ImportWizardAction")
@ActionRegistration(displayName = "Import Master Data...", iconBase = "icons/gui/Database.png")
@ActionReferences({
    @ActionReference(path = "Menu/Master Data", position = 70),
    @ActionReference(path = "Shortcuts", name = "DOS-I")
})
public final class ImportWizardAction implements ActionListener {

    private List<BaseEntity> objectsToOvewrite;
    public static final String LIST = "list";
    public static final String PATH = "path";

    @Override
    public void actionPerformed(ActionEvent e) {
        List<WizardDescriptor.Panel<WizardDescriptor>> wizardDescritorPanels = new ArrayList<>();

        if (!e.getSource().getClass().equals(StartupOptionProcessor.class)) {
            wizardDescritorPanels.add(new ImportFileChooser());
        }
        wizardDescritorPanels.add(new ImportPanel());

        String[] steps = new String[wizardDescritorPanels.size()];
        for (int i = 0; i < wizardDescritorPanels.size(); i++) {
            Component c = wizardDescritorPanels.get(i).getComponent();
            // Default step name to component name of panel.
            steps[i] = c.getName();
            if (c instanceof JComponent) { // assume Swing components
                JComponent component = (JComponent) c;
                component.putClientProperty(WizardDescriptor.PROP_CONTENT_SELECTED_INDEX, i);
                component.putClientProperty(WizardDescriptor.PROP_CONTENT_DATA, steps);
                component.putClientProperty(WizardDescriptor.PROP_AUTO_WIZARD_STYLE, true);
                component.putClientProperty(WizardDescriptor.PROP_CONTENT_DISPLAYED, true);
                component.putClientProperty(WizardDescriptor.PROP_CONTENT_NUMBERED, true);
            }
        }
        WizardDescriptor wizardDescriptor = new WizardDescriptor(new WizardDescriptor.ArrayIterator<>(wizardDescritorPanels));
        if (e.getSource().getClass().equals(StartupOptionProcessor.class)) {
            wizardDescriptor.putProperty(ExportWizardAction.PATH, e.getActionCommand());
        }
        wizardDescriptor.setTitleFormat(new MessageFormat("{0}"));
        wizardDescriptor.setTitle(LIMOResourceBundle.getString("IMPORT_MASTERDATA"));
        if (DialogDisplayer.getDefault().notify(wizardDescriptor) == WizardDescriptor.FINISH_OPTION) {
            handleWizardFinishClick(wizardDescriptor);
        }
    }

    /**
     * Save or update the event based on the inputs.
     *
     * @param wiz - the WizardDescriptor which contains the inputs.
     */
    private void handleWizardFinishClick(final WizardDescriptor wizardDescriptor) {
        objectsToOvewrite = (List<BaseEntity>) wizardDescriptor.getProperty(LIST);
        objectsToOvewrite.stream().map((entity) -> {
            if (entity instanceof ProcedureCategory) {
                ImportWizardAction.<ProcedureCategory>updateItem((ProcedureCategory) entity, ProcedureCategoryDAO.class);
            }
            return entity;
        }).map((entity) -> {
            if (entity instanceof LegType) {
                ImportWizardAction.<LegType>updateItem((LegType) entity, LegTypeService.class);
            }
            return entity;
        }).map((entity) -> {
            if (entity instanceof HubType) {
                ImportWizardAction.<HubType>updateItem((HubType) entity, HubTypeService.class);
            }
            return entity;
        }).map((entity) -> {
            if (entity instanceof Hub) {
                ImportWizardAction.<Hub>updateItem((Hub) entity, HubService.class);
            }
            return entity;
        }).map((entity) -> {
            if (entity instanceof Event) {
                ImportWizardAction.<Event>updateItem((Event) entity, EventService.class);
            }
            return entity;
        }).filter((entity) -> (entity instanceof Procedure)).forEach((entity) -> {
            ImportWizardAction.<Procedure>updateItem((Procedure) entity, ProcedureService.class);
        });
    }

    /**
     * Updates an entity inside the database.
     *
     * @param <T> The Class of the entitiy. Needs to extend BaseEntity
     * @param item The item with the new information that has to overwrite the
     * old one.
     * @param serviceClass The Class of the service that communicates with the
     * database.
     */
    private static <T extends BaseEntity> void updateItem(T item, Class serviceClass) {
        DAO service = (DAO) Lookup.getDefault().lookup(serviceClass);
        service.update(item);
    }
}