#include <stdio.h>

#include <stdlib.h>

#include <math.h>

#include <sys/types.h>

#include <sys/stat.h>

#include <fcntl.h>

#include <assert.h>

#include <limits.h>

#include <unistd.h>

#include <iwlib.h>

#include "config.h"

char *eights[] = { "ø", "▁", "▂", "▃", "▄", "▅", "▆","▇", "█" };

char *color_stop;

char *color_red;

char *color_orange;

char *color_green;

char *color_none;

char *color(int low,int mid,int rev,int value) {

    if (value>mid) {

        return rev ? color_red : color_green;

    } else if (value>low) {

        return color_orange;

    }

    return rev ? color_green : color_red;

}

int main(int argc, char *argv[]) {

    if (argc<=1) {

        fprintf(stderr, "Usage: %s <wlan0> [term|pango]\n", argv[0]);

        fflush(stderr);

        return 1;

    }

    int skfd;

    if((skfd = iw_sockets_open()) < 0) {

        perror("socket");

        exit(-1);

    }

    while (1) {

        if (argc>2 && !strcmp(argv[2],"term")) {

            color_stop = "\x1b[0m";

            color_red = "\x1b[1;31m";

            color_green = "\x1b[1;32m";

            color_orange = "\x1b[1;33m";

            color_none = color_stop;

        } else if (argc>2 && !strcmp(argv[2],"pango")) {

            color_stop = "</span>";

            color_none = "<span>";

        } else {

            color_stop = color_red = color_green = color_orange = color_none = "";

        }

        iwstats stats;

        iwrange range;

        int down = 0;

        wireless_config info;

        if (iw_get_basic_config(skfd,argv[1],&info) < 0) {

            perror("iw_get_basic_config");

            exit(-2);

        }

        if (iw_get_range_info(skfd,argv[1],&range) < 0) {

            perror("iw_get_range_info");

            exit(-2);

        }

        if (iw_get_stats(skfd,argv[1],&stats,&range,1) < 0) {

            if (argc>2)

                down = 1;

            else {

                perror("iw_get_stats");

                exit(-2);

            }

        }

        if (down) {

            if (!strcmp(info.essid,"")) {

                printf("%sDOWN%s\n",color_orange,color_stop);

            } else {

                printf("%s [%sXX/XX%s] XX\n",info.essid,color_red,color_stop);

            }

        } else {

            int ql = 100 * stats.qual.qual / range.max_qual.qual;

            char *qual_color = color(20,40,0,ql);

            if (stats.qual.qual<range.avg_qual.qual) qual_color = color_red;

            signed char dbm;

            dbm = (signed char)stats.qual.level;

            int perc;

            if (dbm>-35)

                perc=100;

            else if(dbm<=-95)

                perc = 1;

            else

            perc = (dbm+95)*100/60;

            printf("%s [%s%02d/%02d%s] ",info.essid,qual_color,stats.qual.qual,range.max_qual.qual,color_stop);

            printf("%s%hhd%s ",color(-83,-77,0,dbm),dbm,color_stop);

            printf("%s%d%%%s\n",color(20,40,0,perc),perc,color_stop);

        }

        fflush(stdout);

        sleep(UPDATE_PERIOD);

    }

    close(skfd);

}