|
|
|
@ -11,6 +11,8 @@ import io.ktor.request.* |
|
|
|
import io.ktor.response.* |
|
|
|
import io.ktor.response.* |
|
|
|
import io.ktor.routing.* |
|
|
|
import io.ktor.routing.* |
|
|
|
import io.ktor.websocket.* |
|
|
|
import io.ktor.websocket.* |
|
|
|
|
|
|
|
import kotlinx.coroutines.Job |
|
|
|
|
|
|
|
import kotlinx.coroutines.cancel |
|
|
|
import kotlinx.coroutines.channels.Channel |
|
|
|
import kotlinx.coroutines.channels.Channel |
|
|
|
import kotlinx.coroutines.launch |
|
|
|
import kotlinx.coroutines.launch |
|
|
|
import kotlinx.html.HTML |
|
|
|
import kotlinx.html.HTML |
|
|
|
@ -25,6 +27,7 @@ const val website = "https://gammaplus.takeshobo.co.jp" |
|
|
|
var taskChannel = Channel<ParseTask>() |
|
|
|
var taskChannel = Channel<ParseTask>() |
|
|
|
val urlResultChannel = Channel<UrlResult>() |
|
|
|
val urlResultChannel = Channel<UrlResult>() |
|
|
|
val htmlChannel= Channel<UrlParam>() |
|
|
|
val htmlChannel= Channel<UrlParam>() |
|
|
|
|
|
|
|
var currentJob: Job?=null |
|
|
|
|
|
|
|
|
|
|
|
fun Application.parse(){ |
|
|
|
fun Application.parse(){ |
|
|
|
log.info("初始化解析任务") |
|
|
|
log.info("初始化解析任务") |
|
|
|
@ -35,57 +38,64 @@ fun Application.parse(){ |
|
|
|
launch { |
|
|
|
launch { |
|
|
|
while (true){ |
|
|
|
while (true){ |
|
|
|
val resHtml= htmlChannel.receive() |
|
|
|
val resHtml= htmlChannel.receive() |
|
|
|
launch { |
|
|
|
|
|
|
|
log.info("开始解析:${resHtml.url}") |
|
|
|
|
|
|
|
val client = HttpClient(CIO) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
val titlePrefix="title>" |
|
|
|
currentJob?.let { |
|
|
|
val title=Regex("$titlePrefix[\\u4e00-\\u9fa5\\u0800-\\u4e00\\s\\d\\uff00-\\uffef]+").find(resHtml.html)?.value?.replace(titlePrefix,"")?:throw IllegalArgumentException("无法解析漫画标题") |
|
|
|
if(it.isActive) this.cancel() |
|
|
|
val romajiPrefix="/manga" |
|
|
|
} |
|
|
|
val romajiTitle=Regex("$romajiPrefix/\\w+").find(resHtml.url)?.value?.replace(romajiPrefix,"")?:throw IllegalArgumentException("无法解析漫画罗马音标题") |
|
|
|
|
|
|
|
log.info("解析漫画标题") |
|
|
|
launch { |
|
|
|
val imageDir= File(filePath,romajiTitle).apply { if(!exists()) mkdir() } |
|
|
|
log.info("开始解析:${resHtml.url}") |
|
|
|
log.info("漫画图片存储到:${imageDir.absolutePath}") |
|
|
|
val client = HttpClient(CIO) |
|
|
|
|
|
|
|
|
|
|
|
Regex("data/.*.json").findAll(resHtml.html) |
|
|
|
val titlePrefix="title>" |
|
|
|
.apply { |
|
|
|
val title=Regex("$titlePrefix[\\u4e00-\\u9fa5\\u0800-\\u4e00\\s\\d\\uff00-\\uffef]+").find(resHtml.html)?.value?.replace(titlePrefix,"")?:throw IllegalArgumentException("无法解析漫画标题") |
|
|
|
withIndex() |
|
|
|
val romajiPrefix="/manga" |
|
|
|
|
|
|
|
val romajiTitle=Regex("$romajiPrefix/\\w+").find(resHtml.url)?.value?.replace(romajiPrefix,"")?:throw IllegalArgumentException("无法解析漫画罗马音标题") |
|
|
|
|
|
|
|
log.info("解析漫画标题") |
|
|
|
|
|
|
|
val imageDir= File(filePath,romajiTitle).apply { if(!exists()) mkdir() } |
|
|
|
|
|
|
|
log.info("漫画图片存储到:${imageDir.absolutePath}") |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Regex("data/.*.json").findAll(resHtml.html) |
|
|
|
|
|
|
|
.apply { |
|
|
|
|
|
|
|
withIndex() |
|
|
|
//TODO .forEach { |
|
|
|
//TODO .forEach { |
|
|
|
.first().let { |
|
|
|
.first().let { |
|
|
|
log.info("开始解析：${it.value.value}") |
|
|
|
log.info("开始解析：${it.value.value}") |
|
|
|
val urlPath = "${resHtml.url}/${it.value.value}" |
|
|
|
val urlPath = "${resHtml.url}/${it.value.value}" |
|
|
|
val jsonRes: HttpResponse = client.get(urlPath) |
|
|
|
val jsonRes: HttpResponse = client.get(urlPath) |
|
|
|
if (jsonRes.status == HttpStatusCode.OK) { |
|
|
|
if (jsonRes.status == HttpStatusCode.OK) { |
|
|
|
log.info("url:${urlPath} request OK") |
|
|
|
log.info("url:${urlPath} request OK") |
|
|
|
val t:t=Json.decodeFromString(jsonRes.readText()) |
|
|
|
val t:t=Json.decodeFromString(jsonRes.readText()) |
|
|
|
val originImagePath="/data/${t.resources.i.src}" |
|
|
|
val originImagePath="/data/${t.resources.i.src}" |
|
|
|
val imageUrl="${resHtml.url}${originImagePath}" |
|
|
|
val imageUrl="${resHtml.url}${originImagePath}" |
|
|
|
val imgRes:HttpResponse=client.get(imageUrl) |
|
|
|
val imgRes:HttpResponse=client.get(imageUrl) |
|
|
|
if(imgRes.status==HttpStatusCode.OK&&imgRes.contentType()==ContentType.Image.JPEG){ |
|
|
|
if(imgRes.status==HttpStatusCode.OK&&imgRes.contentType()==ContentType.Image.JPEG){ |
|
|
|
val filename="${it.index}.jpg" |
|
|
|
val filename="${it.index}.jpg" |
|
|
|
val file=File(imageDir,filename).apply { |
|
|
|
val file=File(imageDir,filename).apply { |
|
|
|
writeBytes(imgRes.readBytes()) |
|
|
|
writeBytes(imgRes.readBytes()) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
log.info("存储漫画图片：${file.absolutePath}") |
|
|
|
|
|
|
|
val serverImagePath="/${uploadDir}/${romajiTitle}/${filename}" |
|
|
|
|
|
|
|
log.info("serverImagePath:${serverImagePath}") |
|
|
|
|
|
|
|
urlResultChannel.send(UrlResult(originImagePath = originImagePath, t = t, |
|
|
|
|
|
|
|
serverImagePath = serverImagePath)) |
|
|
|
|
|
|
|
}else{ |
|
|
|
|
|
|
|
log.warn("image url:${imageUrl} 响应码${jsonRes.status} 响应类型${imgRes.contentType()}") |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} else { |
|
|
|
|
|
|
|
log.warn("json url:${urlPath} 响应码:${jsonRes.status}") |
|
|
|
} |
|
|
|
} |
|
|
|
log.info("存储漫画图片：${file.absolutePath}") |
|
|
|
taskChannel.send(ParseTask(total = count(), finish = it.index + 1,percentage = if(count()==it.index+1) 100F else String.format("%.2f",(it.index+1)*100F/count()).toFloat())) |
|
|
|
val serverImagePath="/${uploadDir}/${romajiTitle}/${filename}" |
|
|
|
|
|
|
|
log.info("serverImagePath:${serverImagePath}") |
|
|
|
|
|
|
|
urlResultChannel.send(UrlResult(originImagePath = originImagePath, t = t, |
|
|
|
|
|
|
|
serverImagePath = serverImagePath)) |
|
|
|
|
|
|
|
}else{ |
|
|
|
|
|
|
|
log.warn("image url:${imageUrl} 响应码${jsonRes.status} 响应类型${imgRes.contentType()}") |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
} else { |
|
|
|
|
|
|
|
log.warn("json url:${urlPath} 响应码:${jsonRes.status}") |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
taskChannel.send(ParseTask(total = count(), finish = it.index + 1,percentage = if(count()==it.index+1) 100F else String.format("%.2f",(it.index+1)*100F/count()).toFloat())) |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
client.close() |
|
|
|
|
|
|
|
}.apply { |
|
|
|
|
|
|
|
currentJob=this |
|
|
|
|
|
|
|
invokeOnCompletion { |
|
|
|
|
|
|
|
log.info("${resHtml.url}解析完成") |
|
|
|
} |
|
|
|
} |
|
|
|
client.close() |
|
|
|
|
|
|
|
}.apply { |
|
|
|
|
|
|
|
invokeOnCompletion { |
|
|
|
|
|
|
|
log.info("${resHtml.url}解析完成") |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
@ -94,7 +104,7 @@ fun Application.parse(){ |
|
|
|
fun Application.configureRouting() { |
|
|
|
fun Application.configureRouting() { |
|
|
|
|
|
|
|
|
|
|
|
routing { |
|
|
|
routing { |
|
|
|
val uploadDir=environment.config.property("ktor.deployment.filePath").getString() |
|
|
|
val uploadDir = environment.config.property("ktor.deployment.filePath").getString() |
|
|
|
static(uploadDir) { |
|
|
|
static(uploadDir) { |
|
|
|
resources(uploadDir) |
|
|
|
resources(uploadDir) |
|
|
|
} |
|
|
|
} |
|
|
|
@ -111,8 +121,8 @@ fun Application.configureRouting() { |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
launch { |
|
|
|
launch { |
|
|
|
while (true){ |
|
|
|
while (true) { |
|
|
|
val task=taskChannel.receive() |
|
|
|
val task = taskChannel.receive() |
|
|
|
outgoing.send(Frame.Text(Json.encodeToString(task))) |
|
|
|
outgoing.send(Frame.Text(Json.encodeToString(task))) |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
@ -120,11 +130,19 @@ fun Application.configureRouting() { |
|
|
|
when (frame) { |
|
|
|
when (frame) { |
|
|
|
is Frame.Text -> { |
|
|
|
is Frame.Text -> { |
|
|
|
val text = frame.readText() |
|
|
|
val text = frame.readText() |
|
|
|
outgoing.send(Frame.Text("YOU SAID: $text")) |
|
|
|
when { |
|
|
|
if (text.equals("bye", ignoreCase = true)) { |
|
|
|
"cancel" == text -> { |
|
|
|
close(CloseReason(CloseReason.Codes.NORMAL, "Client said BYE")) |
|
|
|
if (currentJob?.isActive == true) { |
|
|
|
|
|
|
|
currentJob?.cancel() |
|
|
|
|
|
|
|
outgoing.send(Frame.Text("当前服务器解析任务已停止")) |
|
|
|
|
|
|
|
} else { |
|
|
|
|
|
|
|
outgoing.send(Frame.Text("当前服务器没有正在运行的解析任务")) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
"exit" == text -> close(CloseReason(CloseReason.Codes.NORMAL, "Client said exit")) |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
else -> log.warn("无法处理${frame.frameType}类型消息") |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
@ -142,7 +160,7 @@ fun Application.configureRouting() { |
|
|
|
val response: HttpResponse = client.get(urlParam) |
|
|
|
val response: HttpResponse = client.get(urlParam) |
|
|
|
if (response.status == HttpStatusCode.OK) { |
|
|
|
if (response.status == HttpStatusCode.OK) { |
|
|
|
val resHtml = response.readText() |
|
|
|
val resHtml = response.readText() |
|
|
|
htmlChannel.send(UrlParam(url = urlParam,html = resHtml)) |
|
|
|
htmlChannel.send(UrlParam(url = urlParam, html = resHtml)) |
|
|
|
call.respond(MessageResponse(message = "开始执行解析任务")) |
|
|
|
call.respond(MessageResponse(message = "开始执行解析任务")) |
|
|
|
} else { |
|
|
|
} else { |
|
|
|
log.warn("http code:${response.status}") |
|
|
|
log.warn("http code:${response.status}") |
|
|
|
|
pan
3 years ago