io.sitoolkit.wt.gui.pres.EditorTabController Maven / Gradle / Ivy

Go to download
package io.sitoolkit.wt.gui.pres;

import java.io.File;
import java.nio.file.Path;
import java.util.Optional;

import org.controlsfx.control.spreadsheet.SpreadsheetView;

import io.sitoolkit.wt.domain.debug.DebugListener;
import io.sitoolkit.wt.domain.testscript.TestScript;
import io.sitoolkit.wt.gui.app.script.ScriptFileType;
import io.sitoolkit.wt.gui.app.script.ScriptService;
import io.sitoolkit.wt.gui.infra.fx.FxContext;
import io.sitoolkit.wt.gui.infra.fx.ScriptDialog;
import io.sitoolkit.wt.gui.pres.editor.TestScriptEditor;
import javafx.application.Platform;
import javafx.beans.property.BooleanProperty;
import javafx.beans.property.SimpleBooleanProperty;
import javafx.scene.control.SingleSelectionModel;
import javafx.scene.control.Tab;
import javafx.scene.control.TabPane;
import javafx.scene.control.Tooltip;
import lombok.Setter;

public class EditorTabController implements FileOpenable, DebugListener {

    @Setter
    TabPane tabs;

    ScriptService scriptService;

    TestScriptEditor testScriptEditor = new TestScriptEditor();

    ScriptDialog scriptDialog = new ScriptDialog();

    private BooleanProperty empty = new SimpleBooleanProperty(true);

    public void initialize() {
        tabs.getSelectionModel().selectedItemProperty()
                .addListener((observable, oldVal, newVal) -> {
                    if (newVal == null) {
                        empty.set(true);
                    } else {
                        empty.set(false);
                    }
                });
    }

    @Override
    public void open(File file) {
        open(file, Optional.empty());
    }

    private void open(File file, Optional fileType) {
        Optional scriptTab = tabs.getTabs().stream()
                .filter(tab -> tab.getTooltip().getText().equals(file.getAbsolutePath())).findFirst();

        if (scriptTab.isPresent()) {
            SingleSelectionModel selectionModel = tabs.getSelectionModel();
            selectionModel.select(scriptTab.get());

        } else {
            TestScript testScript = scriptService.read(file, fileType);
            Tab tab = new Tab(file.getName());
            SpreadsheetView spreadSheet = testScriptEditor.buildSpreadsheet(testScript);
            tab.setContent(spreadSheet);
            tabs.getSelectionModel().select(tab);

            Tooltip tooltip = new Tooltip();
            tooltip.setText(file.getAbsolutePath());
            tab.setTooltip(tooltip);
            tabs.getTabs().add(tab);
        }
    }

    public void save() {
        Optional selectedTab = tabs.getTabs().stream().filter(Tab::isSelected).findFirst();

        SpreadsheetView spreadSheet = (SpreadsheetView) selectedTab.get().getContent();
        TestScript testScript = testScriptEditor.buildTestscript(spreadSheet);
        scriptService.write(testScript);
    }

    public void open() {
        File file = scriptDialog.showOpenDialog(FxContext.getPrimaryStage());
        if (file != null) {
            Optional fileType = scriptDialog.getSelectedFileType();
            open(file, fileType);
        }
    }

    public void saveAs() {
        File file = scriptDialog.showSaveDialog(FxContext.getPrimaryStage());
        if (file != null) {

            Optional fileType = scriptDialog.getSelectedFileType();
            Tab selectedTab = tabs.getTabs().stream().filter(Tab::isSelected).findFirst().get();

            SpreadsheetView spreadSheet = (SpreadsheetView) selectedTab.getContent();
            TestScript testScript = testScriptEditor.buildTestscript(spreadSheet);
            testScript.setScriptFile(file);
            scriptService.write(testScript, fileType);
            selectedTab.setText(file.getName());
            selectedTab.getTooltip().setText(file.getAbsolutePath());
            spreadSheet.setId(file.getAbsolutePath());
        }

    }

    public BooleanProperty isEmpty() {
        return this.empty;
    }


    @Override
    public void onDebugging(Path scriptPath, int stepIndex, int caseIndex) {
        Platform.runLater(() -> {
            Optional targetView = getSpreadSheet(scriptPath);
            targetView.ifPresent(view -> {
                testScriptEditor.setDebugStyle(view, stepIndex, caseIndex);
            });
        });
    }

    @Override
    public void onCaseEnd(Path scriptPath, int caseIndex) {
        Platform.runLater(() -> {
            Optional targetView = getSpreadSheet(scriptPath);
            targetView.ifPresent(view -> {
                testScriptEditor.removeDebugStyle(view);
            });
        });
    }

    @Override
    public void onClose() {
        Platform.runLater(() -> {
            tabs.getTabs().stream().map(Tab::getContent).map(content -> (SpreadsheetView) content)
                    .forEach(testScriptEditor::removeDebugStyle);
        });
    }

    private Optional getTab(Path scriptPath) {
        String absolutePath = scriptPath.toAbsolutePath().toString();
        return tabs.getTabs().stream()
                .filter(tab -> tab.getTooltip().getText().equals(absolutePath)).findFirst();
    }

    private Optional getSpreadSheet(Path scriptPath) {
        return getTab(scriptPath)
                .map(Tab::getContent)
                .map(content -> (SpreadsheetView) content);
    }

}