org.logdoc.fairhttp.helpers.CookieKeeper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fair-http-client Show documentation

LogDoc.org project http client

There is a newer version: 1.0.7

package org.logdoc.fairhttp.helpers;

import org.logdoc.fairhttp.structs.traits.Headers;
import org.logdoc.helpers.Digits;

import java.net.HttpCookie;
import java.net.URLConnection;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.regex.Pattern;
import java.util.stream.Collectors;

import static org.logdoc.helpers.Texts.isEmpty;
import static org.logdoc.helpers.Texts.notNull;

/**
 * @author Denis Danilin | [email protected]
 * 05.12.2022 17:03
 * fairhttp ☭ sweat and blood
 */
public class CookieKeeper {
    private final Set store = ConcurrentHashMap.newKeySet();

    public void save(final URLConnection connection) {
        String headerName;

        for (int i = 1; (headerName = connection.getHeaderFieldKey(i)) != null; i++)
            if (headerName.equalsIgnoreCase(Headers.GetCookies)) {
                save(notNull(connection.getHeaderField(i)));
                break;
            }
    }

    public void save(final String cookieHeader) {
        store.addAll(HttpCookie.parse(notNull(cookieHeader)));
    }

    public String load(final String host, final int port, final String path) {
        final String coma = Pattern.quote(",");

        return store.stream()
                .filter(c ->
                        !c.hasExpired()
                                && (isEmpty(c.getDomain()) || HttpCookie.domainMatches(c.getDomain(), host))
                                && (isEmpty(c.getPath()) || path.startsWith(c.getPath()))
                                && (isEmpty(c.getPortlist()) || Arrays.stream(c.getPortlist().split(coma)).map(Digits::getInt).collect(Collectors.toSet()).contains(port))
                )
                .map(c -> c.getName() + "=" + c.getValue())
                .collect(Collectors.joining(";"));
    }

    public List cookie(final String name) {
        return isEmpty(name) ? Collections.emptyList() :
                store.stream()
                        .filter(c -> c.getName().equalsIgnoreCase(name))
                        .collect(Collectors.toList());
    }
}