mastodon-app-ufficiale-ipho.../CoreDataStack/Entity/Status.swift

341 lines
12 KiB
Swift
Raw Normal View History

2021-01-27 07:50:13 +01:00
//
2021-01-28 09:10:30 +01:00
// Toot.swift
2021-01-27 07:50:13 +01:00
// CoreDataStack
//
// Created by MainasuK Cirno on 2021/1/27.
//
import CoreData
import Foundation
2021-01-27 07:50:13 +01:00
2021-04-01 08:39:15 +02:00
public final class Status: NSManagedObject {
2021-01-27 07:50:13 +01:00
public typealias ID = String
2021-01-27 07:50:13 +01:00
@NSManaged public private(set) var identifier: ID
@NSManaged public private(set) var domain: String
@NSManaged public private(set) var id: String
@NSManaged public private(set) var uri: String
@NSManaged public private(set) var createdAt: Date
2021-01-27 07:50:13 +01:00
@NSManaged public private(set) var content: String
@NSManaged public private(set) var visibility: String?
@NSManaged public private(set) var sensitive: Bool
@NSManaged public private(set) var spoilerText: String?
@NSManaged public private(set) var application: Application?
// Informational
@NSManaged public private(set) var reblogsCount: NSNumber
@NSManaged public private(set) var favouritesCount: NSNumber
@NSManaged public private(set) var repliesCount: NSNumber?
@NSManaged public private(set) var url: String?
2021-04-01 08:39:15 +02:00
@NSManaged public private(set) var inReplyToID: Status.ID?
@NSManaged public private(set) var inReplyToAccountID: MastodonUser.ID?
@NSManaged public private(set) var language: String? // (ISO 639 Part 1 two-letter language code)
@NSManaged public private(set) var text: String?
// many-to-one relastionship
@NSManaged public private(set) var author: MastodonUser
2021-04-01 08:39:15 +02:00
@NSManaged public private(set) var reblog: Status?
@NSManaged public private(set) var replyTo: Status?
// many-to-many relastionship
2021-02-07 07:42:50 +01:00
@NSManaged public private(set) var favouritedBy: Set<MastodonUser>?
@NSManaged public private(set) var rebloggedBy: Set<MastodonUser>?
@NSManaged public private(set) var mutedBy: Set<MastodonUser>?
@NSManaged public private(set) var bookmarkedBy: Set<MastodonUser>?
// one-to-one relastionship
@NSManaged public private(set) var pinnedBy: MastodonUser?
@NSManaged public private(set) var poll: Poll?
// one-to-many relationship
2021-04-01 08:39:15 +02:00
@NSManaged public private(set) var reblogFrom: Set<Status>?
@NSManaged public private(set) var mentions: Set<Mention>?
@NSManaged public private(set) var emojis: Set<Emoji>?
@NSManaged public private(set) var tags: Set<Tag>?
2021-01-27 07:50:13 +01:00
@NSManaged public private(set) var homeTimelineIndexes: Set<HomeTimelineIndex>?
@NSManaged public private(set) var mediaAttachments: Set<Attachment>?
2021-04-01 08:39:15 +02:00
@NSManaged public private(set) var replyFrom: Set<Status>?
@NSManaged public private(set) var updatedAt: Date
@NSManaged public private(set) var deletedAt: Date?
2021-01-27 07:50:13 +01:00
}
2021-04-01 08:39:15 +02:00
public extension Status {
2021-01-27 07:50:13 +01:00
@discardableResult
static func insert(
2021-01-27 07:50:13 +01:00
into context: NSManagedObjectContext,
property: Property,
author: MastodonUser,
2021-04-01 08:39:15 +02:00
reblog: Status?,
application: Application?,
2021-04-01 08:39:15 +02:00
replyTo: Status?,
poll: Poll?,
mentions: [Mention]?,
emojis: [Emoji]?,
tags: [Tag]?,
mediaAttachments: [Attachment]?,
favouritedBy: MastodonUser?,
rebloggedBy: MastodonUser?,
mutedBy: MastodonUser?,
bookmarkedBy: MastodonUser?,
pinnedBy: MastodonUser?
2021-04-01 08:39:15 +02:00
) -> Status {
let toot: Status = context.insertObject()
2021-01-27 07:50:13 +01:00
toot.identifier = property.identifier
toot.domain = property.domain
toot.id = property.id
toot.uri = property.uri
toot.createdAt = property.createdAt
toot.content = property.content
2021-01-27 07:50:13 +01:00
toot.visibility = property.visibility
toot.sensitive = property.sensitive
toot.spoilerText = property.spoilerText
toot.application = application
2021-01-27 07:50:13 +01:00
toot.reblogsCount = property.reblogsCount
toot.favouritesCount = property.favouritesCount
toot.repliesCount = property.repliesCount
2021-01-27 07:50:13 +01:00
toot.url = property.url
toot.inReplyToID = property.inReplyToID
toot.inReplyToAccountID = property.inReplyToAccountID
toot.language = property.language
toot.text = property.text
2021-01-27 07:50:13 +01:00
toot.author = author
toot.reblog = reblog
2021-02-07 07:42:50 +01:00
toot.pinnedBy = pinnedBy
toot.poll = poll
2021-02-07 07:42:50 +01:00
if let mentions = mentions {
2021-04-01 08:39:15 +02:00
toot.mutableSetValue(forKey: #keyPath(Status.mentions)).addObjects(from: mentions)
}
if let emojis = emojis {
2021-04-01 08:39:15 +02:00
toot.mutableSetValue(forKey: #keyPath(Status.emojis)).addObjects(from: emojis)
}
if let tags = tags {
2021-04-01 08:39:15 +02:00
toot.mutableSetValue(forKey: #keyPath(Status.tags)).addObjects(from: tags)
}
if let mediaAttachments = mediaAttachments {
2021-04-01 08:39:15 +02:00
toot.mutableSetValue(forKey: #keyPath(Status.mediaAttachments)).addObjects(from: mediaAttachments)
}
if let favouritedBy = favouritedBy {
2021-04-01 08:39:15 +02:00
toot.mutableSetValue(forKey: #keyPath(Status.favouritedBy)).add(favouritedBy)
}
if let rebloggedBy = rebloggedBy {
2021-04-01 08:39:15 +02:00
toot.mutableSetValue(forKey: #keyPath(Status.rebloggedBy)).add(rebloggedBy)
}
if let mutedBy = mutedBy {
2021-04-01 08:39:15 +02:00
toot.mutableSetValue(forKey: #keyPath(Status.mutedBy)).add(mutedBy)
}
if let bookmarkedBy = bookmarkedBy {
2021-04-01 08:39:15 +02:00
toot.mutableSetValue(forKey: #keyPath(Status.bookmarkedBy)).add(bookmarkedBy)
}
toot.updatedAt = property.networkDate
return toot
}
2021-03-10 12:12:53 +01:00
2021-02-03 09:37:18 +01:00
func update(reblogsCount: NSNumber) {
if self.reblogsCount.intValue != reblogsCount.intValue {
self.reblogsCount = reblogsCount
}
}
2021-03-10 12:12:53 +01:00
2021-02-03 09:37:18 +01:00
func update(favouritesCount: NSNumber) {
if self.favouritesCount.intValue != favouritesCount.intValue {
self.favouritesCount = favouritesCount
}
}
2021-03-10 12:12:53 +01:00
2021-02-03 09:37:18 +01:00
func update(repliesCount: NSNumber?) {
guard let count = repliesCount else {
return
}
if self.repliesCount?.intValue != count.intValue {
self.repliesCount = repliesCount
}
}
2021-03-10 12:12:53 +01:00
2021-04-01 08:39:15 +02:00
func update(replyTo: Status?) {
2021-03-10 12:12:53 +01:00
if self.replyTo != replyTo {
self.replyTo = replyTo
}
}
2021-04-01 08:39:15 +02:00
func update(liked: Bool, by mastodonUser: MastodonUser) {
2021-02-08 11:29:27 +01:00
if liked {
if !(self.favouritedBy ?? Set()).contains(mastodonUser) {
2021-04-01 08:39:15 +02:00
self.mutableSetValue(forKey: #keyPath(Status.favouritedBy)).add(mastodonUser)
2021-02-08 11:29:27 +01:00
}
} else {
if (self.favouritedBy ?? Set()).contains(mastodonUser) {
2021-04-01 08:39:15 +02:00
self.mutableSetValue(forKey: #keyPath(Status.favouritedBy)).remove(mastodonUser)
2021-02-08 11:29:27 +01:00
}
}
}
2021-03-10 12:12:53 +01:00
2021-04-01 08:39:15 +02:00
func update(reblogged: Bool, by mastodonUser: MastodonUser) {
if reblogged {
if !(self.rebloggedBy ?? Set()).contains(mastodonUser) {
2021-04-01 08:39:15 +02:00
self.mutableSetValue(forKey: #keyPath(Status.rebloggedBy)).add(mastodonUser)
}
} else {
if (self.rebloggedBy ?? Set()).contains(mastodonUser) {
2021-04-01 08:39:15 +02:00
self.mutableSetValue(forKey: #keyPath(Status.rebloggedBy)).remove(mastodonUser)
}
}
}
2021-04-01 08:39:15 +02:00
func update(muted: Bool, by mastodonUser: MastodonUser) {
if muted {
if !(self.mutedBy ?? Set()).contains(mastodonUser) {
2021-04-01 08:39:15 +02:00
self.mutableSetValue(forKey: #keyPath(Status.mutedBy)).add(mastodonUser)
}
} else {
if (self.mutedBy ?? Set()).contains(mastodonUser) {
2021-04-01 08:39:15 +02:00
self.mutableSetValue(forKey: #keyPath(Status.mutedBy)).remove(mastodonUser)
}
}
}
2021-04-01 08:39:15 +02:00
func update(bookmarked: Bool, by mastodonUser: MastodonUser) {
if bookmarked {
if !(self.bookmarkedBy ?? Set()).contains(mastodonUser) {
2021-04-01 08:39:15 +02:00
self.mutableSetValue(forKey: #keyPath(Status.bookmarkedBy)).add(mastodonUser)
}
} else {
if (self.bookmarkedBy ?? Set()).contains(mastodonUser) {
2021-04-01 08:39:15 +02:00
self.mutableSetValue(forKey: #keyPath(Status.bookmarkedBy)).remove(mastodonUser)
}
}
}
2021-02-03 09:37:18 +01:00
func didUpdate(at networkDate: Date) {
self.updatedAt = networkDate
}
}
2021-04-01 08:39:15 +02:00
public extension Status {
struct Property {
public let identifier: ID
public let domain: String
public let id: String
public let uri: String
public let createdAt: Date
public let content: String
public let visibility: String?
public let sensitive: Bool
public let spoilerText: String?
public let reblogsCount: NSNumber
public let favouritesCount: NSNumber
public let repliesCount: NSNumber?
public let url: String?
2021-04-01 08:39:15 +02:00
public let inReplyToID: Status.ID?
public let inReplyToAccountID: MastodonUser.ID?
public let language: String? // (ISO 639 Part @1 two-letter language code)
public let text: String?
public let networkDate: Date
2021-01-27 07:50:13 +01:00
public init(
domain: String,
id: String,
uri: String,
2021-01-27 07:50:13 +01:00
createdAt: Date,
content: String,
visibility: String?,
sensitive: Bool,
spoilerText: String?,
reblogsCount: NSNumber,
favouritesCount: NSNumber,
repliesCount: NSNumber?,
url: String?,
2021-04-01 08:39:15 +02:00
inReplyToID: Status.ID?,
inReplyToAccountID: MastodonUser.ID?,
language: String?,
text: String?,
networkDate: Date
) {
2021-01-27 07:50:13 +01:00
self.identifier = id + "@" + domain
self.domain = domain
self.id = id
self.uri = uri
2021-01-27 07:50:13 +01:00
self.createdAt = createdAt
self.content = content
self.visibility = visibility
self.sensitive = sensitive
self.spoilerText = spoilerText
self.reblogsCount = reblogsCount
self.favouritesCount = favouritesCount
self.repliesCount = repliesCount
self.url = url
self.inReplyToID = inReplyToID
self.inReplyToAccountID = inReplyToAccountID
self.language = language
self.text = text
self.networkDate = networkDate
2021-01-27 07:50:13 +01:00
}
2021-01-27 07:50:13 +01:00
}
}
2021-04-01 08:39:15 +02:00
extension Status: Managed {
2021-01-27 07:50:13 +01:00
public static var defaultSortDescriptors: [NSSortDescriptor] {
2021-04-01 08:39:15 +02:00
return [NSSortDescriptor(keyPath: \Status.createdAt, ascending: false)]
2021-01-27 07:50:13 +01:00
}
}
2021-04-01 08:39:15 +02:00
extension Status {
static func predicate(domain: String) -> NSPredicate {
2021-04-01 08:39:15 +02:00
return NSPredicate(format: "%K == %@", #keyPath(Status.domain), domain)
}
static func predicate(id: String) -> NSPredicate {
2021-04-01 08:39:15 +02:00
return NSPredicate(format: "%K == %@", #keyPath(Status.id), id)
}
public static func predicate(domain: String, id: String) -> NSPredicate {
return NSCompoundPredicate(andPredicateWithSubpredicates: [
predicate(domain: domain),
predicate(id: id)
])
}
static func predicate(ids: [String]) -> NSPredicate {
2021-04-01 08:39:15 +02:00
return NSPredicate(format: "%K IN %@", #keyPath(Status.id), ids)
2021-01-28 09:10:30 +01:00
}
public static func predicate(domain: String, ids: [String]) -> NSPredicate {
return NSCompoundPredicate(andPredicateWithSubpredicates: [
predicate(domain: domain),
predicate(ids: ids)
])
2021-01-28 09:10:30 +01:00
}
public static func notDeleted() -> NSPredicate {
2021-04-01 08:39:15 +02:00
return NSPredicate(format: "%K == nil", #keyPath(Status.deletedAt))
2021-01-28 09:10:30 +01:00
}
public static func deleted() -> NSPredicate {
2021-04-01 08:39:15 +02:00
return NSPredicate(format: "%K != nil", #keyPath(Status.deletedAt))
2021-01-28 09:10:30 +01:00
}
}