Initial commit

src/main/java/be/jaud/Main.java

@@ -0,0 +1,128 @@
+package be.jaud;
+
+import com.sun.net.httpserver.HttpExchange;
+import com.sun.net.httpserver.HttpHandler;
+import com.sun.net.httpserver.HttpServer;
+import org.json.JSONArray;
+import org.json.JSONObject;
+
+import java.io.IOException;
+import java.io.OutputStream;
+import java.net.InetSocketAddress;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.Timer;
+import java.util.TimerTask;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+
+public class Main {
+    private static final HashMap<String, Integer> data = new HashMap<String, Integer>();
+    static OebbCheck check = new OebbCheck();
+
+    public static void main(String[] args) throws IOException {
+        refresh();
+        HttpServer server = HttpServer.create(new InetSocketAddress(8000), 0);
+        server.createContext("/oebbapi", new OebbHandler());
+        server.setExecutor(null); // creates a default executor
+        server.start();
+
+        Runnable task = new Runnable() {
+            @Override
+            public void run() {
+                try {
+                    refresh();
+                } catch (IOException e) {
+                    throw new RuntimeException(e);
+                }
+            }
+        };
+        try (ScheduledExecutorService service = Executors.newSingleThreadScheduledExecutor()) {
+            service.scheduleAtFixedRate(task, 60, 60, TimeUnit.SECONDS);
+        }
+    }
+
+    private static void refresh() throws IOException {
+        data.put("06:16",check.getStatus("1250201","6:16", "Freilassing (Oberbayern)","S 3"));
+        data.put("06:29",check.getStatus("1250201","6:29", "Salzburg Hbf","REX 3"));
+        data.put("06:50",check.getStatus("8100002","6:50", "Freilassing (Oberbayern)","REX 5052"));
+        data.put("06:39",check.getStatus("1250201","6:39", "Freilassing (Oberbayern)","R 3"));
+    }
+
+    static class OebbHandler implements HttpHandler {
+        @Override
+        public void handle(HttpExchange t){
+            String response = createOutput();
+            t.getResponseHeaders().put("Content-Type", Collections.singletonList("application/json"));
+            try {
+                t.sendResponseHeaders(200, response.getBytes().length);
+                OutputStream os = t.getResponseBody();
+                os.write(response.getBytes());
+                os.close();
+            } catch (IOException e) {
+                throw new RuntimeException(e);
+            }
+        }
+    }
+
+    private static String createOutput(){
+        JSONArray dat = new JSONArray();
+
+        JSONObject monitor1 = new JSONObject();
+        monitor1.put("label","Zug");
+        monitor1.put("data","Abfahrten");
+        monitor1.put("labelColor","red");
+        monitor1.put("color","red");
+        monitor1.put("background","silver");
+        dat.put(monitor1);
+
+        JSONObject monitor2 = new JSONObject();
+        monitor2.put("label","S3 - 06:16");
+        monitor2.put("labelColor","green");
+        int zug1 = data.get("06:16");
+        makeLabel(monitor2, zug1);
+        dat.put(monitor2);
+
+        JSONObject monitor3 = new JSONObject();
+        monitor3.put("label","REX3 - 06:29");
+        monitor3.put("labelColor","green");
+        int zug2 = data.get("06:29");
+        makeLabel(monitor3, zug2);
+        dat.put(monitor3);
+
+        JSONObject monitor4 = new JSONObject();
+        monitor4.put("label","REX5052 - 06:50");
+        monitor4.put("labelColor","green");
+        int zug3 = data.get("06:50");
+        makeLabel(monitor4, zug3);
+        dat.put(monitor4);
+
+        JSONObject monitor5 = new JSONObject();
+        monitor5.put("label","R3 - 06:39");
+        monitor5.put("labelColor","green");
+        int zug4 = data.get("06:39");
+        makeLabel(monitor5, zug4);
+        dat.put(monitor5);
+
+        return dat.toString();
+    }
+
+    private static void makeLabel(JSONObject Monitor, int zeit){
+        if(zeit != 0){
+            Monitor.put("color","red");
+            if(zeit > 0 && zeit < 9999){
+                Monitor.put("data",zeit);
+            }else if(zeit == 9999){
+                Monitor.put("data","AUSFALL!");
+            }else{
+                Monitor.put("data","Fehler!");
+            }
+            Monitor.put("background","yellow");
+        } else {
+            Monitor.put("color","green");
+            Monitor.put("data","On time");
+            Monitor.put("background","silver");
+        }
+    }
+}