EnginesController.java

/*
 * Copyright (C) 2015-2017 PÂRIS Quentin
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

package org.phoenicis.javafx.controller.engines;

import javafx.application.Platform;
import jdk.nashorn.api.scripting.ScriptObjectMirror;
import org.apache.commons.lang.StringUtils;
import org.phoenicis.engines.EnginesSource;
import org.phoenicis.engines.dto.EngineCategoryDTO;
import org.phoenicis.engines.dto.EngineDTO;
import org.phoenicis.javafx.controller.apps.AppsController;
import org.phoenicis.javafx.views.common.ConfirmMessage;
import org.phoenicis.javafx.views.common.ErrorMessage;
import org.phoenicis.javafx.views.common.ThemeManager;
import org.phoenicis.javafx.views.mainwindow.engines.ViewEngines;
import org.phoenicis.scripts.interpreter.InteractiveScriptSession;
import org.phoenicis.scripts.interpreter.ScriptInterpreter;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.IOException;
import java.net.URI;
import java.net.URL;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.List;
import java.util.function.Consumer;

public class EnginesController {
    private final org.slf4j.Logger LOGGER = LoggerFactory.getLogger(AppsController.class);
    private final ViewEngines viewEngines;
    private final EnginesSource enginesSource;
    private final ScriptInterpreter scriptInterpreter;
    private ThemeManager themeManager;

    public EnginesController(ViewEngines viewEngines, EnginesSource enginesSource, ScriptInterpreter scriptInterpreter,
            ThemeManager themeManager) {
        this.viewEngines = viewEngines;
        this.enginesSource = enginesSource;
        this.scriptInterpreter = scriptInterpreter;
        this.themeManager = themeManager;

        this.viewEngines.setOnInstallEngine(engineDTO -> {
            new ConfirmMessage("Install " + engineDTO.getVersion(),
                    "Are you sure you want to install " + engineDTO.getVersion() + "?").ask(() -> {
                        installEngine(engineDTO, e -> Platform.runLater(() -> new ErrorMessage("Error", e).show()));
                    });
        });

        this.viewEngines.setOnDeleteEngine(engineDTO -> {
            new ConfirmMessage("Delete " + engineDTO.getVersion(),
                    "Are you sure you want to delete " + engineDTO.getVersion() + "?").ask(() -> {
                        deleteEngine(engineDTO, e -> Platform.runLater(() -> new ErrorMessage("Error", e).show()));
                    });
        });
    }

    public ViewEngines getView() {
        return viewEngines;
    }

    public void loadEngines() {
        enginesSource.fetchAvailableEngines(versions -> Platform.runLater(() -> populateView(versions)));
    }

    private void installEngine(EngineDTO engineDTO, Consumer<Exception> errorCallback) {
        final InteractiveScriptSession interactiveScriptSession = scriptInterpreter.createInteractiveSession();

        interactiveScriptSession.eval("include([\"Functions\", \"Engines\", \"" + engineDTO.getCategory() + "\"]);",
                ignored -> interactiveScriptSession.eval("new Wine()", output -> {
                    final ScriptObjectMirror wine = (ScriptObjectMirror) output;
                    wine.callMember("install", engineDTO.getCategory(), engineDTO.getSubCategory(),
                            engineDTO.getVersion(), engineDTO.getUserData());
                }, errorCallback), errorCallback);
    }

    private void deleteEngine(EngineDTO engineDTO, Consumer<Exception> errorCallback) {
        final InteractiveScriptSession interactiveScriptSession = scriptInterpreter.createInteractiveSession();

        interactiveScriptSession.eval("include([\"Functions\", \"Engines\", \"" + engineDTO.getCategory() + "\"]);",
                ignored -> interactiveScriptSession.eval("new Wine()", output -> {
                    final ScriptObjectMirror wine = (ScriptObjectMirror) output;
                    wine.callMember("delete", engineDTO.getCategory(), engineDTO.getSubCategory(),
                            engineDTO.getVersion(), engineDTO.getUserData());
                }, errorCallback), errorCallback);
    }

    private void populateView(List<EngineCategoryDTO> engineCategoryDTOS) {
        setDefaultEngineIcons(engineCategoryDTOS);
        this.viewEngines.populate(engineCategoryDTOS);
    }

    private void setDefaultEngineIcons(List<EngineCategoryDTO> engineCategoryDTOS) {
        try {
            StringBuilder cssBuilder = new StringBuilder();
            for (EngineCategoryDTO category : engineCategoryDTOS) {
                cssBuilder.append("#" + category.getName().toLowerCase() + "Button{\n");
                URI categoryIcon = category.getIcon();
                if (categoryIcon == null) {
                    cssBuilder
                            .append("-fx-background-image: url('/org/phoenicis/javafx/views/common/phoenicis.png');\n");
                } else {
                    cssBuilder.append("-fx-background-image: url('" + categoryIcon + "');\n");
                }
                cssBuilder.append("}\n");
            }
            String css = cssBuilder.toString();
            Path temp = Files.createTempFile("defaultEngineIcons", ".css").toAbsolutePath();
            File tempFile = temp.toFile();
            tempFile.deleteOnExit();
            Files.write(temp, css.getBytes());
            String defaultEngineIconsCss = temp.toUri().toString();
            themeManager.setDefaultEngineIconsCss(defaultEngineIconsCss);

            // apply current theme again to fix hierarchy
            final String shortName = themeManager.getCurrentTheme().getShortName();
            final String url = String.format("/org/phoenicis/javafx/themes/%s/main.css", shortName);
            final URL style = this.getClass().getResource(url);
            this.viewEngines.getTabPane().getScene().getStylesheets().clear();
            this.viewEngines.getTabPane().getScene().getStylesheets().addAll(themeManager.getDefaultCategoryIconsCss(),
                    defaultEngineIconsCss, style.toExternalForm());
        } catch (IOException e) {
            LOGGER.warn("Could not set default engine icons.", e);
        }
    }
}