/*********************************************************************************
 *      Copyright:  (C) 2019 LingYun IoT System Studio
 *                  All rights reserved.
 *
 *       Filename:  main.c
 *    Description:  This file
 *
 *        Version:  1.0.0(29/01/19)
 *         Author:  Guo Wenxue <guowenxue@gmail.com>
 *      ChangeLog:  1, Release initial version on "29/01/19 15:34:41"
 *
 ********************************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <time.h>
#include <getopt.h>
#include <libgen.h>
#include <string.h>
#include <errno.h>

#include <mosquitto.h>
#include <cjson/cJSON.h>

#include "logger.h"
#include "util_proc.h"
#include "config.h"
#include "tsl2561.h"

#define PROG_VERSION               "v1.0.0"
#define DAEMON_PIDFILE             "/tmp/.lightd.pid"

int check_sample_time(time_t *last_time, int interval);
void *thingsboard_subsciber(void *args);
void *thingsboard_publisher(void *args);

static void program_usage(char *progname)
{
    printf("Usage: %s [OPTION]...\n", progname);
    printf(" %s is LingYun studio MQTT daemon program running on RaspberryPi\n", progname);

    printf("\nMandatory arguments to long options are mandatory for short options too:\n");
    printf(" -b[daemon  ]  Running in daemon mode\n");
    printf(" -d[debug   ]  Running in debug mode\n");
    printf(" -c[conf    ]  Specify configure file\n");
    printf(" -h[help    ]  Display this help information\n");
    printf(" -v[version ]  Display the program version\n");

    printf("\n%s version %s\n", progname, PROG_VERSION);
    return;
}

int main (int argc, char **argv)
{
    int                daemon = 0;
    char               *conf_file="/etc/lightd.conf";
    int                debug = 0;
    int                opt;
    char              *progname=NULL;
    pthread_t          tid;

    int                rv;
    float              lux = 0.0;
    time_t             last_time = 0;
    iotd_ctx_t         ctx;
    hwinfo_t          *hwinfo = &ctx.hwinfo;

    struct option long_options[] = {
        {"conf", required_argument, NULL, 'c'},
        {"daemon", no_argument, NULL, 'b'},
        {"debug", no_argument, NULL, 'd'},
        {"version", no_argument, NULL, 'v'},
        {"help", no_argument, NULL, 'h'},
        {NULL, 0, NULL, 0}
    };

    progname = (char *)basename(argv[0]);

    /* parser the command line parameters */
    while ((opt = getopt_long(argc, argv, "c:bdvh", long_options, NULL)) != -1)
    {
        switch (opt)
        {
            case 'c': /* Set configure file */
                conf_file = optarg;
                break;

            case 'b': /* Set daemon running */
                daemon = 1;
                break;

            case 'd': /* Set debug running */
                debug = 1;
                break;

            case 'v':  /* Get software version */
                printf("%s version %s\n", progname, PROG_VERSION);
                return 0;

            case 'h':  /* Get help information */
                program_usage(progname);
                return 0;

            default:
                break;
        }

    }

    /* parser configure file */
    if( parser_conf(conf_file, &ctx, debug)<0 )
    {
        fprintf(stderr, "Parser thingsboard configure file failure\n");
        return -2;
    }

    /* install signal proc handler */
    install_default_signal();

    /* check program already running or not, if already running then exit, or set running as daemon */
    if( check_set_program_running(daemon, DAEMON_PIDFILE) < 0 )
        goto cleanup;

    /* initial mosquitto library */
    mosquitto_lib_init();

    /* initial gpio hardware */
    gpio_init(&hwinfo->gpio);

    /*
     * +--------------------------------+
     * |    MQTT Subscriber Thread      |
     * +--------------------------------+
     */
    if( thread_start(&tid, thingsboard_subsciber, &ctx.mqtt ) < 0 )
    {
        log_error("Start MQTT subsciber worker thread failure\n");
        goto cleanup;
    }
    log_info("Start MQTT subsciber worker thread ok\n");

    /*
     * +--------------------------------+
     * |   Auto Light Control Thread    |
     * +--------------------------------+
     */
    log_info("Auto light control thread start\n");
    while( ! g_signal.stop )
    {
        /* TSL2561 update lux in every 5 minutes  */
        if(check_sample_time(&last_time, 300))
        {
            if( tsl2561_get_lux(&lux) < 0 )
            {
                log_error("TSL2561 sample Lux failed\n");
                lux = hwinfo->lux_threshold + 100.0;
            }

            if( lux > hwinfo->lux_threshold )
                log_info("Lux[%.3f] > Treshold[%.3f], don't auto light\n", lux, hwinfo->lux_threshold);
            else
                log_info("Lux[%.3f] <= Treshold[%.3f], need auto light\n", lux, hwinfo->lux_threshold);
        }

        if( lux > hwinfo->lux_threshold )
        {
            sleep(1);
            continue;
        }

        rv = infrared_detect();
        if( rv != 0 )
        {
            log_warn("Someone incoming detected by %s infrared\n", rv&FLAG_INFRARED_INDOOR?"indoor":"hallway");

            turn_light("indoor", ON);
            turn_light("hallway", ON);

            sleep(hwinfo->light_intval);

            turn_light("indoor", OFF);
            turn_light("hallway", OFF);
        }

        msleep(100);
    }

