Obfuscator.swift

//
//  Obfuscator.swift
//  SwiftObfuscatorExample
//
//  Created by Mathieu White on 2016-07-03.
//  Copyright © 2016 Mathieu White. All rights reserved.
//
import Foundation

class Obfuscator: AnyObject {
    
    // MARK: - Variables
    
    /// The salt used to obfuscate and reveal the string.
    private var salt: String = ""
    
    
    // MARK: - Initialization
    
    init(withSalt salt: String) {
        self.salt = salt
    }
    
    
    // MARK: - Instance Methods
    
    /**
     This method obfuscates the string passed in using the salt
     that was used when the Obfuscator was initialized.
     
     - parameter string: the string to obfuscate
     
     - returns: the obfuscated string in a byte array
     */
    func bytesByObfuscatingString(string: String) -> [UInt8] {
        let text = [UInt8](string.utf8)
        let cipher = [UInt8](self.salt.utf8)
        let length = cipher.count
        
        var encrypted = [UInt8]()
        var count = 0
        for t in text.enumerated() {
            encrypted.append(t.element ^ cipher[count % length])
            count += 1
        }
        
        #if DEVELOPMENT
            print("Salt used: \(self.salt)\n")
            print("Swift Code:\n************")
            print("// Original \"\(string)\"")
            print("let key: [UInt8] = \(encrypted)\n")
        #endif
        
        return encrypted
    }
    
    /**
     This method reveals the original string from the obfuscated
     byte array passed in. The salt must be the same as the one
     used to encrypt it in the first place.
     
     - parameter key: the byte array to reveal
     
     - returns: the original string
     */
    func reveal(key: [UInt8]) -> String {
        let cipher = [UInt8](self.salt.utf8)
        let length = cipher.count
        
        var decrypted = [UInt8]()
        
        var count = 0
        for k in key.enumerated() {
            decrypted.append(k.element ^ cipher[count % length])
            count += 1
        }
        
        return String(bytes: decrypted, encoding: String.Encoding.utf8)!
    }
}