From b7d38ef1c1a6ef4e051124779b7637a344d3a223 Mon Sep 17 00:00:00 2001
From: Masoud Ashrafzadeh <m.ashrafzadeh@kian.digital>
Date: Sun, 23 Feb 2020 14:41:05 +0330
Subject: [PATCH] selection color bug fix

---
 .../java/com/sma6871/cardentry/CardEntry.kt    | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/cardentry/src/main/java/com/sma6871/cardentry/CardEntry.kt b/cardentry/src/main/java/com/sma6871/cardentry/CardEntry.kt
index e0664a8..4ea8bb7 100644
--- a/cardentry/src/main/java/com/sma6871/cardentry/CardEntry.kt
+++ b/cardentry/src/main/java/com/sma6871/cardentry/CardEntry.kt
@@ -32,13 +32,13 @@ class CardEntry : AppCompatEditText {
     private val spaceSize
         get() = getCharSize(" ")
     private var spaceCount = 0
-    private val mSpace
-        get() = spaceCount * spaceSize
+    private val mSpace: Float by lazy { spaceCount * spaceSize }
 
     private val mPartLength
         get() = maxLength / partCount
-    private val mPartSize
-        get() = mCharSize * mPartLength
+
+    private val mPartSize: Float by lazy { mCharSize * mPartLength }
+
     private var mLineSpacing = toPxF(12)
     private var mLineSpacingAnimated = toPxF(12)
 
@@ -76,6 +76,10 @@ class CardEntry : AppCompatEditText {
         }
     var oldText = ""
 
+    private val partNumbersCount:Int by lazy {
+        maxLength/partCount
+    }
+
     private fun String.getChunked(): String {
         return replace(" ", "").chunked(4).joinToString(separator = spaces)
     }
@@ -243,7 +247,7 @@ class CardEntry : AppCompatEditText {
 
     private fun copySelectedNumberWithoutSpaces() {
         val clipboardManager = context?.getSystemService(Context.CLIPBOARD_SERVICE) as ClipboardManager
-        val copiedText = text?.substring(selectionStart until  selectionEnd)?.replace(" ", "")
+        val copiedText = text?.substring(selectionStart until selectionEnd)?.replace(" ", "")
         clipboardManager.setPrimaryClip(ClipData.newPlainText(copiedText, copiedText))
     }
 
@@ -333,7 +337,9 @@ class CardEntry : AppCompatEditText {
     }
 
     private fun drawNumber(canvas: Canvas, text: CharSequence, i: Int, middle: Float, top: Int, animated: Boolean) {
-        val isSelected = i in selectionStart until selectionEnd
+        val diff = spaceCount * (selectionStart / (spaceCount + partNumbersCount))
+        val diffEnd = spaceCount * (selectionEnd / (spaceCount + partNumbersCount))
+        val isSelected = i in selectionStart - diff until selectionEnd - diffEnd
         if (animated) {
             paint.alpha = animatedAlpha
         } else {