Net/net/src/main/java/com/drake/net/body/BodyExtension.kt

package com.drake.net.body

import com.drake.net.interfaces.ProgressListener
import okhttp3.MultipartBody
import okhttp3.RequestBody
import okhttp3.ResponseBody
import okhttp3.ResponseBody.Companion.asResponseBody
import okio.Buffer
import java.util.concurrent.ConcurrentLinkedQueue

fun RequestBody.toNetRequestBody(listeners: ConcurrentLinkedQueue<ProgressListener>? = null) = run {
    NetRequestBody(this, listeners)
}

fun ResponseBody.toNetResponseBody(
    listeners: ConcurrentLinkedQueue<ProgressListener>? = null,
    complete: (() -> Unit)? = null
) = run { NetResponseBody(this, listeners, complete) }

/**
 * 复制一段指定长度的字符串内容
 * @param byteCount 复制的字节长度. 如果-1则返回完整的字符串内容
 */
@JvmName("peekString")
fun RequestBody?.peekString(byteCount: Long = 1024 * 1024, discard: Boolean = false): String? {
    return when (this) {
        null -> return null
        is NetRequestBody -> peekString(byteCount, discard)
        else -> {
            val buffer = Buffer()
            writeTo(buffer)
            if (discard && buffer.size > byteCount) return ""
            val byteCountFinal = if (byteCount < 0) buffer.size else minOf(buffer.size, byteCount)
            buffer.readUtf8(byteCountFinal)
        }
    }
}

/**
 * 复制一段指定长度的字符串内容
 * @param byteCount 复制的字节长度. 如果-1则返回完整的字符串内容
 */
fun ResponseBody.peekString(byteCount: Long = 1024 * 1024 * 4, discard: Boolean = false): String {
    return when (this) {
        is NetResponseBody -> peekString(byteCount, discard)
        else -> {
            val peeked = source().peek()
            val buffer = Buffer()
            peeked.request(byteCount)
            val byteCountFinal =
                if (byteCount < 0) peeked.buffer.size else minOf(byteCount, peeked.buffer.size)
            buffer.write(peeked, byteCountFinal)
            if (discard && buffer.size > byteCount) return ""
            buffer.asResponseBody(contentType(), byteCountFinal).string()
        }
    }
}

/**
 * 通过判断[okhttp3.Headers]里面的Content-Disposition是否存在filename属性来确定是否为文件类型[MultipartBody.Part]
 */
fun MultipartBody.Part.isFile(): Boolean {
    val contentDisposition = headers?.get("Content-Disposition") ?: return false
    return ";\\s${"filename"}=\"(.+?)\"".toRegex().find(contentDisposition)?.groupValues?.getOrNull(1) != null
}

/**
 * 返回Content-Disposition里面的字段名称
 */
fun MultipartBody.Part.name(): String? {
    val contentDisposition = headers?.get("Content-Disposition") ?: return null
    return ";\\s${"name"}=\"(.+?)\"".toRegex().find(contentDisposition)?.groupValues?.getOrNull(1) ?: ""
}

/**
 * 将[MultipartBody.Part.body]作为字符串返回
 * 如果[MultipartBody.Part]是文件类型则返回的是文件名称, 确定文件类型请参考[MultipartBody.Part.isFile]
 */
fun MultipartBody.Part.value(): String? {
    val contentDisposition = headers?.get("Content-Disposition") ?: return null
    return ";\\s${"filename"}=\"(.+?)\"".toRegex().find(contentDisposition)?.groupValues?.getOrNull(1)
        ?: body.peekString()
}