package com.xw.common
|
|
import android.view.Window
|
import android.view.WindowManager
|
import java.lang.ref.WeakReference
|
|
/**
|
* 统一管理 APP 防截屏（FLAG_SECURE）开关。
|
*
|
* 默认行为：非 DEBUG 包禁止截屏，DEBUG 包允许截屏，保持与历史逻辑一致。
|
*
|
* 内部用弱引用集合保存所有受管理的 [Window]（避免持有 Activity/Dialog 导致内存泄漏）。
|
* 当 [allowScreenShot] 在 true/false 之间发生变化时，会自动刷新所有存活 window 的
|
* FLAG_SECURE 标记，从而实时启用 / 禁用截屏。
|
*/
|
object SecureScreenShot {
|
|
/** 受管理的 window 弱引用集合 */
|
private val windowRefs = mutableListOf<WeakReference<Window>>()
|
|
/**
|
* 运行时是否允许截屏。默认 false（即开启防截屏）。
|
* 赋值发生变化时，会自动刷新所有已登记 window 的截屏标记。
|
*/
|
@Volatile
|
var allowScreenShot: Boolean = false
|
set(value) {
|
if (field == value) return
|
field = value
|
refreshAll()
|
}
|
|
/**
|
* 登记并立即为指定 [window] 应用当前的截屏配置。
|
* 同一个 window 不会重复登记，失效的弱引用会被顺带清理。
|
*/
|
@Synchronized
|
fun apply(window: Window?) {
|
window ?: return
|
var exists = false
|
val iterator = windowRefs.iterator()
|
while (iterator.hasNext()) {
|
val w = iterator.next().get()
|
if (w == null) {
|
iterator.remove()
|
} else if (w === window) {
|
exists = true
|
}
|
}
|
if (!exists) {
|
windowRefs.add(WeakReference(window))
|
}
|
applyFlag(window)
|
}
|
|
/**
|
* 刷新所有存活 window 的截屏标记，并清理已失效的弱引用。
|
*/
|
@Synchronized
|
private fun refreshAll() {
|
val iterator = windowRefs.iterator()
|
while (iterator.hasNext()) {
|
val w = iterator.next().get()
|
if (w == null) {
|
iterator.remove()
|
} else {
|
applyFlag(w)
|
}
|
}
|
}
|
|
/**
|
* 根据当前配置为指定 [window] 应用或清除防截屏标记。
|
*
|
* - DEBUG 包或 [allowScreenShot] 为 true 时：清除 FLAG_SECURE，允许截屏
|
* - 否则：添加 FLAG_SECURE，禁止截屏
|
*/
|
private fun applyFlag(window: Window) {
|
if (BuildConfig.DEBUG || allowScreenShot) {
|
window.clearFlags(WindowManager.LayoutParams.FLAG_SECURE)
|
} else {
|
window.addFlags(WindowManager.LayoutParams.FLAG_SECURE)//禁止截屏
|
}
|
}
|
}
|
