1 files changed, 13 insertions, 1 deletions
diff --git a/app/src/main/java/me/brysonsteck/wiimmfiwatcher/wiimmfi/fragments/RoomFragment.java b/app/src/main/java/me/brysonsteck/wiimmfiwatcher/wiimmfi/fragments/RoomFragment.java
index d6326f7..90c97da 100644
--- a/app/src/main/java/me/brysonsteck/wiimmfiwatcher/wiimmfi/fragments/RoomFragment.java
+++ b/app/src/main/java/me/brysonsteck/wiimmfiwatcher/wiimmfi/fragments/RoomFragment.java
@@ -2,7 +2,9 @@ package me.brysonsteck.wiimmfiwatcher.wiimmfi.fragments;
 
 import android.os.Bundle;
 import android.view.View;
+import android.widget.ImageView;
 import android.widget.TextView;
+import android.widget.Toast;
 
 import androidx.annotation.NonNull;
 import androidx.annotation.Nullable;
@@ -18,6 +20,7 @@ import java.util.ArrayList;
 import me.brysonsteck.wiimmfiwatcher.R;
 import me.brysonsteck.wiimmfiwatcher.wiimmfi.Player;
 import me.brysonsteck.wiimmfiwatcher.wiimmfi.RoomData;
+import me.brysonsteck.wiimmfiwatcher.wiimmfi.SetImageView;
 
 public class RoomFragment extends Fragment {
     String display;
@@ -26,8 +29,9 @@ public class RoomFragment extends Fragment {
     ArrayList<Player> players;
     RoomData roomData;
     MaterialToolbar toolbar;
+    ImageView image;
 
-    public RoomFragment(String friendCode, ArrayList<Player> players, String playerLink, String display, MaterialToolbar toolbar) {
+    public RoomFragment(String friendCode, ArrayList<Player> players, String playerLink, String display, MaterialToolbar toolbar, ImageView image) {
         super(R.layout.room_fragment);
         this.roomData = new RoomData(players, friendCode);
         new Thread(() -> {
@@ -37,6 +41,7 @@ public class RoomFragment extends Fragment {
         this.players = players;
         this.playerLink = playerLink;
         this.toolbar = toolbar;
+        this.image = image;
     }
     @Override
     public void onViewCreated(@NonNull View view, @Nullable Bundle savedInstanceState) {
@@ -46,20 +51,24 @@ public class RoomFragment extends Fragment {
         if (header == null) {
             headerTextView.setText(R.string.header_null_error);
             toolbar.setNavigationIcon(null);
+            SetImageView setImageView = new SetImageView(image, header, true);
         }
         if (roomData.error != null) {
             headerTextView.setText(getResources().getString(R.string.jsoup_error, roomData.error));
             toolbar.setNavigationIcon(null);
+            SetImageView setImageView = new SetImageView(image, header, true);
         }
         if (roomData.error == null && header != null) {
             headerTextView.setText(header);
             toolbar.setNavigationIcon(R.drawable.ic_baseline_menu_24);
+            SetImageView setImageView = new SetImageView(image, header, false);
         }
         RecyclerView recyclerView = view.findViewById(R.id.player_data_recycler_view);
         recyclerView.setLayoutManager(new LinearLayoutManager(getContext()));
         recyclerView.setAdapter(new RoomAdapter(display, playerLink, header, players, getContext()));
 
         refreshButton.setOnClickListener((buttonView) -> {
+            Toast.makeText(getContext(), "Refreshing!", Toast.LENGTH_LONG).show();
             this.players.clear();
             this.header = "";
             this.roomData = roomData.refresh();
@@ -69,14 +78,17 @@ public class RoomFragment extends Fragment {
             if (header == null) {
                 headerTextView.setText(R.string.header_null_error);
                 toolbar.setNavigationIcon(null);
+                SetImageView setImageView = new SetImageView(image, header, true);
             }
             if (newRoomData.error instanceof java.net.SocketTimeoutException || newRoomData.error instanceof java.net.UnknownHostException) {
                 headerTextView.setText(getResources().getString(R.string.jsoup_error, roomData.error));
                 toolbar.setNavigationIcon(null);
+                SetImageView setImageView = new SetImageView(image, header, true);
             }
             if (roomData.error == null && header != null) {
                 headerTextView.setText(header);
                 toolbar.setNavigationIcon(R.drawable.ic_baseline_menu_24);
+                SetImageView setImageView = new SetImageView(image, header, false);
             }
             recyclerView.setLayoutManager(new LinearLayoutManager(getContext()));
             recyclerView.setAdapter(new RoomAdapter(display, playerLink, header, players, getContext()));