dnrops
/
Aperture
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Prevent the computer from sleeping while recording (#64)

This commit is contained in:
Sindre Sorhus 2019-02-20 13:07:20 +07:00 committed by GitHub
parent 732c766a86
commit 67d0743770
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
Sources/Aperture/Aperture.swift
View File

@ -13,6 +13,7 @@ public final class Aperture: NSObject {
private let destination: URL private let destination: URL
private let session: AVCaptureSession private let session: AVCaptureSession
private let output: AVCaptureMovieFileOutput private let output: AVCaptureMovieFileOutput
private var activity: NSObjectProtocol?
public var onStart: (() -> Void)? public var onStart: (() -> Void)?
public var onFinish: (() -> Void)? public var onFinish: (() -> Void)?
@ -118,11 +119,28 @@ public final class Aperture: NSObject {
} }
extension Aperture: AVCaptureFileOutputRecordingDelegate { extension Aperture: AVCaptureFileOutputRecordingDelegate {
private var shouldPreventSleep: Bool {
get {
return activity != nil
}
set {
if newValue {
activity = ProcessInfo.processInfo.beginActivity(options: .idleSystemSleepDisabled, reason: "Recording screen")
} else if let activity = activity {
ProcessInfo.processInfo.endActivity(activity)
self.activity = nil
}
}
}
public func fileOutput(_ captureOutput: AVCaptureFileOutput, didStartRecordingTo fileURL: URL, from connections: [AVCaptureConnection]) { public func fileOutput(_ captureOutput: AVCaptureFileOutput, didStartRecordingTo fileURL: URL, from connections: [AVCaptureConnection]) {
shouldPreventSleep = true
onStart?() onStart?()
} }
public func fileOutput(_ captureOutput: AVCaptureFileOutput, didFinishRecordingTo outputFileURL: URL, from connections: [AVCaptureConnection], error: Error?) { public func fileOutput(_ captureOutput: AVCaptureFileOutput, didFinishRecordingTo outputFileURL: URL, from connections: [AVCaptureConnection], error: Error?) {
shouldPreventSleep = false
let FINISHED_RECORDING_ERROR_CODE = -11_806 let FINISHED_RECORDING_ERROR_CODE = -11_806
if let error = error, error._code != FINISHED_RECORDING_ERROR_CODE { if let error = error, error._code != FINISHED_RECORDING_ERROR_CODE {
@ -133,10 +151,12 @@ extension Aperture: AVCaptureFileOutputRecordingDelegate {
} }
public func fileOutput(_ output: AVCaptureFileOutput, didPauseRecordingTo fileURL: URL, from connections: [AVCaptureConnection]) { public func fileOutput(_ output: AVCaptureFileOutput, didPauseRecordingTo fileURL: URL, from connections: [AVCaptureConnection]) {
shouldPreventSleep = false
onPause?() onPause?()
} }
public func fileOutput(_ output: AVCaptureFileOutput, didResumeRecordingTo fileURL: URL, from connections: [AVCaptureConnection]) { public func fileOutput(_ output: AVCaptureFileOutput, didResumeRecordingTo fileURL: URL, from connections: [AVCaptureConnection]) {
shouldPreventSleep = true
onResume?() onResume?()
} }