/**
 * Test replication-manager
 */

#include "testconnections.h"
#include <termios.h>

void prepare()
{
    struct termios t;
    tcgetattr(STDIN_FILENO, &t);
    t.c_iflag |= IGNBRK;
    t.c_iflag &= ~ECHO;
    tcsetattr(STDIN_FILENO, TCSANOW, &t);
}

void get_output(TestConnections& test)
{
    test.tprintf("Maxadmin output:");
    char *output = test.ssh_maxscale_output(true, "maxadmin list servers");
    test.tprintf("%s", output);
    free(output);

    test.tprintf("MaxScale output:");
    output = test.ssh_maxscale_output(true,
                                      "cat /var/log/maxscale/maxscale.log && sudo truncate -s 0 /var/log/maxscale/maxscale.log");
    test.tprintf("%s", output);
    free(output);
}

static int inserts = 0;

void check(TestConnections& test)
{
    MYSQL *conn = test.open_rwsplit_connection();
    const char *query1 = "INSERT INTO test.t1 VALUES (%d)";
    const char *query2 = "SELECT * FROM test.t1";

    test.try_query(conn, "BEGIN");
    test.tprintf(query1, inserts);
    test.try_query(conn, query1, inserts++);
    mysql_query(conn, query2);

    MYSQL_RES *res = mysql_store_result(conn);
    test.add_result(res == NULL, "Query shoud return a result set");

    if (res)
    {
        std::string values;
        MYSQL_ROW row;
        int num_rows = mysql_num_rows(res);
        test.add_result(num_rows != inserts, "Query returned %d rows when %d rows were expected", num_rows, inserts);
        const char *separator = "";

        while ((row = mysql_fetch_row(res)))
        {
            values += separator;
            values += row[0];
            separator = ", ";
        }

        test.tprintf("%s: %s", query2, values.c_str());
    }

    test.try_query(conn, "COMMIT");
    mysql_close(conn);
}

int get_server_id(TestConnections& test)
{
    MYSQL *conn = test.open_rwsplit_connection();
    int id = -1;
    char str[1024];

    if (find_field(conn, "SELECT @@server_id", "@@server_id", str) == 0)
    {
        id = atoi(str);
    }

    mysql_close(conn);
    return id;
}

static bool interactive = false;

void get_input()
{
    if (interactive)
    {
        printf("--- Press any key to confinue ---\n");
        getchar();
    }
}

int main(int argc, char** argv)
{
    interactive = strcmp(argv[argc - 1], "interactive") == 0;
    prepare();

    TestConnections test(argc, argv);

    // TODO: Figure out how to do this without having replication-manager pre-installed on the system
    //
    // test.tprintf("Installing replication-manager");
    // int rc = system("./manage_mrm.sh install > manage_mrm.log");
    // if (!WIFEXITED(rc) || WEXITSTATUS(rc) != 0)
    // {
    //     test.tprintf("Failed to install replication-manager, see manage_mrm.log for more details");
    //     return -1;
    // }

    // // Wait a few seconds
    // sleep(5);

    test.tprintf("Creating table and inserting data");
    get_input();
    test.connect_maxscale();
    test.try_query(test.conn_rwsplit, "CREATE OR REPLACE TABLE test.t1(id INT)");
    test.repl->sync_slaves();

    check(test);
    get_output(test);

    test.tprintf("Stopping master and waiting for it to fail over");
    get_input();
    test.repl->stop_node(0);
    sleep(10);

    check(test);
    get_output(test);

    test.tprintf("Stopping another node and waiting for replication-manager to detect it");
    get_input();
    test.repl->stop_node(1);
    sleep(10);

    check(test);
    get_output(test);
    get_input();

    test.tprintf("Stopping all but one node and waiting for replication-manager to detect it");
    get_input();
    test.repl->stop_node(2);
    sleep(10);

    check(test);
    get_output(test);


    test.tprintf("Fix replication and recreate table");
    test.close_maxscale_connections();
    test.repl->fix_replication();
    test.connect_maxscale();
    test.try_query(test.conn_rwsplit, "CREATE OR REPLACE TABLE test.t1(id INT)");
    test.repl->sync_slaves();
    inserts = 0;

    check(test);
    get_output(test);

    test.tprintf("Disable replication on a slave and kill master, check that it is not promoted");
    execute_query(test.repl->nodes[1], "STOP SLAVE; RESET SLAVE; RESET SLAVE ALL;");
    test.repl->stop_node(0);
    sleep(10);

    check(test);
    get_output(test);

    int id = get_server_id(test);
    test.add_result(id == test.repl->get_server_id(1), "Invalid slave should not be used");

    // TODO: Figure this also out, remove the component if it's not needed
    // test.tprintf("Removing replication-manager");
    // get_input();
    // system("./manage_mrm.sh remove >> manage_mrm.log");

    test.repl->fix_replication();
    return test.global_result;
}