Buran/app/src/main/java/corewala/buran/ui/GemViewModel.kt

package corewala.buran.ui

import android.net.Uri
import androidx.lifecycle.ViewModel
import corewala.buran.io.gemini.Datasource
import corewala.buran.io.GemState
import corewala.buran.io.database.BuranDatabase
import corewala.buran.io.gemini.GeminiDatasource
import java.net.URI

class GemViewModel: ViewModel() {

    private lateinit var gemini: Datasource
    private lateinit var db: BuranDatabase
    private var onState: (state: GemState) -> Unit = {}

    fun initialise(home: String, gemini: Datasource, db: BuranDatabase, onState: (state: GemState) -> Unit){
        this.gemini = gemini
        this.db = db
        this.onState = onState

        if(home.startsWith("gemini://") and !home.contains(" ")){
            request(home, null, null)
        }
    }

    fun request(address: String, clientCertPassword: String?, alternativeRequest: String?) {
        gemini.request(address, false, clientCertPassword, alternativeRequest){ state ->
            onState(state)
        }
    }

    fun isRequesting(): Boolean{
        return gemini.isRequesting()
    }

    fun cancel(){
        gemini.cancel()
    }

    fun requestBinaryDownload(uri: URI, clientCertPassword: String?, alternativeRequest: String?) {
        gemini.request(uri.toString(), true, clientCertPassword, alternativeRequest){ state ->
            onState(state)
        }
    }

    //todo - same action as above... refactor
    fun requestInlineImage(uri: URI, clientCertPassword: String?, onImageReady: (cacheUri: Uri?) -> Unit){
        gemini.request(uri.toString(), false, clientCertPassword, null){ state ->
            when (state) {
                is GemState.ResponseImage -> onImageReady(state.cacheUri)
                else -> onState(state)
            }
        }
    }

    //If user changes client cert prefs in Settings this awful hack causes it to refresh state on next request
    fun invalidateDatasource() {
        if(gemini is GeminiDatasource){
            (gemini as GeminiDatasource).invalidate()
        }
    }
}