1 files changed, 30 insertions, 30 deletions
diff --git a/app/src/main/kotlin/com/pitchedapps/frost/injectors/CssAssets.kt b/app/src/main/kotlin/com/pitchedapps/frost/injectors/CssAssets.kt
index 3155b11a..d9542cba 100644
--- a/app/src/main/kotlin/com/pitchedapps/frost/injectors/CssAssets.kt
+++ b/app/src/main/kotlin/com/pitchedapps/frost/injectors/CssAssets.kt
@@ -2,6 +2,7 @@ package com.pitchedapps.frost.injectors
 
 import android.graphics.Color
 import android.webkit.WebView
+import ca.allanwang.kau.kotlin.lazyContext
 import ca.allanwang.kau.utils.*
 import com.pitchedapps.frost.utils.L
 import com.pitchedapps.frost.utils.Prefs
@@ -17,42 +18,41 @@ enum class CssAssets(val folder: String = "themes") : InjectorContract {
     MATERIAL_LIGHT, MATERIAL_DARK, MATERIAL_AMOLED, MATERIAL_GLASS, CUSTOM, ROUND_ICONS("components")
     ;
 
-    var file = "${name.toLowerCase(Locale.CANADA)}.compact.css"
-    var injector: JsInjector? = null
-
-    override fun inject(webView: WebView, callback: ((String) -> Unit)?) {
-        if (injector == null) {
-            try {
-                var content = webView.context.assets.open("css/$folder/$file").bufferedReader().use { it.readText() }
-                if (this == CUSTOM) {
-                    val bt: String
-                    if (Color.alpha(Prefs.bgColor) == 255) {
-                        bt = Prefs.bgColor.toRgbaString()
-                    } else {
-                        bt = "transparent"
-                    }
-                    content = content
-                            .replace("\$T\$", Prefs.textColor.toRgbaString())
-                            .replace("\$TT\$", Prefs.textColor.colorToBackground(0.05f).toRgbaString())
-                            .replace("\$A\$", Prefs.accentColor.toRgbaString())
-                            .replace("\$B\$", Prefs.bgColor.toRgbaString())
-                            .replace("\$BT\$", bt)
-                            .replace("\$BBT\$", Prefs.bgColor.withAlpha(51).colorToForeground(0.35f).toRgbaString())
-                            .replace("\$O\$", Prefs.bgColor.withAlpha(255).toRgbaString())
-                            .replace("\$OO\$", Prefs.bgColor.withAlpha(255).colorToForeground(0.35f).toRgbaString())
-                            .replace("\$D\$", Prefs.textColor.adjustAlpha(0.3f).toRgbaString())
+    var file = "${name.toLowerCase(Locale.CANADA)}.css"
+    var injector = lazyContext {
+        try {
+            var content = it.assets.open("css/$folder/$file").bufferedReader().use { it.readText() }
+            if (this == CUSTOM) {
+                val bt: String
+                if (Color.alpha(Prefs.bgColor) == 255) {
+                    bt = Prefs.bgColor.toRgbaString()
+                } else {
+                    bt = "transparent"
                 }
-                injector = JsBuilder().css(content).build()
-            } catch (e: FileNotFoundException) {
-                L.e(e, "CssAssets file not found")
-                injector = JsInjector(JsActions.EMPTY.function)
+                content = content
+                        .replace("\$T\$", Prefs.textColor.toRgbaString())
+                        .replace("\$TT\$", Prefs.textColor.colorToBackground(0.05f).toRgbaString())
+                        .replace("\$A\$", Prefs.accentColor.toRgbaString())
+                        .replace("\$B\$", Prefs.bgColor.toRgbaString())
+                        .replace("\$BT\$", bt)
+                        .replace("\$BBT\$", Prefs.bgColor.withAlpha(51).colorToForeground(0.35f).toRgbaString())
+                        .replace("\$O\$", Prefs.bgColor.withAlpha(255).toRgbaString())
+                        .replace("\$OO\$", Prefs.bgColor.withAlpha(255).colorToForeground(0.35f).toRgbaString())
+                        .replace("\$D\$", Prefs.textColor.adjustAlpha(0.3f).toRgbaString())
             }
+            JsBuilder().css(content).build()
+        } catch (e: FileNotFoundException) {
+            L.e(e, "CssAssets file not found")
+            JsInjector(JsActions.EMPTY.function)
         }
-        injector!!.inject(webView, callback)
+    }
+
+    override fun inject(webView: WebView, callback: ((String) -> Unit)?) {
+        injector(webView.context).inject(webView, callback)
     }
 
     fun reset() {
-        injector = null
+        injector.invalidate()
     }
 
 }