mobile-ios/CovidSafe/API/MessageAPI.swift

//
//  MessageAPI.swift
//  CovidSafe
//
//  Copyright © 2020 Australian Government. All rights reserved.
//

import Foundation
import Alamofire
import KeychainSwift

class MessageAPI {
    
    static let keyLastApiUpdate = "keyLastApiUpdate"
    static let keyLastVersionChecked = "keyLastVersionChecked"

    static func getMessagesIfNeeded(completion: @escaping (MessageResponse?, Swift.Error?) -> Void) {
        if shouldGetMessages() {
            guard let token = UserDefaults.standard.string(forKey: "deviceTokenForAPN") else {
                return
            }
            //Get relevat encounter data
            guard let persistentContainer =
                EncounterDB.shared.persistentContainer else {
                    return
            }
            let managedContext = persistentContainer.newBackgroundContext()
            guard let encounterLastWeekRequest = Encounter.fetchEncountersInLast(days: 7) else {
                return
            }
            
            do {
                //fetch last week encounters count
                let weekEncounters = try managedContext.count(for: encounterLastWeekRequest)
                let healthcheck = (BluetraceManager.shared.isBluetoothOn() &&
                    BluetraceManager.shared.isBluetoothAuthorized() &&
                    weekEncounters > 0 ? healthCheckParamValue.OK : healthCheckParamValue.POSSIBLE_ERROR)
                
                // Make API call to get messages
                let messageRequest = MessageRequest(remotePushToken: token, healthcheck: healthcheck)
                getMessages(msgRequest: messageRequest, completion: completion)

            } catch let error as NSError {
                DLog("Could not fetch encounter(s) from db. \(error), \(error.userInfo)")
            }
        }
    }
    
    private static func shouldGetMessages() -> Bool {
        let lastChecked = UserDefaults.standard.double(forKey: keyLastApiUpdate)
        let versionChecked = UserDefaults.standard.integer(forKey: keyLastVersionChecked)
        
        var shouldGetMessages = true
        
        let calendar = NSCalendar.current
        let currentDate = calendar.startOfDay(for: Date())
        
        // if the current version is newer than the last version checked, allow messages call
        if let currVersionStr = Bundle.main.version, let currVersion = Int(currVersionStr), currVersion > versionChecked {
            return true
        }
        
        if lastChecked > 0 {
            let lastCheckedDate = Date(timeIntervalSince1970: lastChecked)
            let components = calendar.dateComponents([.day], from: lastCheckedDate, to: currentDate)
            
            if let numDays = components.day {
                shouldGetMessages = numDays > 0
            }
        }
        
        return shouldGetMessages
    }
    
    private static func getMessages(msgRequest: MessageRequest,
                            completion: @escaping (MessageResponse?, Swift.Error?) -> Void) {
        let keychain = KeychainSwift()
        guard let apiHost = PlistHelper.getvalueFromInfoPlist(withKey: "API_Host", plistName: "CovidSafe-config") else {
            return
        }
        
        guard let token = keychain.get("JWT_TOKEN") else {
            completion(nil, nil)
            return
        }
        let headers: HTTPHeaders = [
            "Authorization": "Bearer \(token)"
        ]
        
        var params: [String : Any] = [
            "os" : "ios-\(UIDevice.current.systemVersion)",
            "healthcheck" : msgRequest.healthcheck.rawValue,
            "preferredlanguages": Locale.preferredLanguages
            ]

        if let buildString = Bundle.main.version {
            params["appversion"] = "\(buildString)"
        }
        if let remoteToken = msgRequest.remotePushToken {
            params["token"] = remoteToken
        }
        CovidNetworking.shared.session.request("\(apiHost)/messages",
            method: .get,
            parameters: params,
            headers: headers
        ).validate().responseDecodable(of: MessageResponse.self) { (response) in
                switch response.result {
                case .success:
                    guard let messageResponse = response.value else { return }

                    // save successful timestamp
                    let calendar = NSCalendar.current
                    let currentDate = calendar.startOfDay(for: Date())
                    UserDefaults.standard.set(currentDate.timeIntervalSince1970, forKey: keyLastApiUpdate)
                    UserDefaults.standard.set(Bundle.main.version, forKey: keyLastVersionChecked)

                    completion(messageResponse, nil)
                case let .failure(error):
                    completion(nil, error)
                }
        }
    }
}

enum healthCheckParamValue: String {
    case OK = "OK"
    case POSSIBLE_ERROR = "POSSIBLE_ERROR"
    case ERROR = "ERROR"
}

struct MessageRequest {
    var remotePushToken: String?
    var healthcheck: healthCheckParamValue
}

struct MessageResponse: Decodable {
    let messages: [Message]?
    let forceappupgrade: Bool
    
    enum CodingKeys: String, CodingKey {
        case messages
        case forceappupgrade
    }
}

struct Message: Decodable {
    let title: String?
    let body: String?
    let destination: String?
    
    enum CodingKeys: String, CodingKey {
        case title
        case body
        case destination
    }
}