From b0212d4b00a1e7f26c50b4402b2a9f80b03e9dbc Mon Sep 17 00:00:00 2001
From: Jianhui Zhao <jianhuizhao329@gmail.com>
Date: Wed, 17 Jan 2018 23:55:12 +0800
Subject: [PATCH] Support Query online devices

Signed-off-by: Jianhui Zhao <jianhuizhao329@gmail.com>
---
 rttys.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/rttys.go b/rttys.go
index a7d0250..5985c70 100755
--- a/rttys.go
+++ b/rttys.go
@@ -240,6 +240,16 @@ func serveWs(w http.ResponseWriter, r *http.Request) {
     go wsConn.wsWriteLoop()
 }
 
+func handlerList(w http.ResponseWriter, r *http.Request) {
+    devs := make([]string, 0)
+    for k, _ := range dev2wsConnection {
+        devs = append(devs, k)
+    }
+
+    js, _ := json.Marshal(devs)
+    fmt.Fprintf(w, "%s", js)
+}
+
 func main() {
     port := flag.Int("port", 5912, "http service port")
     flag.Parse()
@@ -254,6 +264,7 @@ func main() {
 
     http.HandleFunc("/ws/device", serveWs)
     http.HandleFunc("/ws/browser", serveWs)
+    http.HandleFunc("/list", handlerList)
     http.Handle("/", http.FileServer(statikFS))
     log.Fatal(http.ListenAndServe(":" + strconv.Itoa(*port), nil))
 }