CryptoSwift/CryptoSwift/Cipher.swift

//
//  Cipher.swift
//  CryptoSwift
//
//  Created by Marcin Krzyzanowski on 30/08/14.
//  Copyright (c) 2014 Marcin Krzyzanowski. All rights reserved.
//

import Foundation

public enum Cipher {
    case ChaCha20(key: NSData, iv: NSData)
    case AES(key: NSData, iv: NSData)
    
    public func encrypt(message: NSData) -> NSData? {
        switch (self) {
            case .ChaCha20(let key, let iv):
                var chacha = CryptoSwift.ChaCha20(key: key, iv: iv)
                return chacha?.encrypt(message)
            case .AES(let key, let iv):
                var aes = CryptoSwift.AES(key: key, iv: iv)
                return aes?.encrypt(message)
            
        }
    }
    
    public func decrypt(message: NSData) -> NSData? {
        switch (self) {
            case .ChaCha20(let key, let iv):
                var chacha = CryptoSwift.ChaCha20(key: key, iv: iv);
                return chacha?.decrypt(message)
            case .AES(let key, let iv):
                var aes = CryptoSwift.AES(key: key, iv: iv);
                return aes?.decrypt(message)
        }
    }

}