#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <stdbool.h>
#include <string.h>
#include <time.h>
#include <getopt.h>

#include "usbms.h"

#include "config.h"

#define VERSION "1.0"

static int setpassword(const char *password)
{
    unsigned long slen = strlen(password);
    if (slen < 2 || slen > 12) {
        fprintf(stderr, "Password has to be at least 2 but no more than 12 characters long.\n");
        return 1;
    }

    for (int i = 0; i < slen; i++) {
        switch (password[i]) {
            case '/':
            case ':':
            case '@':
            case '[':
            case '`':
            case '{':
                fprintf(stderr, "Use of invalid character in password. Do not use `%c'.\n",
                        password[i]);
                fprintf(stderr, "Invalid character for passwords are \"/:@[`{\".\n");
                return 1;
        }
    }
    printf("Updating password to `%s'... ", password);
    if (send_command(ACTIONPRO_OPCODE_SETPASSWORD, (const uint8_t *)password, slen) == ACTIONPRO_CMD_OK)
        printf("OK\n");
    else
        printf("ERROR\n");

    return 0;
}

static int setssid(const char *ssid)
{
    unsigned long slen = strlen(ssid);
    if (slen < 2 || slen > 12) {
        fprintf(stderr, "SSID has to be at least 2 but no more than 12 characters long.\n");
        return 1;
    }

    for (int i = 0; i < slen; i++) {
        switch (ssid[i]) {
            case ' ':
            case '~':
                fprintf(stderr, "Use of invalid character in SSID. Do not use `%c'.\n",
                        ssid[i]);
                fprintf(stderr, "Invalid character for SSID are \" ~\".\n");
                return 1;
        }
    }
    printf("Updating SSID to `%s'... ", ssid);
    if (send_command(ACTIONPRO_OPCODE_SETSSID, (const uint8_t *)ssid, slen) == ACTIONPRO_CMD_OK)
        printf("OK\n");
    else
        printf("ERROR\n");


    return 0;
}

static int settime()
{
    printf("Setting device time... ");

    time_t curtime;
    struct tm *timeinfo;


    if (time(&curtime)) {
        timeinfo = localtime(&curtime);
        if (timeinfo) {
            uint8_t newtime[8] = {0};

            const uint16_t year = (timeinfo->tm_year + 1900) >> 8 | (timeinfo->tm_year + 1900) << 8;
            const uint8_t month = timeinfo->tm_mon + 1;

            memcpy(newtime+0, &year, 2);
            memcpy(newtime+2, &month, 1);
            memcpy(newtime+3, &timeinfo->tm_mday,  1);
            memcpy(newtime+4, &timeinfo->tm_hour,  1);
            memcpy(newtime+5, &timeinfo->tm_min,  1);
            memcpy(newtime+6, &timeinfo->tm_sec,  1);

            if (send_command(ACTIONPRO_OPCODE_SETTIME, newtime, 8) == ACTIONPRO_CMD_OK)
                printf("OK\n");
            else
                printf("ERROR\n");

            return 0;
        }
    }

    return 1;
}

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

    bool show_help = false;
    char *new_password = NULL;
    char *new_ssid = NULL;
    bool sync_time = false;
    bool show_version = false;

    printf("ACTIONPRO configuration utility\n");

    while (1) {
        static struct option long_options[] = {
            {"help"    , no_argument      , 0, 'p'},
            {"password", required_argument, 0, 'p'},
            {"ssid"    , required_argument, 0, 's'},
            {"time"    , no_argument      , 0, 't'},
            {"version" , no_argument      , 0, 'v'},
            {0, 0, 0, 0}
        };

        int option_index = 0;

        c = getopt_long(argc, argv, "hp:s:tv", long_options, &option_index);

        if (argc < 2) {
            fprintf(stderr, "At least one option is required. See -h, --help for instructions.\n");
            exit(EXIT_FAILURE);
        }

        if (c == -1)
            break;

        switch (c) {
            case 'h':
                show_help = true;
                break;
            case 'p':
                new_password = optarg;
                break;
            case 's':
                new_ssid = optarg;
                break;
            case 't':
                sync_time = true;
                break;
            case 'v':
                show_version = true;
            case '?':
                fprintf(stderr, "Use option -h, --help for instructions.\n");
                exit(EXIT_FAILURE);
            default:
                abort();
        }
    }

    if (show_help) {
        printf("Usage: %s [OPTION]\n", argv[0]);
        printf("  -C, --config-file=FILE     use this user configuration file\n");
        printf("  -h, --help                 give this help list\n");
        printf("  -p, --password=PASSWORD    sets the access point authentication PASSWORD\n");
        printf("  -s, --ssid=SSID            sets the access point SSID\n");
        printf("  -t, --time                 synchronize the camera time\n");
        printf("  -v, --version              display version number\n");
        printf("\nThis program requires write access to usb devices and might be run as root.\n");

        exit(EXIT_SUCCESS);
    }

    if (show_version) {
        printf("Actionpro-cli version: %s\n", VERSION);
        exit(EXIT_SUCCESS);
    }

    if (open_device() == ACTIONPRO_OK) {
        if (new_password)
            setpassword(new_password);

        if (new_ssid)
            setssid(new_ssid);

        if (sync_time)
            settime();
    }
    close_device();

    exit(EXIT_SUCCESS);
}