# implemented activity context injection for Beans injected by Views

SchwarzIT · Jun 2, 2020 · 670c78f · 670c78f
1 parent 45ff6f0
commit 670c78f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/kokain-di/src/main/java/com/schwarz/kokain/di/ActivityContextGuard.kt b/kokain-di/src/main/java/com/schwarz/kokain/di/ActivityContextGuard.kt
@@ -3,6 +3,7 @@ package com.schwarz.kokain.di
 import android.app.Activity
 import android.app.Application
 import android.content.Context
+import android.view.View
 import androidx.activity.ComponentActivity
 import androidx.fragment.app.Fragment
 import androidx.lifecycle.LifecycleObserver
@@ -35,6 +36,9 @@ class ActivityContextGuard(applicationContext: Application) : LifecycleObserver
         if (thisRef is Activity) {
             return thisRef?.equals(currentRef?.get()?.toString())
         }
+        if(thisRef is View){
+            return thisRef?.context?.equals(currentRef?.get()?.toString()) ?: false
+        }
         return false
     }
 
@@ -48,6 +52,9 @@ class ActivityContextGuard(applicationContext: Application) : LifecycleObserver
             if (thisRef is Activity) {
                 bean.activityRef = thisRef?.toString()
             }
+            if(thisRef is View){
+                bean.activityRef = thisRef?.context?.toString()
+            }
         }