com.automation.remarks.remote.hub.Video Maven / Gradle / Ivy

Go to download
package com.automation.remarks.remote.hub;

import com.automation.remarks.remote.node.VideoServlet;
import com.google.common.base.Joiner;
import org.apache.http.HttpStatus;
import org.openqa.grid.internal.ProxySet;
import org.openqa.grid.internal.Registry;
import org.openqa.grid.internal.RemoteProxy;
import org.openqa.grid.web.servlet.RegistryBasedServlet;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;

import static com.automation.remarks.remote.utils.RestUtils.sendRecordingRequest;
import static com.automation.remarks.remote.utils.RestUtils.updateResponse;

/**
 * Created by Serhii_Pirohov on 11.05.2016.
 */
public class Video extends RegistryBasedServlet {

    public Video() {
        this(null);
    }

    public Video(Registry registry) {
        super(registry);
    }

    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        doPost(req, resp);
    }

    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        ProxySet allProxies = getRegistry().getAllProxies();
        String path = req.getPathInfo();
        String params = mapToParams(req);

        try {
            for (RemoteProxy proxy : allProxies) {
                String proxyId = proxy.getId();
                final String url = proxyId +
                        "/extra/" + VideoServlet.class.getSimpleName() + path + "?" + params;
                String response = sendRecordingRequest(url);
                updateResponse(resp, HttpStatus.SC_OK, proxyId + " video command " + path + " " + response);
            }
        } catch (Exception ex) {
            updateResponse(resp, HttpStatus.SC_INTERNAL_SERVER_ERROR,
                    "Internal server error occurred while trying to start / stop recording on hub: " + ex);
        }
    }

    private String mapToParams(HttpServletRequest req) {
        Enumeration parameterNames = req.getParameterNames();
        Map map = new HashMap<>();
        while (parameterNames.hasMoreElements()){
            String parameter = parameterNames.nextElement();
            String value = req.getParameter(parameter);
            map.put(parameter,value);
        }
        return Joiner.on("&").withKeyValueSeparator("=").join(map);
    }

}