safe-ios/Multisig/Features/Connect to Web/Logic/Model/WebConnection.swift

//
// Created by Dmitry Bespalov on 10.02.22.
// Copyright (c) 2022 Gnosis Ltd. All rights reserved.
//

import Foundation

class WebConnection {
    /// Identifier of the connection. Also a URL for establishing the connection. Required.
    var connectionURL: WebConnectionURL

    /// Current state of the connection
    var status: WebConnectionStatus = .initial

    /// Currently selected chain id. Required.
    var chainId: Int? = nil

    /// Currently selected account addresses. Empty for pending connections.
    var accounts: [Address] = []

    /// Date when the connection was established. Nil if the connection pending.
    var createdDate: Date? = nil

    /// Date when the connection will expire. Nil if the connection pending.
    var expirationDate: Date? = nil

    /// Date of last activity on the connection. Activity means receiving or sending any requests to the remote peer. Nil if connection is pending.
    var lastActivityDate: Date? = nil

    /// Information about the 'self' or this application. Nil before connection is established.
    var localPeer: WebConnectionPeerInfo? = nil

    /// Information about the remote application. Nil if not yet fetched.
    var remotePeer: WebConnectionPeerInfo? = nil

    /// Error that occurred during operation
    var lastError: String? = nil

    init(connectionURL: WebConnectionURL) {
        self.connectionURL = connectionURL
    }

    init(connectionURL: WebConnectionURL, status: WebConnectionStatus, chainId: Int?, accounts: [Address], createdDate: Date?, expirationDate: Date?, lastActivityDate: Date?, localPeer: WebConnectionPeerInfo?, remotePeer: WebConnectionPeerInfo?, lastError: String?) {
        self.connectionURL = connectionURL
        self.status = status
        self.chainId = chainId
        self.accounts = accounts
        self.createdDate = createdDate
        self.expirationDate = expirationDate
        self.lastActivityDate = lastActivityDate
        self.localPeer = localPeer
        self.remotePeer = remotePeer
        self.lastError = lastError
    }
    
    var connectedAsDapp: Bool {
        if status == .opened, let localPeer = localPeer, let remotePeer = remotePeer {
            return localPeer.role == WebConnectionPeerRole.dapp && remotePeer.role == WebConnectionPeerRole.wallet
        }
        return false
    }

    var connectedAsWallet: Bool {
        if status == .opened, let localPeer = localPeer, let remotePeer = remotePeer {
            return localPeer.role == WebConnectionPeerRole.wallet && remotePeer.role == WebConnectionPeerRole.dapp
        }
        return false
    }
}

enum WebConnectionStatus: Int16 {
    // connection with a url set
    case initial 

    // connection that started to receive the connection request
    case handshaking

    // connection received connection request, waits for user response
    case approving

    // connection will send ok response
    case approved

    // connection will send rejected response
    case rejected

    // connection successfully sent connected response, ready to receive other requests
    case opened

    // connection is closed
    case closed

    // connection is about to be deleted
    case final 

    /// compatibility for future versions
    case unknown = -1
}