laoyuyu
2 years ago
parent
ef754f04cc
commit
1415de11c6
@ -1,93 +0,0 @@ |
|||||||
/* |
|
||||||
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria) |
|
||||||
* |
|
||||||
* 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.arialyy.dua.group |
|
||||||
|
|
||||||
import android.net.Uri |
|
||||||
import com.arialyy.aria.core.DuaContext |
|
||||||
import com.arialyy.aria.core.task.ITask |
|
||||||
import com.arialyy.aria.core.task.ITaskInterceptor |
|
||||||
import com.arialyy.aria.core.task.TaskChain |
|
||||||
import com.arialyy.aria.core.task.TaskResp |
|
||||||
import com.arialyy.aria.http.HttpTaskOption |
|
||||||
import com.arialyy.aria.orm.entity.DEntity |
|
||||||
import com.arialyy.aria.orm.entity.DGEntity |
|
||||||
import com.arialyy.aria.util.FileUri |
|
||||||
import timber.log.Timber |
|
||||||
import java.io.File |
|
||||||
|
|
||||||
/** |
|
||||||
* @Author laoyuyu |
|
||||||
* @Description |
|
||||||
* @Date 7:27 PM 2023/3/8 |
|
||||||
**/ |
|
||||||
internal class HttpDGSubTaskInterceptor : ITaskInterceptor { |
|
||||||
|
|
||||||
private val dgDao by lazy { |
|
||||||
DuaContext.getServiceManager().getDbService().getDuaDb().getDGEntityDao() |
|
||||||
} |
|
||||||
|
|
||||||
private lateinit var task: ITask |
|
||||||
private lateinit var option: HttpTaskOption |
|
||||||
private lateinit var dgOption: HttpDGOptionAdapter |
|
||||||
|
|
||||||
override suspend fun interceptor(chain: TaskChain): TaskResp { |
|
||||||
task = chain.getTask() |
|
||||||
option = task.getTaskOption(HttpTaskOption::class.java) |
|
||||||
dgOption = option.getOptionAdapter(HttpDGOptionAdapter::class.java) |
|
||||||
if (!checkRecord()) { |
|
||||||
return TaskResp(TaskResp.CODE_INTERRUPT) |
|
||||||
} |
|
||||||
|
|
||||||
|
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* check dg task record, if dfEntiry exist, return false |
|
||||||
* otherwise, save new record |
|
||||||
*/ |
|
||||||
private suspend fun checkRecord(): Boolean { |
|
||||||
val entity = dgDao.getDGEntityByPath(option.savePathUri.toString()) |
|
||||||
if (entity != null) { |
|
||||||
Timber.e("task already exist, filePath: ${option.savePathUri}") |
|
||||||
return false |
|
||||||
} |
|
||||||
// create sub task record |
|
||||||
val dgEntity = DGEntity( |
|
||||||
savePath = option.savePathUri!!, |
|
||||||
urls = dgOption.subUrlList.toList(), |
|
||||||
subNameList = dgOption.subNameList |
|
||||||
) |
|
||||||
|
|
||||||
val subTask = mutableListOf<DEntity>() |
|
||||||
val dir = File(FileUri.getPathByUri(option.savePathUri)!!) |
|
||||||
dgOption.subUrlList.forEachIndexed { index, it -> |
|
||||||
val subFile = File( |
|
||||||
dir.path, |
|
||||||
if (dgOption.subNameList.isNotEmpty()) dgOption.subNameList[index] else "dgTask${index}" |
|
||||||
) |
|
||||||
subTask.add( |
|
||||||
DEntity( |
|
||||||
sourceUrl = it, |
|
||||||
savePath = Uri.parse(subFile.toString()), |
|
||||||
) |
|
||||||
) |
|
||||||
} |
|
||||||
dgEntity.subList.addAll(subTask) |
|
||||||
|
|
||||||
dgDao.insert(dgEntity) |
|
||||||
return true |
|
||||||
} |
|
||||||
} |
|
||||||
@ -0,0 +1,37 @@ |
|||||||
|
/* |
||||||
|
* Copyright (C) 2016 AriaLyy(https://github.com/AriaLyy/Aria) |
||||||
|
* |
||||||
|
* 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.arialyy.aria.util |
||||||
|
|
||||||
|
import java.security.MessageDigest |
||||||
|
|
||||||
|
/** |
||||||
|
* @Author laoyuyu |
||||||
|
* @Description |
||||||
|
* @Date 21:38 2023/3/11 |
||||||
|
**/ |
||||||
|
object DuaUtil { |
||||||
|
fun getMD5Hash(str: String): String { |
||||||
|
val md = MessageDigest.getInstance("MD5") |
||||||
|
val messageDigest = md.digest(str.toByteArray()) |
||||||
|
val no = messageDigest.toBigInteger() // 使用 Kotlin 扩展函数将字节数组转换为 BigInteger |
||||||
|
var hashText = no.toString(16) |
||||||
|
while (hashText.length < 32) { |
||||||
|
hashText = "0$hashText" |
||||||
|
} |
||||||
|
return hashText |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
Loading…
Reference in new issue