From 8adce506154748345fc83a4448a211eb91860a4b Mon Sep 17 00:00:00 2001
From: Jonathan Trowbridge <jonathan.trowbridge@gmail.com>
Date: Fri, 10 Aug 2018 00:40:59 -0400
Subject: [PATCH] session-changes - tweaks to adapter.

---
 .../adapters/TorrentPieceAdapter.kt                   | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/app/src/main/java/com/masterwok/demosimpletorrentandroid/adapters/TorrentPieceAdapter.kt b/app/src/main/java/com/masterwok/demosimpletorrentandroid/adapters/TorrentPieceAdapter.kt
index 337806e..f5c57b7 100644
--- a/app/src/main/java/com/masterwok/demosimpletorrentandroid/adapters/TorrentPieceAdapter.kt
+++ b/app/src/main/java/com/masterwok/demosimpletorrentandroid/adapters/TorrentPieceAdapter.kt
@@ -51,8 +51,13 @@ class TorrentPieceAdapter : RecyclerView.Adapter<TorrentPieceAdapter.PieceViewHo
             , position: Int
     ): Int {
         val isDownloaded = torrentSessionBuffer.isPieceDownloaded(position)
+        val isHeadIndex = torrentSessionBuffer.bufferHeadIndex == position
 
         if (isDownloaded) {
+            if (isHeadIndex) {
+                return ContextCompat.getColor(context, R.color.blue)
+            }
+
             return ContextCompat.getColor(context, R.color.green)
         }
 
@@ -60,11 +65,7 @@ class TorrentPieceAdapter : RecyclerView.Adapter<TorrentPieceAdapter.PieceViewHo
             return ContextCompat.getColor(context, R.color.purple)
         }
 
-        if (torrentSessionBuffer.bufferHeadIndex == position) {
-            if (isDownloaded) {
-                return ContextCompat.getColor(context, R.color.blue)
-            }
-
+        if (isHeadIndex) {
             return ContextCompat.getColor(context, R.color.red)
         }