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

import Foundation
import Sodium
import GRDB
import Curve25519Kit
import SessionUtilitiesKit
import SessionMessagingKit

enum Onboarding {
    
    enum Flow {
        case register, recover, link
        
        func preregister(with seed: Data, ed25519KeyPair: Sign.KeyPair, x25519KeyPair: ECKeyPair) {
            let userDefaults = UserDefaults.standard
            Identity.store(seed: seed, ed25519KeyPair: ed25519KeyPair, x25519KeyPair: x25519KeyPair)
            let x25519PublicKey = x25519KeyPair.hexEncodedPublicKey
            
            GRDBStorage.shared.write { db in
                try Contact(id: x25519PublicKey)
                    .with(
                        isApproved: true,
                        didApproveMe: true
                    )
                    .save(db)
            }
            
            switch self {
                case .register:
                    userDefaults[.hasViewedSeed] = false
                    // Set hasSyncedInitialConfiguration to true so that when we hit the
                    // home screen a configuration sync is triggered (yes, the logic is a
                    // bit weird). This is needed so that if the user registers and
                    // immediately links a device, there'll be a configuration in their swarm.
                    userDefaults[.hasSyncedInitialConfiguration] = true
                        
                case .recover, .link:
                    // No need to show it again if the user is restoring or linking
                    userDefaults[.hasViewedSeed] = true
                    userDefaults[.hasSyncedInitialConfiguration] = false
            }
            
            switch self {
                case .register, .recover:
                    // Set both lastDisplayNameUpdate and lastProfilePictureUpdate to the
                    // current date, so that we don't overwrite what the user set in the
                    // display name step with whatever we find in their swarm.
                    userDefaults[.lastDisplayNameUpdate] = Date()
                    userDefaults[.lastProfilePictureUpdate] = Date()
                    
                case .link: break
            }
        }
    }
}