LibraryController.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.library;

import javafx.application.Platform;
import org.phoenicis.javafx.controller.library.console.ConsoleController;
import org.phoenicis.javafx.views.common.ConfirmMessage;
import org.phoenicis.javafx.views.common.ErrorMessage;
import org.phoenicis.javafx.views.mainwindow.library.ViewLibrary;
import org.phoenicis.library.LibraryManager;
import org.phoenicis.library.ShortcutManager;
import org.phoenicis.library.ShortcutRunner;
import org.phoenicis.library.dto.ShortcutCategoryDTO;
import org.phoenicis.library.dto.ShortcutDTO;
import org.phoenicis.scripts.interpreter.ScriptInterpreter;

import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;

public class LibraryController {
    private final ViewLibrary viewLibrary;
    private final ConsoleController consoleController;
    private final LibraryManager libraryManager;
    private final ShortcutRunner shortcutRunner;
    private final ShortcutManager shortcutManager;
    private final ScriptInterpreter scriptInterpreter;

    private String keywords = "";

    public LibraryController(ViewLibrary viewLibrary, ConsoleController consoleController,
            LibraryManager libraryManager, ShortcutRunner shortcutRunner, ShortcutManager shortcutManager,
            ScriptInterpreter scriptInterpreter) {
        this.consoleController = consoleController;

        this.viewLibrary = viewLibrary;
        this.libraryManager = libraryManager;
        this.shortcutRunner = shortcutRunner;
        this.shortcutManager = shortcutManager;
        this.scriptInterpreter = scriptInterpreter;
        this.viewLibrary.populate(libraryManager.fetchShortcuts());

        libraryManager.setOnUpdate(this::updateLibrary);

        this.viewLibrary.setOnSearch(searchKeyword -> {
            keywords = searchKeyword;
            this.updateLibrary();
        });

        this.viewLibrary.setOnShortcutRun(this::runShortcut);
        this.viewLibrary.setOnShortcutDoubleClicked(this::runShortcut);
        this.viewLibrary
                .setOnShortcutStop(shortcutDTO -> shortcutRunner.stop(shortcutDTO, e -> new ErrorMessage("Error", e)));

        this.viewLibrary.setOnShortcutUninstall(shortcutDTO -> {
            new ConfirmMessage("Uninstall " + shortcutDTO.getName(),
                    "Are you sure you want to uninstall " + shortcutDTO.getName() + "?")
                            .ask(() -> shortcutManager.uninstallFromShortcut(shortcutDTO,
                                    e -> new ErrorMessage("Error while uninstalling " + shortcutDTO.getName(), e)));
        });

        this.viewLibrary.setOnOpenConsole(() -> {
            viewLibrary.createNewTab(consoleController.createConsole());
        });

        this.viewLibrary.setOnScriptRun(file -> {
            scriptInterpreter.runScript(file,
                    e -> Platform.runLater(() -> new ErrorMessage("Error while running script", e)));
        });
    }

    private void runShortcut(ShortcutDTO shortcutDTO) {
        shortcutRunner.run(shortcutDTO, Collections.emptyList(), e -> new ErrorMessage("Error", e));
    }

    public void setOnTabOpened(Runnable onTabOpened) {
        this.viewLibrary.setOnTabOpened(onTabOpened);
    }

    private void updateLibrary() {
        final List<ShortcutCategoryDTO> categories = libraryManager.fetchShortcuts();
        final List<ShortcutCategoryDTO> shortcutsCorrespondingToKeywords = categories.stream()
                .filter(shortcutDTO -> shortcutDTO.getName().toLowerCase().contains(keywords.toLowerCase().trim()))
                .collect(Collectors.toList());

        Platform.runLater(() -> this.viewLibrary.populate(shortcutsCorrespondingToKeywords));
    }

    public ViewLibrary getView() {
        return viewLibrary;
    }
}