#include "./database.h"

int initializeTrailsTable()
{
    sqlite3 *db;
    char *err_msg;
    int rc;
    sqlite3_stmt *stmt;

    rc = sqlite3_open("robotgowest.db", &db);

    // char *sql = "CREATE TABLE Test(Id INT)";

    char *sql =
        "CREATE TABLE Trials(Id INTEGER PRIMARY KEY AUTOINCREMENT, Position TEXT, Start INT, Frequency NUMBER, Duration INT);";

    rc = sqlite3_prepare_v2(db, sql, -1, &stmt, NULL);
    rc = sqlite3_step(stmt);
    rc = sqlite3_finalize(stmt);
    sqlite3_close(db);
}

int initializeCaptorMetadataTable()
{
    sqlite3 *db;
    char *err_msg;
    int rc;
    sqlite3_stmt *stmt;

    rc = sqlite3_open("robotgowest.db", &db);

    // char *sql = "CREATE TABLE Test(Id INT)";

    char *sql =
        "CREATE TABLE CaptorMetadata(Id INTEGER PRIMARY KEY AUTOINCREMENT, CaptorNumber INT, TrialId INT, FOREIGN KEY (TrialId) REFERENCES Trials(Id));";

    rc = sqlite3_prepare_v2(db, sql, -1, &stmt, NULL);
    rc = sqlite3_step(stmt);
    rc = sqlite3_finalize(stmt);
    sqlite3_close(db);
}

int createCaptorDataTable(int trialId, int captorNumber)
{
    sqlite3 *db;
    char *err_msg;
    int rc;
    sqlite3_stmt *stmt;

    // Create Table Name
    char tableName[8];
    strcpy(tableName, "T");
    strcat(tableName, convertIntegerToChar(trialId));
    strcat(tableName, "C");
    strcat(tableName, convertIntegerToChar(captorNumber));
    strcat(tableName, "Data");

    rc = sqlite3_open("robotgowest.db", &db);
    if (rc != SQLITE_OK)
    {
        return 1;
    }

    char sql[101] = "CREATE TABLE ";
    strcat(sql, tableName);
    strcat(sql, " (Id INTEGER PRIMARY KEY AUTOINCREMENT, Lux NUMBER, Mean NUMBER , Power NUMBER);");

    rc = sqlite3_prepare_v2(db, sql, -1, &stmt, NULL);
    rc = sqlite3_step(stmt);
    rc = sqlite3_finalize(stmt);
    sqlite3_close(db);

    return 0;
}

/**
 * @brief convert an interger N into a char*
 *
 * @param N
 * @return char*
 */
char *convertIntegerToChar(int N)
{
    // Count digits in number N
    int m = N;
    int digit = 0;
    while (m)
    {
        digit++;
        m /= 10;
    }
    char *arr;
    char arr1[digit];
    arr = (char *)malloc(digit * sizeof(char));
    int index = 0;
    while (N)
    {
        arr1[++index] = N % 10 + '0';
        N /= 10;
    }
    int i;
    for (i = 0; i < index; i++)
    {
        arr[i] = arr1[index - i];
    }
    arr[i] = '\0';
    return (char *)arr;
}

int main(void)
{
    sqlite3_initialize();

    initializeTrailsTable();
    initializeCaptorMetadataTable();
    createCaptorDataTable(12, 8);
    sqlite3_shutdown();
}