69 lines
2.0 KiB
Kotlin
69 lines
2.0 KiB
Kotlin
/*
|
|
* Copyright 2017 Hippo Seven
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
package com.hippo.network
|
|
|
|
import okhttp3.Cookie
|
|
import okhttp3.HttpUrl
|
|
|
|
internal class CookieSet {
|
|
private val map: MutableMap<Key, Cookie> = HashMap()
|
|
|
|
/**
|
|
* Adds a cookie to this `CookieSet`.
|
|
* Returns a previous cookie with
|
|
* the same name, domain and path or `null`.
|
|
*/
|
|
fun add(cookie: Cookie): Cookie? {
|
|
return map.put(Key(cookie), cookie)
|
|
}
|
|
|
|
/**
|
|
* Removes a cookie with the same name,
|
|
* domain and path as the cookie.
|
|
* Returns the removed cookie or `null`.
|
|
*/
|
|
fun remove(cookie: Cookie): Cookie? {
|
|
return map.remove(Key(cookie))
|
|
}
|
|
|
|
/**
|
|
* Get cookies for the url. Fill `accepted` and `expired`.
|
|
*/
|
|
operator fun get(url: HttpUrl, accepted: MutableList<Cookie>, expired: MutableList<Cookie>) {
|
|
val now = System.currentTimeMillis()
|
|
val iterator = map.entries.iterator()
|
|
while (iterator.hasNext()) {
|
|
val cookie = iterator.next().value
|
|
if (cookie.expiresAt <= now) {
|
|
iterator.remove()
|
|
expired.add(cookie)
|
|
} else if (cookie.matches(url)) {
|
|
accepted.add(cookie)
|
|
}
|
|
}
|
|
}
|
|
|
|
fun get(name: String, domain: String, path: String) = map[Key(name, domain, path)]
|
|
|
|
internal data class Key(
|
|
val name: String,
|
|
val domain: String,
|
|
val path: String,
|
|
) {
|
|
constructor(cookie: Cookie) : this(cookie.name, cookie.domain, cookie.path)
|
|
}
|
|
}
|