cleanup:
    mosquitto_lib_cleanup();
    log_close();

    return 0;
}

int check_sample_time(time_t *last_time, int interval)
{
    int                  need = 0; /* no need sample now */
    time_t               now;

    time(&now);

    if( difftime(now, *last_time)>interval )
    {
        need = 1; /* need sample now  */
        *last_time = now;
    }

    return need;
}

void sub_connect_callback(struct mosquitto *mosq, void *userdata, int result)
{
    mqtt_ctx_t             *ctx = (mqtt_ctx_t *)userdata;

    if( result )
    {
        log_error("Subscriber connect to broker server failed, rv=%d\n", result);
        return ;
    }

    log_info("Subscriber connect to broker server[%s:%d] successfully\n", ctx->host, ctx->port);
    mosquitto_subscribe(mosq, NULL, ctx->subTopic, ctx->subQos);
}

void sub_disconnect_callback(struct mosquitto *mosq, void *userdata, int result)
{
    mqtt_ctx_t             *ctx = (mqtt_ctx_t *)userdata;

    log_warn("Subscriber disconnect to broker server[%s:%d], reason=%d\n", ctx->host, ctx->port, result);
}

void proc_json_items(cJSON *root, mqtt_ctx_t *ctx)
{
    cJSON *item_method;
    cJSON *item_params;
    cJSON *item_device;
    cJSON *item_status;

    char *method;
    char *device;
    int status;

    /* 获取 method 字段 */
    item_method = cJSON_GetObjectItem(root, "method");
    if (item_method == NULL || !cJSON_IsString(item_method))
    {
        log_error("JSON missing method\n");
        return;
    }
    method = item_method->valuestring;

    /* 只处理 setValue 命令 */
    if (strcmp(method, "setValue") != 0)
    {
        log_error("Ignore method: %s\n", method);
        return;
    }

    /* 获取 params 对象 */
    item_params = cJSON_GetObjectItem(root, "params");
    if (item_params == NULL || !cJSON_IsObject(item_params))
    {
        log_error("JSON missing params\n");
        return;
    }

    /* 获取 params.device */
    item_device = cJSON_GetObjectItem(item_params, "device");
    if (item_device == NULL || !cJSON_IsString(item_device))
    {
        log_error("JSON missing params.device\n");
        return;
    }
    device = item_device->valuestring;

    /* 获取 params.status (布尔值) */
    item_status = cJSON_GetObjectItem(item_params, "status");
    if (item_status == NULL || !(cJSON_IsBool(item_status)))
    {
        log_error("JSON missing params.status\n");
        return;
    }
    status = cJSON_IsTrue(item_status) ? ON : OFF;

    log_info("parser JSON message to control '%s' status '%d'\n", device, status);

    /* 映射 Light 名称并调用控制函数 */
    if ( strstr(device, "Indoor") || strstr(device, "indoor"))
    {
        turn_light("indoor", status);
    }
    else if ( strstr(device, "Hallway") || strstr(device, "hallway") )
    {
        turn_light("hallway", status);
    }

    return;
}

void sub_message_callback(struct mosquitto *mosq, void *userdata, const struct mosquitto_message *message)
{
    mqtt_ctx_t             *ctx = (mqtt_ctx_t *)userdata;

    cJSON                  *root = NULL;

    if ( !message->payloadlen )
    {
        log_error("%s (null)\n", message->topic);
        return ;
    }

    log_info("Subscriber receive message: '%s'\n", message->payload);

    root = cJSON_Parse(message->payload);
    if( !root )
    {
        log_error("cJSON_Parse parser failure: %s\n", cJSON_GetErrorPtr());
        return ;
    }

    proc_json_items(root, ctx);

    /* must delete it, or it will result memory leak */
    cJSON_Delete(root);

    return ;
}

void *thingsboard_subsciber(void *args)
{
    mqtt_ctx_t             *ctx = (mqtt_ctx_t *)args;
    struct mosquitto       *mosq;
    bool                    session = true;

    mosq = mosquitto_new(ctx->devid, session, ctx);
    if( !mosq )
    {
        log_error("mosquitto_new failure\n");
        return NULL;
    }

    /* connnect to broker by token or uid/pwd */
    if( strlen(ctx->token ) > 0)
    {
        log_info("Using token authentication\n");
        mosquitto_username_pw_set(mosq, ctx->token, NULL);
    }
    else if( strlen(ctx->uid)> 0  && strlen(ctx->pwd)> 0 )
    {
        log_info("Using username/password authentication\n");
        mosquitto_username_pw_set(mosq, ctx->uid, ctx->pwd);
    }

    /* set callback functions */
    mosquitto_connect_callback_set(mosq, sub_connect_callback);
    mosquitto_disconnect_callback_set(mosq, sub_disconnect_callback);
    mosquitto_message_callback_set(mosq, sub_message_callback);

    while( !g_signal.stop )
    {
        /* connect to MQTT broker  */
        if( mosquitto_connect(mosq, ctx->host, ctx->port, ctx->keepalive) )
        {
            log_error("Subscriber connect to broker[%s:%d] failure: %s\n", ctx->host, ctx->port, strerror(errno));
            msleep(1000);
            continue;
        }

        /* -1: use default timeout 1000ms  1: unused */
        mosquitto_loop_forever(mosq, -1, 1);
    }

    mosquitto_destroy(mosq);
    return NULL;
}