feat: load persisted plugins

Signed-off-by: 82Flex <[email protected]>

Author: Lessica

TrollFools/EjectListModel.swift

@@ -15,6 +15,9 @@ final class EjectListModel: ObservableObject {
     @Published var filter = FilterOptions()
     @Published var filteredPlugIns: [InjectedPlugIn] = []
 
+    @Published var isOkToEnableAll = false
+    @Published var isOkToDisableAll = false
+
     private var cancellables = Set<AnyCancellable>()
 
     init(_ app: App) {
@@ -30,8 +33,18 @@ final class EjectListModel: ObservableObject {
     }
 
     func reload() {
-        self.injectedPlugIns = InjectorV3.main.injectedAssetURLsInBundle(app.url)
+        var plugIns = [InjectedPlugIn]()
+        plugIns += InjectorV3.main.injectedAssetURLsInBundle(app.url)
             .map { InjectedPlugIn(url: $0, isEnabled: true) }
+
+        let enabledNames = plugIns.map { $0.url.lastPathComponent }
+        plugIns += InjectorV3.main.persistedAssetURLs(id: app.id)
+            .filter { !enabledNames.contains($0.lastPathComponent) }
+            .map { InjectedPlugIn(url: $0, isEnabled: false) }
+
+        injectedPlugIns = plugIns
+            .sorted { $0.url.lastPathComponent.localizedStandardCompare($1.url.lastPathComponent) == .orderedAscending }
+
         performFilter()
     }
 
@@ -45,5 +58,7 @@ final class EjectListModel: ObservableObject {
         }
 
         self.filteredPlugIns = filteredPlugIns
+        self.isOkToEnableAll = filteredPlugIns.contains { !$0.isEnabled }
+        self.isOkToDisableAll = filteredPlugIns.contains { $0.isEnabled }
     }
 }

TrollFools/EjectListView.swift

@@ -126,11 +126,11 @@ struct EjectListView: View {
             if !ejectList.filter.isSearching && !ejectList.filteredPlugIns.isEmpty {
                 Section {
                     enableAllButton
-                        .disabled(isEnablingAll)
+                        .disabled(isEnablingAll || !ejectList.isOkToEnableAll)
                         .foregroundColor(isEnablingAll ? .secondary : .accentColor)
 
                     disableAllButton
-                        .disabled(isDisablingAll)
+                        .disabled(isDisablingAll || !ejectList.isOkToDisableAll)
                         .foregroundColor(isDisablingAll ? .secondary : .accentColor)
                 }
 

TrollFools/IndexableScroller.swift

@@ -31,7 +31,7 @@ struct IndexableScroller: View {
                     .updating($dragLocation) { value, state, _ in
                         state = value.location
                     }
-                    .onEnded { value in
+                    .onEnded { _ in
                         currentIndex = nil
                     }
             )

TrollFools/InjectorV3+Bundle.swift

@@ -102,7 +102,7 @@ extension InjectorV3 {
             .sorted(by: { $0.lastPathComponent.localizedStandardCompare($1.lastPathComponent) == .orderedAscending })
     }
 
-    func injectedBundleURLsInBundle(_ target: URL) -> [URL] {
+    fileprivate func injectedBundleURLsInBundle(_ target: URL) -> [URL] {
         precondition(checkIsBundle(target), "Not a bundle: \(target.path)")
 
         guard let bundleContentURLs = try? FileManager.default.contentsOfDirectory(at: target, includingPropertiesForKeys: [.isDirectoryKey]) else {
@@ -125,7 +125,7 @@ extension InjectorV3 {
         return bundleURLs
     }
 
-    func injectedDylibAndFrameworkURLsInBundle(_ target: URL) -> [URL] {
+    fileprivate func injectedDylibAndFrameworkURLsInBundle(_ target: URL) -> [URL] {
         precondition(checkIsBundle(target), "Not a bundle: \(target.path)")
 
         let frameworksURL = target.appendingPathComponent("Frameworks")

TrollFools/InjectorV3+Persistent.swift

@@ -12,6 +12,7 @@ extension InjectorV3 {
         for filteredURL in filteredURLs(assetURLs) {
             let destURL = persistentPlugInsDirectoryURL.appendingPathComponent(filteredURL.lastPathComponent)
             try cmdCopy(from: filteredURL, to: destURL, overwrite: true)
+            try cmdChangeOwner(destURL, owner: 501, groupOwner: 501, recursively: checkIsDirectory(destURL))
         }
     }
 
@@ -34,6 +35,16 @@ extension InjectorV3 {
         }
     }
 
+    func persistedAssetURLs(id: String) -> [URL] {
+        let base = Self.persistentPlugInsRootURL.appendingPathComponent(id, isDirectory: true)
+        guard let contents = try? FileManager.default.contentsOfDirectory(atPath: base.path) else {
+            return []
+        }
+        return contents
+            .sorted { $0.localizedStandardCompare($1) == .orderedAscending }
+            .map { base.appendingPathComponent($0) }
+    }
+
     fileprivate func filteredURLs(_ assetURLs: [URL]) -> [URL] {
         assetURLs.filter {
             $0.pathExtension.lowercased() == "bundle" ||

TrollFools/InjectorV3.swift

@@ -119,7 +119,7 @@ final class InjectorV3 {
 
     // MARK: - Persistent
 
-    fileprivate static let persistentPlugInsRootURL: URL = {
+    static let persistentPlugInsRootURL: URL = {
         let url = URL(fileURLWithPath: "/var/mobile/Library/TrollFools/PersistentPlugins")
         try? FileManager.default.createDirectory(at: url, withIntermediateDirectories: true)
         return url

TrollFools/PlugInCell.swift

@@ -20,14 +20,13 @@ struct PlugInCell: View {
     @Environment(\.verticalSizeClass) var verticalSizeClass
 
     @Binding var quickLookExport: URL?
-    @State var isEnabled: Bool = true
+    @State var isEnabled: Bool = false
 
     let plugIn: InjectedPlugIn
 
     init(_ plugIn: InjectedPlugIn, quickLookExport: Binding<URL?>) {
         self.plugIn = plugIn
         _quickLookExport = quickLookExport
-        isEnabled = plugIn.isEnabled
     }
 
     @available(iOS 15, *)
@@ -82,6 +81,9 @@ struct PlugInCell: View {
                 }
             }
         }
+        .onAppear {
+            isEnabled = plugIn.isEnabled
+        }
         .contextMenu {
             if #available(iOS 16.4, *) {
                 ShareLink(item: plugIn.url) {