Implementing /wd/hub/sessions endpoint in selenium server

Author: barancev

java/server/src/org/openqa/selenium/remote/server/ActiveSessions.java

@@ -25,6 +25,7 @@
 import org.openqa.selenium.remote.server.log.LoggingManager;
 import org.openqa.selenium.remote.server.log.PerSessionLogHandler;
 
+import java.util.Collection;
 import java.util.LinkedList;
 import java.util.List;
 import java.util.concurrent.TimeUnit;
@@ -85,6 +86,10 @@ public void invalidate(SessionId id) {
     allSessions.invalidate(id);
   }
 
+  public Collection<ActiveSession> getAllSessions() {
+    return allSessions.asMap().values();
+  }
+
   public void addListener(ActiveSessionListener listener) {
     listeners.add(listener);
   }

java/server/src/org/openqa/selenium/remote/server/AllHandlers.java

@@ -33,6 +33,7 @@
 import org.openqa.selenium.remote.server.commandhandler.NoHandler;
 import org.openqa.selenium.remote.server.commandhandler.NoSessionHandler;
 import org.openqa.selenium.remote.server.commandhandler.Status;
+import org.openqa.selenium.remote.server.commandhandler.GetAllSessions;
 
 import java.lang.reflect.Constructor;
 import java.util.List;
@@ -61,6 +62,7 @@ public AllHandlers(ActiveSessions allSessions) {
         HttpMethod.DELETE, ImmutableList.of(),
         HttpMethod.GET, ImmutableList.of(
             handler("/session/{sessionId}/log/types", GetLogTypes.class),
+            handler("/sessions", GetAllSessions.class),
             handler("/status", Status.class)
         ),
         HttpMethod.POST, ImmutableList.of(

java/server/src/org/openqa/selenium/remote/server/commandhandler/GetAllSessions.java

@@ -0,0 +1,69 @@
+// Licensed to the Software Freedom Conservancy (SFC) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The SFC licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+package org.openqa.selenium.remote.server.commandhandler;
+
+import static com.google.common.net.MediaType.JSON_UTF_8;
+import static java.net.HttpURLConnection.HTTP_OK;
+import static java.nio.charset.StandardCharsets.UTF_8;
+import static org.openqa.selenium.remote.ErrorCodes.SUCCESS;
+
+import com.google.common.collect.ImmutableMap;
+import com.google.gson.GsonBuilder;
+
+import org.openqa.selenium.remote.http.HttpRequest;
+import org.openqa.selenium.remote.http.HttpResponse;
+import org.openqa.selenium.remote.server.ActiveSessions;
+import org.openqa.selenium.remote.server.CommandHandler;
+
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+
+public class GetAllSessions implements CommandHandler {
+
+  private volatile ActiveSessions allSessions;
+
+  public GetAllSessions(ActiveSessions allSessions) {
+    this.allSessions = allSessions;
+  }
+
+  @Override
+  public void execute(HttpRequest req, HttpResponse resp) throws IOException {
+    List<Map<String, Object>> value = new ArrayList<>();
+
+    allSessions.getAllSessions().forEach(s -> value.add(
+        ImmutableMap.of("id", s.getId().toString(), "capabilities", s.getCapabilities())));
+
+    Map<String, Object> payloadObj = ImmutableMap.of(
+        "status", SUCCESS,
+        "value", value);
+
+    // Write out a minimal W3C status response.
+    byte[] payload = new GsonBuilder()
+        .serializeNulls()
+        .create()
+        .toJson(payloadObj).getBytes(UTF_8);
+
+    resp.setStatus(HTTP_OK);
+    resp.setHeader("Content-Type", JSON_UTF_8.toString());
+    resp.setHeader("Content-Length", String.valueOf(payload.length));
+
+    resp.setContent(payload);
+  }
+}