#include "options.hpp"

int Options::validateStringOption( int argc, char** argv, int argumentIndex ) {
    std::string argument = argv[argumentIndex];
    unsigned int index = 0;
    while( index < validArgumentCount ) {
        std::string& check = validStringArguments[index];
        for( unsigned int i=0;i<check.length();i++ ) {
            if ( check[i] != argument[i+2] ) {
                break;
            }
            if ( i == check.length()-1 ) {
                if ( !isFlagArgument[index] && argument.find("=") == std::string::npos ) {
                    throw new std::invalid_argument("Expected `=` after " + arguments[i]);
                }
                if ( argument[i+3] != '=' ) {
                    break;
                }
                return parseStringOption( argc, argv, argumentIndex, index );
            }
        }
        index++;
    }
    throw new std::invalid_argument("Invalid argument " + argument);
    return 0;
}

int Options::parseCharOption( int argc, char** argv, int argumentIndex, int validIndex ) {
    std::string argument = argv[argumentIndex];
    // If we're a flag, we take no arguments, nor do we allow = signs or whatever.
    if ( isFlagArgument[validIndex] ) {
        if ( argument != std::string()+"-"+validCharArguments[validIndex] ) {
            throw new std::invalid_argument( std::string()+"Unexpected characters around flag `" + argument + "`." );
            return 0;
        } else {
            arguments.push_back( std::string()+argument[1] );
            values.push_back("");
            return 1;
        }
    }
    arguments.push_back( std::string()+argument[1] );
    // If they supplied the parameters with spaces
    if ( argument == std::string()+"-"+validCharArguments[validIndex] ) {
        values.push_back(argv[argumentIndex+1]);
        return 2;
    }
    // If they didn't supply the parameters with spaces
    if ( argument[2] == '=' ) {
        values.push_back(argument.substr(3));
        return 1;
    }
    values.push_back(argument.substr(2));
    return 1;
}

int Options::parseStringOption( int argc, char** argv, int argumentIndex, int validIndex ) {
    std::string argument = argv[argumentIndex];
    if ( isFlagArgument[validIndex] ) {
        arguments.push_back( argument.substr(2) );
        values.push_back("");
        return 1;
    }
    arguments.push_back( argument.substr(2,argument.find_first_of('=')) );
    values.push_back(argument.substr(argument.find_first_of('=')));
    return 1;
}

int Options::validateCharOption( int argc, char** argv, int argumentIndex ) {
    std::string argument = argv[argumentIndex];
    unsigned int index = 0;
    while( index < validArgumentCount ) {
        char check = validCharArguments[index];
        if ( argument.length() < 2 ) {
            continue;
        }
        if ( check == argument[1] && ( argument.find("=") == 2 || argument.find('=') == std::string::npos ) && argument[0] == '-' ) {
            return parseCharOption( argc, argv, argumentIndex, index );
        }
        index++;
    }
    throw new std::invalid_argument("Invalid argument `" + argument + "`.");
    return 0;
}

void Options::validate( int argc, char** argv ) {
    for ( int i=1;i<argc;) {
        std::string argument = argv[i];
        if ( argument[0] != '-' ) {
            floatingValues.push_back( argument );
            if ( floatingValues.size() > maxFloatingValues ) {
                throw new std::invalid_argument("Unexpected floating value `" + argument + "`. Forget to specify an option?" );
            }
        }
        if ( argument[0] == '-' && argument[1] == '-' ) {
            i += validateStringOption( argc, argv, i );
            continue;
        }
        i += validateCharOption( argc, argv, i );
    }
}

Options::Options( int argc, char** argv ) {
    validate( argc, argv );
}

bool Options::getFloat( std::string name, char namec, float& found ) {
    for( unsigned int i=0;i<arguments.size();i++ ) {
        if ( arguments[i] == name || arguments[i] == std::string("")+namec ) {
            std::string::size_type sz;
            float retvar;
            try {
                retvar = std::stof(values[i],&sz);
            } catch ( ... ) {
                throw new std::invalid_argument("Unable to parse `" + arguments[i] + "`'s value `" + values[i] + "` as a float.");
            }
            if ( sz != values[i].length() ) {
                throw new std::invalid_argument("Unable to parse `" + arguments[i] + "`'s value `" + values[i] + "` as a float.");
            }
            found = retvar;
            return true;
        }
    }
    return false;
}

bool Options::getInt( std::string name, char namec, int& found ) {
    for( unsigned int i=0;i<arguments.size();i++ ) {
        if ( arguments[i] == name || arguments[i] == std::string("")+namec ) {
            if ( arguments[i].size() > 1 && arguments[i].find("=") == std::string::npos ) {
                throw new std::invalid_argument("Expected `=` after " + arguments[i]);
            }
            std::string::size_type sz;
            float retvar;
            try {
                retvar = std::stoi(values[i],&sz);
            } catch ( ... ) {
                throw new std::invalid_argument("Unable to parse " + arguments[i] + "'s value " + values[i] + " as an integer.");
            }
            if ( sz != values[i].length() ) {
                throw new std::invalid_argument("Unable to parse " + arguments[i] + "'s value " + values[i] + " as an integer.");
            }
            found = retvar;
            return true;
        }
    }
    return false;
}

bool Options::getString( std::string name, char namec, std::string& found ) {
    for( unsigned int i=0;i<arguments.size();i++ ) {
        if ( arguments[i] == name || arguments[i] == std::string("")+namec ) {
            found = values[i];
            return true;
        }
    }
    return false;
}

bool Options::getColor( std::string name, char namec, glm::vec4& found ) {
    for( unsigned int i=0;i<arguments.size();i++ ) {
        if ( arguments[i] == name || arguments[i] == std::string("")+namec ) {
            std::string::size_type sz;
            std::string value = values[i];
            try {
                found[0] = std::stof(value,&sz);
                value = value.substr(sz+1);
                found[1] = std::stof(value,&sz);
                value = value.substr(sz+1);
                found[2] = std::stof(value,&sz);
                if ( value.size() != sz ) {
                    value = value.substr(sz+1);
                    found[3] = std::stof(value,&sz);
                    if ( value.size() != sz ) {
                        throw "dur";
                    }
                } else {
                    found[3] = 1;
                }
            } catch ( ... ) {
                throw new std::invalid_argument("Unable to parse `" + arguments[i] + "`'s value `" + values[i] + "` as a color. Should be in the format r,g,b or r,g,b,a. Like 1,1,1,1.");
            }
            return true;
        }
    }
    return false;
}

bool Options::getBool( std::string name, char namec, bool& found ) {
    for( unsigned int i=0;i<arguments.size();i++ ) {
        if ( arguments[i] == name || arguments[i] == std::string("")+namec ) {
            if ( values[i] != "" ) {
                throw new std::invalid_argument("Unexpected value `" + values[i] + "` for flag argument `" + arguments[i] + "`.");
            }
            found = true;
            return true;
        }
    }
    return false;
}