325 lines
11 KiB
Plaintext
325 lines
11 KiB
Plaintext
import { NotificationCenter } from 'Foundation';
|
||
import { CGRect } from "CoreFoundation";
|
||
import { UIApplication, UIView, UITextField, UIScreen, UIDevice } from "UIKit"
|
||
import { UTSiOS } from "DCloudUTSFoundation"
|
||
import { DispatchQueue } from 'Dispatch';
|
||
import { SetUserCaptureScreenOptions, OnUserCaptureScreenCallbackResult, OnUserCaptureScreen, OffUserCaptureScreen, SetUserCaptureScreen, UserCaptureScreenCallback, SetUserCaptureScreenSuccess } from "../interface.uts"
|
||
import { SetUserCaptureScreenFailImpl } from "../unierror.uts"
|
||
/**
|
||
* 定义监听截屏事件工具类
|
||
*/
|
||
class CaptureScreenTool {
|
||
static listener : UserCaptureScreenCallback | null;
|
||
static secureView : UIView | null;
|
||
|
||
// ===== 仅防录屏（动态切换）相关状态 =====
|
||
// 保留 UITextField，使 isSecureTextEntry 可在录屏开始/结束时反复切换
|
||
static secureField : UITextField | null;
|
||
// field 的安全画布（已挂入视图层级，承载 App 根视图）
|
||
static secureCanvas : UIView | null;
|
||
// 被安全画布包裹的 App 根视图（用于卸载时还原）
|
||
static secureRootView : UIView | null;
|
||
// 是否已为"仅防录屏"挂载安全层
|
||
static recordWrapped : boolean = false;
|
||
|
||
// 监听截屏
|
||
static listenCaptureScreen(callback : UserCaptureScreenCallback | null) {
|
||
this.listener = callback
|
||
|
||
// 注册监听截屏事件及回调方法
|
||
// target-action 回调方法需要通过 Selector("方法名") 构建
|
||
const method = Selector("userDidTakeScreenshot")
|
||
NotificationCenter.default.addObserver(this, selector = method, name = UIApplication.userDidTakeScreenshotNotification, object = null)
|
||
}
|
||
|
||
// 捕获截屏回调的方法
|
||
// target-action 的方法前需要添加 @objc 前缀
|
||
@objc static userDidTakeScreenshot() {
|
||
// 回调
|
||
const res: OnUserCaptureScreenCallbackResult = {
|
||
}
|
||
this.listener?.(res)
|
||
}
|
||
|
||
// 移除监听事件
|
||
static removeListen(callback : UserCaptureScreenCallback | null) {
|
||
this.listener = null
|
||
NotificationCenter.default.removeObserver(this)
|
||
}
|
||
|
||
static createSecureView() : UIView | null {
|
||
let field = new UITextField(frame = CGRect.zero)
|
||
field.isSecureTextEntry = true
|
||
if (field.subviews.length > 0 && UIDevice.current.systemVersion != '15.1') {
|
||
let view = field.subviews[0]
|
||
view.subviews.forEach((item) => {
|
||
item.removeFromSuperview()
|
||
})
|
||
view.isUserInteractionEnabled = true
|
||
return view
|
||
}
|
||
return null
|
||
}
|
||
|
||
// 应用防截屏遮罩（secureView：实时显示正常，但被截屏/录屏捕获到的内容为黑屏）
|
||
// 不触发回调，供内部复用；注意：需在主线程调用
|
||
static applySecureViewInternal() {
|
||
if (this.secureView != null) {
|
||
// 已应用，避免重复包裹
|
||
return
|
||
}
|
||
let secureView = this.createSecureView()
|
||
let window = UTSiOS.getKeyWindow()
|
||
let rootView = window.rootViewController == null ? null : window.rootViewController!.view
|
||
if (secureView != null && rootView != null) {
|
||
let rootSuperview = rootView!.superview
|
||
if (rootSuperview != null) {
|
||
this.secureView = secureView
|
||
rootSuperview!.addSubview(secureView!)
|
||
rootView!.removeFromSuperview()
|
||
secureView!.addSubview(rootView!)
|
||
// secureView 充满父视图并随父视图宽高自适应（横竖屏、全屏切换时自动跟随）
|
||
secureView!.frame = rootSuperview!.bounds
|
||
secureView!.autoresizingMask = [UIView.AutoresizingMask.flexibleWidth, UIView.AutoresizingMask.flexibleHeight]
|
||
// rootView 充满 secureView 并自适应，避免全屏时被固定在竖屏尺寸而缩到角落
|
||
rootView!.frame = secureView!.bounds
|
||
rootView!.autoresizingMask = [UIView.AutoresizingMask.flexibleWidth, UIView.AutoresizingMask.flexibleHeight]
|
||
}
|
||
}
|
||
}
|
||
|
||
// 开启防截屏
|
||
static onAntiScreenshot(option : SetUserCaptureScreenOptions) {
|
||
// uts方法默认会在子线程中执行，涉及 UI 操作必须在主线程中运行，通过 DispatchQueue.main.async 方法可将代码在主线程中运行
|
||
DispatchQueue.main.async(execute = () : void => {
|
||
this.applySecureViewInternal()
|
||
let res: SetUserCaptureScreenSuccess = {
|
||
}
|
||
option.success?.(res)
|
||
option.complete?.(res)
|
||
})
|
||
}
|
||
|
||
// ===== 仅防录屏：提前挂载安全层 + 翻转 isSecureTextEntry =====
|
||
|
||
// 提前挂载安全层（应在录屏开始前调用，例如进入页面时）。
|
||
// 挂载后默认不保护（isSecureTextEntry = false）→ 截屏正常、实时正常。
|
||
// 注意：需在主线程调用。
|
||
static setupRecordSecureWrapper() {
|
||
if (this.recordWrapped) {
|
||
return
|
||
}
|
||
let field = new UITextField(frame = CGRect.zero)
|
||
field.isSecureTextEntry = true
|
||
// 15.1 系统 secureView 不可用；拿不到安全画布则放弃
|
||
if (field.subviews.length == 0 || UIDevice.current.systemVersion == '15.1') {
|
||
return
|
||
}
|
||
let canvas = field.subviews[0]
|
||
canvas.subviews.forEach((item) => {
|
||
item.removeFromSuperview()
|
||
})
|
||
canvas.isUserInteractionEnabled = true
|
||
|
||
let window = UTSiOS.getKeyWindow()
|
||
let rootView = window.rootViewController == null ? null : window.rootViewController!.view
|
||
if (rootView == null) {
|
||
return
|
||
}
|
||
let rootSuperview = rootView!.superview
|
||
if (rootSuperview == null) {
|
||
return
|
||
}
|
||
|
||
// 把安全画布插入原层级，并把 App 根视图移入画布
|
||
rootSuperview!.addSubview(canvas)
|
||
rootView!.removeFromSuperview()
|
||
canvas.addSubview(rootView!)
|
||
canvas.frame = rootSuperview!.bounds
|
||
canvas.autoresizingMask = [UIView.AutoresizingMask.flexibleWidth, UIView.AutoresizingMask.flexibleHeight]
|
||
rootView!.frame = canvas.bounds
|
||
rootView!.autoresizingMask = [UIView.AutoresizingMask.flexibleWidth, UIView.AutoresizingMask.flexibleHeight]
|
||
|
||
// 关键：保留 field，后续才能切换 isSecureTextEntry
|
||
this.secureField = field
|
||
this.secureCanvas = canvas
|
||
this.secureRootView = rootView
|
||
this.recordWrapped = true
|
||
|
||
// 默认不保护：允许截屏、实时正常
|
||
field.isSecureTextEntry = false
|
||
}
|
||
|
||
// 切换录屏保护：true=保护（录屏文件黑、实时正常、此刻截屏也黑）；false=不保护（允许截屏）
|
||
// 注意：需在主线程调用。
|
||
static setRecordSecure(secure : boolean) {
|
||
if (this.secureField != null) {
|
||
this.secureField!.isSecureTextEntry = secure
|
||
}
|
||
}
|
||
|
||
// 卸载安全层，恢复原始视图层级。注意：需在主线程调用。
|
||
static teardownRecordSecureWrapper() {
|
||
if (!this.recordWrapped) {
|
||
return
|
||
}
|
||
if (this.secureField != null) {
|
||
this.secureField!.isSecureTextEntry = false
|
||
}
|
||
if (this.secureCanvas != null && this.secureRootView != null) {
|
||
let rootSuperview = this.secureCanvas!.superview
|
||
if (rootSuperview != null) {
|
||
rootSuperview!.addSubview(this.secureRootView!)
|
||
this.secureRootView!.frame = rootSuperview!.bounds
|
||
this.secureCanvas!.removeFromSuperview()
|
||
}
|
||
}
|
||
this.secureField = null
|
||
this.secureCanvas = null
|
||
this.secureRootView = null
|
||
this.recordWrapped = false
|
||
}
|
||
|
||
// 移除防截屏遮罩视图（不触发回调，供内部复用）
|
||
// 注意：需在主线程调用
|
||
static removeSecureViewInternal() {
|
||
if (this.secureView != null) {
|
||
let window = UTSiOS.getKeyWindow()
|
||
let rootView = window.rootViewController == null ? null : window.rootViewController!.view
|
||
if (rootView != null && this.secureView!.superview != null) {
|
||
let rootSuperview = this.secureView!.superview
|
||
if (rootSuperview != null) {
|
||
rootSuperview!.addSubview(rootView!)
|
||
this.secureView!.removeFromSuperview()
|
||
}
|
||
}
|
||
this.secureView = null
|
||
}
|
||
}
|
||
|
||
// 关闭防截屏
|
||
static offAntiScreenshot(option : SetUserCaptureScreenOptions) {
|
||
DispatchQueue.main.async(execute = () : void => {
|
||
this.removeSecureViewInternal()
|
||
let res: SetUserCaptureScreenSuccess = {
|
||
}
|
||
option.success?.(res)
|
||
option.complete?.(res)
|
||
})
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 防录屏工具类：基于 UIScreen.isCaptured 监听录屏/投屏状态，动态切换安全层。
|
||
* 截屏不会触发 isCaptured，因此可做到"允许截屏、仅在录屏时保护"。
|
||
*
|
||
* 实现要点（避免实时画面变黑）：
|
||
* 1) 进入时（录屏开始前）就提前挂好安全层，且默认不保护 → 截屏正常、实时正常；
|
||
* 2) 仅在录屏期间翻转 isSecureTextEntry = true → 录屏文件黑、实时正常、此刻截屏也黑；
|
||
* 3) 录屏结束翻回 false → 截屏恢复正常。
|
||
*/
|
||
class ScreenRecordTool {
|
||
// 是否已注册监听
|
||
static monitoring : boolean = false;
|
||
|
||
// 开启防录屏：提前挂载安全层并监听录屏状态变化
|
||
static onAntiScreenRecord() {
|
||
DispatchQueue.main.async(execute = () : void => {
|
||
// 关键：录屏开始前就把安全层挂好（默认不保护）
|
||
CaptureScreenTool.setupRecordSecureWrapper()
|
||
if (!this.monitoring) {
|
||
this.monitoring = true
|
||
const method = Selector("screenCaptureDidChange")
|
||
NotificationCenter.default.addObserver(this, selector = method, name = UIScreen.capturedDidChangeNotification, object = null)
|
||
}
|
||
// 进入时若已经在录屏，立即按当前状态保护
|
||
this.applyIfNeeded()
|
||
})
|
||
}
|
||
|
||
// 录屏状态变化回调
|
||
@objc static screenCaptureDidChange() {
|
||
DispatchQueue.main.async(execute = () : void => {
|
||
this.applyIfNeeded()
|
||
})
|
||
}
|
||
|
||
// 根据当前录屏状态翻转安全层开关
|
||
static applyIfNeeded() {
|
||
CaptureScreenTool.setRecordSecure(UIScreen.main.isCaptured)
|
||
}
|
||
|
||
// 关闭防录屏：移除监听并卸载安全层
|
||
static offAntiScreenRecord() {
|
||
DispatchQueue.main.async(execute = () : void => {
|
||
if (this.monitoring) {
|
||
NotificationCenter.default.removeObserver(this, name = UIScreen.capturedDidChangeNotification, object = null)
|
||
this.monitoring = false
|
||
}
|
||
CaptureScreenTool.teardownRecordSecureWrapper()
|
||
})
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 开启截图监听
|
||
*/
|
||
export const onUserCaptureScreen : OnUserCaptureScreen = function (callback : UserCaptureScreenCallback | null) {
|
||
CaptureScreenTool.listenCaptureScreen(callback)
|
||
}
|
||
|
||
/**
|
||
* 关闭截屏监听
|
||
*/
|
||
export const offUserCaptureScreen : OffUserCaptureScreen = function (callback : UserCaptureScreenCallback | null) {
|
||
CaptureScreenTool.removeListen(callback)
|
||
}
|
||
|
||
/**
|
||
* 开启/关闭防截屏
|
||
*/
|
||
export const setUserCaptureScreen : SetUserCaptureScreen = function (options : SetUserCaptureScreenOptions) {
|
||
const systemVersion = UIDevice.current.systemVersion
|
||
|
||
if (options.enable == true) {
|
||
// 允许截屏与录屏：无条件清理录屏遮罩，再移除防截屏遮罩。
|
||
// 清理动作不受版本限制，避免遗留监听/遮罩。
|
||
ScreenRecordTool.offAntiScreenRecord()
|
||
CaptureScreenTool.offAntiScreenshot(options)
|
||
return
|
||
}
|
||
|
||
if (options.antiRecordOnly == true) {
|
||
// 仅禁止录屏、允许截屏：进入页面时提前挂好安全层（默认不保护），
|
||
// 用 UIScreen.isCaptured（iOS 11+）监听录屏/投屏，仅在录屏期间翻转 isSecureTextEntry
|
||
// —— 实时显示正常、录到的画面为黑屏，且平时不影响截屏。
|
||
// 注意：secureView 在 iOS 15.1 上不可用，该版本录屏保护会失效（属系统已知问题）。
|
||
if (systemVersion < "11.0") {
|
||
let res = new SetUserCaptureScreenFailImpl(12001)
|
||
options.fail?.(res)
|
||
options.complete?.(res)
|
||
return
|
||
}
|
||
ScreenRecordTool.onAntiScreenRecord()
|
||
let res: SetUserCaptureScreenSuccess = {
|
||
}
|
||
options.success?.(res)
|
||
options.complete?.(res)
|
||
return
|
||
}
|
||
|
||
// 默认：截屏与录屏一起禁止（secureView 方案，受系统版本限制）
|
||
if (systemVersion < "13.0") {
|
||
let res = new SetUserCaptureScreenFailImpl(12001)
|
||
options.fail?.(res)
|
||
options.complete?.(res)
|
||
} else if (systemVersion == "15.1") {
|
||
let res = new SetUserCaptureScreenFailImpl(12010)
|
||
options.fail?.(res)
|
||
options.complete?.(res)
|
||
} else {
|
||
CaptureScreenTool.onAntiScreenshot(options)
|
||
}
|
||
}
|
||
|