pull/977/head
gedoor
4 years ago
parent
da0d070bfe
commit
d9bcb7506a
@ -1,49 +0,0 @@ |
|||||||
package io.legado.app.help.http |
|
||||||
|
|
||||||
import io.legado.app.utils.EncodingDetect |
|
||||||
import io.legado.app.utils.UTF8BOMFighter |
|
||||||
import kotlinx.coroutines.suspendCancellableCoroutine |
|
||||||
import okhttp3.Call |
|
||||||
import okhttp3.Callback |
|
||||||
import okhttp3.Response |
|
||||||
import okhttp3.ResponseBody |
|
||||||
import java.io.IOException |
|
||||||
import java.nio.charset.Charset |
|
||||||
import kotlin.coroutines.resume |
|
||||||
import kotlin.coroutines.resumeWithException |
|
||||||
|
|
||||||
suspend fun Call.await(): Response = suspendCancellableCoroutine { block -> |
|
||||||
|
|
||||||
block.invokeOnCancellation { |
|
||||||
cancel() |
|
||||||
} |
|
||||||
|
|
||||||
enqueue(object : Callback { |
|
||||||
override fun onFailure(call: Call, e: IOException) { |
|
||||||
block.resumeWithException(e) |
|
||||||
} |
|
||||||
|
|
||||||
override fun onResponse(call: Call, response: Response) { |
|
||||||
block.resume(response) |
|
||||||
} |
|
||||||
}) |
|
||||||
|
|
||||||
} |
|
||||||
|
|
||||||
fun ResponseBody.text(encode: String? = null): String { |
|
||||||
val responseBytes = UTF8BOMFighter.removeUTF8BOM(bytes()) |
|
||||||
var charsetName: String? = encode |
|
||||||
|
|
||||||
charsetName?.let { |
|
||||||
return String(responseBytes, Charset.forName(charsetName)) |
|
||||||
} |
|
||||||
|
|
||||||
//根据http头判断 |
|
||||||
contentType()?.charset()?.let { |
|
||||||
return String(responseBytes, it) |
|
||||||
} |
|
||||||
|
|
||||||
//根据内容判断 |
|
||||||
charsetName = EncodingDetect.getHtmlEncode(responseBytes) |
|
||||||
return String(responseBytes, Charset.forName(charsetName)) |
|
||||||
} |
|
||||||
@ -0,0 +1,131 @@ |
|||||||
|
package io.legado.app.help.http |
||||||
|
|
||||||
|
import io.legado.app.utils.EncodingDetect |
||||||
|
import io.legado.app.utils.UTF8BOMFighter |
||||||
|
import kotlinx.coroutines.suspendCancellableCoroutine |
||||||
|
import okhttp3.* |
||||||
|
import java.io.IOException |
||||||
|
import java.net.InetSocketAddress |
||||||
|
import java.net.Proxy |
||||||
|
import java.nio.charset.Charset |
||||||
|
import java.util.concurrent.ConcurrentHashMap |
||||||
|
import java.util.concurrent.TimeUnit |
||||||
|
import kotlin.coroutines.resume |
||||||
|
import kotlin.coroutines.resumeWithException |
||||||
|
|
||||||
|
val okHttpClient: OkHttpClient by lazy { |
||||||
|
|
||||||
|
val specs = arrayListOf( |
||||||
|
ConnectionSpec.MODERN_TLS, |
||||||
|
ConnectionSpec.COMPATIBLE_TLS, |
||||||
|
ConnectionSpec.CLEARTEXT |
||||||
|
) |
||||||
|
|
||||||
|
val builder = OkHttpClient.Builder() |
||||||
|
.connectTimeout(15, TimeUnit.SECONDS) |
||||||
|
.writeTimeout(15, TimeUnit.SECONDS) |
||||||
|
.readTimeout(15, TimeUnit.SECONDS) |
||||||
|
.sslSocketFactory(SSLHelper.unsafeSSLSocketFactory, SSLHelper.unsafeTrustManager) |
||||||
|
.retryOnConnectionFailure(true) |
||||||
|
.hostnameVerifier(SSLHelper.unsafeHostnameVerifier) |
||||||
|
.connectionSpecs(specs) |
||||||
|
.followRedirects(true) |
||||||
|
.followSslRedirects(true) |
||||||
|
.addInterceptor(Interceptor { chain -> |
||||||
|
val request = chain.request() |
||||||
|
.newBuilder() |
||||||
|
.addHeader("Keep-Alive", "300") |
||||||
|
.addHeader("Connection", "Keep-Alive") |
||||||
|
.addHeader("Cache-Control", "no-cache") |
||||||
|
.build() |
||||||
|
chain.proceed(request) |
||||||
|
}) |
||||||
|
|
||||||
|
builder.build() |
||||||
|
} |
||||||
|
|
||||||
|
object OkHttpHelper { |
||||||
|
private val proxyClientCache: ConcurrentHashMap<String, OkHttpClient> by lazy { |
||||||
|
ConcurrentHashMap() |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 缓存代理okHttp |
||||||
|
*/ |
||||||
|
fun getProxyClient(proxy: String? = null): OkHttpClient { |
||||||
|
if (proxy.isNullOrBlank()) { |
||||||
|
return okHttpClient |
||||||
|
} |
||||||
|
proxyClientCache[proxy]?.let { |
||||||
|
return it |
||||||
|
} |
||||||
|
val r = Regex("(http|socks4|socks5)://(.*):(\\d{2,5})(@.*@.*)?") |
||||||
|
val ms = r.findAll(proxy) |
||||||
|
val group = ms.first() |
||||||
|
var username = "" //代理服务器验证用户名 |
||||||
|
var password = "" //代理服务器验证密码 |
||||||
|
val type = if (group.groupValues[1] == "http") "http" else "socks" |
||||||
|
val host = group.groupValues[2] |
||||||
|
val port = group.groupValues[3].toInt() |
||||||
|
if (group.groupValues[4] != "") { |
||||||
|
username = group.groupValues[4].split("@")[1] |
||||||
|
password = group.groupValues[4].split("@")[2] |
||||||
|
} |
||||||
|
if (type != "direct" && host != "") { |
||||||
|
val builder = okHttpClient.newBuilder() |
||||||
|
if (type == "http") { |
||||||
|
builder.proxy(Proxy(Proxy.Type.HTTP, InetSocketAddress(host, port))) |
||||||
|
} else { |
||||||
|
builder.proxy(Proxy(Proxy.Type.SOCKS, InetSocketAddress(host, port))) |
||||||
|
} |
||||||
|
if (username != "" && password != "") { |
||||||
|
builder.proxyAuthenticator { _, response -> //设置代理服务器账号密码 |
||||||
|
val credential: String = Credentials.basic(username, password) |
||||||
|
response.request.newBuilder() |
||||||
|
.header("Proxy-Authorization", credential) |
||||||
|
.build() |
||||||
|
} |
||||||
|
} |
||||||
|
val proxyClient = builder.build() |
||||||
|
proxyClientCache[proxy] = proxyClient |
||||||
|
return proxyClient |
||||||
|
} |
||||||
|
return okHttpClient |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
suspend fun Call.await(): Response = suspendCancellableCoroutine { block -> |
||||||
|
|
||||||
|
block.invokeOnCancellation { |
||||||
|
cancel() |
||||||
|
} |
||||||
|
|
||||||
|
enqueue(object : Callback { |
||||||
|
override fun onFailure(call: Call, e: IOException) { |
||||||
|
block.resumeWithException(e) |
||||||
|
} |
||||||
|
|
||||||
|
override fun onResponse(call: Call, response: Response) { |
||||||
|
block.resume(response) |
||||||
|
} |
||||||
|
}) |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
fun ResponseBody.text(encode: String? = null): String { |
||||||
|
val responseBytes = UTF8BOMFighter.removeUTF8BOM(bytes()) |
||||||
|
var charsetName: String? = encode |
||||||
|
|
||||||
|
charsetName?.let { |
||||||
|
return String(responseBytes, Charset.forName(charsetName)) |
||||||
|
} |
||||||
|
|
||||||
|
//根据http头判断 |
||||||
|
contentType()?.charset()?.let { |
||||||
|
return String(responseBytes, it) |
||||||
|
} |
||||||
|
|
||||||
|
//根据内容判断 |
||||||
|
charsetName = EncodingDetect.getHtmlEncode(responseBytes) |
||||||
|
return String(responseBytes, Charset.forName(charsetName)) |
||||||
|
} |
||||||
Loading…
Reference in new issue