#include <sys/types.h>
#include <ctype.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>

#include "script_parser.h"
#include "debug.h"


#define ITEM_NAME_MAX_LEN               32

#define KEY_MAX_LEN                     32
#define VALUE_MAX_LEN                   128

#define ITEM_MAX_COUNT                  128

#define LINE_MAX_LEN                    512

#define LINE_ERROR                      -1
#define LINE_COMMENT                    0
#define LINE_NULL                       1
#define LINE_MAINKEY                    2
#define LINE_SUBKEY                     3

#define DATA_TYPE_SINGLE_WORD           1
#define DATA_TYPE_STRING                2
#define DATA_TYPE_MULTI_WORD            3
#define DATA_TYPE_GPIO                  4

struct script_head
{
    int mainkey_cnt;
    int version[3];
};

struct script_item
{
    char name[32];
    int length;
    int offset;
};

/*
 * get value data type.
 */
static int __get_str2int(char *buf, int value[])
{
    char *src;
    char ch;
    unsigned int temp;
    int sign;
    int i;
    int idx;
    char str[128];

    src = buf;

    if (strncasecmp(src, "port:p", 6) == 0) {
        /* handle gpio */
        src += 6;
        if (strncasecmp(src, "ower", 4) == 0) {
            value[0] = 0xffff;
            src += 4;
        }
        else {
            ch = *src++;
            if (islower(ch))
                value[0] = ch - 'a' + 1;
            else if (isupper(ch))
                value[0] = ch - 'A' + 1;
            else
                return -1;
        }

        temp = 0;
        ch = *src++;
        while (ch != '<') {
            if (isdigit(ch)) {
                temp = temp * 10 + (ch - '0');
                ch = *src++;
            }
            else if (ch == '\0') {
                src--;
                break;
            }
            else {
                return -1;
            }
        }
        value[1] = temp;

        idx = 2;
        ch = *src++;
        while (ch != '\0') {
            i = 0;
            memset(str, 0, sizeof(str));
            while (ch != '>') {
                if (isupper(ch))
                    ch = tolower(ch);
                str[i++] = ch;
                ch = *src++;
            }

            if (strcmp(str, "default") == 0 || 
                strcmp(str, "none") == 0 || 
                strcmp(str, "null") == 0 || 
                strcmp(str, "-1") == 0) {
                value[idx] = -1;
            }
            else {
                i = 0;
                ch = str[i++];
                temp = 0;
                if (ch == '-') {
                    sign = -1;
                    ch = str[i++];
                }
                else {
                    sign = 1;
                }

                while (ch != '\0') {
                    if (isdigit(ch))
                        temp = temp * 10 + (ch - '0');
                    else
                        return -1;

                    ch = str[i++];
                }

                value[idx] = temp * sign;
            }

            idx++;
            ch = *src++;
            if (ch == '<') {
                ch = *src++;
            }
            else if (ch == '\0') {
                ;
            }
            else {
                return -1;
            }
        }

        switch (idx) {
            case 3:
                value[3] = -1;
            case 4:
                value[4] = -1;
            case 5:
                value[5] = -1;
            case 6:
                break;
            default:
                return -1;
        }

        return DATA_TYPE_GPIO;
    }
    else if (strncasecmp(src, "string:", 7) == 0) {
        src += 7;
        idx = 0;
        while (src[idx] != '\0') {
            idx++;
            if (idx > 127)
                break;
        }

        if (idx & 0x03)
            idx = (idx & (~0x03)) + 4;

        value[0] = idx >> 2;
        value[1] = 1;

        return DATA_TYPE_STRING;
    }
    else if (src[0] == '"') {
        src += 1;
        idx = 0;
        while (src[idx] != '"') {
            idx++;
            if (idx > 127)
                break;
        }

        src[idx] = '\0';
        if (idx & 0x03)
            idx = (idx & (~0x03)) + 4;
        
        value[0] = idx >> 2;
        value[1] = 2;

        return DATA_TYPE_STRING;
    }
    else if (isdigit(src[0]) && (src[1] == 'x' || src[2] == 'X')) {
        temp = 0;
        ch = *src++;
        while (ch != '\0') {
            if (isdigit(ch)) {
                temp = temp * 16 + (ch - '0');
                ch = *src++;
            }
            else if (isupper(ch)) {
                temp = temp * 16 + (ch - 'A' + 10);
                ch = *src++;
            }
            else if (islower(ch)) {
                temp = temp * 16 + (ch - 'a' + 10);
                ch = *src++;
            }
            else {
                break;
            }
        }
        value[0] = temp;

        return DATA_TYPE_SINGLE_WORD;
    }
    else if (isdigit(src[0]) || 
            (isdigit(src[1]) && src[0] == '-')) {
        if (src[0] == '-') {
            sign = -1;
            ch = *src++;
        }
        else {
            sign = 1;
        }

        temp = 0;
        ch = *src++;
        while (ch != '\0') {
            if (isdigit(ch)) {
                temp = temp * 10 + (ch - '0');
                ch = *src++;
            }
            else {
                break;
            }
        }
        value[0] = temp * sign;

        return DATA_TYPE_SINGLE_WORD;
    }
    else {
        idx = 0;
        while (src[idx] != '\0') {
            idx++;
            if (idx > 127)
                break;
        }

        if (idx & 0x03)
            idx = (idx & (~0x03)) + 4;

        value[0] = idx >> 2;

        return DATA_TYPE_STRING;
    }
}

