Apple Silicon Support

com.davidwernhart.Helper/HelperTool.swift

@@ -0,0 +1,126 @@
+//
+//  HelperTool.swift
+//  com.davidwernhart.Helper
+//
+//  Created by David Wernhart on 13.02.20.
+//  Copyright © 2020 David Wernhart. All rights reserved.
+//
+
+import Foundation
+import IOKit.pwr_mgt
+
+final class HelperTool: NSObject, HelperToolProtocol {
+    
+    static let instance = HelperTool()
+    
+    var modifiedKeys: [String: UInt8] = [:]
+    var openAssertions: [IOPMAssertionID] = []
+    
+    func getVersion(withReply reply: (String) -> Void) {
+//        let version = Bundle.main.object(forInfoDictionaryKey: "CFBundleShortVersionString" as String) as? String ?? "(unknown version)"
+//        let build = Bundle.main.object(forInfoDictionaryKey: kCFBundleVersionKey as String) as? String ?? "(unknown build)"
+//        reply("v\(version) (\(build))")
+        reply(helperVersion)
+
+    }
+
+    func setSMCByte(key: String, value: UInt8) {
+        do {
+            try SMCKit.open()
+        } catch {
+            print(error)
+            exit(EX_UNAVAILABLE)
+        }
+        let smcKey = SMCKit.getKey(key, type: DataTypes.UInt8)
+        let bytes: SMCBytes = (value, UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0),
+        UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0),
+        UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0),
+        UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0),
+        UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0),
+        UInt8(0), UInt8(0))
+        
+        if(self.modifiedKeys[key] == nil){
+            readSMCByte(key: key) { (originalValue) in
+                self.modifiedKeys[key] = originalValue
+                _ = try? SMCKit.writeData(smcKey, data: bytes)
+            }
+        }
+        else{
+            _ = try? SMCKit.writeData(smcKey, data: bytes)
+        }
+
+        
+    }
+
+    func readSMCByte(key: String, withReply reply: @escaping (UInt8) -> Void) {
+        do {
+            try SMCKit.open()
+        } catch {
+            print(error)
+            exit(EX_UNAVAILABLE)
+        }
+
+        let smcKey = SMCKit.getKey(key, type: DataTypes.UInt8)
+        do {
+            let status = try SMCKit.readData(smcKey).0
+            reply(status)
+        } catch {
+            reply(0)
+        }
+    }
+    
+    func readSMCUInt32(key: String, withReply reply: @escaping (UInt32) -> Void) {
+        do {
+            try SMCKit.open()
+        } catch {
+            print(error)
+            exit(EX_UNAVAILABLE)
+        }
+
+        let smcKey = SMCKit.getKey(key, type: DataTypes.UInt32)
+        do {
+            let data = try SMCKit.readData(smcKey)
+            reply(UInt32(fromBytes: (data.0, data.1, data.2, data.3)))
+        } catch {
+            reply(0)
+        }
+    }
+    
+    func createAssertion(assertion:String, withReply reply: @escaping (IOPMAssertionID) -> Void){
+        var assertionID : IOPMAssertionID = IOPMAssertionID(0)
+        let reason:CFString = "AlDente" as NSString
+        let cfAssertion:CFString = assertion as NSString
+        let success = IOPMAssertionCreateWithName(cfAssertion,
+                        IOPMAssertionLevel(kIOPMAssertionLevelOn),
+                        reason,
+                        &assertionID)
+        if success == kIOReturnSuccess {
+            openAssertions.append(assertionID)
+            reply(assertionID)
+        }
+        else{
+            reply (UInt32(kCFNumberNaN))
+        }
+    }
+    
+    func releaseAssertion(assertionID:IOPMAssertionID){
+        IOPMAssertionRelease(assertionID)
+        openAssertions.remove(at: openAssertions.firstIndex(of: assertionID)!)
+    }
+    
+    func setResetVal(key:String, value: UInt8){
+        modifiedKeys[key]=value
+    }
+    
+    func reset(){
+        for (key, value) in modifiedKeys{
+            setSMCByte(key: key, value: value)
+        }
+        
+        for assertionID in openAssertions{
+            releaseAssertion(assertionID: assertionID)
+        }
+        modifiedKeys.removeAll()
+        openAssertions.removeAll()
+    }
+}