mirror of
https://github.com/Novaloop-AG/element-x-ios.git
synced 2026-11-03 16:10:20 +00:00
Remove unused ScreenshotDetector
This commit is contained in:
committed by
Stefan Ceriu
parent
7a1cbdf6b1
commit
1815b0a7ac
@@ -1,114 +0,0 @@
|
|||||||
//
|
|
||||||
// Copyright 2022 New Vector Ltd
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
// you may not use this file except in compliance with the License.
|
|
||||||
// You may obtain a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
// See the License for the specific language governing permissions and
|
|
||||||
// limitations under the License.
|
|
||||||
//
|
|
||||||
|
|
||||||
import Foundation
|
|
||||||
import Photos
|
|
||||||
import UIKit
|
|
||||||
|
|
||||||
enum ScreenshotDetectorError: String, Error {
|
|
||||||
case loadFailed
|
|
||||||
case notAuthorized
|
|
||||||
}
|
|
||||||
|
|
||||||
@MainActor
|
|
||||||
class ScreenshotDetector {
|
|
||||||
var callback: (@MainActor (UIImage?, Error?) -> Void)?
|
|
||||||
|
|
||||||
/// Flag to whether ask for photos authorization by default if needed.
|
|
||||||
var autoRequestPHAuthorization = false
|
|
||||||
|
|
||||||
init() {
|
|
||||||
startObservingScreenshots()
|
|
||||||
}
|
|
||||||
|
|
||||||
private func startObservingScreenshots() {
|
|
||||||
NotificationCenter.default.addObserver(self,
|
|
||||||
selector: #selector(userDidTakeScreenshot),
|
|
||||||
name: UIApplication.userDidTakeScreenshotNotification,
|
|
||||||
object: nil)
|
|
||||||
}
|
|
||||||
|
|
||||||
@objc private func userDidTakeScreenshot() {
|
|
||||||
let authStatus = PHPhotoLibrary.authorizationStatus(for: .readWrite)
|
|
||||||
if authStatus == .authorized {
|
|
||||||
findScreenshot()
|
|
||||||
} else if authStatus == .notDetermined, autoRequestPHAuthorization {
|
|
||||||
Task {
|
|
||||||
await self.handleAuthStatus(PHPhotoLibrary.requestAuthorization(for: .readWrite))
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
fail(withError: ScreenshotDetectorError.notAuthorized)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private func handleAuthStatus(_ status: PHAuthorizationStatus) {
|
|
||||||
if status == .authorized {
|
|
||||||
findScreenshot()
|
|
||||||
} else {
|
|
||||||
fail(withError: ScreenshotDetectorError.notAuthorized)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private func findScreenshot() {
|
|
||||||
if let asset = PHAsset.fetchLastScreenshot() {
|
|
||||||
let imageManager = PHImageManager()
|
|
||||||
imageManager.requestImage(for: asset,
|
|
||||||
targetSize: PHImageManagerMaximumSize,
|
|
||||||
contentMode: .default,
|
|
||||||
options: PHImageRequestOptions.highQualitySyncLocal) { [weak self] image, _ in
|
|
||||||
guard let image else {
|
|
||||||
self?.fail(withError: ScreenshotDetectorError.loadFailed)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
self?.succeed(withImage: image)
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
fail(withError: ScreenshotDetectorError.loadFailed)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func succeed(withImage image: UIImage) {
|
|
||||||
callback?(image, nil)
|
|
||||||
}
|
|
||||||
|
|
||||||
func fail(withError error: Error) {
|
|
||||||
callback?(nil, error)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
extension PHAsset {
|
|
||||||
static func fetchLastScreenshot() -> PHAsset? {
|
|
||||||
let options = PHFetchOptions()
|
|
||||||
|
|
||||||
options.fetchLimit = 1
|
|
||||||
options.includeAssetSourceTypes = [.typeUserLibrary]
|
|
||||||
options.wantsIncrementalChangeDetails = false
|
|
||||||
options.predicate = NSPredicate(format: "(mediaSubtype & %d) != 0", PHAssetMediaSubtype.photoScreenshot.rawValue)
|
|
||||||
options.sortDescriptors = [NSSortDescriptor(key: "creationDate", ascending: false)]
|
|
||||||
|
|
||||||
return PHAsset.fetchAssets(with: .image, options: options).firstObject
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private extension PHImageRequestOptions {
|
|
||||||
static var highQualitySyncLocal: PHImageRequestOptions {
|
|
||||||
let options = PHImageRequestOptions()
|
|
||||||
options.deliveryMode = .highQualityFormat
|
|
||||||
options.isNetworkAccessAllowed = false
|
|
||||||
options.isSynchronous = true
|
|
||||||
return options
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,89 +0,0 @@
|
|||||||
//
|
|
||||||
// Copyright 2022 New Vector Ltd
|
|
||||||
//
|
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
// you may not use this file except in compliance with the License.
|
|
||||||
// You may obtain a copy of the License at
|
|
||||||
//
|
|
||||||
// http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
//
|
|
||||||
// Unless required by applicable law or agreed to in writing, software
|
|
||||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
// See the License for the specific language governing permissions and
|
|
||||||
// limitations under the License.
|
|
||||||
//
|
|
||||||
|
|
||||||
@testable import ElementX
|
|
||||||
import Foundation
|
|
||||||
import Photos
|
|
||||||
import XCTest
|
|
||||||
|
|
||||||
class ScreenshotDetectorTests: XCTestCase {
|
|
||||||
@MainActor func testDetection() async {
|
|
||||||
async { expectation in
|
|
||||||
let detector = ScreenshotDetector()
|
|
||||||
// disable auto request authorization
|
|
||||||
detector.autoRequestPHAuthorization = false
|
|
||||||
detector.callback = { image, error in
|
|
||||||
|
|
||||||
defer {
|
|
||||||
expectation.fulfill()
|
|
||||||
}
|
|
||||||
|
|
||||||
if PHPhotoLibrary.authorizationStatus(for: .readWrite) == .authorized {
|
|
||||||
// if Photos already authorized on the simulator
|
|
||||||
|
|
||||||
if PHAsset.fetchLastScreenshot() != nil {
|
|
||||||
// we should get an image
|
|
||||||
XCTAssertNotNil(image)
|
|
||||||
|
|
||||||
// we should not get an error
|
|
||||||
XCTAssertNil(error)
|
|
||||||
} else {
|
|
||||||
// otherwise we should not get an image
|
|
||||||
XCTAssertNil(image)
|
|
||||||
|
|
||||||
// and get an error
|
|
||||||
guard let error else {
|
|
||||||
XCTFail("Should get an error")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
switch error {
|
|
||||||
case ScreenshotDetectorError.loadFailed:
|
|
||||||
break
|
|
||||||
default:
|
|
||||||
XCTFail("Unknown error")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
// otherwise we should not get an image
|
|
||||||
XCTAssertNil(image)
|
|
||||||
|
|
||||||
// and get an error
|
|
||||||
guard let error else {
|
|
||||||
XCTFail("Should get an error")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
switch error {
|
|
||||||
case ScreenshotDetectorError.notAuthorized:
|
|
||||||
break
|
|
||||||
default:
|
|
||||||
XCTFail("Unknown error")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
NotificationCenter.default.post(name: UIApplication.userDidTakeScreenshotNotification, object: nil)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private func async(_ timeout: TimeInterval = 0.5, _ block: @escaping (XCTestExpectation) -> Void) {
|
|
||||||
let waiter = XCTWaiter()
|
|
||||||
let expectation = XCTestExpectation(description: "Async operation expectation")
|
|
||||||
block(expectation)
|
|
||||||
waiter.wait(for: [expectation], timeout: timeout)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Reference in New Issue
Block a user