sO thoughtBot;

static int longPollTick = 100;
static int longPollMaxWait = 1000*60;

sclass Msg {
  long time;
  bool fromUser;
  S text;
  L<S> buttons;
  
  *() {}
  *(bool *fromUser, S *text) { time = now(); }
}

concept Conversation {
  S cookie;
  new LL<Msg> oldDialogs;
  new L<Msg> msgs;

  void add(Msg m) {
    msgs.add(m);
    change();
  }
  
  int allCount() { ret lengthLevel2(oldDialogs) + l(msgs); }
  int archiveSize() { ret lengthLevel2(oldDialogs); }
}

svoid pWebChatBot {
  db();
  thoughtBot = runDependent(thoughtBotID);
  Class envType = fieldType(thoughtBot, "env");
  if (envType != null)
    setOpt(thoughtBot, "env", proxy(envType, (O) mc()));
}

static void sayAsync(S session, S text) {
  lock dbLock();
  Conversation conv = getConv(session);
  conv.add(new Msg(false, text));
  print("sayAsync " + session + ", new size=" + conv.allCount());
}

html {
 _registerThread();
 registerVisitor();
 fS cookie = cookieSent();
 try {
  Conversation conv;
  {
    lock dbLock();
    if (eq(uri, "/stats"))
      ret "Threads: " + ul_htmlEncode(getThreadNames(registeredThreads()));
      
    if (eq(uri, "/logs"))
      ret webChatBotLogsHTML();

    S message = trim(params.get("btn"));
    if (empty(message)) message = trim(params.get("message"));
    conv = getConv(cookie);
    
    if (match("clear", message)) {
      conv.oldDialogs.add(conv.msgs);
      cset(conv, msgs := new L);
      conv.change();
      callOpt(thoughtBot, "clearSession", conv.cookie);
      message = null;
    }
    
    call(thoughtBot, "setSession", cookieSent());
    
    if (empty(conv.msgs))
      conv.add(new Msg(false, initialMessage()));
      
    if (nempty(message) && !lastUserMessageWas(conv, message))
      conv.add(new Msg(true, message));
  
    if (nempty(conv.msgs) && last(conv.msgs).fromUser) {
      S reply = "I'm sorry, I don't understand";
      L<S> buttons = null;
      pcall {
        reply = or2(makeReply(last(conv.msgs).text), reply);
        buttons = (L<S>) getThreadLocal(thoughtBot, "buttons");
      }
      Msg msg = new Msg(false, reply);
      msg.buttons = buttons;
      conv.add(msg);
    }
  } // locked
  
  if (eq(uri, "/msg")) ret "OK";
  
  if (eq(uri, "/incremental")) {
    int a = parseInt(params.get("a")), origA = a;
    //int b = parseInt(params.get("b"));
    int as = conv.archiveSize();
    a -= as;
    //b -= as;
    
    long start = sysNow();
    L msgs;
    bool first = true;
    while (licensed() && sysNow() < start+longPollMaxWait) {
      msgs = subList(conv.msgs, a);
      if (empty(msgs)) {
        if (first) {
          print("Long poll starting on " + cookie + ", " + origA + "/" + a);
          first = false;
        }
        sleep(longPollTick);
      } else {
        if (first) print("Long poll ended.");
        new StringBuilder buf;
        renderMessages(buf, msgs);
        ret "<!-- " + conv.allCount() + "-->\n"  + buf;
      }
    }
    ret "";
  }
  
  {
    lock dbLock();
    S html = loadSnippet(templateID); // TODO: cache
    new StringBuilder buf;
    
    renderMessages(buf, conv.msgs);
  
    html = html.replace("#N#", str(conv.allCount()));
    html = html.replace("#INCREMENTALURL#", relativeRawBotLink(programID(), "incremental?a=");
    html = html.replace("#MSGURL#", relativeRawBotLink(programID(), "msg?message="));
    if (nempty(params.get("debug")))
      html = html.replace("var showActions = false;", "var showActions = true;");
  
    html = html.replace("$HEADING", heading);
    html = html.replace("<!-- MSGS HERE -->", str(buf));
    html = hreplaceTitle(html, heading);
    html = hmobilefix(html);
    ret html;
  }
 } finally {
  _unregisterThread();
 }
}

svoid renderMessages(StringBuilder buf, L<Msg> msgs) {
  for (Msg m : msgs)
    if (m.fromUser || neq(m.text, "-"))
      appendMsg(buf, m.fromUser ? "Esteemed Customer" : botName, formatTime(m.time), htmlEncode2_nlToBr(trim(m.text)), !m.fromUser);
      
  L<S> buttons = last(msgs).buttons;
  if (nempty(buttons))
    appendButtons(buf, buttons);
}

svoid appendMsg(StringBuilder buf, S name, S time, S text, bool bot) {
  S imgLink = snippetImgLink(bot ? #1008323 : #1008359);
  buf.append([[<div class="direct-chat-msg doted-border">
    <div class="direct-chat-info clearfix">
    <span class="direct-chat-name pull-left">$NAME</span>
    </div>
    <img alt="message user image" src="$IMG" class="direct-chat-img">
    <div class="direct-chat-text">
      $TEXT
    </div>
    <div class="direct-chat-info clearfix">
      <span class="direct-chat-timestamp pull-right">$TIME</span>
    </div>
  </div>
    ]].replace("$IMG", imgLink)
      .replace("$NAME", name)
      .replace("$TIME", time)
      .replace("$TEXT", text));
}

svoid appendButtons(StringBuilder buf, L<S> buttons) {
  S buttonsHtml = lines(map(buttons, func(S text) {
    hsubmit(text, name := "btn")
  }));
  buf.append([[<div class="direct-chat-msg doted-border">
    <div class="direct-chat-buttons">
      $BUTTONS
    </div>
  </div>
    ]].replace("$BUTTONS", buttonsHtml);
}

svoid appendDate(StringBuilder buf, S date) {
  buf.append([[
  <div class="chat-box-single-line">
    <abbr class="timestamp">DATE</abbr>
  </div>]].replace("DATE", date));
}

sS initialMessage() {
  ret (S) call(thoughtBot, "initialMessage"); 
}

static bool lastUserMessageWas(Conversation conv, S message) {
  Msg m = last(conv.msgs);
  ret m != null && m.fromUser && eq(m.text, message);
}

sS makeReply(S message) {
  ret callStaticAnswerMethod(thoughtBot, message);
}

sS formatTime(long time) {
  ret formatGMT_24(time);
}

sS formatDialog(S id, L<Msg> msgs) {
  new L<S> lc;
  for (Msg m : msgs)
    lc.add(htmlencode((m.fromUser ? "> " : "< ") + m.text));
  ret id + ul(lc);
}

static Conversation getConv(fS cookie) {
  ret withDBLock(func -> Conversation {
    uniq(Conversation, +cookie)
  });
}

Began life as a copy of #1008316