io.webfolder.ui4j.api.interceptor.Request Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ui4j-api Show documentation
Web Automation for Java
There is a newer version: 4.0.0
package io.webfolder.ui4j.api.interceptor;

import java.net.HttpCookie;
import java.net.URLConnection;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;


import java.util.Optional;

import static java.util.Collections.unmodifiableMap;
import static java.util.Arrays.asList;
import static java.util.Collections.emptyList;

public class Request {

    private String url;

    private Map> headers = new HashMap<>();

    private URLConnection urlConnection;

    public Request(String url) {
        this.url = url;
    }

    public Map> getHeaders() {
        return unmodifiableMap(headers);
    }

    public Optional getHeader(String name) {
        List values = headers.get(name);
        if (values != null && !values.isEmpty()) {
            return Optional.of(values.get(0));
        }
        return Optional.empty();
    }

    public String getUrl() {
        return url;
    }

    public Request setHeader(String name, String... values) {
        headers.put(name, asList(values));
        return this;
    }

    public Request setCookie(HttpCookie cookie) {
        List cookies = new ArrayList<>();
        cookies.add(cookie);
        setCookies(cookies);
        return this;
    }

    public boolean removeHeader(String name) {
        List list = headers.remove(name);
        return list != null ? true : false;
    }

    public boolean removeHeaders(List names) {
        boolean removed = false;
        if (names != null) {
            for (String name : names) {
                removed &= removeHeader(name);
            }
        }
        return removed;
    }

    public List getCookies() {
        List cookies = headers.get("Cookie");
        if (cookies == null || cookies.isEmpty()) {
            return emptyList();
        }
        String cookie = cookies.get(0);
        if (cookie == null || cookie.trim().isEmpty()) {
            return emptyList();
        }
        List list = new ArrayList<>();
        for (String next : cookie.split(";")) {
            list.addAll(HttpCookie.parse(next));
        }
        return list;
    }

    public boolean removeCookie(String name) {
        List cookies = getCookies();
        List newCookies = new ArrayList<>();
        if (name != null && !name.trim().isEmpty()) {
            for (HttpCookie cookie : cookies) {
                if (!cookie.getName().equals(name)) {
                    newCookies.add(cookie);
                }
            }
        }
        setCookies(newCookies);
        return false;
    }

    public boolean removeCookies(List names) {
        boolean removed = false;
        if (names != null) {
            for (String name : names) {
                removed &= removeCookie(name);
            }
        }
        return removed;
    }

    public Request setCookies(List cookies) {
        StringBuilder builder = new StringBuilder();
        for (HttpCookie cookie : cookies) {
            builder
                .append(cookie.getName())
                .append("=")
                .append(cookie.getValue())
                .append(";");
        }
        setHeader("Cookie", builder.toString());
        return this;
    }

    public URLConnection getUrlConnection() {
        return urlConnection;
    }

    public void setUrlConnection(URLConnection urlConnection) {
        this.urlConnection = urlConnection;
    }

    @Override
    public String toString() {
        return "Request [url=" + url + ", headers=" + headers
                + ", urlConnection=" + urlConnection + "]";
    }
}