pull/981/head
gedoor 4 years ago
parent c6cffe7008
commit bdd75cc157
  1. 23
      app/src/main/java/io/legado/app/help/http/OkHttpUtils.kt
  2. 42
      app/src/main/java/io/legado/app/lib/webdav/WebDav.kt

@ -124,3 +124,26 @@ fun Request.Builder.postJson(json: String?) {
fun Request.Builder.mkCol() { fun Request.Builder.mkCol() {
method("MKCOL", null) method("MKCOL", null)
} }
fun Request.Builder.propFind(fields: List<String>?) {
var propFind =
"""<?xml version="1.0"?>
<a:propfind xmlns:a="DAV:">
<a:prop>
%s
</a:prop>
</a:propfind>"""
// 添加RequestBody对象,可以只返回的属性。如果设为null,则会返回全部属性
// 注意:尽量手动指定需要返回的属性。若返回全部属性,可能后由于Prop.java里没有该属性名,而崩溃。
if (fields.isNullOrEmpty()) {
method("PROPFIND", null)
} else {
val requestProps = StringBuilder()
for (p in fields) {
requestProps.append("<a:").append(p).append("/>\n")
}
propFind = String.format(propFind, requestProps.toString() + "\n")
val requestBody = propFind.toRequestBody("text/plain".toMediaType())
method("PROPFIND", requestBody)
}
}

@ -1,9 +1,6 @@
package io.legado.app.lib.webdav package io.legado.app.lib.webdav
import io.legado.app.help.http.mkCol import io.legado.app.help.http.*
import io.legado.app.help.http.newCall
import io.legado.app.help.http.okHttpClient
import io.legado.app.help.http.text
import okhttp3.* import okhttp3.*
import okhttp3.MediaType.Companion.toMediaType import okhttp3.MediaType.Companion.toMediaType
import okhttp3.RequestBody.Companion.asRequestBody import okhttp3.RequestBody.Companion.asRequestBody
@ -18,21 +15,6 @@ import java.util.*
@Suppress("unused", "MemberVisibilityCanBePrivate") @Suppress("unused", "MemberVisibilityCanBePrivate")
class WebDav(urlStr: String) { class WebDav(urlStr: String) {
companion object {
// 指定返回哪些属性
private const val DIR =
"""<?xml version="1.0"?>
<a:propfind xmlns:a="DAV:">
<a:prop>
<a:displayname/>
<a:resourcetype/>
<a:getcontentlength/>
<a:creationdate/>
<a:getlastmodified/>
%s
</a:prop>
</a:propfind>"""
}
private val url: URL = URL(urlStr) private val url: URL = URL(urlStr)
private val httpUrl: String? by lazy { private val httpUrl: String? by lazy {
@ -64,26 +46,17 @@ class WebDav(urlStr: String) {
/** /**
* 列出当前路径下的文件 * 列出当前路径下的文件
* *
* @param propsList 指定列出文件的哪些属性 * @param fields 指定列出文件的哪些属性
* @return 文件列表 * @return 文件列表
*/ */
suspend fun listFiles(propsList: ArrayList<String> = ArrayList()): List<WebDav> { suspend fun listFiles(fields: ArrayList<String> = ArrayList()): List<WebDav> {
propFindResponse(propsList)?.let { body -> propFindResponse(fields)?.let { body ->
return parseDir(body) return parseDir(body)
} }
return ArrayList() return ArrayList()
} }
private suspend fun propFindResponse(propsList: ArrayList<String>): String? { private suspend fun propFindResponse(fields: ArrayList<String>?): String? {
val requestProps = StringBuilder()
for (p in propsList) {
requestProps.append("<a:").append(p).append("/>\n")
}
val requestPropsStr: String = if (requestProps.toString().isEmpty()) {
DIR.replace("%s", "")
} else {
String.format(DIR, requestProps.toString() + "\n")
}
val url = httpUrl val url = httpUrl
val auth = HttpAuth.auth val auth = HttpAuth.auth
if (url != null && auth != null) { if (url != null && auth != null) {
@ -91,10 +64,7 @@ class WebDav(urlStr: String) {
okHttpClient.newCall { okHttpClient.newCall {
url(url) url(url)
addHeader("Authorization", Credentials.basic(auth.user, auth.pass)) addHeader("Authorization", Credentials.basic(auth.user, auth.pass))
// 添加RequestBody对象,可以只返回的属性。如果设为null,则会返回全部属性 propFind(fields)
// 注意:尽量手动指定需要返回的属性。若返回全部属性,可能后由于Prop.java里没有该属性名,而崩溃。
val requestBody = requestPropsStr.toRequestBody("text/plain".toMediaType())
method("PROPFIND", requestBody)
}.text() }.text()
}.onFailure { }.onFailure {
it.printStackTrace() it.printStackTrace()

Loading…
Cancel
Save