mobile-ios/CovidSafe/InfoViewController.swift

import UIKit
import CoreData
import KeychainSwift

final class InfoViewController: UIViewController {
    @IBOutlet weak var identifierLabel: UILabel!
    @IBOutlet weak var devicesEncounteredLabel: UILabel!
    @IBOutlet weak var clearLogsButton: UIButton!
    @IBOutlet weak var advertisementSwitch: UISwitch!
    @IBOutlet weak var scanningSwitch: UISwitch!
    @IBOutlet weak var centralStateLabel: UILabel!
    @IBOutlet weak var obtainBluetoothStateButton: UIButton!
    @IBOutlet weak var peripheralStateLabel: UILabel!
    @IBOutlet weak var discoveredPeripheralsCountLabel: UILabel!
    @IBOutlet weak var silentNotificationsCountLabel: UILabel!
    @IBOutlet weak var apnTokenLabel: UILabel!
    private var devicesEncounteredCount: Int?
    @IBOutlet weak var messagesAPILastDateLabel: UILabel!
    @IBOutlet weak var messagesAPILastVersionLabel: UILabel!
    @IBOutlet weak var bleSensorStateLabel: UILabel!
    @IBOutlet weak var awakeSensorStateLabel: UILabel!
    @IBOutlet weak var jwtExpiryLabel: UILabel!
    @IBOutlet weak var jwtSubjectLabel: UILabel!
    @IBOutlet weak var refreshExpiryLabel: UILabel!
    
    @IBOutlet weak var versionNumLabel: UILabel!
    
    let dateFormatter = DateFormatter()
    var isSensorOn = true
    
    
    override func viewWillAppear(_ animated: Bool) {
        super.viewWillAppear(animated)
        fetchDevicesEncounteredCount()
        self.identifierLabel.text = DeviceIdentifier.getID()
        self.versionNumLabel.text = "\(PlistHelper.getvalueFromInfoPlist(withKey: kCFBundleVersionKey as String) ?? "no commit hash")"
        
        let lastAPICall = UserDefaults.standard.double(forKey: MessageAPI.keyLastApiUpdate)
        guard let lastVersion = UserDefaults.standard.string(forKey: MessageAPI.keyLastVersionChecked), lastAPICall > 0 else {
            return
        }
        messagesAPILastVersionLabel.text = lastVersion
        messagesAPILastDateLabel.text = dateFormatter.string(from:  Date(timeIntervalSince1970: lastAPICall))
    }
    
