#include <stdio.h>
#include <string.h>
#include <unistd.h>

#include <arpa/inet.h>
#include <net/if.h>
#include <sys/ioctl.h>

#include "wifi_scan.h"

#define COLOR_CONNECTED "#859900"
#define COLOR_DISCONNECTED "#dc322f"
#define ICON_CONNECTED "f05a9"
#define ICON_DISCONNECTED "f16bc"

int main() {
    const char *ip_addr = NULL;
    const char *ssid = NULL;
    const char *color = NULL;
    const char *icon = NULL;
    const char *fmt = "<span color=\"%s\"><span font=\"Symbols Nerd Font\">&#x%s;</span> %s %s</span>\n";

#ifdef DEBUG
    for (int i = 0; true; ++i) {
#else
    for (;;) {
#endif

#ifdef DEBUG
        printf("\nloop #%d:\n", ++i);
#endif

        color = COLOR_DISCONNECTED;
        icon = ICON_DISCONNECTED;

        // check if wlan0 has an IP address
        struct ifreq ifr;
        ifr.ifr_addr.sa_family = AF_INET;
        strncpy(ifr.ifr_name, "wlan0", IFNAMSIZ - 1);

        int station_fd = socket(AF_INET, SOCK_DGRAM, 0);
        int ioctl_res = ioctl(station_fd, SIOCGIFADDR, &ifr);

#ifdef DEBUG
        if (ioctl_res < 0) {
            perror("ioctl()");
        }
#endif

        close(station_fd);
        ip_addr = inet_ntoa(((struct sockaddr_in *)&ifr.ifr_addr)->sin_addr);

#ifdef DEBUG
        printf("ip_addr set to: %s\n", ip_addr);
#endif

        if (ioctl_res != 0 || strcmp(ip_addr, "0.0.0.0") == 0) {
            ssid = "-";
            ip_addr = "-";

#ifdef DEBUG
            printf("ioctl() failed or ip set to 0:0:0:0, ssid set to: %s, ip_addr set to: %s\n", ssid, ip_addr);
#endif
        } else {
            // get ssid in case we have an IP
            struct station_info station;
            wifi_scan_station(wifi_scan_init("wlan0"), &station);
            ssid = station.ssid;

#ifdef DEBUG
            printf("ssid set to: %s\n", ssid);
#endif

            // lastly, check if we can ping google
            struct sockaddr_in sock_in;
            sock_in.sin_family = AF_INET;
            sock_in.sin_port = htons(443);
            inet_pton(AF_INET, "8.8.8.8", &sock_in.sin_addr);

            int remote_fd = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
            int connect_res = connect(remote_fd, (struct sockaddr *)&sock_in, sizeof(sock_in));

#ifdef DEBUG
            if (connect_res < 0) {
                perror("connect()");
            }
#endif

            fd_set fdset;
            FD_ZERO(&fdset);
            FD_SET(remote_fd, &fdset);

            struct timeval tv = { .tv_sec = 1, .tv_usec = 0 };
            int select_res = select(remote_fd + 1, NULL, &fdset, NULL, &tv);

#ifdef DEBUG
            if (select_res < 0) {
                perror("select()");
            }
#endif

            if (select_res == 1) {
#ifdef DEBUG
                printf("select() returned 1\n");
#endif

                int so_error;
                socklen_t len = sizeof so_error;
                getsockopt(remote_fd, SOL_SOCKET, SO_ERROR, &so_error, &len);

                // we have a good connection
                if (so_error == 0) {
#ifdef DEBUG
                    printf("so_error set to 0, we must have internet\n");
#endif

                    color = COLOR_CONNECTED;
                    icon = ICON_CONNECTED;
                }
            }

            close(remote_fd);
        }

        // print results
#ifndef DEBUG
        printf(fmt, color, icon, ssid, ip_addr);
#endif
        fflush(stdout);
        sleep(1);
    }

    return 0;
}