add option for -custom-sofa-feed-url

macadmins · Jul 22, 2024 · efb9ecc · efb9ecc
1 parent 73656af
commit efb9ecc
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/Nudge/3rd Party Assets/sofa.swift b/Nudge/3rd Party Assets/sofa.swift
@@ -223,6 +223,15 @@ extension MacOSDataFeed {
 
 class SOFA: NSObject, URLSessionDelegate {
     func URLSync(url: URL, maxRetries: Int = 3) -> (data: Data?, response: URLResponse?, error: Error?, responseCode: Int?, eTag: String?) {
+        if url.scheme == "file" {
+            do {
+                let data = try Data(contentsOf: url)
+                return (data, nil, nil, 200, nil)
+            } catch {
+                return (nil, nil, error, nil, nil)
+            }
+        }
+
         let semaphore = DispatchSemaphore(value: 0)
         let lastEtag = Globals.nudgeDefaults.string(forKey: "LastEtag") ?? ""
         var request = URLRequest(url: url)

diff --git a/Nudge/Preferences/DefaultPreferencesNudge.swift b/Nudge/Preferences/DefaultPreferencesNudge.swift
@@ -145,6 +145,7 @@ struct OptionalFeatureVariables {
     }
 
     static var customSOFAFeedURL: String {
+        CommandLineUtilities().customSOFAFeedURLOption() ??
         optionalFeaturesProfile?["customSOFAFeedURL"] as? String ??
         optionalFeaturesJSON?.customSOFAFeedURL ??
         "https://sofafeed.macadmins.io/v1/macos_data_feed.json"

diff --git a/Nudge/Utilities/Utils.swift b/Nudge/Utilities/Utils.swift
@@ -366,6 +366,10 @@ struct CommandLineUtilities {
         return checkAndLogArgument("-bundle-mode-profile", logStateKey: &nudgeLogState.hasLoggedBundleMode)
     }
 
+    func customSOFAFeedURLOption() -> String? {
+        return valueForArgument("-custom-sofa-feed-url")
+    }
+
     func debugUIModeEnabled() -> Bool {
         return checkAndLogArgument("-debug-ui-mode", logStateKey: &nudgeLogState.afterFirstRun)
     }