    override func viewDidLoad() {
        super.viewDidLoad()
        BluetraceManager.shared.addDelegateToSensors(delegate: self)
        dateFormatter.dateFormat = "dd/MM/yyyy"
        advertisementSwitch.addTarget(self, action: #selector(self.advertisementSwitchChanged), for: UIControl.Event.valueChanged)
        scanningSwitch.addTarget(self, action: #selector(self.scanningSwitchChanged), for: UIControl.Event.valueChanged)
        clearLogsButton.addTarget(self, action:#selector(self.clearLogsButtonClicked), for: .touchUpInside)
        silentNotificationsCountLabel.text = "\(UserDefaults.standard.integer(forKey: "debugSilentNotificationCount"))"
        apnTokenLabel.text = UserDefaults.standard.string(forKey: "deviceTokenForAPN")
        
        let keychain = KeychainSwift()
        let dateFormatter = DateFormatter()
        dateFormatter.dateStyle = .short
        dateFormatter.timeStyle = .medium
        
        if let token = keychain.get("JWT_TOKEN"),
           let jwtExpiry = AuthenticationToken(token: token).getExpiry(),
           let jwtSubject = AuthenticationToken(token: token).getSubject()
        {
            jwtExpiryLabel.text = dateFormatter.string(from: jwtExpiry)
            jwtSubjectLabel.text = jwtSubject
        } else {
            jwtExpiryLabel.text = "N/A"
        }
        
        if let refreshToken = keychain.get("REFRESH_TOKEN") {
            if let refreshExpiry = AuthenticationToken(token: refreshToken).getExpiry() {
                refreshExpiryLabel.text = dateFormatter.string(from: refreshExpiry)
            } else {
                refreshExpiryLabel.text = "exists, exp N/A"
            }
            
        } else {
            refreshExpiryLabel.text = "N/A, N/A"
        }
        
    }
    
    @IBAction func logoutBtn(_ sender: UIButton) {
        do {
            // do a logout if we still leave this button in
            performSegue(withIdentifier: "logoutToInitialVCSegue", sender: nil)
        } catch {
            print("Unable to log out")
        }
        
    }
    
    @IBAction func requestUploadOTP(_ sender: UIButton) {
        let keychain = KeychainSwift()
        guard let jwt = keychain.get("JWT_TOKEN") else {
            DLog("Error trying to upload when not logged in")
            return
        }
        InitiateUploadAPI.requestUploadOTP(session: jwt) { (success, error) in
            DLog("success? \(success) error \(String(describing: error))")
        }
    }
    
    func fetchDevicesEncounteredCount() {
                    guard let persistentContainer =
            EncounterDB.shared.persistentContainer else {
                return
        }
        let managedContext = persistentContainer.viewContext
        let fetchRequest = Encounter.fetchRequestForRecords()
        
        do {
            let devicesEncountered = try managedContext.fetch(fetchRequest)
            let uniqueIDs = Set(devicesEncountered.map { $0.timestamp })
            self.devicesEncounteredLabel.text = String(uniqueIDs.count)
        } catch let error as NSError {
            print("Could not fetch. \(error), \(error.userInfo)")
        }
    }
    
    @objc
    func advertisementSwitchChanged(mySwitch: UISwitch) {
        BluetraceManager.shared.toggleAdvertisement(mySwitch.isOn)
    }
    
    @objc
    func scanningSwitchChanged(mySwitch: UISwitch) {
        BluetraceManager.shared.toggleScanning(mySwitch.isOn)
    }
    
    @IBAction func dumpDBpressed(_ sender: UIButton) {
        var activityItems: [Any] = []
        let localStoreUrl = CovidPersistentContainer.defaultDirectoryURL().appendingPathComponent("tracer", isDirectory: false).appendingPathExtension("sqlite")
        activityItems.append(localStoreUrl)
        let localStoreUrlshm = CovidPersistentContainer.defaultDirectoryURL().appendingPathComponent("tracer", isDirectory: false).appendingPathExtension("sqlite-shm")
        if FileManager.default.fileExists(atPath: localStoreUrlshm.path) {
            activityItems.append(localStoreUrlshm)
        }
        let localStoreUrlwal = CovidPersistentContainer.defaultDirectoryURL().appendingPathComponent("tracer", isDirectory: false).appendingPathExtension("sqlite-wal")
        if FileManager.default.fileExists(atPath: localStoreUrlwal.path) {
            activityItems.append(localStoreUrlwal)
        }
        let activity = UIActivityViewController(activityItems: activityItems, applicationActivities: nil)
        present(activity, animated: true, completion: nil)
    }
    
    @objc
    func clearLogsButtonClicked() {
        guard let persistentContainer =
            EncounterDB.shared.persistentContainer else {
                return
        }
        let managedContext = persistentContainer.viewContext
        let fetchRequest = NSFetchRequest<Encounter>(entityName: "Encounter")
        fetchRequest.includesPropertyValues = false
        do {
            let encounters = try managedContext.fetch(fetchRequest)
            for encounter in encounters {
                managedContext.delete(encounter)
            }
            try managedContext.save()
        } catch {
            print("Could not perform delete. \(error)")
        }
        
        guard let logPersistentContainer =
            BLELogDB.shared.persistentContainer else {
                return
        }
        let logManagedContext = logPersistentContainer.viewContext
        let logFetchRequest = NSFetchRequest<BLELog>(entityName: "BLELog")
        logFetchRequest.includesPropertyValues = false
        do {
            let logs = try logManagedContext.fetch(logFetchRequest)
            for bleLog in logs {
                logManagedContext.delete(bleLog)
            }
            try logManagedContext.save()
        } catch {
            print("Could not perform delete. \(error)")
        }
    }
    
    @IBAction func resetSilentNotificationsCounter(_ sender: Any) {
        UserDefaults.standard.set(0, forKey: "debugSilentNotificationCount")
        silentNotificationsCountLabel.text = "0"
    }
    
    @IBAction func resetMessagesAPILocks(_ sender: Any) {
        UserDefaults.standard.removeObject(forKey: MessageAPI.keyLastApiUpdate)
        UserDefaults.standard.removeObject(forKey: MessageAPI.keyLastVersionChecked)
        messagesAPILastDateLabel.text = "-"
        messagesAPILastVersionLabel.text = "-"
    }
    
    @IBAction func resetDisclaimerUpdateMessage(_ sender: Any) {
        UserDefaults.standard.removeObject(forKey: "latestPolicyUpdateVersionShown")
    }
    
    @IBAction func setReauthenticationNeeded(_ sender: Any) {
        let keychain = KeychainSwift()
        keychain.set("corruptedjwt", forKey: "JWT_TOKEN", withAccess: .accessibleAfterFirstUnlock)
    }
    
    @IBAction func toogleSensorsTapped(_ sender: Any) {
        
        BluetraceManager.shared.toggleScanning(!isSensorOn)
    }
}

extension InfoViewController: SensorDelegate {
    func sensor(_ sensor: SensorType, didUpdateState: SensorState) {
        isSensorOn = didUpdateState == .on
        
        DispatchQueue.main.async {
            if sensor == .BLE {
                switch didUpdateState {
                case .off:
                    self.bleSensorStateLabel.text = "BLE Sensor State: OFF"
                case .on:
                    self.bleSensorStateLabel.text = "BLE Sensor State: ON"
                default:
                    self.bleSensorStateLabel.text = "BLE Sensor State: N/A"
                }
            }
            
            if sensor == .AWAKE {
                switch didUpdateState {
                case .off:
                    self.awakeSensorStateLabel.text = "Awake Sensor State: OFF"
                case .on:
                    self.awakeSensorStateLabel.text = "Awake Sensor State: ON"
                default:
                    self.awakeSensorStateLabel.text = "Awake Sensor State: N/A"
                }
            }
        }
    }
}