feat: adding pause rate limiter

Sources/hyper-focus/hyper_focus.swift

@@ -21,6 +21,7 @@ struct SwitchingActivity: CustomStringConvertible {
 
 struct Configuration: Codable {
     struct ScheduleItem: Codable, Equatable {
+        var pause_limit: Int?
         var start: Int?
         var start_minute: Int?
         var end: Int?

Sources/hyper-focus/rate_limiter.swift

@@ -0,0 +1,31 @@
+import Foundation
+
+class RateLimiter {
+    private var timestamps: [String: [Date]] = [:]
+
+    // actions per hour
+    var limit: Int?
+
+    func allowed(type: String) -> Bool {
+        if limit == nil {
+            debug("no limit set, allowing action '\(type)'")
+            return true
+        }
+
+        let now = Date()
+        let oneHourAgo = now.addingTimeInterval(-3600)
+
+        var typeStamps = timestamps[type, default: []]
+        typeStamps = typeStamps.filter { $0 > oneHourAgo }
+
+        if typeStamps.count < limit! {
+            typeStamps.append(now)
+            timestamps[type] = typeStamps
+            return true
+        }
+
+        // Update the timestamps to remove stale entries
+        timestamps[type] = typeStamps
+        return false
+    }
+}

Sources/hyper-focus/schedule_manager.swift

@@ -2,6 +2,7 @@ import Foundation
 
 class ScheduleManager {
     var configuration: Configuration
+    var pauseLimiter = RateLimiter()
 
     var schedule: Configuration.ScheduleItem?
     var endOverride: Date?
@@ -83,9 +84,15 @@ class ScheduleManager {
         }
     }
 
-    func pauseBlocking(_ end: Date) {
+    func pauseBlocking(_ end: Date) -> Bool {
+        if !pauseLimiter.allowed(type: "pause") {
+            error("pause limit reached, not pausing")
+            return false
+        }
+
         log("pause blocking until \(end)")
         endPause = end
+        return true
     }
 
     func resumeBlocking() {
@@ -122,6 +129,11 @@ class ScheduleManager {
         // TODO: there's probably some race condition risk here, but I'm too lazy to understand swift concurrency locking
         self.schedule = schedule
 
+        if schedule != nil {
+            // we want to set limit to nil if not set in schedule
+            pauseLimiter.limit = schedule!.pause_limit
+        }
+
         if schedule != nil, schedule!.start_script != nil {
             TaskRunner.executeTaskWithName(schedule!.start_script!, "start_script")
         }

Sources/hyper-focus/server.swift

@@ -147,9 +147,12 @@ class ApiServer {
 
                 // convert pauseUtil to date
                 let untilDate = Date(timeIntervalSince1970: TimeInterval(pauseUtil!))
-                self.scheduleManager.pauseBlocking(untilDate)
 
-                return ["status": "ok"]
+                if self.scheduleManager.pauseBlocking(untilDate) {
+                    return ["status": "ok"]
+                } else {
+                    return ["status": "error", "message": "blocked"]
+                }
             }
 
             try! server.start()

schema.json

@@ -27,6 +27,9 @@
           "name": {
             "type": "string"
           },
+          "pause_limit": {
+            "type": "number"
+          },
           "schedule_only": {
             "type": "boolean"
           },