Fix consume flag when using listener

Fixes #102
chrisbanes · Mar 16, 2021 · de19152 · de19152
1 parent 2724d46
commit de19152
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/library/src/main/java/dev/chrisbanes/insetter/Insetter.kt b/library/src/main/java/dev/chrisbanes/insetter/Insetter.kt
@@ -370,8 +370,10 @@ class Insetter private constructor(builder: Builder) {
                 // If we have an onApplyInsetsListener, invoke it
                 onApplyInsetsListener.onApplyInsets(v, insets, initialState)
                 // We don't know what sides have been applied, so we assume all
-                return@setOnApplyWindowInsetsListener if (consume != CONSUME_NONE) insets
-                else WindowInsetsCompat.CONSUMED
+                return@setOnApplyWindowInsetsListener when (consume) {
+                    CONSUME_NONE -> insets
+                    else -> WindowInsetsCompat.CONSUMED
+                }
             }
 
             // Otherwise we applied through applyInsetsToView()