#include <string>
#include <csignal>
#include <sys/ioctl.h>
#include <chrono>

int w, h;

struct meminfo{
    int total = 0;
    int free = 0;
    int available = 0;
    void loadmem();
    int used(){return total-free;}
    int taken(){return total-available;}
    int percent(){if (total == 0){return 0;} else return 100*taken()/total;}
    int bufpercent(){if (total == 0){return 0;} else return 100*used()/total;}
};
void meminfo::loadmem() {
    FILE* file = fopen("/proc/meminfo", "r");
    int line = 0, n = 0;
    char c[84];
    fread(&c, 1, 84, file);
    fclose(file);
    for (int i = 0; i < 84; i++) {
        if (c[i] > 47 && c[i] < 58) ((n *= 10) += c[i] - 48);
        else if (c[i] == 10) {
            n /= 1000;
            switch (line) {
                case 0: total = n;
                case 1: free = n;
                case 2: available = n;
            }
            n = 0;
            line++;
        }
    }
}

struct cpuinfo{
    int temp = 0;
    int loadavg = 0;
    void loadcpu();
    int getloadavg();
    int gettemp();
};
int cpuinfo::getloadavg() {
    long double a[4], b[4];
    FILE *fp;
    char dump[50];
    fp = fopen("/proc/stat","r");
    fscanf(fp,"%*s %Lf %Lf %Lf %Lf",&a[0],&a[1],&a[2],&a[3]);
    fclose(fp);
    usleep(1000000);
    fp = fopen("/proc/stat","r");
    fscanf(fp,"%*s %Lf %Lf %Lf %Lf",&b[0],&b[1],&b[2],&b[3]);
    fclose(fp);
    return 100 * ((b[0]+b[1]+b[2]) - (a[0]+a[1]+a[2])) / ((b[0]+b[1]+b[2]+b[3]) - (a[0]+a[1]+a[2]+a[3]));
}

struct cpufile{
    std::string filepath;
    std::string filename;
    std::string comp;
    std::string usepath(int i) {
        std::string s = filepath;
        s.append(std::to_string(i));
        s.append(filename);
        return s;
    }
};

int findcpu(cpufile& ct)  {
    FILE* file;
    char buf[ct.comp.length()+1];
    int i = 0;
    if (ct.filename == "_label") i = 1;
    for (;;i++) {
        file = fopen(ct.usepath(i).c_str(), "r");
        if (file == NULL) return -1;
        fread(&buf, 1, ct.comp.length(), file);
        fclose(file);
        if (std::string(buf).substr(0, ct.comp.length()) == ct.comp) break;
    }
    return i;
}

int cpuinfo::gettemp() {
    FILE* file;
    char buf[2];
    cpufile ct = {"/sys/class/thermal/thermal_zone", "/type", "x86_pkg_temp"};
    int fileNo = findcpu(ct);
    if (fileNo == -1) {
        ct = {"/sys/class/hwmon/hwmon", "/name", "k10temp"};
        fileNo = findcpu(ct);
        if (fileNo==-1) return 0;
        ct.filename = "_label";
        ct.filepath.append(std::to_string(fileNo)).append("/temp");
        ct.comp = "Tdie";
        fileNo = findcpu(ct);
        if (fileNo==-1) return 0;
        ct.filename = "_input";
    } else {
        ct.filename = "/temp";
    }
    file = fopen(ct.usepath(fileNo).c_str(), "r");
    if (file == NULL) return 0;
    fread(&buf, 1, 2, file);
    fclose(file);
    return (buf[0]-48)*10 + buf[1] - 48;
}

void cpuinfo::loadcpu(){
    loadavg = getloadavg();
    temp = gettemp();
}

void termsize(){
    struct winsize size;
    ioctl(STDOUT_FILENO, TIOCGWINSZ, &size);
    w = size.ws_col;
    h = size.ws_row;
}

void signal_callback_handler(int _) {
    printf("\033[?1049l\033[?25h");
    exit(0);
}

std::string padint(int n, int len){
    std::string s = std::to_string(n);
    for (; s.length() < len; s = " " + s);
    return s;
}

int graphscale(int n) {return h-1-(n*(h-6)/100);}

void plot(int x, int y, int color, std::string* s){
    if (color > 7 || color < 1) color = 7;
    if (x%2 == 0) color+=8;
    std::string print = std::string(";").append(std::to_string(x)).append("H\033[48;5;").append(std::to_string(color)).append("m \033[0m");
    for (;y<h;y++) s->append("\033[").append(std::to_string(y)).append(print);
}

