session-ios/SessionMessagingKit/Database/Migrations/_011_DisappearingMessagesConfiguration.swift

// Copyright © 2022 Rangeproof Pty Ltd. All rights reserved.

import Foundation
import GRDB
import SessionUtilitiesKit

enum _011_DisappearingMessagesConfiguration: Migration {
    static let target: TargetMigrations.Identifier = .messagingKit
    static let identifier: String = "DisappearingMessagesWithTypes"
    static let needsConfigSync: Bool = false
    static let minExpectedRunDuration: TimeInterval = 0.1
    
    static func migrate(_ db: GRDB.Database) throws {
        try db.alter(table: DisappearingMessagesConfiguration.self) { t in
            t.add(.type, .integer)
            t.add(.lastChangeTimestampMs, .integer)
                .defaults(to: 0)
        }
        
        try db.alter(table: Contact.self) { t in
            t.add(.lastKnownClientVersion, .integer)
        }
        
        /// Add index on interaction table for wasRead and variant
        /// 
        /// This is due to new disappearing messages will need some info messages to be able to be unread,
        /// but we only want to count the unread message number by incoming visible messages and call messages.
        try db.create(
            index: "interaction_on_wasRead_and_variant",
            on: Interaction.databaseTableName,
            columns: [Interaction.Columns.wasRead, Interaction.Columns.variant].map { $0.name }
        )
        
        func updateDisappearingMessageType(_ db: GRDB.Database, id: String, type: DisappearingMessagesConfiguration.DisappearingMessageType) throws {
            _ = try DisappearingMessagesConfiguration
                .filter(DisappearingMessagesConfiguration.Columns.threadId == id)
                .updateAll(
                    db,
                    DisappearingMessagesConfiguration.Columns.type.set(to: type)
                )
        }
        
        try DisappearingMessagesConfiguration
            .filter(DisappearingMessagesConfiguration.Columns.isEnabled == true)
            .fetchAll(db)
            .forEach { config in
                if let thread = try? SessionThread.fetchOne(db, id: config.threadId) {
                    guard !thread.isNoteToSelf(db) else {
                        try updateDisappearingMessageType(db, id: config.threadId, type: .disappearAfterSend)
                        return
                    }
                    
                    switch thread.variant {
                        case .contact: try updateDisappearingMessageType(db, id: config.threadId, type: .disappearAfterRead)
                        case .closedGroup: try updateDisappearingMessageType(db, id: config.threadId, type: .disappearAfterSend)
                        case .openGroup: return
                    }
                }
            }
        
        Storage.update(progress: 1, for: self, in: target) // In case this is the last migration
    }
}