From 60a4ccd7f65940721117ed51f17795017c73b65e Mon Sep 17 00:00:00 2001
From: yinjiacheng <yinjiacheng@dcloud.io>
Date: Thu, 16 Feb 2023 15:30:35 +0800
Subject: [PATCH] =?UTF-8?q?uni-usercapturescreen=20Android=20=E6=96=B0?=
 =?UTF-8?q?=E5=A2=9E=E7=A6=81=E6=AD=A2=E6=88=AA=E5=B1=8FAPI?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pages/index/index.vue                         | 29 ++++++++++++
 .../uni-usercapturescreen/package.json        |  3 +-
 .../utssdk/app-android/index.uts              | 46 +++++++++++++++++++
 3 files changed, 77 insertions(+), 1 deletion(-)

diff --git a/pages/index/index.vue b/pages/index/index.vue
index 5bac7fa..4e24a5f 100644
--- a/pages/index/index.vue
+++ b/pages/index/index.vue
@@ -6,6 +6,8 @@
 		</view>
 		<button @tap="testScreenShotListen">开启截屏监听</button>
 		<button @tap="testScreenShotOff">关闭截屏监听</button>
+		<button @tap="testSetUserCaptureScreen">{{setUserCaptureScreenText}}</button>
+		
 		<button @tap="testGetBatteryInfo">获取电池电量</button>
 		<button @tap="testonMemoryWarning">开启内存不足告警监听</button>
 		<button @tap="testoffMemoryWarning">关闭内存不足告警监听</button>
@@ -25,6 +27,8 @@
 			return {
 				title: 'Hello',
 				memListener:null,
+				setUserCaptureScreenFlag: true,
+				setUserCaptureScreenText: '禁止截屏'
 			}
 		},
 		onLoad() {
@@ -190,6 +194,31 @@
 					}
 				})
 			},
+			testSetUserCaptureScreen() {
+				let flag = this.setUserCaptureScreenFlag;
+				uni.setUserCaptureScreen({
+					open: flag,
+					success: (res) => {
+						console.log("setUserCaptureScreen open: " + flag + " success: " + JSON.stringify(res));
+					},
+					fail: (res) => {
+						console.log("setUserCaptureScreen open: " + flag + " fail: " + JSON.stringify(res));
+					},
+					complete: (res) => {
+						console.log("setUserCaptureScreen open: " + flag + " complete: " + JSON.stringify(res));
+					}
+				});
+				uni.showToast({
+					icon:"none",
+					title: this.setUserCaptureScreenText
+				});
+				this.setUserCaptureScreenFlag = !this.setUserCaptureScreenFlag;
+				if (this.setUserCaptureScreenFlag) {
+					this.setUserCaptureScreenText = '禁止截屏';
+				} else {
+					this.setUserCaptureScreenText = '允许截屏';
+				}
+			},
 		}
 	}
 </script>
diff --git a/uni_modules/uni-usercapturescreen/package.json b/uni_modules/uni-usercapturescreen/package.json
index c2c1d59..461dc0a 100644
--- a/uni_modules/uni-usercapturescreen/package.json
+++ b/uni_modules/uni-usercapturescreen/package.json
@@ -34,7 +34,8 @@
 		"uni-ext-api":{
 			"uni": {
 				"onUserCaptureScreen": "onUserCaptureScreen",
-				"offUserCaptureScreen": "offUserCaptureScreen"
+				"offUserCaptureScreen": "offUserCaptureScreen",
+				"setUserCaptureScreen": "setUserCaptureScreen"
 			}
 		},
 		"dependencies": [],
diff --git a/uni_modules/uni-usercapturescreen/utssdk/app-android/index.uts b/uni_modules/uni-usercapturescreen/utssdk/app-android/index.uts
index 6ff073c..776bc27 100644
--- a/uni_modules/uni-usercapturescreen/utssdk/app-android/index.uts
+++ b/uni_modules/uni-usercapturescreen/utssdk/app-android/index.uts
@@ -13,7 +13,17 @@ import FileObserver from "android.os.FileObserver";
 import File from "java.io.File";
 import Environment from "android.os.Environment";
 import System from 'java.lang.System';
+import WindowManager from 'android.view.WindowManager';
 
+/**
+ * setUserCaptureScreen 参数定义
+ */
+type SetUserCaptureScreenOption = {
+	open: boolean;
+	success?: (res: UTSJSONObject) => void;
+	fail?: (res: UTSJSONObject) => void;
+	complete?: (res: UTSJSONObject) => void;
+}
 
 
 /**
@@ -145,5 +155,41 @@ export function offUserCaptureScreen(success: (res: any) => void) {
 	success({});
 }
 
+/**
+ * 设置是否禁止截屏
+ */
+export function setUserCaptureScreen(option: SetUserCaptureScreenOption) {
+	const res = {
+		errCode: 0,
+		errMsg: "setUserCaptureScreen:ok",
+		errSubject: "uni-setUserCaptureScreen"
+	};
+	UTSAndroid.getUniActivity()?.runOnUiThread(new SetUserCaptureScreenRunnable(option.open));
+	option.success?.(res);
+	option.complete?.(res);
+}
+
+class SetUserCaptureScreenRunnable extends Runnable {
+	
+	/**
+	 * true 表示禁止截屏
+	 * false 表示允许截屏
+	 */
+	private open: boolean = false;
+
+	constructor(open: boolean) {
+		super();
+		this.open = open;
+	}
+
+	override run(): void {
+		if (this.open) {
+			UTSAndroid.getUniActivity()?.getWindow()?.addFlags(WindowManager.LayoutParams.FLAG_SECURE);
+		} else {
+			UTSAndroid.getUniActivity()?.getWindow()?.clearFlags(WindowManager.LayoutParams.FLAG_SECURE);
+		}
+	}
+}
+
 
 
-- 
GitLab