From b2a87e63e23b4f5ab3584ba6ba7a8648c9dcc58d Mon Sep 17 00:00:00 2001
From: Adhidarma Hadiwinoto <me@adhisimon.org>
Date: Mon, 31 Oct 2016 20:12:34 +0700
Subject: [PATCH] _updateLastMessage

---
 matrix-util.js | 29 +++++++----------------------
 1 file changed, 7 insertions(+), 22 deletions(-)

diff --git a/matrix-util.js b/matrix-util.js
index 98cf323..eba84f5 100644
--- a/matrix-util.js
+++ b/matrix-util.js
@@ -122,7 +122,7 @@ MatrixUtil.prototype.isPartnerOffline = function(partner) {
     return true;
 }
 
-MatrixUtil.prototype.updateLastIncoming = function(partner, msg) {
+MatrixUtil.prototype._updateLastMessage = function(partner, msg, direction) {
     if (!partner) { return; }
     partner = _cleanPartnerId(partner);
 
@@ -140,32 +140,17 @@ MatrixUtil.prototype.updateLastIncoming = function(partner, msg) {
         matrix.buddies[partner] = {};
     }
 
-    matrix.buddies[partner].lastIncoming = {
+    matrix.buddies[destination]['last_' + direction] = {
         msg: msg,
         lastUpdate: moment().format('YYYY-MM-DD HH:mm:ss')
     }
 }
 
-MatrixUtil.prototype.updateLastOutgoing = function(partner, msg) {
-    if (!partner) { return; }
-    partner = _cleanPartnerId(partner);
-
-    let matrix = this.matrix;
-
-    if (!matrix) {
-        return;
-    }
 
-    if (!matrix.buddies) {
-        matrix.buddies = {};
-    }
-
-    if (!matrix.buddies[partner]) {
-        matrix.buddies[partner] = {};
-    }
+MatrixUtil.prototype.updateLastIncoming = function(partner, msg) {
+    this._updateLastMessage(partner, msg, 'incoming');
+}
 
-    matrix.buddies[destination].lastOutgoing = {
-        msg: msg,
-        lastUpdate: moment().format('YYYY-MM-DD HH:mm:ss')
-    }
+MatrixUtil.prototype.updateLastOutgoing = function(partner, msg) {
+    this._updateLastMessage(partner, msg, 'outgoing');
 }
-- 
1.9.0