class VoiceRecognizerBox(appKey:String, token:String) extends HBox{

  val START = "START"
  val END = "END"

  val spaceLabel = new Label("检测间隔时间(秒)：")
  val spaceTime: TextField = {
    val t = new TextField(); t.setPromptText("检测间隔时间（秒）"); t.setText("2")
    t.setPrefWidth(50); t
  }

  val regResultString = new SimpleStringProperty("Result")
  val regResult: Label = {
    val t = new Label(); t.textProperty().bind(regResultString);t
  }

  var recognizer: AliVoiceRecognizer = _

  var completedConsumer: Consumer[SpeechRecognizerResponse] = _

  val controlBtn: Button = {
    val t = new Button(START); t.setOnAction(_ => {
      doAction()
    }); t
  }

  getChildren.addAll(spaceLabel, spaceTime, controlBtn, regResult)

  this.setAlignment(Pos.CENTER)
  this.setSpacing(10)

  def endSetStatus(): Unit = {
      controlBtn.setText(START)
      recognizer.requestShutdown()
  }

  def startSetStatus(): Unit = controlBtn.setText(END)


  def startRecognizer(): Unit = {
    recognizer = new AliVoiceRecognizer(appKey, token)
    if (completedConsumer != null)
      recognizer.completedConsumer = completedConsumer
    new Thread(() =>
      recognizer.doRecognition(spaceTime.getText.trim.toInt)).start()
  }

  def doAction(): Unit = {
    if (controlBtn.getText == START) {
      Platform.runLater(() => startRecognizer())
      startSetStatus()
    } else endSetStatus()
  }

  def setFinishCallbackAction(op: SpeechRecognizerResponse => Unit): Unit =
    completedConsumer = (t: SpeechRecognizerResponse) => op(t)


  def setAfterBtnClickedBeforeRecognizerStartedAction(op: => Unit): Unit = {
    controlBtn.setOnAction(_ => {
      if (controlBtn.getText == START) op
      doAction()
    })
  }

  def requestShutdown(): Unit = if (recognizer != null) recognizer.requestShutdown()

}

class Recorder(mainStage:Stage, showNoBtn: Boolean = false) {

  private[this] val logger = LoggerFactory.getLogger(getClass)

  import Utils._

  val key = new SimpleStringProperty("Waiting...")
  var saving = false
  val waiting: TextField = {
    val t = new TextField("5"); t.setPrefWidth(50); t
  }
  val edit = new Label("检测到修改")
  val save = new Label("正在保存")
  val enter = new Button("开始")
  val data = new ArrayBuffer[(Instant,String)]()

  var fileName: String = if (showNoBtn) "NoName_record.csv" else showAlertAndGetName

  val voiceBox = new VoiceRecognizerBox("PZiaqW1PVQE4","f6860ddbd72643a29")

  voiceBox.setFinishCallbackAction(resp => {
    val res = resp.getRecognizedText.trim
    Platform.runLater(() => {
      voiceBox.regResultString.set(res)
      tryHint(res.trim)
    })
  })

  voiceBox.setAfterBtnClickedBeforeRecognizerStartedAction {
    logger.info(s"Reload Recognizer File from $configLoadFile now...")
    loadWordsToGroup(configLoadFile)
  }

  var g2: Array[String] = _
  var g3: Array[String] = _
  var g4: Array[String] = _
  var g5: Array[String] = _
  var g6: Array[String] = _

  val configLoadFile: Path = Paths.get(System.getProperty("user.dir") + File.separator + "words.yml")

  val root: Parent = {
    val pane = new BorderPane()
    val info = new Text()
    pane.setTop(info)

    val keyPressed = new Label("")
    keyPressed.textProperty().bind(key)
    keyPressed.setTextFill(Color.RED)
    keyPressed.setFont(Font.font(80))
    pane.setCenter(keyPressed)

    val waitingLabel = new Label("可修改延迟时间(秒)：")
    val allBox = new VBox(); allBox.setSpacing(10); allBox.setAlignment(Pos.CENTER_LEFT)
    val funcBox = new HBox(); funcBox.setAlignment(Pos.CENTER_LEFT); funcBox.setSpacing(15)
    voiceBox.setAlignment(Pos.CENTER_LEFT)
    allBox.getChildren.addAll(funcBox, voiceBox)

    info.textProperty().bind(new SimpleStringProperty("在英文输入下，记录按键，按键后 ")
      .concat(waiting.textProperty()).concat(" 秒内可修改，以防止主试记录的错误。\n" +
      "按下 Enter 记录为空，按下数字记录对应数字"))
    edit.setTextFill(Color.RED)
    edit.setVisible(false)
    save.setTextFill(Color.GREEN)
    save.setVisible(false)
    funcBox.getChildren.addAll(waitingLabel, waiting, enter, edit, save)
    BorderPane.setAlignment(info, Pos.CENTER)
    BorderPane.setMargin(info, new Insets(20,20,20,20))
    BorderPane.setMargin(allBox, new Insets(20,20,20,20))
    pane.setBottom(allBox)
    pane
  }

