109
            

             /*
 * File: prompt.cpp
 * This program is used to generate a bash prompt with rainbow colors.
 * It can also be used to generate the necessary environment variables
 *
 */

#include <string>
#include <chrono>
#include <cstdio>
#include <iostream>
#include <ctime>
#include <sys/ioctl.h>
#include <unistd.h>
#include <vector>
#include <exception>

#include "rainbow.hpp"

typedef std::string str;

int randint(int max) {
    srand(std::chrono::duration_cast<std::chrono::milliseconds>(
    std::chrono::high_resolution_clock::now().time_since_epoch()).count());
    return rand() % max;
}

const str emotes[] = {":)", ":3", ";D",":]", ":D", "xD",";3", ";)", "=)",":P", ":O", ":b"};

str emote() {
    return emotes[randint(12)];
}

str intpad(int n, int len) {
    str s = std::to_string(n);
    while (s.length() < len) s = "0" + s;
    return s;
}

int str2int(str &s) {
    int n = 0;
    while (!s.empty()) {
        if (s[0] < '0' || s[0] > '9') {
            s.erase(0, 1);
            break;
        }
        n = n * 10 + s[0] - '0';
        s.erase(0, 1);
    }
    return n;
}

str time_hhmm() {
    std::chrono::time_point now = std::chrono::system_clock::now();
    time_t in_time_t = std::chrono::system_clock::to_time_t(now);
    auto tt = std::localtime(&in_time_t);
    return intpad((int)(tt->tm_hour),2) + ":" + intpad((int)(tt->tm_min),2);
}

str docmd(std::string inputted) {
    FILE* file = popen(inputted.c_str(), "r");
    char buff[1024];
    int n = fread(buff, 1, sizeof(buff), file);
    pclose(file);
    str r = std::string(buff, n);
    if (r.back() == '\n') r.pop_back();
    return r;
}

str envorcmd(str env, str cmd) {
    str ret = "";
    try { ret = std::string(getenv(env.c_str())); }
    catch (std::exception e) { ret = docmd(cmd); }
    return ret;
}


int main(int argc, char* argv[]) {
    struct winsize term;
    if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &term) == -1) return 1;

    rainbow r;

    try {
        str rainenv = std::string(getenv("RAINBOW"));
        for(int i = 0; i < 3; i++) {
            r.c.indexAdd(i, str2int(rainenv));
        }
        r.s = str2int(rainenv);
        r.next();
    } catch (std::exception e) {
        r.init((int)term.ws_row-10+randint(10));
    }

    str output="";
    str user = envorcmd("USER", "whoami");
    str ip = docmd("hostname -I | awk -F '.' ' { for(i=1;i<5;i++){printf(\"%.3d\", $i);}; } ' ");
    for(int i=0;i<user.length();i++){
        output+="\\e[1;38;5;"+ip.substr((i%4)*3,3)+"m"+user.substr(i,1)+"\\e[0m";
    }
    printf("%s", output.c_str());
    /*str pwd;
    try { pwd = std::string(getenv("PWD")); }
    catch (std::exception e) { pwd = ""; }
    if (user.back()=='\n')user.pop_back();
    output += " ";
    output += pwd;*/
    return 0;
}