support for installing extensions, working extensions filesystem

src/META-INF/MANIFEST.MF

@@ -0,0 +1,3 @@
+Manifest-Version: 1.0
+Main-Class: main.extensions.examples.AdminOnConnect
+

src/main/extensions/Extension.java

@@ -4,10 +4,7 @@ import main.protocol.HMessage;
 import main.protocol.HPacket;
 import main.ui.extensions.Extensions;
 
-import java.io.DataInputStream;
+import java.io.*;
-import java.io.IOException;
-import java.io.InputStream;
-import java.io.OutputStream;
 import java.net.Socket;
 import java.util.ArrayList;
 import java.util.HashMap;
@@ -63,7 +60,15 @@ public abstract class Extension {
 
             while (!gEarthExtensionServer.isClosed()) {
 
-                int length = dIn.readInt();
+                int length;
+                try {
+                    length = dIn.readInt();
+                }
+                catch(EOFException exception) {
+                    //g-earth closed the extension
+                    break;
+                }
+
                 byte[] headerandbody = new byte[length + 4];
 
                 int amountRead = 0;
@@ -142,10 +147,11 @@ public abstract class Extension {
 
                 }
             }
-
+            System.out.println("Extension closed");
 
         } catch (IOException | ArrayIndexOutOfBoundsException e) {
-            e.printStackTrace();
+            System.err.println("Connection failed; is G-Earth open?");
+//            e.printStackTrace();
         }
         finally {
             if (gEarthExtensionServer != null && !gEarthExtensionServer.isClosed()) {

src/main/extensions/examples/AdminOnConnect.java

@@ -20,7 +20,7 @@ public class AdminOnConnect extends Extension {
     private boolean done = true;
 
     protected void init() {
-        intercept(HMessage.Side.TOCLIENT, -1, message -> {
+        intercept(HMessage.Side.TOCLIENT, message -> {
             if (!done) {
                 HPacket packet = message.getPacket();
                 if (packet.length() == 11) {

src/main/ui/extensions/Extensions.fxml

@@ -83,7 +83,7 @@
                                 <Insets left="3.0" />
                             </GridPane.margin>
                         </Label>
-                        <Button fx:id="btn_install" disable="true" maxHeight="1.7976931348623157E308" maxWidth="1.7976931348623157E308" mnemonicParsing="false" onAction="#installBtnClicked" text="Install" GridPane.columnIndex="3" />
+                        <Button fx:id="btn_install" maxHeight="1.7976931348623157E308" maxWidth="1.7976931348623157E308" mnemonicParsing="false" onAction="#installBtnClicked" text="Install" GridPane.columnIndex="3" />
                     </children>
                 </GridPane>
             </children>

src/main/ui/extensions/Extensions.java

@@ -9,10 +9,19 @@ import javafx.scene.control.ScrollPane;
 import javafx.scene.control.TextField;
 import javafx.scene.layout.GridPane;
 import javafx.scene.layout.VBox;
+import javafx.stage.FileChooser;
 import main.Main;
 import main.protocol.*;
 import main.ui.SubForm;
+import main.ui.extensions.executer.ExecutionInfo;
+import main.ui.extensions.executer.ExtensionRunner;
+import main.ui.extensions.executer.ExtensionRunnerFactory;
+import main.ui.scheduler.ScheduleItem;
+import sun.misc.ExtensionInfo;
 
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.FileReader;
 import java.io.IOException;
 import java.net.ServerSocket;
 import java.util.*;
@@ -108,6 +117,9 @@ public class Extensions extends SubForm {
     public GridPane header_ext;
     public ScrollPane scroller;
 
+    private ExtensionRunner extensionRunner = null;
+    private GEarthExtensionsRegistrer extensionsRegistrer = null;
+
     public static class OUTGOING_MESSAGES_IDS {
         public static final int ONDOUBLECLICK = 1;
         public static final int INFOREQUEST = 2;        // backend: implemented
@@ -138,6 +150,8 @@ public class Extensions extends SubForm {
     }
 
     protected void onParentSet() {
+
+
         getHConnection().addStateChangeListener((oldState, newState) -> {
             if (newState == HConnection.State.CONNECTED) {
                 for (GEarthExtension extension : gEarthExtensions) {
@@ -217,7 +231,6 @@ public class Extensions extends SubForm {
         });
 
 
-        GEarthExtensionsRegistrer extensionsRegistrer = null;
         HashMap<GEarthExtension, GEarthExtension.ReceiveMessageListener> messageListeners = new HashMap<>();
         try {
             extensionsRegistrer = new GEarthExtensionsRegistrer(new GEarthExtensionsRegistrer.ExtensionRegisterObserver() {
@@ -287,9 +300,22 @@ public class Extensions extends SubForm {
 
         ext_port.setText(extensionsRegistrer.getPort()+"");
 //        System.out.println("Extension server registered on port: " + extensionsRegistrer.getPort());
+
+        extensionRunner = ExtensionRunnerFactory.get();
+        extensionRunner.runAllExtensions(extensionsRegistrer.getPort());
     }
 
 
     public void installBtnClicked(ActionEvent actionEvent) {
+        List<ScheduleItem> list = new ArrayList<>();
+
+        FileChooser fileChooser = new FileChooser();
+        fileChooser.setTitle("Install extension");
+        fileChooser.getExtensionFilters().addAll(
+                new FileChooser.ExtensionFilter("G-Earth extensions", ExecutionInfo.ALLOWEDEXTENSIONTYPES));
+        File selectedFile = fileChooser.showOpenDialog(parentController.getStage());
+        if (selectedFile != null) {
+            extensionRunner.installAndRunExtension(selectedFile.getPath(), extensionsRegistrer.getPort());
+        }
     }
 }

src/main/ui/extensions/executer/ExecutionInfo.java

@@ -0,0 +1,32 @@
+package main.ui.extensions.executer;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * Created by Jonas on 22/09/18.
+ */
+public class ExecutionInfo {
+
+    private static Map<String, String> extensionTypeToExecutionCommand;
+    public final static List<String> ALLOWEDEXTENSIONTYPES;
+    public final static String EXTENSIONSDIRECTORY = "Extensions";
+
+    static {
+        extensionTypeToExecutionCommand = new HashMap<>();
+        extensionTypeToExecutionCommand.put("*.jar","java -jar {path} -p {port}");
+        extensionTypeToExecutionCommand.put("*.py","python {path} -p {port}");
+        extensionTypeToExecutionCommand.put("*.py3","python3 {path} -p {port}");
+        extensionTypeToExecutionCommand.put("*.sh","{path} -p {port}");
+        extensionTypeToExecutionCommand.put("*.exe","{path} -p {port}");
+
+        ALLOWEDEXTENSIONTYPES = new ArrayList<>(extensionTypeToExecutionCommand.keySet());
+    }
+
+    public static String getExecutionCommand(String type) {
+        return extensionTypeToExecutionCommand.get(type);
+    }
+
+}

src/main/ui/extensions/executer/ExtensionRunner.java

@@ -0,0 +1,18 @@
+package main.ui.extensions.executer;
+
+import java.nio.file.Path;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * Created by Jonas on 21/09/18.
+ */
+public interface ExtensionRunner {
+
+    void runAllExtensions(int port);
+
+    void installAndRunExtension(String path, int port);
+
+}

src/main/ui/extensions/executer/ExtensionRunnerFactory.java

@@ -0,0 +1,14 @@
+package main.ui.extensions.executer;
+
+import main.misc.OSValidator;
+
+/**
+ * Created by Jonas on 22/09/18.
+ */
+public class ExtensionRunnerFactory {
+
+    public static ExtensionRunner get() {
+
+        return new NormalExtensionRunner();
+    }
+}

src/main/ui/extensions/executer/NormalExtensionRunner.java

@@ -0,0 +1,110 @@
+package main.ui.extensions.executer;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.*;
+import java.util.Arrays;
+import java.util.Random;
+
+/**
+ * Created by Jonas on 22/09/18.
+ */
+public class NormalExtensionRunner implements ExtensionRunner {
+
+    @Override
+    public void runAllExtensions(int port) {
+        if (dirExists(ExecutionInfo.EXTENSIONSDIRECTORY)){
+            File folder =
+                    new File(FileSystems.getDefault().getPath(".").toString() +
+                            FileSystems.getDefault().getSeparator() +
+                            ExecutionInfo.EXTENSIONSDIRECTORY);
+
+            File[] childs = folder.listFiles();
+
+            for (File file : childs) {
+                tryRunExtension(file.getPath(), port);
+            }
+        }
+    }
+
+    @Override
+    public void installAndRunExtension(String path, int port) {
+        if (!dirExists(ExecutionInfo.EXTENSIONSDIRECTORY)) {
+            createDirectory(ExecutionInfo.EXTENSIONSDIRECTORY);
+        }
+
+
+        String name = Paths.get(path).getFileName().toString();
+        String[] split = name.split("\\.");
+        String ext = "*." + split[split.length - 1];
+
+        String realname = String.join(".",Arrays.copyOf(split, split.length-1));
+        String newname = realname + "-" + getRandomString() + ext.substring(1);
+
+
+        Path originalPath = Paths.get(path);
+        Path newPath = Paths.get(
+                FileSystems.getDefault().getPath(".").toString(),
+                ExecutionInfo.EXTENSIONSDIRECTORY,
+                newname
+        );
+
+        try {
+            Files.copy(
+                    originalPath,
+                    newPath
+            );
+
+            addExecPermission(newPath.toString());
+            tryRunExtension(newPath.toString(), port);
+
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+
+
+    }
+
+    private void tryRunExtension(String path, int port) {
+        try {
+            Runtime.getRuntime().exec(
+                    ExecutionInfo.getExecutionCommand(getFileExtension(path))
+                    .replace("{path}", path)
+                    .replace("{port}", port+"")
+            );
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+    }
+
+    private void addExecPermission(String path) {
+        //not needed at first sight
+    }
+
+    private String getFileExtension(String path) {
+        String name = Paths.get(path).getFileName().toString();
+        String[] split = name.split("\\.");
+        return "*." + split[split.length - 1];
+    }
+    private boolean dirExists(String dir) {
+        return Files.isDirectory(Paths.get(FileSystems.getDefault().getPath(".").toString(), dir));
+    }
+    private void createDirectory(String dir) {
+        if (!dirExists(dir)) {
+            try {
+                Files.createDirectories(Paths.get(FileSystems.getDefault().getPath(".").toString(), dir));
+            } catch (IOException e) {
+                e.printStackTrace();
+            }
+        }
+    }
+    private String getRandomString() {
+        StringBuilder builder = new StringBuilder();
+        Random r = new Random();
+        for (int i = 0; i < 10; i++) {
+            builder.append(r.nextInt(10));
+        }
+
+        return builder.toString();
+    }
+}