44 lines
1.7 KiB
Kotlin
44 lines
1.7 KiB
Kotlin
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
||
|
License, v. 2.0. If a copy of the MPL was not distributed with this
|
||
|
file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||
|
|
||
|
package org.mozilla.fenix
|
||
|
|
||
|
import android.content.Context
|
||
|
import android.content.SharedPreferences
|
||
|
import androidx.preference.PreferenceManager
|
||
|
import com.squareup.leakcanary.AndroidHeapDumper
|
||
|
import com.squareup.leakcanary.HeapDumper
|
||
|
import com.squareup.leakcanary.LeakCanary
|
||
|
import com.squareup.leakcanary.internal.LeakCanaryInternals
|
||
|
import org.mozilla.fenix.R.string.pref_key_leakcanary
|
||
|
import org.mozilla.fenix.ext.getPreferenceKey
|
||
|
import java.io.File
|
||
|
|
||
|
class DebugFenixApplication : FenixApplication() {
|
||
|
|
||
|
private var heapDumper: ToggleableHeapDumper? = null
|
||
|
|
||
|
override fun setupLeakCanary() {
|
||
|
val leakDirectoryProvider = LeakCanaryInternals.getLeakDirectoryProvider(this)
|
||
|
val defaultDumper = AndroidHeapDumper(this, leakDirectoryProvider)
|
||
|
heapDumper = ToggleableHeapDumper(this, defaultDumper)
|
||
|
LeakCanary.refWatcher(this)
|
||
|
.heapDumper(heapDumper)
|
||
|
.buildAndInstall()
|
||
|
}
|
||
|
|
||
|
override fun toggleLeakCanary(newValue: Boolean) {
|
||
|
heapDumper?.enabled = newValue
|
||
|
}
|
||
|
|
||
|
internal class ToggleableHeapDumper(
|
||
|
private val context: Context,
|
||
|
private val defaultDumper: HeapDumper
|
||
|
) : HeapDumper {
|
||
|
var prefs: SharedPreferences? = PreferenceManager.getDefaultSharedPreferences(context)
|
||
|
var enabled = prefs?.getBoolean(context.getPreferenceKey(pref_key_leakcanary), true) ?: true
|
||
|
override fun dumpHeap(): File = if (enabled) defaultDumper.dumpHeap() else HeapDumper.RETRY_LATER
|
||
|
}
|
||
|
}
|