purchases-ios/Sources/FoundationExtensions/Data+Extensions.swift

//
//  Copyright RevenueCat Inc. All Rights Reserved.
//
//  Licensed under the MIT License (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at
//
//      https://opensource.org/licenses/MIT
//
//  Data+Extensions.swift
//  Purchases
//
//  Created by Josh Holtz on 6/28/21.
//

import CommonCrypto
import CryptoKit
import Foundation

extension Data {

    var asString: String {
        return Self.hexString(
            self
                .lazy
                .map { $0 } // Extract byte
                .makeIterator()
        )
    }

    /// - Returns: `UUID` from the first 16 bytes of the underlying data.
    var uuid: UUID? {
        /// This implementation is equivalent to `return NSUUID(uuidBytes: [UInt8](self)) as UUID`
        /// but ensures that the `Data` isn't unnecessarily copied in memory.
        return self.dataWithMinLengthForUUID.withUnsafeBytes {
            guard let baseAddress = $0.bindMemory(to: UInt8.self).baseAddress else {
                return nil
            }

            return NSUUID(uuidBytes: baseAddress) as UUID
        }
    }

    /// - Returns: a string representing a fetch token.
    var asFetchToken: String {
        return self.base64EncodedString()
    }

    /// - Returns: a hash representation of the underlying bytes, using SHA256.
    var hashString: String {
        var sha256 = SHA256()
        return self.hashString(with: &sha256)
    }

    /// - Returns: the SHA1 hash of the underlying bytes.
    var sha1: Data {
        var sha1 = Insecure.SHA1()
        return self.hash(with: &sha1)
    }

    var sha256: Data {
        var sha256 = SHA256()
        return self.hash(with: &sha256)
    }

    var sha384: Data {
        var sha384 = SHA384()
        return self.hash(with: &sha384)
    }

    var sha512: Data {
        var sha512 = SHA512()
        return self.hash(with: &sha512)
    }

    var sha1String: String {
        var sha1 = Insecure.SHA1()
        return self.hashString(with: &sha1)
    }

    var sha256String: String {
        var sha256 = SHA256()
        return self.hashString(with: &sha256)
    }

    var sha384String: String {
        var sha384 = SHA384()
        return self.hashString(with: &sha384)
    }

    var sha512String: String {
        var sha512 = SHA512()
        return self.hashString(with: &sha512)
    }

    var md5String: String {
        var md5 = Insecure.MD5()
        return self.hashString(with: &md5)
    }

    fileprivate static func hexString(_ iterator: Array<UInt8>.Iterator) -> String {
        return iterator
            .lazy
            .map { String(format: "%02x", $0) }
            .joined()
    }

    private var dataWithMinLengthForUUID: Data {
        let uuidMemorySize = MemoryLayout<UUID>.size
        guard self.count >= uuidMemorySize else {
            return self + Data(count: uuidMemorySize - self.count)
        }
        return self
    }
}

extension Data {

    static func randomNonce() -> Data {
        return Data(ChaChaPoly.Nonce())
    }

    static let nonceLength = 12

}

// MARK: - Hashing

extension HashFunction {

    func toString() -> String {
        return Data.hexString(self.finalize().makeIterator())
    }

}

private extension Data {

    func hashString<T: HashFunction>(with digest: inout T) -> String {
        digest.update(data: self)
        return digest.toString()
    }

    func hash<T: HashFunction>(with digest: inout T) -> Data {
        digest.update(data: self)

        return Data(digest.finalize())
    }

}