  val scene = new Scene(root, 600, 400)

  val stage: Stage = initDataStage(mainStage, scene)

  stage.setTitle(fileName)

  scene.setOnKeyPressed(e => {
    withTimeAndSaveKeyCode(e.getCode)
  })

  waiting.textProperty().addListener {
    enter.requestFocus()
  }

  def loadWordsToGroup(path: Path): Unit = {
    val map = VoiceRecognizerBox.readYamlToKeyVoiceMatchMap(path)
    import collection.JavaConverters._
    g2 = map.getOrDefault("two",List("二","额","啊").asJava).asScala.toArray
    g3 = map.getOrDefault("three",List("三","伞","山").asJava).asScala.toArray
    g4 = map.getOrDefault("four",List("四","是","事").asJava).asScala.toArray
    g5 = map.getOrDefault("five",List("五","无","物","我").asJava).asScala.toArray
    g6 = map.getOrDefault("six",List("六","哎呦","呦","没有").asJava).asScala.toArray
  }

  private[this] def tryHint(from: String): Unit = {
    synchronized {
      println("Try Hit " + from)
      from match {
        case i2 if g2.exists(g => i2.contains(g)) => withTimeAndSaveKeyCode(KeyCode.NUMPAD2)
        case i3 if g3.exists(g => i3.contains(g)) => withTimeAndSaveKeyCode(KeyCode.NUMPAD3)
        case i4 if g4.exists(g => i4.contains(g)) => withTimeAndSaveKeyCode(KeyCode.NUMPAD4)
        case i5 if g5.exists(g => i5.contains(g)) => withTimeAndSaveKeyCode(KeyCode.NUMPAD5)
        case i6 if g6.exists(g => i6.contains(g)) => withTimeAndSaveKeyCode(KeyCode.NUMPAD6)
        case _ => println("No match for Hit " + from)
      }
    }
  }

  private[this] def showAlertAndGetName: String = {
    val alert = new TextInputDialog()
    alert.setContentText("输入被试编号")
    alert.setHeaderText("将保存被试回答到 编号.csv 中")
    alert.showAndWait()
    alert.getEditor.getText().trim match {
      case i if !i.isEmpty => i + "_record.csv"
      case _ => "NoName_record.csv"
    }
  }

  private[this] def initDataStage(mainStage: Stage, scene: Scene): Stage = {
    val stage = new Stage()
    stage.setScene(scene)
    stage.initOwner(mainStage)
    //stage.initModality(Modality.APPLICATION_MODAL)
    stage.setOnCloseRequest(e => {
      voiceBox.requestShutdown()
      println("Closing Stage Now...")
      doSave(fileName)
    })
    stage
  }

  private[this] def withTimeAndSaveKeyCode(from:KeyCode): Unit = {
    println("Pull trigger with " + from)
    key.set(from.toString)
    //如果没有启动保存进程，则启动，反之，则不启动，算作更改
    if (!saving) {
      edit.setVisible(false)
      save.setVisible(true)
      saving = true
      val current = (Instant.now(), from)
      val task = new Task[Double] {
        override def call(): Double = {
          //n s 后自动保存
          TimeUnit.SECONDS.sleep(waiting.getText().trim.toInt)
          //如果检测到更改，则以最后修改为准
          val real = (current._1, key.get())
          data.append(real)
          println("Saved " + real)
          saving = false
          Platform.runLater(() => {
            key.set("Waiting...")
            save.setVisible(false)
            edit.setVisible(false)
            stage.setTitle(fileName + " [ Collected - " + data.length + " ]")
          })
          0.0
        }
      }
      new Thread(task).start()
    } else edit.setVisible(true)
  }

  private[this] def doSave(name: String): Unit = {
    DataUtils.saveTo(Paths.get(name)) {
      val sb = new StringBuilder
      data.foreach(i => {
        sb.append(i._1.toString).append(", ").append(i._2).append("\n")
      })
      sb
    }
  }
}