org.protege.editor.owl.ui.ontology.imports.missing.MissingImportHandlerUI Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of protege-editor-owl
OWL ontology editing infrastructure used by the Protege desktop application.
The newest version!
package org.protege.editor.owl.ui.ontology.imports.missing;

import org.protege.editor.core.ui.util.UIUtil;
import org.protege.editor.owl.OWLEditorKit;
import org.protege.editor.owl.model.MissingImportHandler;
import org.protege.editor.owl.model.library.OntologyCatalogManager;
import org.protege.editor.owl.ui.UIHelper;
import org.protege.xmlcatalog.CatalogUtilities;
import org.protege.xmlcatalog.XMLCatalog;
import org.protege.xmlcatalog.entry.UriEntry;
import org.semanticweb.owlapi.model.IRI;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.annotation.Nonnull;
import javax.swing.*;
import java.awt.*;
import java.io.File;
import java.io.IOException;
import java.net.URI;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.FutureTask;


/**
 * Author: Matthew Horridge

 * The University Of Manchester

 * Medical Informatics Group

 * Date: 31-Aug-2006



 * [email protected]

 * www.cs.man.ac.uk/~horridgm


 */
public class MissingImportHandlerUI implements MissingImportHandler {

    private final Logger logger = LoggerFactory.getLogger(MissingImportHandlerUI.class);

    private final OWLEditorKit owlEditorKit;


    public MissingImportHandlerUI(OWLEditorKit owlEditorKit) {
        this.owlEditorKit = owlEditorKit;
    }


    public IRI getDocumentIRI(@Nonnull IRI ontologyIRI) {
        FutureTask futureTask = new FutureTask<>(() -> {
            int ret = JOptionPane.showConfirmDialog(null,
                    "The system couldn't locate the ontology

" + ontologyIRI.toString() + "

" +

                            "Would you like to attempt to resolve the missing import?",
                    "Resolve missing import?",
                    JOptionPane.YES_NO_OPTION,
                    JOptionPane.WARNING_MESSAGE);
            if (ret != JOptionPane.YES_OPTION) {
                return null;
            }
            UIHelper helper = new UIHelper(owlEditorKit);
            File file = helper.chooseOWLFile("Please select an ontology file");
            if (file == null) {
                return ontologyIRI;
            }
            updateActiveCatalog(ontologyIRI, file);
            return IRI.create(file);
        });

        SwingUtilities.invokeLater(futureTask);
        try {
            return futureTask.get();
        } catch (InterruptedException e) {
            logger.debug("Resolve import task interrupted");
            return null;
        } catch (ExecutionException e) {
            logger.error("An exception was thrown whilst the user was resolving a missing import: {}", e.getCause().getMessage());
            return null;
        }
    }
    
    private void updateActiveCatalog(IRI ontologyIRI, File file) {
        OntologyCatalogManager catalogManager = owlEditorKit.getOWLModelManager().getOntologyCatalogManager();
        catalogManager.getCurrentCatalog().ifPresent(catalog -> {
            URI relativeFile = CatalogUtilities.relativize(file.toURI(), catalog);
            catalog.addEntry(0, new UriEntry("User Entered Import Resolution", catalog, ontologyIRI.toString(), relativeFile, null));
            File catalogLocation = new File(catalog.getXmlBaseContext().getXmlBase());
            try {
                CatalogUtilities.save(catalog, catalogLocation);
            }
            catch (IOException e) {
                logger.error("Could not save user supplied import redirection to catalog.", e);
            }
        });
    }
}