net.dongliu.requests.SetCookie Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of requests Show documentation
There is a newer version: 5.0.8
package net.dongliu.requests;

import net.dongliu.commons.collection.Pair;

import java.time.Instant;
import java.time.format.DateTimeParseException;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;

/**
 * Http response set cookie header
 */
class SetCookie {

    private List> cookies = new ArrayList<>();
    private String domain;
    private boolean bareDomain;
    private String path;
    private Optional expiry = Optional.empty();

    public List> getCookies() {
        return cookies;
    }

    public void addCookie(Pair cookie) {
        this.cookies.add(cookie);
    }

    public String getDomain() {
        return domain;
    }

    public void setDomain(String domain) {
        this.domain = domain;
    }

    public boolean isBareDomain() {
        return bareDomain;
    }

    public void setBareDomain(boolean bareDomain) {
        this.bareDomain = bareDomain;
    }

    public String getPath() {
        return path;
    }

    public void setPath(String path) {
        this.path = path;
    }

    public Optional getExpiry() {
        return expiry;
    }

    public void setExpiry(Optional expiry) {
        this.expiry = expiry;
    }

    public void setExpiry(Instant expiry) {
        this.expiry = Optional.of(expiry);
    }

    public static SetCookie parse(String value) {
        SetCookie setCookie = new SetCookie();
        for (String item : value.split("; ")) {
            Pair pair = Utils.pairFrom(item);
            switch (pair.getName().toLowerCase()) {
                case "domain":
                    String domain = pair.getValue();
                    if (domain.startsWith(".")) {
                        domain = domain.substring(1);
                    }
                    setCookie.setDomain(domain);
                    setCookie.setBareDomain(false);
                    break;
                case "path":
                    setCookie.setPath(pair.getValue().endsWith("/") ? pair.getValue() : pair.getValue() + "/");
                    break;
                case "expires":
                    try {
                        Instant instant = Utils.parseDate(pair.getValue());
                        setCookie.setExpiry(instant);
                    } catch (DateTimeParseException ignore) {
                    }
                    break;
                case "max-age":
                    try {
                        int seconds = Integer.parseInt(pair.getValue());
                        if (seconds >= 0) {
                            Instant instant = Instant.now().plusSeconds(seconds);
                            setCookie.setExpiry(instant);
                        }
                    } catch (NumberFormatException ignore) {
                    }
                    break;
                case "":
                    if (value.equalsIgnoreCase("HttpOnly") || value.equalsIgnoreCase("Secure")) {
                        break;
                    }
                default:
                    setCookie.addCookie(pair);
            }
        }
        return setCookie;
    }
}