/*
 * get key[32] and value[128], end with '\0'.
 * \retval -1 empty line.
 * \retval 0 OK
 */
static int __get_key_value(char *buf, char *key, char *value)
{
    char *src;
    int key_idx;
    int value_idx;

    /* check line */
    src = buf;
    key_idx = value_idx = 0;
    while (1) {
        if (*src == ' ' || *src == '\t') {
            src++;
        }
        else if (*src == 0x0a || *src == 0x0d) {
            key[key_idx] = '\0';
            value[value_idx] = '\0';
            return -1;
        }
        else {
            break;
        }
    }

    /* get key */
    while (*src != '=') {
        key[key_idx++] = *src++;
        if (key_idx >= 31) {
            key[key_idx] = '\0';
            break;
        }
    }

    for (key_idx--; key_idx > 0; key_idx--) {
        if (key[key_idx] == ' ' || key[key_idx] == '\t') {
            key[key_idx] = '\0';
        }
        else {
            key[key_idx + 1] = '\0';
            break;
        }
    }

    for (; *src != '='; src++);

    /* check line */
    src++;
    while (1) {
        if (*src == ' ' || *src == '\t') {
            src++;
        }
        else if (*src == 0x0a || *src == 0x0d) {
            value[value_idx] = '\0';
            return 0;
        }
        else {
            break;
        }
    }

    /* get value */
    while (*src != 0x0a && *src != 0x0d) {
        value[value_idx++] = *src++;
        if (value_idx >= 127) {
            value[value_idx] = '\0';
            break;
        }
    }

    for (value_idx--; value_idx > 0; value_idx--) {
        if (value[value_idx] == ' ' || value[value_idx] == '\t') {
            value[value_idx] = '\0';
        }
        else {
            value[value_idx + 1] = '\0';
            break;
        }
    }

    return 0;
}

static int __fill_mainkey(char *buf, struct script_item *item)
{
    char *src;
    char ch;
    int i;

    src = buf + 1;
    for (ch = *src++, i = 0; ch != ']'; i++, ch = *src++) {
        item->name[i] = ch;
        if (i + 1 >= ITEM_NAME_MAX_LEN) {
            item->name[i] = '\0';
            break;
        }
    }

    if (item->name[0] == '\0')
        return -1;

    return 0;
}

static int __getline(char *buf, int len, int *flag)
{
    char *src;
    char ch;
    char prev_ch;
    int line_len;

    /* get line flag */
    src = buf;
    ch = *src++;
    switch (ch) {
        case ';':
            *flag = LINE_COMMENT;
            break;

        case 0x0a:
        case 0x0d:
            *flag = LINE_NULL;
            break;

        case '[':
            *flag = LINE_MAINKEY;
            break;

        default:
            *flag = LINE_SUBKEY;
            break;
    }

    /* get line length */
    if (*flag == LINE_NULL) {
        ch = *src++;
        if (ch == 0x0a)
            return 2;
        else
            return 1;
    }

    ch = *src++;
    line_len = 1;
    while (line_len < len) {
        if (ch == 0x0a || ch == 0x0d)
            break;

        ch = *src++;
        line_len++;
        if (line_len >= LINE_MAX_LEN) {
            *flag = LINE_ERROR;
            return 0;
        }
    }
    line_len++;

    prev_ch = ch;
    ch = *src++;
    if (prev_ch == 0x0d) {
        if (ch == 0x0a)
            line_len++;
    }

    return line_len;
}

