com.hiczp.bilibili.api.web.cookie.SimpleCookieJar Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of bilibili-api Show documentation

Bilibili Android client API library for Kotlin

There is a newer version: 0.2.0

package com.hiczp.bilibili.api.web.cookie;

import okhttp3.Cookie;
import okhttp3.CookieJar;
import okhttp3.HttpUrl;

import javax.annotation.Nonnull;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Vector;
import java.util.concurrent.ConcurrentHashMap;
import java.util.stream.Collectors;

public class SimpleCookieJar implements CookieJar {
    private final Map> cookiesMap;

    public SimpleCookieJar() {
        cookiesMap = new ConcurrentHashMap<>();
    }

    public SimpleCookieJar(@Nonnull Map> cookiesMap) {
        this();
        cookiesMap.forEach((domain, cookies) ->
                this.cookiesMap.put(domain, new Vector<>(cookies))
        );
    }

    @Override
    public void saveFromResponse(HttpUrl url, List cookies) {
        cookies.forEach(cookie -> {
            String domain = cookie.domain();
            List savedCookies;
            synchronized (cookiesMap) {
                savedCookies = cookiesMap.get(domain);
                if (savedCookies == null) {
                    savedCookies = new Vector<>();
                    savedCookies.add(cookie);
                    cookiesMap.put(domain, savedCookies);
                    return;
                }
            }
            for (int i = savedCookies.size() - 1; i >= 0; i--) {
                Cookie current = savedCookies.get(i);
                if (current.name().equals(cookie.name())) {
                    savedCookies.remove(current);
                }
            }
            savedCookies.add(cookie);
        });
    }

    @Override
    public List loadForRequest(HttpUrl url) {
        return getCookiesForHost(url.host());
    }

    public List getCookiesForHost(@Nonnull String host) {
        List cookieList = new ArrayList<>();
        cookiesMap.forEach((domain, cookies) -> {
            if (host.endsWith(domain)) {
                //移除过期的 cookies
                for (int i = cookies.size() - 1; i >= 0; i--) {
                    Cookie current = cookies.get(i);
                    if (current.expiresAt() < System.currentTimeMillis()) {
                        cookies.remove(current);
                    }
                }
                cookieList.addAll(cookies);
            }
        });
        return cookieList;
    }

    public String getCookiesStringForHost(String host) {
        return getCookiesForHost(host).stream()
                .map(cookie -> String.format("%s=%s", cookie.name(), cookie.value()))
                .collect(Collectors.joining(";"));
    }

    public Map> getCookiesMap() {
        return cookiesMap;
    }
}