From 4507d8b5882e283c3e634ff11aa64a85f38c0dc0 Mon Sep 17 00:00:00 2001
From: Ammar Githam <ammargitham786@gmail.com>
Date: Mon, 29 Mar 2021 23:02:42 +0900
Subject: [PATCH] Handle unknown dm types. Fixes the crash in
 https://github.com/austinhuang0131/barinsta/issues/891

---
 .../instagrabber/adapters/DirectItemsAdapter.java   | 13 +++++++++++--
 .../instagrabber/models/enums/DirectItemType.java   |  4 ++++
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/app/src/main/java/awais/instagrabber/adapters/DirectItemsAdapter.java b/app/src/main/java/awais/instagrabber/adapters/DirectItemsAdapter.java
index 9fa309f2..f28fb139 100644
--- a/app/src/main/java/awais/instagrabber/adapters/DirectItemsAdapter.java
+++ b/app/src/main/java/awais/instagrabber/adapters/DirectItemsAdapter.java
@@ -146,7 +146,12 @@ public final class DirectItemsAdapter extends RecyclerView.Adapter<RecyclerView.
     @NonNull
     private DirectItemViewHolder getItemViewHolder(final LayoutInflater layoutInflater,
                                                    final LayoutDmBaseBinding baseBinding,
-                                                   @NonNull final DirectItemType directItemType) {
+                                                   final DirectItemType directItemType) {
+        if (directItemType == null) {
+            // For unknown dm types
+            final LayoutDmTextBinding binding = LayoutDmTextBinding.inflate(layoutInflater, baseBinding.message, false);
+            return new DirectItemDefaultViewHolder(baseBinding, binding, currentUser, thread, callback);
+        }
         switch (directItemType) {
             case TEXT: {
                 final LayoutDmTextBinding binding = LayoutDmTextBinding.inflate(layoutInflater, baseBinding.message, false);
@@ -240,7 +245,11 @@ public final class DirectItemsAdapter extends RecyclerView.Adapter<RecyclerView.
         if (itemOrHeader.isHeader()) {
             return -1;
         }
-        return itemOrHeader.item.getItemType().getId();
+        final DirectItemType itemType = itemOrHeader.item.getItemType();
+        if (itemType == null) {
+            return 0;
+        }
+        return itemType.getId();
     }
 
     @Override
diff --git a/app/src/main/java/awais/instagrabber/models/enums/DirectItemType.java b/app/src/main/java/awais/instagrabber/models/enums/DirectItemType.java
index 8f6f5ceb..58821f5c 100755
--- a/app/src/main/java/awais/instagrabber/models/enums/DirectItemType.java
+++ b/app/src/main/java/awais/instagrabber/models/enums/DirectItemType.java
@@ -1,5 +1,7 @@
 package awais.instagrabber.models.enums;
 
+import androidx.annotation.Nullable;
+
 import com.google.gson.annotations.SerializedName;
 
 import java.io.Serializable;
@@ -59,7 +61,9 @@ public enum DirectItemType implements Serializable {
         return id;
     }
 
+    @Nullable
     public static DirectItemType valueOf(final int id) {
+        if (!map.containsKey(id)) return null;
         return map.get(id);
     }