!7

concept Screen {

  S screenID;

  long lastSeen = now();

  S query;

  transient S status;

  // the ongoing search (if any)

  transient LC_ParallelSearch search;

  transient Lock lock = lock();

  toString { ret "Screen " + screenID; }

}

cmodule2 BEALiveInput > DynPrintLogAndEnabled {

  switchable int statusUpdateInterval = 100;

  transient Map<virtual WebSocket, Screen> webSockets = syncWeakHashMap();

  transient Set<virtual WebSocket> liveSearches = weakHashSet();

  start {

    dbIndexing(Screen, 'screenID);

    dm_doEvery(statusUpdateInterval, r updateStati);

  }

  set flag NoNanoHTTPD. 

  O html(IWebRequest request) { try {

    //try object redirectToHTTPS(request);

    S uri = request.uri();

    SS params = request.params();

    print(+params);

    S cookie = request.cookie();

    print("cookie", cookie);

    new Matches m;

    if (eqic(uri, "/about"))

      ret serveText(unindent_mls([[

        What is the BEA Live Input?

        Bla bla bla. Also, bla.

      ]]));

    if (eqic(uri, "/websockettest")) {

      ret htitle("Web Socket Test")

        + hreconnectingWebSockets()

        + hloadjquery()

        + hdiv("bla<br>", id := "dynamic", style := "font-size: 40px")

        + hjavascript([[

          var ws = new ReconnectingWebSocket("wss://wikify.live/test");

          ws.onopen = function(event) {

            ws.send("here i am");

            $("#dynamic").append("Streaming...<br>");

          };

          ws.onmessage = function(event) {

            $("#dynamic").html(event.data);

            document.title = event.data;

          };

        ]]);

    }

    // home page

    S query = "";

    if (swic(uri, "/search/", m))

      query = takeFirst(1000, urldecode(m.rest()));

    else // now we actually accept wikify.live/{any search term}

      query = urldecode(dropSlashPrefix(uri));

    ret hhtml(hmobilefix() + hhead(

        htitle("BEA Live Input [dev.]")

      + hsansserif()

      + hstyle([[

        .queryContainer {

          height: 100%;

          width: 100%;

          display: flex;

          flex-flow: column;

        }

        input.query {

          font-size: 40px;

          padding: 20px;

          width: 400px;

          height: 2em;

          position: relative;

          left: 50%;

          transform: translate(-50%, 0);

          flex: .01 1 auto;

        }

        .filler {

          flex: .5 1 auto;

        }

        .filler.ontop {

          flex: .01 1 auto;

        }

        input.query.ontop {

          transform: translate(-50%, 0);

          margin-top: 20px;

          font-size: 20px;

        }

        .searching {

          display: none;

        }

        .searching.show {

          display: block;

          position: relative;

          left: 50%;

          margin-left: -45px;

        }

        .results {

          font-family: monospace;

          white-space: pre; /* no word-wrapping */

          overflow: auto; /* scrollbars */

          margin-top: 20px;

          border: 2px solid black;

          padding: 0px;

          display: block;

          flex: .01 1 auto;

          opacity: 0%;

          width: 1px;

          margin-left: auto;

          margin-right: auto;

        }

        .results.ontop {

          padding: 20px;

          flex: 1 0 0;  

          opacity: 100%;

          width: calc(100% - 50px);

          left: 0;

          transform: none;

        }

        .about {

          margin-bottom: 10px;

          font-size: 12px;

          text-align: center;

          margin-left: auto;

          margin-right: auto;

        }

        .status {

          display: none;

          margin-top: 10px;

          font-size: 12px;

        }

        .status.ontop {

          display: block;

          margin-left: auto;

          margin-right: auto;

        }

        .query, .results, .filler {

          transition: all 1s;

        }

      ]])

      + hLoadJQuery2()

      + hreconnectingWebSockets())

      + hbody(hdiv(

            hdiv("", class := "filler")

          + hdiv("AI bla bla<br>"

          + "By " + ahref("https://BotCompany.de", "BotCompany") + "."

          + targetBlank("/about", "More info."), class := "about")

          + hinput("", value := query, autofocus := true, class := "query", placeholder := "TYPE ANYTHING")

          + hdiv("", class := "status")

          + hdiv(

              hdiv("", class := "actualResults")

            + hdiv(himgsnippet(#1102949, title := "Searching..."), class := "searching"),

              class := "results")

          + hdiv("", class := "filler"),

          class := "queryContainer")

      + hjs_urlencode()

      + hjs([[

        function queryUpdate() {

          var sel = $("input.query, .queryContainer, .results, .filler, .status");

          var q = $("input.query").val();

          if (q == "") {

            $(".actualResults").html("");

            sel.removeClass("ontop");

          } else {

            sel.addClass("ontop");

          }

          if (history.replaceState)

            history.replaceState(null, "Wikify.LIVE", q == "" ? "/" : "/search/" + urlencode(q));

          sendQuery();

        }

        $("input.query").on('change keydown paste input', queryUpdate);

        var screenID = Math.random().toString(36).substr(2, 9);

        var wsURL = "wss://wikify.live/search?screen=" + urlencode(screenID);

        console.log("WS: " + wsURL);

        var ws = new ReconnectingWebSocket(wsURL);

        var wsReady = false;

        var querySent = "";

        var maxQueryLength = 1000;

        function wsSend(text) {

          if (wsReady) {

            console.log("wsSend: " + text);

            ws.send(text);

          } else

            console.log("wsNotReady: " + text);

        }

        function sendQuery() {

          var q = $("input.query").val().substring(0, maxQueryLength);

          if (querySent != q) {

            querySent = q;

            wsSend(JSON.stringify({ q: q }));

          }

        }

        ws.onopen = function(event) {

          wsReady = true;

          ws.onmessage = function(event) {

            var text = event.data;

            console.log("Got text: " + text);

            var data = JSON.parse(text);

            if ('clearResults' in data)

              $(".actualResults").html("");

            if ('searching' in data)

              $(".searching").addClass("show");

            if ('result' in data)

              $(".actualResults").append(data.result + "<br>");

            if ('searchCancelled' in data)

              $(".searching").removeClass("show");

            if ('searchDone' in data) {

              $(".actualResults").append("<br>[SEARCH DONE]<br>");

              $(".searching").removeClass("show");

            }

            if ('status' in data)

              $(".status").html(data.status);

          };

          queryUpdate();

        };

      ]])

      ));

  } catch e { printStackTrace(e); throw rethrow(e); }

  }

  class Request {

    S cookie;  

  }

  void cleanMeUp_webSockets {

    for (virtual WebSocket ws : cloneAndClearKeys(webSockets)) {

      print("Closing WebSocket");

      close((AutoCloseable) ws);

    }

  }

  void cleanMeUp_searches {

    for (virtual WebSocket ws : cloneAndClearList(liveSearches))

      cancelSearch_noLock(webSockets.get(ws));

  }

  // API for Eleu

  void setEleu(O eleu) {

    setOptMC(mainBot := eleu);

  }

  void handleWebSocket(virtual WebSocket ws) {

    print("Got new websocket!");

    set(ws, onClose := r { webSockets.remove(ws) });

    set(ws, onOpen := rEnter {  

      S uri = cast rcall getUri(ws);

      print("WebSocket opened! uri: " + uri);

      SS params = cast rcall getParms(ws);

      S screenID = takeFirst(params.get("screen"), 100);

      Screen screen = uniq Screen(+screenID);

      screen.status = null;

      webSockets.put(ws, screen);

      cset(screen, lastSeen := now());

      print("Screen ID: " + screenID);

      setFieldToIVF1Proxy(ws, onMessage := msg -> {

        temp enter();

        pcall {

          S text = cast rcall getTextPayload(msg);

          print("Got msg: " + text);

          Map json = jsonDecodeMap(text);

          S q = upper((S) json.get("q"));

          if (q != null && neq(screen.query, q)) {

            temp tempLock("start search", screen.lock);

            cancelSearch_noLock(screen);

            cset(screen, query := q);

            if (nempty(q))

              startSearch_noLock(ws, screen, q);

          }

        }

      });

      if (eq(uri, "/test")) thread {

        repeat 100 {

          call(ws, 'send, "It is " + localTimeWithMilliseconds());

          sleep(100);

        }

      }

      if (eq(uri, "/search")) {

      }

    });

  }

  void cancelSearch_noLock(Screen screen) {

    if (screen == null) ret;

    close(screen.search);

    screen.search = null;

  }

  void startSearch_noLock(virtual WebSocket ws, Screen screen, S query) {

    cset(screen, +query);

    call(ws, "send", jsonEncode(litorderedmap("clearResults", true)));

    if (empty(query)) ret;

    call(ws, "send", jsonEncode(litorderedmap("searching", true)));

    print("Starting search for " + quote(query) + " for " + screen);

    /*LC_ParallelSearch search = new(searchers, query);

    screen.search = search;

    liveSearches.add(ws);

    search.enter = lambda0 enter;

    search.maxResults = defaultResults;

    search.onResult = (iSearcher, iFile, position) -> {

      temp tempLock("onResult", screen.lock);

      if (screen.search != search) ret;

      try {

        //print("Sending result");

        S status = search.status_html();

        call(ws, "send", jsonEncode(litorderedmap(

          result := renderResult(iSearcher, iFile, position, query, screen.search.startTime),

          +status)));

        //print("Sent result");

      } catch print e {

        cancelSearch_noLock(screen);

      }

    };

    search.onSearchDone = r {

      liveSearches.remove(ws);

      print("Search done. " + screen.search + " / " + search);

      temp tempLock("onSearchDone", screen.lock);

      if (screen.search != search) ret;

      updateStatus(ws, screen);

      print("Cancelling search");

      cancelSearch_noLock(screen);

      print("Sending searchDone");

      call(ws, "send", jsonEncode(litorderedmap(

        search.completed() ? "searchDone" : "searchCancelled", true)));

      print("Sent searchDone");

    };

    search.start();*/

  }

  void updateStati enter {

    for (virtual WebSocket ws : cloneList(liveSearches)) pcall {

      Screen screen = webSockets.get(ws);

      updateStatus(ws, screen);

    }

  }

  void updateStatus(virtual WebSocket ws, Screen screen) {

    if (screen == null) ret;

    /*LC_ParallelSearch search = screen.search;

    if (search == null) ret;

    S status = search.status_html();

    if (neq(status, screen.status)) {

      screen.status = status;

      call(ws, "send", jsonEncode(litorderedmap(status := screen.status)));

    }*/

  }

}