From 67a229e309885eeab934e6db50b6c62f4c129835 Mon Sep 17 00:00:00 2001
From: Ian Gulliver <ian@flamingcow.tv>
Date: Sun, 27 Dec 2015 21:36:06 -0800
Subject: [PATCH] Handle changes after load.

---
 static/babystats.js | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/static/babystats.js b/static/babystats.js
index 6a77563..bb3cc7d 100644
--- a/static/babystats.js
+++ b/static/babystats.js
@@ -36,6 +36,10 @@ var BabyStats = function(container) {
 };
 
 
+/**
+ * @param {hogfather.PublicChat} chat
+ * @private
+ */
 BabyStats.prototype.onChatReady_ = function(chat) {
   this.chat_ = chat;
 
@@ -52,9 +56,23 @@ BabyStats.prototype.onChatReady_ = function(chat) {
 
   var messages = this.chat_.getMessages();
   messages.forEach(this.handleMessage_.bind(this));
+  this.chat_.addEventListener('message', this.onMessage_.bind(this));
 };
 
 
+/**
+ * @param {Event} e
+ * @private
+ */
+BabyStats.prototype.onMessage_ = function(e) {
+  this.handleMessage_(e.detail);
+};
+
+
+/**
+ * @param {Cosmopolite.typeMessage} message
+ * @private
+ */
 BabyStats.prototype.handleMessage_ = function(message) {
   switch (message.message.type) {
     case 'child_name_change':
@@ -408,6 +426,9 @@ BabyStats.prototype.checkOverlay_ = function() {
 };
 
 
+/**
+ * @private
+ */
 BabyStats.prototype.onChildNameChange_ = function() {
   this.chat_.sendMessage({
     type: 'child_name_change',