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

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

import Foundation
import GRDB
import SessionUtilitiesKit

enum _048_SessionProChanges: Migration {
    static let identifier: String = "SessionProChanges"
    static let minExpectedRunDuration: TimeInterval = 0.1
    static var createdTables: [(FetchableRecord & TableRecord).Type] = []
    
    static func migrate(_ db: ObservingDatabase, using dependencies: Dependencies) throws {
        try db.alter(table: "interaction") { t in
            t.drop(column: "isProMessage")
            t.add(column: "proMessageFeatures", .integer)
                .notNull()
                .defaults(to: 0)
            t.add(column: "proProfileFeatures", .integer)
                .notNull()
                .defaults(to: 0)
        }
        
        try db.alter(table: "profile") { t in
            t.add(column: "proFeatures", .integer)
                .notNull()
                .defaults(to: 0)
            t.add(column: "proExpiryUnixTimestampMs", .integer)
                .notNull()
                .defaults(to: 0)
            t.add(column: "proGenIndexHashHex", .text)
        }
        
        /// SQLite doesn't retroactively insert default values into columns so we need to add them now
        try db.execute(sql: """
            UPDATE interaction 
            SET proMessageFeatures = 0, proProfileFeatures = 0
        """)

        try db.execute(sql: """
            UPDATE profile 
            SET proFeatures = 0, proExpiryUnixTimestampMs = 0
        """)
        
        MigrationExecution.updateProgress(1)
    }
}