92 lines
2.9 KiB
Kotlin
92 lines
2.9 KiB
Kotlin
/*
|
|
* Copyright 2019 tom5079
|
|
*
|
|
* 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 xyz.quaver.pupil.hitomi
|
|
|
|
import kotlinx.serialization.Serializable
|
|
import org.jsoup.Jsoup
|
|
import java.net.URL
|
|
import java.net.URLDecoder
|
|
import java.nio.ByteBuffer
|
|
import java.nio.ByteOrder
|
|
import javax.net.ssl.HttpsURLConnection
|
|
import kotlin.io.readText
|
|
|
|
//galleryblock.js
|
|
fun fetchNozomi(area: String? = null, tag: String = "index", language: String = "all", start: Int = -1, count: Int = -1) : Pair<List<Int>, Int> {
|
|
val url = when(area) {
|
|
null -> "$protocol//$domain/$tag-$language$nozomiextension"
|
|
else -> "$protocol//$domain/$area/$tag-$language$nozomiextension"
|
|
}
|
|
|
|
with(URL(url).openConnection() as HttpsURLConnection) {
|
|
requestMethod = "GET"
|
|
|
|
if (start != -1 && count != -1) {
|
|
val startByte = start*4
|
|
val endByte = (start+count)*4-1
|
|
|
|
setRequestProperty("Range", "bytes=$startByte-$endByte")
|
|
}
|
|
|
|
connect()
|
|
|
|
val totalItems = getHeaderField("Content-Range")
|
|
.replace(Regex("^[Bb]ytes \\d+-\\d+/"), "").toInt() / 4
|
|
|
|
val nozomi = ArrayList<Int>()
|
|
|
|
val arrayBuffer = ByteBuffer
|
|
.wrap(inputStream.readBytes())
|
|
.order(ByteOrder.BIG_ENDIAN)
|
|
|
|
while (arrayBuffer.hasRemaining())
|
|
nozomi.add(arrayBuffer.int)
|
|
|
|
return Pair(nozomi, totalItems)
|
|
}
|
|
}
|
|
|
|
@Serializable
|
|
data class GalleryBlock(
|
|
val id: Int,
|
|
val galleryUrl: String,
|
|
val thumbnails: List<String>,
|
|
val title: String,
|
|
val artists: List<String>,
|
|
val series: List<String>,
|
|
val type: String,
|
|
val language: String,
|
|
val relatedTags: List<String>,
|
|
val groups: List<String> = emptyList()
|
|
)
|
|
|
|
suspend fun getGalleryBlock(galleryID: Int) : GalleryBlock {
|
|
val info = getGalleryInfo(galleryID)
|
|
|
|
return GalleryBlock(
|
|
galleryID,
|
|
"",
|
|
listOf(urlFromUrlFromHash(galleryID, info.files.first(), "webpbigtn", "webp", "tn")),
|
|
info.title,
|
|
info.artists?.map { it.artist }.orEmpty(),
|
|
info.parodys?.map { it.parody }.orEmpty(),
|
|
info.type,
|
|
info.language.orEmpty(),
|
|
info.tags?.map { "${if (it.female.isNullOrEmpty()) "" else "female:"}${if (it.male.isNullOrEmpty()) "" else "male:"}${it.tag}" }.orEmpty(),
|
|
info.groups?.map { it.group }.orEmpty()
|
|
)
|
|
} |