Lab04/IFTTTServer.java at master · asawas/Lab04 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
import java.io.IOException;
import java.io.OutputStream;
import java.io.InputStream;
import java.net.InetSocketAddress;
import java.net.URI;
import java.util.List;
import java.util.Map;
import java.util.HashMap;
import java.util.Set;

import com.sun.net.httpserver.Headers;
import com.sun.net.httpserver.HttpExchange;
import com.sun.net.httpserver.HttpHandler;
import com.sun.net.httpserver.HttpServer;
import java.net.HttpURLConnection;


public class IFTTTServer {
  public Map<String,String> events = new HashMap<String,String>();
  public final String PAGESTART = "<html><body><center><h1>Known values</h1></center><body><center><table>";
  public final String PAGEEND = "</table></center></body></html>";

  public class Handler implements HttpHandler {
    public void handle(HttpExchange xchg) throws IOException {
      if (xchg.getRequestMethod().equalsIgnoreCase("GET")) {
        try {
          StringBuffer sb = new StringBuffer(PAGESTART);
          synchronized(events) {
            for(String s: events.keySet())
              sb.append("<tr><td>" + s + "</td><td>" + events.get(s) + "</td></tr>");
          }
          String s = sb.toString();
          xchg.sendResponseHeaders(HttpURLConnection.HTTP_OK, s.length());
          OutputStream os = xchg.getResponseBody();
          os.write(s.getBytes());
          os.close();
          xchg.close();
          return;
        } catch(Exception e) {
          System.out.println("Error " + e);
          return;
        }
      }

      if (xchg.getRequestMethod().equalsIgnoreCase("POST")) {
        try {
          Headers requestHeaders = xchg.getRequestHeaders();

          // expect POST as a JSON
          URI uri = xchg.getRequestURI();
          String kind = requestHeaders.getFirst("Content-type");
          if(!"application/json".equals(kind)) {
          System.out.println("bad application kind " + kind);
            xchg.sendResponseHeaders(HttpURLConnection.HTTP_BAD_REQUEST, 0);
            xchg.close();
            return;
          }

          // expect URI in IFTTT format
          String[] parts = uri.toString().split("/");
          if(parts.length < 2) {
          System.out.println("bad application uri " + uri);
            xchg.sendResponseHeaders(HttpURLConnection.HTTP_BAD_REQUEST, 0);
            xchg.close();
            return;
          }

          // obtain POST text as a string
          int contentLength = Integer.parseInt(requestHeaders.getFirst("Content-length"));
          InputStream is = xchg.getRequestBody();
          byte[] data = new byte[contentLength];
          int length = is.read(data);
          String s = new String(data);
          System.out.println(s);

          // update the map
          synchronized(events) {
            events.put(uri.toString(), s);
          }

          xchg.sendResponseHeaders(HttpURLConnection.HTTP_OK, 0);
          xchg.close();

          } catch (Exception e) {
            System.out.println("Error " + e);
          }
      }
    }
  }

  public void serve(int port) throws IOException {
    HttpServer server = HttpServer.create(new InetSocketAddress(port), 0);
    Handler handler = new Handler();
    server.createContext("/", new Handler());
    server.start();
  }

  public static void main(String[] args) throws IOException {
    final int PORT = 8080;
    int port = PORT;
    switch(args.length) {
    case 0:
      break;
    case 1:
      port = Integer.parseInt(args[0]);
      break;
    default :
      System.out.println("Usage: IFTTTServer [port]");
      System.exit(1);
    }
    (new IFTTTServer()).serve(port);
    /*NOTREACHED*/
  }
}