#include <stdio.h>
#include <string.h>
#include <ecdaa-tpm.h>
#include "client.h"
#include "server.h"

typedef enum memberstate {
    ON,
    JOIN,
    APPEND,
    JOINPROCEED,
    JOINED,
    ATTEST,
    PUBLISH
} memberstate_e;

typedef struct member {
    struct ecdaa_member_public_key_FP256BN mpk;
    struct ecdaa_member_public_key_FP256BN msk;
    memberstate_e state;
} member_t;

member_t member;

int process_member(char *buffer);

int member_join(char* buffer);
int member_attest(char* buffer);
int member_publish(char* buffer);

int main() {
    int err = 0;

    if (2 != server_start(&process_member, MEMBERPORT)) {
        printf("server failed\n");
    }
    return 0;
}

int process_member(char *buffer) {
    int ret = 0;

        printf("> MEMBER: %s\n", buffer);

        if (0 == strncasecmp("ATTEST", buffer, 6)) {
            printf("attest()\n");
            bzero(buffer, MAX_BUFSIZE);
            strncpy(buffer, "ATTEST ", 7);
            member_attest(buffer);
        } else if (0 == strncasecmp("PUBLISH", buffer, 7)) {
            printf("publish()\n");
            bzero(buffer, MAX_BUFSIZE);
            strncpy(buffer, "PUBLISH ", 8);
            member_publish(buffer);
        } else if (0 == strncasecmp("JOIN", buffer, 4)) {
            printf("join()\n");
            member.state = JOIN;
            ret = client_connect(&member_join, ISSUERIP, ISSUERPORT);
            if (0 >= ret || JOINED != member.state) {
                printf("process_member: issuer connection failed\n");
                bzero(buffer, MAX_BUFSIZE);
                strncpy(buffer, "ERR\n", 4);
            } else {
                bzero(buffer, MAX_BUFSIZE);
                strncpy(buffer, "OK\n", 3);
            }
            ret = 0;
        } else if (0 == strncasecmp("EXIT", buffer, 4)) {
            printf("exit()\n");
            bzero(buffer, MAX_BUFSIZE);
            strncpy(buffer, "OK\n", 3);
            ret = 1;
        } else if (0 == strncasecmp("SHUTDOWN", buffer, 8)) {
            bzero(buffer, MAX_BUFSIZE);
            strncpy(buffer, "OK\n", 3);
            ret = 2;
        } else {
            printf("error()\n");
            bzero(buffer, MAX_BUFSIZE);
            strncpy(buffer, "ERR\n", 4);
            ret = 0;
        }

        printf("< MEMBER: %s\n", buffer);
    return ret;
}

int member_join(char *buffer) {
    int ret = 0;

    printf("ISSUER > MEMBER: %s", buffer);
    switch (member.state) {
        case JOIN:
            bzero(buffer, MAX_BUFSIZE);
            strncpy(buffer, "JOIN\n", 5);
            member.state = APPEND;
            break;
        case APPEND:
            if (0 == strncasecmp("JOINSTART", buffer, 9)) {
                bzero(buffer, MAX_BUFSIZE);
                strncpy(buffer, "APPEND\n", 7);
                member.state = JOINPROCEED;
            }
            break;
        case JOINPROCEED:
            if (0 == strncasecmp("JOINPROCEED", buffer, 11)) {
                bzero(buffer, MAX_BUFSIZE);
                member.state = JOINED;
                ret = 1;
            }
            break;
        default:
            ret -1;
    }
    if(0 == ret) {
        printf("ISSUER < MEMBER: %s", buffer);
    }
    return ret;
}

int member_attest(char* buffer) {
    strncat(buffer, "<attestval>\n", 13);
    return 0;
}

int member_publish(char* buffer) {
    strncat(buffer, "<publicval>\n", 12);
    return 0;
}