Tusker/Pachyderm/Utilities/NotificationGroup.swift

55 lines
2.0 KiB
Swift
Raw Normal View History

2019-09-05 17:38:04 -04:00
//
// NotificationGroup.swift
// Pachyderm
//
// Created by Shadowfacts on 9/5/19.
// Copyright © 2019 Shadowfacts. All rights reserved.
//
import Foundation
public class NotificationGroup {
2020-04-27 19:20:09 -04:00
public let notifications: [Notification]
2019-09-05 17:38:04 -04:00
public let id: String
public let kind: Notification.Kind
public let statusState: StatusState?
2020-04-27 19:20:09 -04:00
2019-09-05 17:38:04 -04:00
init?(notifications: [Notification]) {
guard !notifications.isEmpty else { return nil }
2020-04-27 19:20:09 -04:00
self.notifications = notifications
2019-09-05 17:38:04 -04:00
self.id = notifications.first!.id
self.kind = notifications.first!.kind
if kind == .mention {
self.statusState = .unknown
} else {
self.statusState = nil
}
2019-09-05 17:38:04 -04:00
}
public static func createGroups(notifications: [Notification], only allowedTypes: [Notification.Kind]) -> [NotificationGroup] {
var groups = [[Notification]]()
for notification in notifications {
if allowedTypes.contains(notification.kind) {
if let lastGroup = groups.last, let firstNotification = lastGroup.first, firstNotification.kind == notification.kind, firstNotification.status?.id == notification.status?.id {
groups[groups.count - 1].append(notification)
continue
} else if groups.count >= 2 {
let secondToLastGroup = groups[groups.count - 2]
if allowedTypes.contains(groups[groups.count - 1][0].kind), let firstNotification = secondToLastGroup.first, firstNotification.kind == notification.kind, firstNotification.status?.id == notification.status?.id {
groups[groups.count - 2].append(notification)
continue
}
}
2019-09-05 17:38:04 -04:00
}
groups.append([notification])
}
return groups.map {
2019-09-05 17:38:04 -04:00
NotificationGroup(notifications: $0)!
}
}
}
extension NotificationGroup: Identifiable {}