import { NotificationCenter } from 'Foundation';
import { UIApplication } from "UIKit"

/**
 * 定义监听截屏事件工具类
 */
class CaptureScreenTool {
	static listener: UTSCallback | null;
	
	// 监听截屏
	static listenCaptureScreen(callback: UTSCallback | 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() {
		// 触发回调
		this.listener?.({})
	}
	
	// 移除监听事件
	static removeListen(callback: UTSCallback | null) {
		this.listener = null
		NotificationCenter.default.removeObserver(this)
		callback?.({})
	}
}

/**
 * 开启截图监听
 */
export function onUserCaptureScreen(callback: UTSCallback | null) {
	CaptureScreenTool.listenCaptureScreen(callback)
}

/**
 * 关闭截屏监听
 */
export function offUserCaptureScreen(callback: UTSCallback | null) {
	CaptureScreenTool.removeListen(callback)
}