learncrypto/source/cipher.d

module cipher;

private import std.range : isInputRange, ElementType;
private import salsa20;
private import chacha20;
import std.stdio;

public:

enum Cipher
{
    salsa20,
    chacha20
}

auto cipherFunction(R)(R range, ubyte[32] key, ubyte[8] nonce, Cipher cipher)
    if(isInputRange!R && is(ElementType!R : ubyte))
{
    final switch(cipher)
    {
        case Cipher.salsa20:
            return range.salsa20Cipher(key, nonce);

        case Cipher.chacha20:
            return range.salsa20Cipher(key, nonce);
    }
}