From 81e6a46f3bd1039cccc3ebae014b0c178ef2dde1 Mon Sep 17 00:00:00 2001
From: Ian Gulliver <flamingcow@gmail.com>
Date: Wed, 4 May 2016 00:02:12 +0000
Subject: [PATCH] Send image type list notification to new websocket

---
 server/https_server.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/server/https_server.py b/server/https_server.py
index 5bbc8e7..7ed7d83 100755
--- a/server/https_server.py
+++ b/server/https_server.py
@@ -52,9 +52,15 @@ parser.add_argument(
 FLAGS = parser.parse_args()
 
 
-def GetWebSocketHandler(websockets):
+def GetWebSocketHandler(image_types, websockets):
   class WebSocketHandler(websocket.WebSocket):
     def opened(self):
+      self.send(json.dumps({
+        'type': 'image_types',
+        'data': {
+          'image_types': list(image_types),
+        },
+      }), False)
       websockets.add(self)
 
     def closed(self, code, reason=None):
@@ -91,7 +97,7 @@ class HTTPRequestHandler(object):
   def __init__(self, image_path, image_types, websockets):
     self._image_path = image_path
     self._image_types = image_types
-    inner_handler = GetWebSocketHandler(websockets)
+    inner_handler = GetWebSocketHandler(image_types, websockets)
     self._websocket_handler = wsgiutils.WebSocketWSGIApplication(handler_cls=inner_handler)
 
   def __call__(self, env, start_response):