package com.simbirsoft.ftsstudyproject.data
import android.content.ContentValues
import android.content.Context
import com.simbirsoft.ftsstudyproject.data.entity.TextLineFts5
import io.requery.android.database.sqlite.SQLiteDatabase
import io.requery.android.database.sqlite.SQLiteOpenHelper
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.flow.Flow
import kotlinx.coroutines.flow.collect
import kotlinx.coroutines.flow.flow
import kotlinx.coroutines.launch
class Fts5DB(private val ctx: Context, private val scope: CoroutineScope, private val assets: AssetsDataProvider) {
   private val dbHelper: SQLiteOpenHelper by lazy {
       object: SQLiteOpenHelper(ctx, NAME, null, VERSION){
           override fun onCreate(db: SQLiteDatabase?) {
               db?.execSQL("CREATE VIRTUAL TABLE IF NOT EXISTS `$TABLE` USING FTS5(`$TEXT`)")
               scope.launch{
                   populate()
               }
           }
           override fun onUpgrade(db: SQLiteDatabase?, oldVersion: Int, newVersion: Int) {
           }
           private suspend fun populate(){
               deleteAll()
               //Текст книги берем из assets
               assets.readFileFromAssets("PeterI.txt").collect {
                   add(TextLineFts5(it))
               }
               optimize()
           }
       }
   }
   private val db: SQLiteDatabase by lazy { dbHelper.writableDatabase }
   //не забываем закрывать соединение с БД
   fun close(){
       dbHelper.close()
   }
   fun find(pattern: String, limit: Int): Flow<List<TextLineFts5>> {
       val query = "SELECT snippet($TABLE, 0, '<b>', '</b>', '<b>...</b>', 14 ) AS $TEXT FROM $TABLE WHERE text MATCH ? || '*' LIMIT ?"
       return flow {
           db.query(query, arrayOf(pattern, limit)).use { cursor ->
               val result = ArrayList<TextLineFts5>(cursor.count)
               while (cursor.moveToNext()) {
                   val text = cursor.getString(0)
                   result.add( TextLineFts5(text) )
               }
               emit(result)
           }
       }
   }
   fun add(textLine: TextLineFts5) {
       val cv = ContentValues()
       cv.put(TEXT, textLine.text)
       db.insert(TABLE, null, cv)
   }
   fun deleteAll(){
       db.delete(TABLE, null, null)
   }
   private fun optimize() {
       db.query("INSERT INTO $TABLE VALUES('optimize')")
   }
   companion object {
       private val NAME = "fts5_db"
       private val VERSION = 1
       private val TABLE = "text_table_fts5"
       private val TEXT = "text"
   }
}