static char*  __parse_script(char *buf, int len)
{
    char *src;
    int rest_len, line_num;
    struct script_item *item_table = NULL;
    struct script_head head;
    unsigned int mainkey_idx, subkey_idx;
    int new_mainkey;
    char key[KEY_MAX_LEN];
    char value[VALUE_MAX_LEN];
    char *key_data = NULL, *key_addr;
    int *value_data = NULL, *value_addr;
    int value_idx;
    int fmt_value[8];
    unsigned int i;
    int ret;
    int shmid;
    char *script_buf, *pos;

    db_debug("the length of script is %d\n", len);

    /* allocate memory for key and value */
    item_table = malloc(sizeof(struct script_item) * ITEM_MAX_COUNT);
    if (item_table == NULL) {
        db_debug("allocate memory for main key failed(%s)\n", strerror(errno));
        ret = -1;
        goto out;
    }
    memset(item_table, 0, sizeof(struct script_item) * ITEM_MAX_COUNT);

    key_data = malloc(512 * 1024);
    if (key_data == NULL) {
        db_debug("allocate memory for key data failed(%s)\n", strerror(errno));
        ret = -1;
        goto out;
    }
    memset(key_data, 0, 512 * 1024);
    key_addr = key_data;

    value_data = malloc(512 * 1024);
    if (value_data == NULL) {
        db_debug("allocate memory for value data failed(%s)\n", 
                strerror(errno));
        ret = -1;
        goto out;
    }
    memset(value_data, 0, 512 * 1024);
    value_addr = value_data;

    /* parse script main loop */
    src = buf;
    rest_len = len;
    mainkey_idx = subkey_idx = value_idx = 0;
    new_mainkey = 0;
    line_num = 0;
    while (rest_len) {
        int line_len;
        int flag;
        
        /* get current line */
        line_len = __getline(src, rest_len, &flag);
        rest_len -= line_len;
        line_num++;
        switch (flag) {
            case LINE_COMMENT:
            case LINE_NULL:
                break;

            case LINE_MAINKEY:
                /* get main key */
                if (__fill_mainkey(src, &item_table[mainkey_idx])) {
                    ret = -1;
                    goto out;
                }

                if (new_mainkey) {
                    item_table[mainkey_idx].offset = 
                        item_table[mainkey_idx-1].offset + 
                        item_table[mainkey_idx-1].length * 10;
                }
                else {
                    /* first main key */
                    new_mainkey = 1;
                    item_table[mainkey_idx].offset = 0;
                }

                mainkey_idx++;
                break;

            case LINE_SUBKEY:
                /* get key[32] and value[128] */
                memset(key, 0, KEY_MAX_LEN);
                memset(value, 0, VALUE_MAX_LEN);
                ret = __get_key_value(src, key, value);
                if (ret == -1)
                    continue;

                /* stores key[32], stores value's offset in the next 4 bytes, 
                 * stores value's length[31;16] and type[0:15] in the next 
                 * next 4 bytes. there are total 40 bytes for one key data.
                 *
                 * the unit of value data length is int
                 */
                strcpy(key_addr, key);
                key_addr += KEY_MAX_LEN;

                /* get value data type */
                memset(fmt_value, 0, sizeof(int) * 8);
                ret = __get_str2int(value, fmt_value);
                switch (ret) {
                    case DATA_TYPE_SINGLE_WORD:
                        *value_addr = fmt_value[0];
                        *(unsigned int *)key_addr = value_idx;
                        key_addr += 4;
                        *(unsigned int *)key_addr = (1 << 0) | 
                            (DATA_TYPE_SINGLE_WORD << 16);
                        value_addr++;
                        value_idx++;
                        break;

                    case DATA_TYPE_STRING:
                        if (fmt_value[0]) {
                            if (fmt_value[1] == 1) {
                                strncpy((char *)value_addr, value + 
                                        sizeof("string:") - 1, fmt_value[0] << 2);
                            }
                            else if (fmt_value[1] == 2) {
                                strncpy((char *)value_addr, value + 1, 
                                        fmt_value[0] << 2);
                            }
                            else{
                                strncpy((char *)value_addr, value, 
                                        fmt_value[0] << 2);
                            }
                        }

                        *(unsigned int *)key_addr = value_idx;
                        key_addr += 4;
                        *(unsigned int *)key_addr = (fmt_value[0] << 0) | 
                            (DATA_TYPE_STRING << 16);
                        value_addr += fmt_value[0];
                        value_idx += fmt_value[0];
                        break;

                    case DATA_TYPE_GPIO:
                        for (i = 0; i < 6; i++)
                            *value_addr++ = fmt_value[i];
                        *(unsigned int *)key_addr = value_idx;
                        key_addr += 4;
                        *(unsigned int *)key_addr = (6 << 0) | 
                            (DATA_TYPE_GPIO << 16);
                        value_idx += 6;
                        break;

                    default:
                        db_debug("%s: L%d: fix me\n", __func__, __LINE__);
                }

                subkey_idx++;
                key_addr += 4;
                item_table[mainkey_idx - 1].length++;
                break;

            default:
                db_debug("script format error at line %d\n", line_num);
                ret = -1;
                goto out;
        }

        src += line_len;
    }

    if (mainkey_idx <= 0) {
        db_debug("mainkey_idx = %d\n", mainkey_idx);
        ret = -1;
        goto out;
    }

    /* recalc first subkey offset */
    for (i = 0; i < mainkey_idx; i++) {
        item_table[i].offset += (sizeof(struct script_head) >> 2) + 
            ((sizeof(struct script_item) * mainkey_idx) >> 2);
    }

    /* recalc every subkey offset */
    key_addr = key_data;
    i = 0;
    while (i < subkey_idx * 10 * sizeof(int)) {
        key_addr += ITEM_NAME_MAX_LEN;
        *(unsigned int *)key_addr += (sizeof(struct script_head) >> 2) + 
            ((sizeof(struct script_item) * mainkey_idx) >> 2) + 
            (subkey_idx * 10);
        i += 10 * sizeof(int); /* the sizeof each subkey data */
        key_addr += 8;
    }

    /* init script head */
    head.mainkey_cnt = mainkey_idx;
    head.version[0] = SCRIPT_VERSION0;
    head.version[1] = SCRIPT_VERSION1;
    head.version[2] = SCRIPT_VERSION2;

    /* allocate share memory segment for script buffer */
    i = sizeof(struct script_head) + sizeof(struct script_item) * 
        mainkey_idx + 10 * sizeof(int) * subkey_idx + sizeof(int) * value_idx;
    #if 0
    shmid = shmget(IPC_PRIVATE, i, IPC_CREAT | 0666);
    if (shmid == -1) {
        db_debug("allocate share memory segment for script buffer "
                "failed(%s)\n", strerror(errno));
        ret = -1;
        goto out;
    }
    db_debug("script shmid = %d\n", shmid);

    script_buf = pos = shmat(shmid, 0, 0);
    if (script_buf == (void *)-1) {
        db_debug("attach the share memory segment failed(%s)\n", 
                strerror(errno));
        shmctl(shmid, IPC_RMID, 0);
        ret = -1;
        goto out;
    }
#else
	script_buf = pos =  malloc(i);
#endif
    /* stores script buffer */
    memcpy(pos, &head, sizeof(struct script_head));
    pos += sizeof(struct script_head);
    memcpy(pos, item_table, sizeof(struct script_item) * mainkey_idx);
    pos += sizeof(struct script_item) * mainkey_idx;
    memcpy(pos, key_data, 10 * sizeof(int) * subkey_idx);
    pos += 10 * sizeof(int) * subkey_idx;
    memcpy(pos, value_data, sizeof(int) * value_idx);

    //shmdt(script_buf);
  //  ret = shmid;
  return script_buf;

out:
    /* free memory */
    if (item_table)
        free(item_table);
    if (key_data)
        free(key_data);
    if (value_data)
        free(value_data);

    return NULL;
}

char * parse_script(const char *name)
{
    FILE *fscript;
    int len;
    char *buf;
    int read_len;
    int shmid;
    char *script_buf;

    if (name == NULL) {
        db_error("script: invalid file name(null)\n");
        return NULL;
    }

    /* read script data */
    fscript = fopen(name, "rb");
    if (fscript == NULL) {
        db_error("script: can't open %s(%s)\n", name, strerror(errno));
        return NULL;
    }

    fseek(fscript, 0, SEEK_END);
    len = ftell(fscript);
    fseek(fscript, 0, SEEK_SET);
 
    buf = malloc(len);
    if (buf == NULL) {
        db_error("script: allocate memory for script data failed(%s)\n", 
                strerror(errno));
        fclose(fscript);
        return NULL;
    }

    read_len = fread(buf, 1, len, fscript);
    if (read_len < len) {
        db_debug("len = %d, read_len = %d, fix me\n", len, read_len);
    }
    fclose(fscript);

    /* parse script */
    script_buf = __parse_script(buf, len);
    free(buf);

    return script_buf;
}

void deparse_script(int shmid)
{
    //shmctl(shmid, IPC_RMID, 0);
}