int helpmenu(int argc, char* argv[]) {
    if (argc < 2) return 0;
    if (!(std::string(argv[1])=="-h" || std::string(argv[1])=="--help")) return 0;
    printf("\nUsage: %s [OPTION]\n\n", argv[0]);
    printf("  -h, --help\t\tDisplay this help message.\n");
    printf("      --log <file>\tLog to file (appends). Log frequency: 1 second. Format:\n");
    printf("            \t\tunix, cpu usage, temp, ram taken, buffer, avialbale, free\n\n");
    return 1;
}

int logarg(int argc, char* argv[], std::string& logdir) {
    if (argc < 3) return 0;
    if (std::string(argv[1]) != "--log") return 0;
    logdir = std::string(argv[2]);
    return 1;
}

void log(std::string& logdir, meminfo& mem, cpuinfo& cpu) {
    std::string logbuf;
    logbuf.clear();
    logbuf.append(std::to_string(std::chrono::duration_cast<std::chrono::seconds>(std::chrono::system_clock::now().time_since_epoch()).count()))
    .append(",").append(std::to_string(cpu.loadavg))
    .append(",").append(std::to_string(cpu.temp))
    .append(",").append(std::to_string(mem.taken()))
    .append(",").append(std::to_string(mem.used()-mem.taken()))
    .append(",").append(std::to_string(mem.available))
    .append(",").append(std::to_string(mem.free))
    .append("\n");
    FILE* f = fopen(logdir.c_str(), "a");
    fputs(logbuf.c_str(), f);
    fclose(f);
}

int main(int argc, char* argv[]){
    if (helpmenu(argc, argv)) return 0;
    std::string logdir;
    int dolog = logarg(argc, argv, logdir);
    meminfo mem;
    cpuinfo cpu;
    int cpugraph[150] = {0};
    int ramgraph[150] = {0};
    int bufgraph[150] = {0};
    int i;
    std::string bg;
    std::string graphnum;
    std::string tmp;
    std::string outbuf;
    outbuf.reserve(1000000);
    printf("\033[?25l\033[?1049h\033[2J");
    signal(SIGINT, signal_callback_handler);
    for (;;) {
        mem.loadmem();
        cpu.loadcpu(); // contains usleep(1000000)
        termsize();
        for (i = 150; i > 0; i--) {
            cpugraph[i] = cpugraph[i-1];
            ramgraph[i] = ramgraph[i-1];
            bufgraph[i] = bufgraph[i-1];
        }
        cpugraph[0] = cpu.loadavg;
        ramgraph[0] = mem.percent();
        bufgraph[0] = mem.bufpercent();

        if (dolog) log(logdir, mem, cpu);


        if (w < 60 || h < 10) {
            printf("\033[2J\033[1;1HWindow too small\033[2;1HMinimum size: 60x10\n");
            continue;
        }
        if (w > 300) w = 300;
        if (h > 106) h = 106;

        outbuf.append(std::string("\033[2J")
        .append("\033[2;3HLoad: ").append(padint(cpu.loadavg, 3))
        .append("% Memory:\033[3;3HTemp: ").append(padint(cpu.temp, 3)).append("C")
        .append("\033[2;22HUsed by apps: ").append(padint(mem.taken(), 5))
        .append(" (" + std::to_string(mem.percent())).append("%)\033[2;48HUsed: ")
        .append(padint(mem.used(), 5)).append("\033[3;22HCan be freed: ").append(padint(mem.available, 5))
        .append("\033[3;48HFree: ").append(padint(mem.free, 5)));

        for (bg = ";3H\033[48;5;235m"; bg.length() < w-4+14; bg.append(" "));
        bg.append("\033[0m");
        for (i = 5; i < h; i++) {
            outbuf.append("\033[").append(std::to_string(i)).append(bg);
        }
        outbuf.append("\0");

        for (i = 3; i < w/2-2+w%2; i++) {
            plot(i, graphscale(cpugraph[w/2+1-i-3]), 2, &outbuf);
            plot(i+w/2+1, graphscale(bufgraph[w/2+1-i-3]), 3, &outbuf);
            plot(i+w/2+1, graphscale(ramgraph[w/2+1-i-3]), 1, &outbuf);
        }
        outbuf.append("\0");

        tmp = ";" + std::to_string(w/2-2) + "H -";
        for (i = 5; i < h; i++) {
            outbuf.append("\033[")
            .append(std::to_string(i))
            .append(tmp)
            .append(padint(100-(i-5)*100/(h-6), 3)).append( + "- ");
        }
        printf("%s\n", outbuf.c_str());
        outbuf.clear();
    }
    return 0;
}