Skip to content

Instantly share code, notes, and snippets.

@sphrak
Forked from maiconhellmann/MaskExtension.kt
Created February 8, 2020 20:52
Show Gist options
  • Save sphrak/79144c2cab27a6e1222cfe4a5877b3a6 to your computer and use it in GitHub Desktop.
Save sphrak/79144c2cab27a6e1222cfe4a5877b3a6 to your computer and use it in GitHub Desktop.
MaskExtension wrote in Kotlin
import android.support.annotation.StringDef
import android.text.Editable
import android.text.TextWatcher
import android.widget.EditText
@StringDef(PHONE_9_MASK, PHONE_8_MASK, CPF_MASK, ZIP_CODE_PT_BR, MONTH_YEAR, CREDIT_CARD)
@Retention(AnnotationRetention.SOURCE)
annotation class MaskType
const val PHONE_9_MASK = "(##) #####-####"
const val PHONE_8_MASK = "(##) ####-####"
const val CPF_MASK = "###.###.###-##"
const val ZIP_CODE_PT_BR = "#####-###"
const val MONTH_YEAR = "##/##"
const val CREDIT_CARD = "#### #### #### ####"
fun String.unmask(): String {
return replace("[\\./\\(\\) \\-\\+]".toRegex(), "")
}
@Suppress("UNUSED")
fun EditText.insert(@MaskType mask: String): TextWatcher {
val textWatcher = MaskTextWatcher(mask)
addTextChangedListener(textWatcher)
return textWatcher
}
fun EditText.insertPhoneMask(): TextWatcher {
val textWatcher = object : MaskTextWatcher() {
override fun getMask(unmaskedValue: String): String {
if (unmaskedValue.length < 11) {
return PHONE_8_MASK
}
return PHONE_9_MASK
}
}
addTextChangedListener(textWatcher)
return textWatcher
}
fun String.formatPhone(): String {
var _phone = this
if (length == 8)
_phone = String.format("%s-%s", substring(0, 4), substring(4, length))
else if (length == 9)
_phone = String.format("%s-%s", substring(0, 5), substring(5, length))
else if (length == 10)
_phone = String.format("%s %s-%s", substring(0, 2), substring(2, 6), substring(6, length))
else if (length == 11)
_phone = String.format("%s %s-%s", substring(0, 2), substring(2, 7), substring(7, length))
return _phone
}
open class MaskTextWatcher(val mask: String = ""): SimpleTextWatcher() {
internal var oldValue = ""
internal var isUpdating: Boolean = false
open fun getMask(unmaskedValue: String): String {
return mask
}
override fun afterTextChanged(edit: Editable) {
val unmaskedString = edit.toString().unmask()
val maskedString = StringBuilder("")
val mask = getMask(unmaskedString)
// EditText was GC'ed
if (isUpdating) {
oldValue = unmaskedString
isUpdating = false
return
}
var i = 0
for (m in mask.toCharArray()) {
if (m != '#' && i < unmaskedString.length) {
maskedString.append(m)
continue
}
try {
maskedString.append(unmaskedString[i])
} catch (e: Exception) {
break
}
i++
}
isUpdating = true
edit.replace(0, edit.length, maskedString)
}
}
open class SimpleTextWatcher : TextWatcher {
override fun beforeTextChanged(s: CharSequence, start: Int, count: Int, after: Int) { }
override fun onTextChanged(s: CharSequence, start: Int, before: Int, count: Int) { }
override fun afterTextChanged(edit: Editable) { }
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment