package com.pylon.RNSpokestack;

import com.pylon.spokestack.SpeechPipeline;
import com.pylon.spokestack.SpeechContext;
import com.pylon.spokestack.SpeechConfig;
import com.pylon.spokestack.OnSpeechEventListener;

import android.os.Bundle;
import android.os.Handler;
import android.util.Log;
import com.facebook.react.modules.core.DeviceEventManagerModule;
import com.facebook.react.bridge.WritableMap;
import com.facebook.react.bridge.ReadableMap;
import com.facebook.react.bridge.ReadableMapKeySetIterator;
import com.facebook.react.bridge.WritableArray;
import com.facebook.react.bridge.Arguments;
import com.facebook.react.bridge.ReactApplicationContext;
import com.facebook.react.bridge.ReactContextBaseJavaModule;
import com.facebook.react.bridge.ReactMethod;

import java.util.ArrayList;
import java.util.Map;
import javax.annotation.Nullable;

public class RNSpokestackModule extends ReactContextBaseJavaModule implements OnSpeechEventListener {

  private final ReactApplicationContext reactContext;
  private SpeechPipeline pipeline;

  public RNSpokestackModule(ReactApplicationContext reactContext) {
    super(reactContext);
    this.reactContext = reactContext;
  }

  @Override
  public String getName() {
    return "Spokestack";
  }

  private void sendEvent(String eventName, @Nullable WritableMap params) {
    this.reactContext
      .getJSModule(DeviceEventManagerModule.RCTDeviceEventEmitter.class)
      .emit(eventName, params);
  }

  @ReactMethod
  public void initialize(ReadableMap config) {
    assert config.hasKey("input") : "'input' key is required in config";
    assert config.hasKey("stages") : "'stages' key is required in config";

    final SpeechPipeline.Builder builder = new SpeechPipeline.Builder();
    builder.setInputClass(config.getString("input"));

    for (Object stage : config.getArray("stages").toArrayList()) {
      builder.addStageClass(stage.toString());
    }

    if (config.hasKey("properties")) {
      Map<String, Object> map = config.getMap("properties").toHashMap();
      for (String k: map.keySet())
        builder.setProperty(k, map.get(k));
    }
    builder.addOnSpeechEventListener(this);

    pipeline = builder.build();
  }

  @ReactMethod
  public void start() throws Exception {
    pipeline.start();
  }

  @ReactMethod
  public void stop () {
    pipeline.stop();
  }

  public void onEvent(SpeechContext.Event event, SpeechContext context) {
    WritableMap react_event = Arguments.createMap();
    react_event.putString("event", event.name());
    react_event.putString("transcript", context.getTranscript());
    react_event.putBoolean("isActive", context.isActive());
    sendEvent("onSpeechEvent", react_event);
  }
}