morgan/macos-stats
1
0
Fork 0
mirror of https://github.com/morgan9e/macos-stats synced 2026-04-14 00:04:15 +09:00
Code Issues Packages Projects Releases Wiki Activity

feat: added a new option to the Network module which allows to auto reset data usage (#542)

Browse Source
This commit is contained in:
Serhiy Mytrovtsiy
2021-09-16 18:56:43 +02:00
parent c3578b8791
commit e6ed7977ec
2 changed files with 78 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
Modules/Net/main.swift
View File

@@ -77,6 +77,7 @@ public class Network: Module {
private var processReader: ProcessReader? = nil
private let ipUpdater = NSBackgroundActivityScheduler(identifier: "eu.exelban.Stats.Network.IP")
private let usageReseter = NSBackgroundActivityScheduler(identifier: "eu.exelban.Stats.Network.Usage")
public init() {
self.settingsView = Settings("Network")
@@ -115,6 +116,9 @@ public class Network: Module {
self.usageReader?.getDetails()
self.usageReader?.read()
}
self.settingsView.usageResetCallback = { [unowned self] in
self.setUsageReset()
}
if let reader = self.usageReader {
self.addReader(reader)
@@ -123,13 +127,8 @@ public class Network: Module {
self.addReader(reader)
}
self.ipUpdater.interval = 60 * 60
self.ipUpdater.repeats = true
self.ipUpdater.schedule { (completion: @escaping NSBackgroundActivityScheduler.CompletionHandler) in
debug("going to automatically refresh IP address...")
NotificationCenter.default.post(name: .refreshPublicIP, object: nil, userInfo: nil)
completion(NSBackgroundActivityScheduler.Result.finished)
}
self.setIPUpdater()
self.setUsageReset()
}
public override func isAvailable() -> Bool {
@@ -157,4 +156,33 @@ public class Network: Module {
}
}
}
private func setIPUpdater() {
self.ipUpdater.interval = 60 * 60
self.ipUpdater.repeats = true
self.ipUpdater.schedule { (completion: @escaping NSBackgroundActivityScheduler.CompletionHandler) in
debug("going to automatically refresh IP address...")
NotificationCenter.default.post(name: .refreshPublicIP, object: nil, userInfo: nil)
completion(NSBackgroundActivityScheduler.Result.finished)
}
}
private func setUsageReset() {
self.usageReseter.invalidate()
switch AppUpdateInterval(rawValue: Store.shared.string(key: "\(self.config.name)_usageReset", defaultValue: AppUpdateInterval.atStart.rawValue)) {
case .oncePerDay: self.usageReseter.interval = 60 * 60 * 24
case .oncePerWeek: self.usageReseter.interval = 60 * 60 * 24 * 7
case .oncePerMonth: self.usageReseter.interval = 60 * 60 * 24 * 30
case .never, .atStart: return
default: return
}
self.usageReseter.repeats = true
self.usageReseter.schedule { (completion: @escaping NSBackgroundActivityScheduler.CompletionHandler) in
debug("going to reset the usage...")
NotificationCenter.default.post(name: .resetTotalNetworkUsage, object: nil, userInfo: nil)
completion(NSBackgroundActivityScheduler.Result.finished)
}
}
}