package com.twasyl.slideshowfx.snippet.executor.go;
import com.sun.javafx.PlatformUtil;
import com.twasyl.slideshowfx.global.configuration.GlobalConfiguration;
import com.twasyl.slideshowfx.snippet.executor.AbstractSnippetExecutor;
import com.twasyl.slideshowfx.snippet.executor.CodeSnippet;
import com.twasyl.slideshowfx.utils.beans.converter.FileStringConverter;
import com.twasyl.slideshowfx.utils.io.DefaultCharsetReader;
import javafx.collections.FXCollections;
import javafx.collections.ObservableList;
import javafx.scene.Node;
import javafx.scene.Parent;
import javafx.scene.control.*;
import javafx.scene.layout.HBox;
import javafx.scene.layout.VBox;
import javafx.stage.DirectoryChooser;
import java.io.*;
import java.util.StringJoiner;
import java.util.logging.Level;
import java.util.logging.Logger;
/**
* An implementation of {@link com.twasyl.slideshowfx.snippet.executor.AbstractSnippetExecutor} that allows to execute
* Go code snippets.
* This implementation is identified with the code {@code GO}.
*
* @author Thierry Wasyczenko
* @version 1.0
* @since SlideshowFX 1.0
*/
public class GoSnippetExecutor extends AbstractSnippetExecutor<GoSnippetExecutorOptions> {
private static final Logger LOGGER = Logger.getLogger(GoSnippetExecutor.class.getName());
protected static final String GO_HOME_PROPERTY_SUFFIX = ".home";
protected static final String WRAP_IN_MAIN_PROPERTY = "wrapInMain";
protected static final String IMPORTS_PROPERTY = "imports";
protected static final String PACKAGE_NAME_PROPERTY = "package";
public GoSnippetExecutor() {
super("GO", "Go", "language-go");
this.setOptions(new GoSnippetExecutorOptions());
final String goHome = GlobalConfiguration.getProperty(this.getConfigurationBaseName().concat(GO_HOME_PROPERTY_SUFFIX));
if(goHome != null) {
try {
this.getOptions().setGoHome(new File(goHome));
} catch (FileNotFoundException e) {
LOGGER.log(Level.SEVERE, "Can not set the GO_HOME", e);
}
}
}
@Override
public Parent getUI(final CodeSnippet codeSnippet) {
final TextField packageTextField = new TextField();
packageTextField.setPromptText("Package name");
packageTextField.setPrefColumnCount(10);
packageTextField.setTooltip(new Tooltip("The package name of this code snippet"));
packageTextField.textProperty().addListener((textValue, oldText, newText) -> {
if(newText == null || newText.isEmpty()) codeSnippet.putProperty(PACKAGE_NAME_PROPERTY, null);
else codeSnippet.putProperty(PACKAGE_NAME_PROPERTY, newText);
});
final CheckBox wrapInMain = new CheckBox("Wrap code snippet in main");
wrapInMain.setTooltip(new Tooltip("Wrap the provided code snippet in a Go main method"));
wrapInMain.selectedProperty().addListener((selectedValue, oldSelected, newSelected) -> {
if(newSelected != null) codeSnippet.putProperty(WRAP_IN_MAIN_PROPERTY, newSelected.toString());
});
final TextArea imports = new TextArea();
imports.setPromptText("Imports");
imports.setTooltip(new Tooltip("Specify packages that must be imported. Provide one package per line, not surrounded by double-quotes"));
imports.setPrefColumnCount(15);
imports.setPrefRowCount(15);
imports.setWrapText(true);
imports.textProperty().addListener((textValue, oldText, newText) -> {
if(newText.isEmpty()) codeSnippet.putProperty(IMPORTS_PROPERTY, null);
else codeSnippet.putProperty(IMPORTS_PROPERTY, newText);
});
final VBox ui = new VBox(5);
ui.getChildren().addAll(wrapInMain, imports);
return ui;
}
@Override
public Node getConfigurationUI() {
this.newOptions = new GoSnippetExecutorOptions();
try {
this.newOptions.setGoHome(this.getOptions().getGoHome());
} catch (FileNotFoundException | NullPointerException e) {
LOGGER.log(Level.FINE, "Can not duplicate GO_HOME", e);
}
final Label label = new Label(this.getLanguage().concat(":"));
final TextField javaHomeField = new TextField();
javaHomeField.textProperty().bindBidirectional(this.newOptions.goHomeProperty(), new FileStringConverter());
javaHomeField.setPrefColumnCount(20);
final Button browse = new Button("...");
browse.setOnAction(event -> {
final DirectoryChooser chooser = new DirectoryChooser();
final File sdkHomeDir = chooser.showDialog(null);
if (sdkHomeDir != null) {
javaHomeField.setText(sdkHomeDir.getAbsolutePath());
}
});
final HBox box = new HBox(5);
box.getChildren().addAll(label, javaHomeField, browse);
return box;
}
@Override
public void saveNewOptions() {
if(this.getNewOptions() != null) {
this.setOptions(this.getNewOptions());
if(this.getOptions().getGoHome() != null) {
GlobalConfiguration.setProperty(this.getConfigurationBaseName().concat(GO_HOME_PROPERTY_SUFFIX),
this.getOptions().getGoHome().getAbsolutePath().replaceAll("\\\\", "/"));
}
}
}
@Override
public ObservableList<String> execute(final CodeSnippet codeSnippet) {
final ObservableList<String> consoleOutput = FXCollections.observableArrayList();
final Thread snippetThread = new Thread(() -> {
File codeFile = null;
try {
codeFile = createSourceCodeFile(codeSnippet);
} catch (IOException e) {
LOGGER.log(Level.SEVERE, "Can not write code to snippet file", e);
consoleOutput.add("ERROR: ".concat(e.getMessage()));
}
// Execute the Go file
final File goExecutable = PlatformUtil.isWindows() ?
new File(this.getOptions().getGoHome(), "bin/go.exe") :
new File(this.getOptions().getGoHome(), "bin/go");
final String[] compilationCommand = {goExecutable.getAbsolutePath(), "run", codeFile.getName()};
Process process = null;
try {
process = new ProcessBuilder()
.redirectErrorStream(true)
.command(compilationCommand)
.directory(this.getTemporaryDirectory())
.start();
try (final BufferedReader errorStream = new BufferedReader(new InputStreamReader(process.getInputStream()))) {
errorStream.lines().forEach(line -> consoleOutput.add(line));
}
} catch (IOException e) {
LOGGER.log(Level.SEVERE, "Can not execute code snippet", e);
consoleOutput.add("ERROR: ".concat(e.getMessage()));
} finally {
if(process != null) {
try {
process.waitFor();
} catch (InterruptedException e) {
LOGGER.log(Level.SEVERE, "Can not wait for process to end", e);
}
}
}
codeFile.delete();
});
snippetThread.start();
return consoleOutput;
}
/**
* Create the source code file for the given code snippet.
* @param codeSnippet The code snippet.
* @return The file created and containing the source code.
*/
protected File createSourceCodeFile(final CodeSnippet codeSnippet) throws IOException {
final File codeFile = new File(this.getTemporaryDirectory(), determinePackageName(codeSnippet).concat(".golo"));
try (final FileWriter codeFileWriter = new FileWriter(codeFile)) {
codeFileWriter.write(buildSourceCode(codeSnippet));
codeFileWriter.flush();
}
return codeFile;
}
/**
*
* Build code file content according properties. The source code can then be written properly inside a file in order
* to be compiled and then executed.
* @param codeSnippet The code snippet to build the source code for.
* @return The content of the source code file.
*/
protected String buildSourceCode(final CodeSnippet codeSnippet) throws IOException {
final StringBuilder sourceCode = new StringBuilder();
sourceCode.append(getStartPackageDefinition(codeSnippet)).append("\n\n");
if(hasImports(codeSnippet)) {
sourceCode.append(getImports(codeSnippet)).append("\n\n");
}
if(mustBeWrappedInMain(codeSnippet)) {
sourceCode.append(getStartMainMethod()).append("\n")
.append(codeSnippet.getCode())
.append("\n").append(getEndMainMethod());
} else {
sourceCode.append(codeSnippet.getCode());
}
sourceCode.append(getEndModuleDefinition(codeSnippet));
return sourceCode.toString();
}
/**
* Get the imports to be added to the source code.
* @param codeSnippet The code snippet.
*/
protected boolean hasImports(final CodeSnippet codeSnippet) {
final String imports = codeSnippet.getProperties().get(IMPORTS_PROPERTY);
return imports != null && !imports.isEmpty();
}
/**
* Get the imports for the code snippets. If some lines of the imports don't contain the {@code import} keyword, it
* will be added properly.
*
* @param codeSnippet The code snippet.
* @return A well formatted string containing all imports.
*/
protected String getImports(final CodeSnippet codeSnippet) throws IOException {
final StringJoiner imports = new StringJoiner("\n");
try (final StringReader stringReader = new StringReader(codeSnippet.getProperties().get(IMPORTS_PROPERTY));
final BufferedReader reader = new DefaultCharsetReader(stringReader)) {
reader.lines()
.filter(line -> !line.trim().isEmpty())
.forEach(line -> imports.add("\t".concat(formatImportLine(line))));
}
return "import (\n".concat(imports.toString()).concat("\n)");
}
/**
* Format an import line by make sure it starts with the {@code import} keyword.
* @param importLine The import line to format.
* @return A well formatted import line.
*/
protected String formatImportLine(final String importLine) {
final String importLineBeginning = "\"";
final String importLineEnding = "\"";
String formattedImportLine;
if(importLine.startsWith(importLineBeginning)) {
formattedImportLine = importLine;
} else {
formattedImportLine = importLineBeginning.concat(importLine);
}
if(!importLine.endsWith(importLineEnding)) {
formattedImportLine = formattedImportLine.concat(importLineEnding);
}
return formattedImportLine;
}
/**
* Get the definition of the class.
* @param codeSnippet The code snippet.
*/
protected String getStartPackageDefinition(final CodeSnippet codeSnippet) {
return "package ".concat(determinePackageName(codeSnippet));
}
/**
* Determine the package name of the code snippet. It looks inside the code snippet's properties and check the value
* of the {@link #PACKAGE_NAME_PROPERTY} property. If {@code null} or empty, {@code Snippet} will be returned.
* @param codeSnippet The code snippet.
* @return The package name of the code snippet.
*/
protected String determinePackageName(final CodeSnippet codeSnippet) {
String packageName = codeSnippet.getProperties().get(PACKAGE_NAME_PROPERTY);
if(packageName == null || packageName.isEmpty()) packageName = "slideshowfx";
return packageName;
}
/**
* Determine if the code snippet must be wrapped inside a main method. It is determined by the presence and value of
* the {@link #WRAP_IN_MAIN_PROPERTY} property.
* @param codeSnippet The code snippet.
* @return {@code true} if the snippet must be wrapped in main, {@code false} otherwhise.
*/
protected boolean mustBeWrappedInMain(final CodeSnippet codeSnippet) {
final Boolean wrapInMain = codeSnippet.getProperties().containsKey(WRAP_IN_MAIN_PROPERTY) ?
Boolean.parseBoolean(codeSnippet.getProperties().get(WRAP_IN_MAIN_PROPERTY)) :
false;
return wrapInMain;
}
/**
* Get the start of the declaration of the main method.
* @return The start of the main method.
*/
protected String getStartMainMethod() {
return "func main() {";
}
/**
* Get the end of the declaration of the main method.
* @return The end of the main method.
*/
protected String getEndMainMethod() {
return "}";
}
/**
* Get the end of the definition of the class.
* @param codeSnippet The code snippet.
*/
protected String getEndModuleDefinition(final CodeSnippet codeSnippet) {
return "";
}
}