!7

sbool autoClose = true;
static int initialDelay = 0;
sbool infoBox, repeatEverything;

static int port;
static O onUtterance; // voidfunc(S)
static L<S> botsToSendUtterancesTo = ll("Voice Actions.", "Show Item Page.");
static L<S> functionsToSendTo = ll();
static L<WebSocket> webSockets = synchroList();
sbool startRecognition;
static java.util.Timer stopper;
sS language = "en-US";
sS myURL;
static JButton btn;
sbool hadAnyWebSockets; // Does Chrome work at all?

p {
  botsToSendUtterancesTo = persistentList("Bots to send utterances to", botsToSendUtterancesTo);
  functionsToSendTo = persistentList("Fuctions to send to", botsToSendUtterancesTo);
  load('language);
  if (isMainProgram())
    infoBox = repeatEverything = true;
  NanoHTTPD.SOCKET_READ_TIMEOUT = 24*3600*1000; // not long enough - TODO: Fix in NanoHTTPD
  
  port = serveHttpWithWebSockets(9999, func(NanoHTTPD.IHTTPSession handshake) {
    WebSocket ws = new WebSocket(handshake) {
      protected void onPong(WebSocketFrame pongFrame) { print("pong"); }
      protected void onMessage(WebSocketFrame messageFrame) {
        cancelTimeout();
        fS s = messageFrame.getTextPayload();
        thread {
          if (repeatEverything) sendOptInNewThread("Mouth.", infoBoxAndReturn(/*switcheroo*/(s)));
          else infoBoxOrPrint("User said: " + s, infoBox);
          pcallF(onUtterance, s);
          for (S bot : cloneList(botsToSendUtterancesTo)) {
            print("Sending to bot " + bot);
            sendOptInNewThread(bot, "User said: *", s);
          }
          for (fS sf : cloneList(functionsToSendTo)) thread {
            makeAndCall(sf, s);
          }
        }
      }
      protected void onClose(WebSocketFrame.CloseCode code, String reason, boolean initiatedByRemote) { webSockets.remove(this); }
      protected void onException(IOException e) { printStackTrace(e); }
    };
    if (startRecognition) {
      startRecognition = false;
      ws.send("start");
    }
    
    // close any other recognizers
    for (WebSocket ws2 : cloneList(webSockets)) {
      pcall { ws2.close(WebSocketFrame.CloseCode.NormalClosure, ""); }
      webSockets.remove(ws2);
    }
    
    hadAnyWebSockets = true;
    
    ret addAndReturn(webSockets, ws);
  });
  myURL = print("http://localhost:" + port + "/popup");
  startChromeApp(myURL);
  makeBot("Chrome Speech.");
  
  showControls(jcenteredline(btn = jbutton("Open Speech Recognizer", r {
    startChromeApp(myURL)
  })));
  awtEvery(btn, 500, r { setEnabled(btn, empty(webSockets)) });
  
  /*thread "Chrome Re-Starter" {
    sleepSeconds(20);
    repeat with sleep 5 {
      if (hadAnyWebSockets && empty(webSockets)) {
        startChromeApp(myURL);
        sleepSeconds(15);
      }
    }
  });*/
}

html {
  if (neq(uri, "/popup"))
    ret hbody("Opening popup..." + hjavascript([[
      window.open('/popup', 'speech_recognizer', 'width=300,height=300,location=no');
      setTimeout(function() { window.close(); }, 10000);
    ]]));

  ret hhtml(hhead(htitle("Speech Recognizer")) + hbody(div(
    h3("Speech Recognizer")
    + loadJQuery()
    + hdiv("Language: " + language, id := 'lang, style := "font-size: 10px")
    + hdiv("Results come here", id := 'results, style := "margin: 10px")
  + hjavascript([[
    var websocket;
    
    function stopRecognition() {
      recognition.stop();
      started = false;
      $("#btn").html("Start recognition");
      document.title = "Speech Recognizer";
    }
    
    function startRecognition() {
      recognition.start();
      started = true;
      $("#btn").html("Stop recognition");
      document.title = "Listening - Speech Recognizer";
    }
    
    function openWebSocket() {
      websocket = new WebSocket("ws://localhost:#PORT#/");
      websocket.onopen = function(event) {
        $("#btn").prop('disabled', false);
      };
    
      websocket.onmessage = function(event) {
        if (event.data == 'start' && !started) startOrStop();
        if (event.data == 'stop' && started) startOrStop();
        if (event.data.substring(0, 9) == 'language ') {
          var l = event.data.substring(9);
          recognition.lang = l;
          $("#lang").html("Language: " + l);
        }
      };
    
      websocket.onclose = function(event) {
        $("#results").html("WebSocket closed");
        if (#AUTOCLOSE#) window.close();
      };
    }
    
    setTimeout(openWebSocket, #INITIALDELAY#);
      
    var recognition = new webkitSpeechRecognition();
    recognition.lang = "#LANGUAGE#";
    
    recognition.onerror = function(event) { 
      $("#results").html("Error: " + event.error);
      stopRecognition();
    }
    
    recognition.onresult = function(event) { 
      var result = event.results[0];
      var transcript = result[0].transcript;
      $("#results").html("Transcript: " + transcript);
      websocket.send(transcript);
      stopRecognition();
    }
    
    recognition.onnomatch = function(event) { 
      $("#results").html("-");
      stopRecognition();
    }
    
    var started = false;
    
    function startOrStop() {
      if (started) stopRecognition(); else {
        stopRecognition(); startRecognition();
      }
    }
    
    window.resizeTo(300, 300);
  ]]).replace("#PORT#", str(port)).replace("#AUTOCLOSE#", autoClose ? "true" : "false").replace("#INITIALDELAY#", str(initialDelay)).replace("#LANGUAGE#", language)
    + tag('button, "Start recognition", onclick := "startOrStop()", type := 'button, id := 'btn, disabled := 'disabled)
    //+ p(ahref("#", "Popup", onClick := "window.open('/', 'speech_recognizer', 'width=300,height=300,location=no'); return false;"));
  , style := "text-align: center"));
}

svoid startRecognition {
  L<WebSocket> l = cloneList(webSockets);
  if (empty(l)) startRecognition = true;
  else {
    //print("Starting recognition." + (l(l) > 1 ? "Weird: Have " + l(l) + " websockets" : ""));
    pcall {
      first(l).send("start");
    }
  }
}

svoid stopRecognition {
  if (startRecognition) startRecognition = false;
  if (nempty(webSockets)) pcall {
    first(webSockets).send("stop");
  }
}

answer {
  if "start recognition timeout *" {
    final int seconds = parseInt($1);
    startRecognition();
    stopper = timerOnce(toMS(seconds), f stopRecognition);
    ret "OK";
  }
  if "start recognition" { startRecognition(); ret "OK"; }
  if "stop recognition" { stopRecognition(); ret "OK"; }
  if "send to bot *" { setAdd(botsToSendUtterancesTo, $1); ret "OK"; }
  if "what bots are you sending to" ret sfu(botsToSendUtterancesTo);
  if "clear send list" { clear(botsToSendUtterancesTo); ret "OK"; }
  
  if "send to function *" { setAdd(functionsToSendTo, $1); ret "OK"; }
  if "what functions are you sending to" ret sfu(functionsToSendTo);
  if "clear functions list" { clear(functionsToSendTo); ret "OK"; }
  if "language *" {
    setAndSave('language, $1);
    pcall { if (nempty(webSockets)) first(webSockets).send("language " + $1); }
    ret "OK";
  }
}

svoid cancelTimeout {
  if (stopper != null) { stopper.cancel(); stopper = null; }
}