Merge branch 'master' of https://github.com/gedoor/legado
atbest
6 years ago
commit
4b8f6e3f9b
@ -0,0 +1 @@ |
||||
## 用于网络备份的WebDav |
||||
@ -0,0 +1,297 @@ |
||||
package io.legado.app.lib.webdav |
||||
|
||||
import io.legado.app.lib.webdav.http.Handler |
||||
import io.legado.app.lib.webdav.http.HttpAuth |
||||
import io.legado.app.lib.webdav.http.OkHttp |
||||
|
||||
import org.jsoup.Jsoup |
||||
import org.jsoup.nodes.Document |
||||
import org.jsoup.nodes.Element |
||||
import org.jsoup.select.Elements |
||||
|
||||
import java.io.File |
||||
import java.io.FileOutputStream |
||||
import java.io.IOException |
||||
import java.io.InputStream |
||||
import java.io.UnsupportedEncodingException |
||||
import java.lang.reflect.Field |
||||
import java.net.MalformedURLException |
||||
import java.net.URL |
||||
import java.net.URLEncoder |
||||
import java.util.ArrayList |
||||
|
||||
import okhttp3.Credentials |
||||
import okhttp3.MediaType |
||||
import okhttp3.OkHttpClient |
||||
import okhttp3.Request |
||||
import okhttp3.RequestBody |
||||
import okhttp3.Response |
||||
|
||||
class WebDav @Throws(MalformedURLException::class) |
||||
constructor(url: String) { |
||||
private val url: URL = URL(null, url, Handler.HANDLER) |
||||
private var httpUrl: String? = null |
||||
|
||||
var displayName: String? = null |
||||
var size: Long = 0 |
||||
private var exists = false |
||||
var parent = "" |
||||
var urlName = "" |
||||
get() { |
||||
if (field.isEmpty()) { |
||||
this.urlName = (if (parent.isEmpty()) url.file else url.toString().replace(parent, "")).replace("/", "") |
||||
} |
||||
return field |
||||
} |
||||
|
||||
private val okHttpClient: OkHttpClient = OkHttp.SingletonHelper.okHttpClient |
||||
|
||||
val path: String |
||||
get() = url.toString() |
||||
|
||||
private val inputStream: InputStream? |
||||
get() = getUrl()?.let {url -> |
||||
val request = Request.Builder().url(url) |
||||
HttpAuth.auth?.let {request.header("Authorization", Credentials.basic(it.user, it.pass)) } |
||||
|
||||
try { |
||||
return okHttpClient.newCall(request.build()).execute().body()?.byteStream() |
||||
} catch (e: IOException) { |
||||
e.printStackTrace() |
||||
} catch (e: IllegalArgumentException) { |
||||
e.printStackTrace() |
||||
} |
||||
return null |
||||
} |
||||
|
||||
val host: String |
||||
get() = url.host |
||||
|
||||
private fun getUrl(): String? { |
||||
if (httpUrl == null) { |
||||
val raw = url.toString().replace("davs://", "https://").replace("dav://", "http://") |
||||
try { |
||||
httpUrl = URLEncoder.encode(raw, "UTF-8") |
||||
.replace("\\+".toRegex(), "%20") |
||||
.replace("%3A".toRegex(), ":") |
||||
.replace("%2F".toRegex(), "/") |
||||
} catch (e: UnsupportedEncodingException) { |
||||
e.printStackTrace() |
||||
} |
||||
|
||||
} |
||||
return httpUrl |
||||
} |
||||
|
||||
/** |
||||
* 填充文件信息。实例化WebDAVFile对象时，并没有将远程文件的信息填充到实例中。需要手动填充！ |
||||
* |
||||
* @return 远程文件是否存在 |
||||
*/ |
||||
@Throws(IOException::class) |
||||
fun indexFileInfo(): Boolean { |
||||
propFindResponse(ArrayList())?.let { response -> |
||||
if (!response.isSuccessful) { |
||||
this.exists = false |
||||
return false |
||||
} |
||||
response.body()?.let { |
||||
if (it.string().isNotEmpty()) { |
||||
return true |
||||
} |
||||
} |
||||
} |
||||
return false |
||||
} |
||||
|
||||
/** |
||||
* 列出当前路径下的文件 |
||||
* |
||||
* @param propsList 指定列出文件的哪些属性 |
||||
* @return 文件列表 |
||||
*/ |
||||
@Throws(IOException::class) |
||||
@JvmOverloads |
||||
fun listFiles(propsList: ArrayList<String> = ArrayList()): List<WebDav> { |
||||
propFindResponse(propsList)?.let {response-> |
||||
if (response.isSuccessful) { |
||||
response.body()?.let {body-> |
||||
return parseDir(body.string()) |
||||
} |
||||
} |
||||
|
||||
} |
||||
return ArrayList() |
||||
} |
||||
|
||||
@Throws(IOException::class) |
||||
private fun propFindResponse(propsList: ArrayList<String>, depth: Int = 1): Response? { |
||||
val requestProps = StringBuilder() |
||||
for (p in propsList) { |
||||
requestProps.append("<a:").append(p).append("/>\n") |
||||
} |
||||
val requestPropsStr: String |
||||
requestPropsStr = if (requestProps.toString().isEmpty()) { |
||||
DIR.replace("%s", "") |
||||
} else { |
||||
String.format(DIR, requestProps.toString() + "\n") |
||||
} |
||||
getUrl()?.let {url-> |
||||
val request = Request.Builder() |
||||
.url(url) |
||||
// 添加RequestBody对象，可以只返回的属性。如果设为null，则会返回全部属性 |
||||
// 注意：尽量手动指定需要返回的属性。若返回全部属性，可能后由于Prop.java里没有该属性名，而崩溃。 |
||||
.method("PROPFIND", RequestBody.create(MediaType.parse("text/plain"), requestPropsStr)) |
||||
|
||||
HttpAuth.auth?.let {request.header("Authorization", Credentials.basic(it.user, it.pass)) } |
||||
|
||||
request.header("Depth", if (depth < 0) "infinity" else Integer.toString(depth)) |
||||
|
||||
return okHttpClient.newCall(request.build()).execute() |
||||
} |
||||
return null |
||||
} |
||||
|
||||
private fun parseDir(s: String): List<WebDav> { |
||||
val list = ArrayList<WebDav>() |
||||
val document = Jsoup.parse(s) |
||||
val elements = document.getElementsByTag("d:response") |
||||
getUrl()?.let { url-> |
||||
val baseUrl = if (url.endsWith("/")) url else "$url/" |
||||
for (element in elements) { |
||||
val href = element.getElementsByTag("d:href")[0].text() |
||||
if (!href.endsWith("/")) { |
||||
val fileName = href.substring(href.lastIndexOf("/") + 1) |
||||
val webDavFile: WebDav |
||||
try { |
||||
webDavFile = WebDav(baseUrl + fileName) |
||||
webDavFile.displayName = fileName |
||||
webDavFile.urlName = href |
||||
list.add(webDavFile) |
||||
} catch (e: MalformedURLException) { |
||||
e.printStackTrace() |
||||
} |
||||
} |
||||
} |
||||
} |
||||
return list |
||||
} |
||||
|
||||
/** |
||||
* 根据自己的URL，在远程处创建对应的文件夹 |
||||
* |
||||
* @return 是否创建成功 |
||||
*/ |
||||
@Throws(IOException::class) |
||||
fun makeAsDir(): Boolean { |
||||
getUrl()?.let { url-> |
||||
val request = Request.Builder() |
||||
.url(url) |
||||
.method("MKCOL", null) |
||||
return execRequest(request) |
||||
} |
||||
return false |
||||
} |
||||
|
||||
/** |
||||
* 下载到本地 |
||||
* |
||||
* @param savedPath 本地的完整路径，包括最后的文件名 |
||||
* @param replaceExisting 是否替换本地的同名文件 |
||||
* @return 下载是否成功 |
||||
*/ |
||||
fun downloadTo(savedPath: String, replaceExisting: Boolean): Boolean { |
||||
if (File(savedPath).exists()) { |
||||
if (!replaceExisting) { |
||||
return false |
||||
} |
||||
} |
||||
val inputS = inputStream ?: return false |
||||
File(savedPath).writeBytes(inputS.readBytes()) |
||||
return true |
||||
} |
||||
|
||||
/** |
||||
* 上传文件 |
||||
*/ |
||||
@Throws(IOException::class) |
||||
@JvmOverloads |
||||
fun upload(localPath: String, contentType: String? = null): Boolean { |
||||
val file = File(localPath) |
||||
if (!file.exists()) return false |
||||
val mediaType = if (contentType == null) null else MediaType.parse(contentType) |
||||
// 务必注意RequestBody不要嵌套，不然上传时内容可能会被追加多余的文件信息 |
||||
val fileBody = RequestBody.create(mediaType, file) |
||||
getUrl()?.let { |
||||
val request = Request.Builder() |
||||
.url(it) |
||||
.put(fileBody) |
||||
return execRequest(request) |
||||
} |
||||
return false |
||||
} |
||||
|
||||
/** |
||||
* 执行请求，获取响应结果 |
||||
* |
||||
* @param requestBuilder 因为还需要追加验证信息，所以此处传递Request.Builder的对象，而不是Request的对象 |
||||
* @return 请求执行的结果 |
||||
*/ |
||||
@Throws(IOException::class) |
||||
private fun execRequest(requestBuilder: Request.Builder): Boolean { |
||||
HttpAuth.auth?.let { requestBuilder.header("Authorization", Credentials.basic(it.user, it.pass)) } |
||||
|
||||
val response = okHttpClient.newCall(requestBuilder.build()).execute() |
||||
return response.isSuccessful |
||||
} |
||||
|
||||
fun canRead(): Boolean { |
||||
return true |
||||
} |
||||
|
||||
fun canWrite(): Boolean { |
||||
return false |
||||
} |
||||
|
||||
fun exists(): Boolean { |
||||
return exists |
||||
} |
||||
|
||||
fun exists(exists: Boolean) { |
||||
this.exists = exists |
||||
} |
||||
|
||||
companion object { |
||||
val TAG = WebDav::class.java.simpleName |
||||
val OBJECT_NOT_EXISTS_TAG = "ObjectNotFound" |
||||
// 指定返回哪些属性 |
||||
private val DIR = "<?xml version=\"1.0\"?>\n" + |
||||
"<a:propfind xmlns:a=\"DAV:\">\n" + |
||||
"<a:prop>\n" + |
||||
"<a:displayname/>\n<a:resourcetype/>\n<a:getcontentlength/>\n<a:creationdate/>\n<a:getlastmodified/>\n%s" + |
||||
"</a:prop>\n" + |
||||
"</a:propfind>" |
||||
|
||||
/** |
||||
* 打印对象内的所有属性 |
||||
*/ |
||||
fun <T> printAllAttrs(className: String, o: Any): T? { |
||||
try { |
||||
val c = Class.forName(className) |
||||
val fields = c.declaredFields |
||||
for (f in fields) { |
||||
f.isAccessible = true |
||||
} |
||||
println("=============$className===============") |
||||
for (f in fields) { |
||||
val field = f.toString().substring(f.toString().lastIndexOf(".") + 1) //取出属性名称 |
||||
println(field + " --> " + f.get(o)) |
||||
} |
||||
} catch (e: Exception) { |
||||
e.printStackTrace() |
||||
} |
||||
|
||||
return null |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,25 @@ |
||||
package io.legado.app.lib.webdav.http |
||||
|
||||
import java.net.URL |
||||
import java.net.URLConnection |
||||
import java.net.URLStreamHandler |
||||
|
||||
class Handler : URLStreamHandler() { |
||||
|
||||
override fun getDefaultPort(): Int { |
||||
return 80 |
||||
} |
||||
|
||||
public override fun openConnection(u: URL): URLConnection? { |
||||
return null |
||||
} |
||||
|
||||
override fun parseURL(url: URL, spec: String, start: Int, end: Int) { |
||||
super.parseURL(url, spec, start, end) |
||||
} |
||||
|
||||
companion object { |
||||
|
||||
val HANDLER: URLStreamHandler = Handler() |
||||
} |
||||
} |
||||
@ -0,0 +1,14 @@ |
||||
package io.legado.app.lib.webdav.http |
||||
|
||||
object HttpAuth { |
||||
|
||||
var auth: Auth? = null |
||||
private set |
||||
|
||||
fun setAuth(user: String, password: String) { |
||||
auth = Auth(user, password) |
||||
} |
||||
|
||||
class Auth internal constructor(val user: String, val pass: String) |
||||
|
||||
} |
||||
@ -0,0 +1,11 @@ |
||||
package io.legado.app.lib.webdav.http |
||||
|
||||
import okhttp3.OkHttpClient |
||||
|
||||
class OkHttp private constructor() { |
||||
|
||||
object SingletonHelper { |
||||
val okHttpClient: OkHttpClient = OkHttpClient.Builder().build() |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue