#include <cstdlib>
#include <string>
#include <ctime>
#include <filesystem>
#include "rainbow.h"

int stoi(const std::string& str, int& i, int mode = 0) {
    int result = 0;
    int mult = 1;
    for (; i < str.length(); i++) {
        if (mode && str[i] == '.') break;
        if (str[i] == '-') mult = -1;
        else if (str[i] >= '0' && str[i] <= '9') result = result * 10 + mult*(str[i] - '0');
    }
    return result;
}

double stof(const std::string& str) {
    int i = 0;
    double whole = stoi(str, i, 1);
    double frac = stoi(str, i);
    for (; frac > 1; frac /= 10);
    return whole + frac;
}

void exportEnv(const std::string& name, double value) {
    printf(" %s=%lf", name.c_str(), value);
}

int getEnv(double& v, const std::string& name) {
    if (char* env = std::getenv(name.c_str())) sscanf(env, "%lf", &v);//v = stof(env);
    else return 1;
    return 0;
}

int randint() {
    std::srand(std::time(nullptr));
    int n = std::rand() & 15;
    n*=3;
    n+=10;
    return n;
}

int getAllEnvs(rainbow& r, int& count){
    if (char* env = std::getenv("RAINBOWPSC")) {sscanf(env, "%d", &count);}
    else count = 0;
    return getEnv(r.s  , "RAINBOWPSS")||getEnv(r.c.r, "RAINBOWPSR")
    ||getEnv(r.c.g, "RAINBOWPSG")||getEnv(r.c.b, "RAINBOWPSB");
}

const std::string emotes[] =
{":)", ":3", "<3",
":/", ":D", ":(",
":|", ";)", ":[",
":P", ":O", ":>"};


std::string makePrompt(int& count) {
    std::srand(std::time(nullptr)*count);
    std::string path = std::filesystem::current_path();
    return std::to_string(count).append(" ")
    .append(emotes[std::rand() % 12]).append(" ")
    .append(path.substr(path.find_last_of('/')+1)).append("$");
}



int main(int argc, char* argv[]) {
    /*Bash prompt should:
        - show current folder
        - count how many commands were executed
        - show random emoticon
        - display in rainbow colors (choose rainbow length with env variable)
    */

    if (argc > 1 && argv[1][0]=='h') {
        printf("To use this program\n");
        printf("add the following lines to your .bashrc:\n");
        printf("export PROMPT_COMMAND='`%s`'\n", argv[0]);
        printf("PS1='`%s \"!\"`'\n", argv[0]);
        return 0;
    }
    rainbow r;
    int count = 0;
    int err = getAllEnvs(r, count);
    if (argc > 1 && argv[1][0] == '!') {
        std::string outbuff = "\033[38;2;255;0;180m<3\033[0m ";
        std::string prompt = makePrompt(count);
        if (err) {
            outbuff.append(prompt);
            printf("%s", outbuff.c_str());
            return 0;
        }
        // r.print2d(prompt, 40+count%20-count%13+count%7);
        r.s=5.0*255.0/(double)(40+count%20-count%13+count%7);
        int i = 0;
        for (; i < prompt.length(); i++) {
            outbuff.append("\033[38;2;")
            .append(r.c.str()).append("m")
            .append(prompt.substr(i, 1))
            .append("\033[0m");
            r.next();
        }
        outbuff.append(" ");
        for (const char* c = outbuff.c_str(); *c; c++) {
            if (*c == 10 || *c == 13 || *c == 0) continue;
            fwrite(c, 1, 1, stdout);
        }
        return 0;
    }

    int i = 0;
    if (argc > 1 && argv[1][0] == 's') r.init(15+stoi(argv[1], i));
    if (err || !(int)r.s || (!(int)r.c.r && !(int)r.c.g && !(int)r.c.b))
        r.init(randint());
    else r.next();

    printf("export");
    exportEnv("RAINBOWPSS", r.s);
    exportEnv("RAINBOWPSR", r.c.r);
    exportEnv("RAINBOWPSG", r.c.g);
    exportEnv("RAINBOWPSB", r.c.b);
    exportEnv("RAINBOWPSC", ++count);
    printf("\n");

    return 0;
}