another.im-ios/ConversationsClassic/AppCore/Database/Database+Migrations.swift

import Foundation
import GRDB

extension Database {
    static var migrator: DatabaseMigrator = {
        var migrator = DatabaseMigrator()

        // flush db on schema change (only in DEV mode)
        #if DEBUG
            migrator.eraseDatabaseOnSchemaChange = true
        #endif

        // 1st migration - basic tables
        migrator.registerMigration("Add basic tables") { db in
            // accounts
            try db.create(table: "accounts", options: [.ifNotExists]) { table in
                table.column("bareJid", .text).notNull().primaryKey().unique(onConflict: .replace)
                table.column("pass", .text).notNull()
                table.column("isActive", .boolean).notNull().defaults(to: true)
                table.column("isTemp", .boolean).notNull().defaults(to: false)
            }

            // rosters
            try db.create(table: "rosterVersions", options: [.ifNotExists]) { table in
                table.column("bareJid", .text).notNull().primaryKey().unique(onConflict: .replace)
                table.column("version", .text).notNull()
            }
            try db.create(table: "rosters", options: [.ifNotExists]) { table in
                table.column("bareJid", .text).notNull()
                table.column("contactBareJid", .text).notNull()
                table.column("name", .text)
                table.column("subscription", .text).notNull()
                table.column("ask", .boolean).notNull().defaults(to: false)
                table.column("data", .text).notNull()
                table.primaryKey(["bareJid", "contactBareJid"], onConflict: .replace)
                table.column("locallyDeleted", .boolean).notNull().defaults(to: false)
            }

            // chats
            try db.create(table: "chats", options: [.ifNotExists]) { table in
                table.column("id", .text).notNull().primaryKey().unique(onConflict: .replace)
                table.column("account", .text).notNull()
                table.column("participant", .text).notNull()
                table.column("type", .integer).notNull()
            }

            // messages
            try db.create(table: "messages", options: [.ifNotExists]) { table in
                table.column("id", .text).notNull().primaryKey().unique(onConflict: .replace)
                table.column("type", .text).notNull()
                table.column("contentType", .text).notNull()
                table.column("from", .text).notNull()
                table.column("to", .text)
                table.column("body", .text)
                table.column("subject", .text)
                table.column("thread", .text)
                table.column("oobUrl", .text)
                table.column("date", .datetime).notNull()
                table.column("pending", .boolean).notNull()
                table.column("sentError", .boolean).notNull()
                table.column("attachmentType", .integer)
                table.column("attachmentLocalName", .text)
                table.column("attachmentRemotePath", .text)
                table.column("attachmentThumbnailName", .text)
                table.column("attachmentDownloadFailed", .boolean).notNull().defaults(to: false)
            }
        }

        // 2nd migration - channels/rooms
        migrator.registerMigration("Add channels/rooms") { db in
            // rooms
            try db.create(table: "rooms", options: [.ifNotExists]) { table in
                table.column("id", .text).notNull().primaryKey().unique(onConflict: .replace)
            }

            // channels
            // try db.create(table: "channels", options: [.ifNotExists]) { table in
            //     table.column("id", .text).notNull().primaryKey().unique(onConflict: .replace)
            //     table.column("account", .text).notNull()
            //     table.column("channel", .text).notNull()
            // }
        }

        // return migrator
        return migrator
    }()
}