1 files changed, 14 insertions, 9 deletions
diff --git a/app/src/main/java/net/lacolaco/smileessence/view/adapter/MessageListAdapter.java b/app/src/main/java/net/lacolaco/smileessence/view/adapter/MessageListAdapter.java
index 114f9e58..c2a53df9 100644
--- a/app/src/main/java/net/lacolaco/smileessence/view/adapter/MessageListAdapter.java
+++ b/app/src/main/java/net/lacolaco/smileessence/view/adapter/MessageListAdapter.java
@@ -25,23 +25,28 @@
 package net.lacolaco.smileessence.view.adapter;
 
 import android.app.Activity;
-import net.lacolaco.smileessence.viewmodel.MessageViewModel;
+import android.view.View;
+import android.view.ViewGroup;
+import net.lacolaco.smileessence.entity.DirectMessage;
+import net.lacolaco.smileessence.view.Partials;
 
-public class MessageListAdapter extends OrderedCustomListAdapter<MessageViewModel> {
-
-    // --------------------------- CONSTRUCTORS ---------------------------
+public class MessageListAdapter extends OrderedCustomListAdapter<DirectMessage> {
+    private final Activity activity;
 
     public MessageListAdapter(Activity activity) {
-        super(activity);
+        this.activity = activity;
     }
 
-    // --------------------- GETTER / SETTER METHODS ---------------------
-
     public long getLastID() {
-        return getItem(getCount() - 1).getDirectMessage().getId();
+        return getItem(getCount() - 1).getId();
     }
 
     public long getTopID() {
-        return getItem(0).getDirectMessage().getId();
+        return getItem(0).getId();
+    }
+
+    @Override
+    public View getView(int position, View convertView, ViewGroup parent) {
+        return Partials.getDirectMessageView(getItem(position), activity, convertView);
     }
 }