本地书籍采用MediaStore保存到Download/books

pull/1486/head
gedoor 3 years ago
parent 4a2e62d413
commit b6b370e9c7
  1. 50
      app/src/main/java/io/legado/app/model/localBook/TextFile.kt

@ -1,7 +1,6 @@
package io.legado.app.model.localBook package io.legado.app.model.localBook
import android.net.Uri import android.net.Uri
import android.system.Os
import androidx.documentfile.provider.DocumentFile import androidx.documentfile.provider.DocumentFile
import io.legado.app.data.appDb import io.legado.app.data.appDb
import io.legado.app.data.entities.Book import io.legado.app.data.entities.Book
@ -10,7 +9,10 @@ import io.legado.app.data.entities.TxtTocRule
import io.legado.app.help.DefaultData import io.legado.app.help.DefaultData
import io.legado.app.utils.* import io.legado.app.utils.*
import splitties.init.appCtx import splitties.init.appCtx
import java.io.* import java.io.File
import java.io.FileInputStream
import java.io.FileNotFoundException
import java.io.InputStream
import java.nio.charset.Charset import java.nio.charset.Charset
import java.util.regex.Matcher import java.util.regex.Matcher
import java.util.regex.Pattern import java.util.regex.Pattern
@ -33,30 +35,32 @@ class TextFile(private val book: Book) {
tocRules.addAll(getTocRules()) tocRules.addAll(getTocRules())
null null
} }
//获取文件流 return getBookInputStream(book).use {
val bookStream = RandomAccessFile(bookFile, "r") analyze(it, book, rulePattern)
return analyze(bookStream, book, rulePattern) }
} }
@Throws(Exception::class) @Throws(Exception::class)
private fun analyze( private fun analyze(
bookStream: RandomAccessFile, bookIs: InputStream,
book: Book, book: Book,
pattern: Pattern? pattern: Pattern?
): ArrayList<BookChapter> { ): ArrayList<BookChapter> {
bookStream.seek(0)
val toc = arrayListOf<BookChapter>() val toc = arrayListOf<BookChapter>()
var tocRule: TxtTocRule? = null var tocRule: TxtTocRule? = null
val buffer = ByteArray(BUFFER_SIZE)
val rulePattern = pattern ?: let { val rulePattern = pattern ?: let {
tocRule = getTocRule(bookStream) bookIs.skip(0)
val length = bookIs.read(buffer, 0, buffer.size)
val content = String(buffer, 0, length, charset)
tocRule = getTocRule(content)
tocRule?.let { tocRule?.let {
Pattern.compile(it.rule, Pattern.MULTILINE) Pattern.compile(it.rule, Pattern.MULTILINE)
} }
} }
//加载章节 //加载章节
val buffer = ByteArray(BUFFER_SIZE)
//获取到的块起始点,在文件中的位置 //获取到的块起始点,在文件中的位置
bookStream.seek(0) bookIs.skip(0)
var curOffset: Long = 0 var curOffset: Long = 0
//block的个数 //block的个数
var blockPos = 0 var blockPos = 0
@ -65,7 +69,7 @@ class TextFile(private val book: Book) {
var allLength = 0 var allLength = 0
//获取文件中的数据到buffer,直到没有数据为止 //获取文件中的数据到buffer,直到没有数据为止
while (bookStream.read(buffer).also { length = it } > 0) { while (bookIs.read(buffer, 0, BUFFER_SIZE).also { length = it } > 0) {
blockPos++ blockPos++
//如果存在Chapter //如果存在Chapter
if (rulePattern != null) { if (rulePattern != null) {
@ -76,7 +80,7 @@ class TextFile(private val book: Book) {
blockContent = blockContent.substring(0, lastN) blockContent = blockContent.substring(0, lastN)
length = blockContent.toByteArray(charset).size length = blockContent.toByteArray(charset).size
allLength += length allLength += length
bookStream.seek(allLength.toLong()) bookIs.skip(allLength.toLong())
} }
//当前Block下使过的String的指针 //当前Block下使过的String的指针
var seekPos = 0 var seekPos = 0
@ -92,7 +96,7 @@ class TextFile(private val book: Book) {
if (curOffset + chapterLength - lastStart > 50000 && pattern == null) { if (curOffset + chapterLength - lastStart > 50000 && pattern == null) {
//移除不匹配的规则 //移除不匹配的规则
tocRules.remove(tocRule) tocRules.remove(tocRule)
return analyze(bookStream, book, null) return analyze(bookIs, book, null)
} }
//如果 seekPos == 0 && nextChapterPos != 0 表示当前block处前面有一段内容 //如果 seekPos == 0 && nextChapterPos != 0 表示当前block处前面有一段内容
//第一种情况一定是序章 第二种情况是上一个章节的内容 //第一种情况一定是序章 第二种情况是上一个章节的内容
@ -148,7 +152,7 @@ class TextFile(private val book: Book) {
if (seekPos == 0 && length > 50000 && pattern == null) { if (seekPos == 0 && length > 50000 && pattern == null) {
//移除不匹配的规则 //移除不匹配的规则
tocRules.remove(tocRule) tocRules.remove(tocRule)
return analyze(bookStream, book, null) return analyze(bookIs, book, null)
} }
} else { //进行本地虚拟分章 } else { //进行本地虚拟分章
//章节在buffer的偏移量 //章节在buffer的偏移量
@ -204,7 +208,6 @@ class TextFile(private val book: Book) {
System.runFinalization() System.runFinalization()
} }
} }
bookStream.close()
for (i in toc.indices) { for (i in toc.indices) {
val bean = toc[i] val bean = toc[i]
bean.index = i bean.index = i
@ -222,12 +225,8 @@ class TextFile(private val book: Book) {
return toc return toc
} }
private fun getTocRule(bookStream: RandomAccessFile): TxtTocRule? { private fun getTocRule(content: String): TxtTocRule? {
var txtTocRule: TxtTocRule? = null var txtTocRule: TxtTocRule? = null
//首先获取128k的数据
val buffer = ByteArray(BUFFER_SIZE / 4)
val length = bookStream.read(buffer, 0, buffer.size)
val content = String(buffer, 0, length, charset)
for (tocRule in tocRules) { for (tocRule in tocRules) {
val pattern = Pattern.compile(tocRule.rule, Pattern.MULTILINE) val pattern = Pattern.compile(tocRule.rule, Pattern.MULTILINE)
val matcher = pattern.matcher(content) val matcher = pattern.matcher(content)
@ -258,19 +257,22 @@ class TextFile(private val book: Book) {
fun getContent(book: Book, bookChapter: BookChapter): String { fun getContent(book: Book, bookChapter: BookChapter): String {
val count = (bookChapter.end!! - bookChapter.start!!).toInt() val count = (bookChapter.end!! - bookChapter.start!!).toInt()
val content = ByteArray(count) val content = ByteArray(count)
Os.read(getBookFD(book), content, bookChapter.start!!.toInt(), count) getBookInputStream(book).use {
it.skip(bookChapter.start!!)
it.read(content, 0, count)
}
return String(content, book.fileCharset()) return String(content, book.fileCharset())
.substringAfter(bookChapter.title) .substringAfter(bookChapter.title)
.replace("^[\\n\\s]+".toRegex(), "  ") .replace("^[\\n\\s]+".toRegex(), "  ")
} }
@Throws(FileNotFoundException::class) @Throws(FileNotFoundException::class)
private fun getBookFD(book: Book): FileDescriptor? { private fun getBookInputStream(book: Book): InputStream {
if (book.bookUrl.isContentScheme()) { if (book.bookUrl.isContentScheme()) {
val uri = Uri.parse(book.bookUrl) val uri = Uri.parse(book.bookUrl)
return appCtx.contentResolver.openFileDescriptor(uri, "r")?.fileDescriptor return appCtx.contentResolver.openInputStream(uri)!!
} }
return FileInputStream(File(book.bookUrl)).fd return FileInputStream(File(book.bookUrl))
} }
private fun getBookFile(book: Book): File { private fun getBookFile(book: Book): File {

Loading…
Cancel
Save