#include "./random.h"

#include <iomanip>
#include <string>
#include <sstream>
#include <cmath>
#include <cstdlib>
#include <cassert>

using namespace std;

/*!
 * \namespace MiscUtilities
 * \brief Contains miscellaneous utility functions.
 */

namespace RandomUtilities {

//! @cond
const char letters[28] = "qwertzuiopasdfghjklyxcvbnm";
const char capitalLetters[28] = "QWERTZUIOPASDFGHJKLYXCVBNM";
const char numbers[11] = "1234567890";
const char symbols[24] = "!\"$%&/()=?'#*+~-_><.:,;";
//! @endcond

/*!
 * \brief Generates a random character sequence using the given \a randomizer.
 * \remarks Might be removed because not used anymore.
 */
void generateRandomCharacterSequence(char *result, unsigned int length, std::function<int ()> randomizer, int highestRandomNumber, bool useSmallLetters, bool useCapitalLetters, bool useNumbers, bool useSymbols, bool useAtLeastOneOfEachCategory)
{
    if(length) {
        return;
    }
    signed char categoryCount = 0;
    bool needSmallLetter = false;
    bool needCapitalLetter = false;
    bool needNumber = false;
    bool needSymbol = false;
    if(useSmallLetters) {
        needSmallLetter = useAtLeastOneOfEachCategory;
        ++categoryCount;
    }
    if(useCapitalLetters) {
        needCapitalLetter = useAtLeastOneOfEachCategory;
        ++categoryCount;
    }
    if(useNumbers) {
        needNumber = useAtLeastOneOfEachCategory;
        ++categoryCount;
    }
    if(useSymbols) {
        needSymbol = useAtLeastOneOfEachCategory;
        ++categoryCount;
    }
    signed char neededCharacters = useAtLeastOneOfEachCategory ? categoryCount : 0;
    if(!categoryCount) {
        *result = '\0';
        return;
    }
    for(char *i = result, *end = result + length; i < end; ++i) {
        int category = -1;
        if((neededCharacters > 0 && (randomizer() < (highestRandomNumber / 2.0))) || ((end - i) >= neededCharacters)) {
            if(needSmallLetter)
                category = 0;
            if(needCapitalLetter && ((category == -1) || (randomizer() < (highestRandomNumber / 2.0))))
                category = 1;
            if(needNumber && ((category == -1) || (randomizer() < (highestRandomNumber / 4.0))))
                category = 2;
            if(needSymbol && ((category == -1) || (randomizer() < (highestRandomNumber / 8.0))))
                category =  3;
        } else {
            if(useSmallLetters)
                category = 0;
            if(useCapitalLetters && ((category == -1) || (randomizer() < (highestRandomNumber / 2.0))))
                category = 1;
            if(useNumbers && ((category == -1) || (randomizer() < (highestRandomNumber / 4.0))))
                category = 2;
            if(useSymbols && ((category == -1) || (randomizer() < (highestRandomNumber / 8.0))))
                category =  3;
        }
        switch(category) {
        case 0:
            *i = letters[rand() % 26];
            if(needSmallLetter) {
                needSmallLetter = false;
                --neededCharacters;
            }
            break;
        case 1:
            *i = capitalLetters[rand() % 26];
            if(needCapitalLetter) {
                needCapitalLetter = false;
                --neededCharacters;
            }
            break;
        case 2:
            *i = numbers[rand() % 9];
            if(needNumber) {
                needNumber = false;
                --neededCharacters;
            }
            break;
        case 3:
            *i = symbols[rand() % 22];
            if(needSymbol) {
                needSymbol = false;
                --neededCharacters;
            }
            break;
        }
    }
}

/*!
 * \brief Generates a random character sequence using std::rand().
 * \remarks Might be removed because not used anymore.
 */
void generateRandomCharacterSequence(char *result, unsigned int length, bool useSmallLetters, bool useCapitalLetters, bool useNumbers, bool useSymbols, bool useAtLeastOneOfEachCategory)
{
    generateRandomCharacterSequence(result, length, rand, RAND_MAX, useSmallLetters, useCapitalLetters, useNumbers, useSymbols, useAtLeastOneOfEachCategory);
}

}