package com.visioncameratextrecognition

import android.net.Uri
import com.facebook.react.bridge.Promise
import com.facebook.react.bridge.ReactApplicationContext
import com.facebook.react.bridge.ReactContextBaseJavaModule
import com.facebook.react.bridge.ReactMethod
import com.facebook.react.bridge.WritableNativeMap
import com.google.android.gms.tasks.Task
import com.google.android.gms.tasks.Tasks
import com.google.mlkit.vision.common.InputImage
import com.google.mlkit.vision.text.Text
import com.google.mlkit.vision.text.TextRecognition
import com.google.mlkit.vision.text.latin.TextRecognizerOptions

class PhotoRecognizerModule(reactContext: ReactApplicationContext) :
    ReactContextBaseJavaModule(reactContext) {

   private val recognizer = TextRecognition.getClient(TextRecognizerOptions.DEFAULT_OPTIONS)

    @ReactMethod
    fun process(uri:String,promise:Promise){
        val parsedUri = Uri.parse(uri)
        val data = WritableNativeMap()
        val image = InputImage.fromFilePath(this.reactApplicationContext,parsedUri)
        val task: Task<Text> = recognizer.process(image)
        try {
            val text: Text = Tasks.await(task)
            if (text.text.isEmpty()) {
                promise.resolve(WritableNativeMap())
            }
            data.putString("resultText", text.text)
            data.putArray("blocks", VisionCameraTextRecognitionPlugin.getBlocks(text.textBlocks))
             promise.resolve(data)
        } catch (e: Exception) {
            e.printStackTrace()
           promise.reject("Error", "Error processing image")
        }

    promise.resolve(true)

    }
    override fun getName(): String {
        return NAME
    }
    companion object {
        const val NAME = "PhotoRecognizerModule"
    }
}