mirror of https://github.com/oxen-io/session-ios
Ryan Zhao
10 months ago
16 changed files with 212 additions and 59 deletions
@ -0,0 +1,46 @@
|
||||
// Copyright © 2022 Rangeproof Pty Ltd. All rights reserved. |
||||
|
||||
import Foundation |
||||
|
||||
@objc(SNOpenGroupServerIdLookup) |
||||
public final class OpenGroupServerIdLookup: NSObject, NSCoding { // NSObject/NSCoding conformance is needed for YapDatabase compatibility |
||||
@objc public let id: String |
||||
@objc public let serverId: UInt64 |
||||
@objc public let tsMessageId: String |
||||
|
||||
// MARK: - Initialization |
||||
|
||||
@objc public init(server: String, room: String, serverId: UInt64, tsMessageId: String) { |
||||
self.id = OpenGroupServerIdLookup.id(serverId: serverId, in: room, on: server) |
||||
self.serverId = serverId |
||||
self.tsMessageId = tsMessageId |
||||
|
||||
super.init() |
||||
} |
||||
|
||||
private override init() { preconditionFailure("Use init(blindedId:sessionId:) instead.") } |
||||
|
||||
// MARK: - Coding |
||||
|
||||
public required init?(coder: NSCoder) { |
||||
guard let id: String = coder.decodeObject(forKey: "id") as! String? else { return nil } |
||||
guard let serverId: UInt64 = coder.decodeObject(forKey: "serverId") as! UInt64? else { return nil } |
||||
guard let tsMessageId: String = coder.decodeObject(forKey: "tsMessageId") as! String? else { return nil } |
||||
|
||||
self.id = id |
||||
self.serverId = serverId |
||||
self.tsMessageId = tsMessageId |
||||
} |
||||
|
||||
public func encode(with coder: NSCoder) { |
||||
coder.encode(id, forKey: "id") |
||||
coder.encode(serverId, forKey: "serverId") |
||||
coder.encode(tsMessageId, forKey: "tsMessageId") |
||||
} |
||||
|
||||
// MARK: - Convenience |
||||
|
||||
static func id(serverId: UInt64, in room: String, on server: String) -> String { |
||||
return "\(server).\(room).\(serverId)" |
||||
} |
||||
} |
||||
@ -0,0 +1,48 @@
|
||||
// Copyright © 2022 Rangeproof Pty Ltd. All rights reserved. |
||||
|
||||
import Foundation |
||||
|
||||
@objc(SNOpenGroupServerIdLookupMigration) |
||||
public class OpenGroupServerIdLookupMigration: OWSDatabaseMigration { |
||||
@objc |
||||
class func migrationId() -> String { |
||||
return "003" |
||||
} |
||||
|
||||
override public func runUp(completion: @escaping OWSDatabaseMigrationCompletion) { |
||||
self.doMigrationAsync(completion: completion) |
||||
} |
||||
|
||||
private func doMigrationAsync(completion: @escaping OWSDatabaseMigrationCompletion) { |
||||
var lookups: [OpenGroupServerIdLookup] = [] |
||||
|
||||
Storage.write(with: { transaction in |
||||
TSGroupThread.enumerateCollectionObjects(with: transaction) { object, _ in |
||||
guard let thread: TSGroupThread = object as? TSGroupThread else { return } |
||||
guard let threadId: String = thread.uniqueId else { return } |
||||
guard let openGroup: OpenGroupV2 = Storage.shared.getV2OpenGroup(for: threadId) else { return } |
||||
|
||||
thread.enumerateInteractions(with: transaction) { interaction, _ in |
||||
guard let tsMessage: TSMessage = interaction as? TSMessage else { return } |
||||
guard let tsMessageId: String = tsMessage.uniqueId else { return } |
||||
|
||||
lookups.append( |
||||
OpenGroupServerIdLookup( |
||||
server: openGroup.server, |
||||
room: openGroup.room, |
||||
serverId: tsMessage.openGroupServerMessageID, |
||||
tsMessageId: tsMessageId |
||||
) |
||||
) |
||||
} |
||||
} |
||||
|
||||
lookups.forEach { lookup in |
||||
Storage.shared.addOpenGroupServerIdLookup(lookup, using: transaction) |
||||
} |
||||
self.save(with: transaction) // Intentionally capture self |
||||
}, completion: { |
||||
completion() |
||||
}) |
||||
} |
||||
} |
||||
Loading…
Reference in new issue