/*
 *  The Mana World Server
 *  Copyright 2004 The Mana World Development Team
 *
 *  This file is part of The Mana World.
 *
 *  The Mana World is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  any later version.
 *
 *  The Mana World is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with The Mana World; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 *  $Id$
 */

#include "sqlitestorage.h"
#include <iostream>
#include <sstream>

/* Values for user level could be:
 *  0: Normal user
 *  1: Moderator (has medium level rights)
 *  2: Administrator (can do basically anything)
 */
const char sqlAccountTable[] =
  "create table tmw_accounts ("
    "id        int          unique primary key not null,"
    "username  varchar(32)  not null,"
    "password  varchar(32)  not null,"
    "email     varchar(128) not null,"
    "level     int          not null,"  // User level (normal, admin, etc.)
    "banned    int          not null"   // The UNIX time of unban (0 default)
  ")";

/* Note: The stats will need to be thought over, as we'll be implementing a
 *  much more elaborate skill based system. We should probably have a separate
 *  table for storing the skill levels.
 *
 * Gender is 0 for male, 1 for female.
 */
const char sqlCharacterTable[] =
  "create table tmw_characters ("
    "id        int          unique primary key not null,"
    "user_id   int          not null,"
    "name      varchar(32)  not null,"
    "gender    int          not null,"  // Player information
    "level     int          not null,"
    "money     int          not null,"
    "x         int          not null,"  // Location
    "y         int          not null,"
    "map       text         not null,"
    "str       int          not null,"  // Stats
    "agi       int          not null,"
    "vit       int          not null,"
    "int       int          not null,"
    "dex       int          not null,"
    "luck      int          not null,"
    "foreign key(user_id) references tmw_accounts(id)"
  ")";

/*
 * All items in the game world are stored in this table.
 */
const char sqlItemTable[] =
  "create table tmw_items ("
    "id        int          unique primary key not null,"
    "amount    int          not null,"  // Items of same kind can stack
    "type      int          not null,"  // Type as defined in item database
    "state     text"                    // Optional item state saved by script
  ")";

/*
 * Items on the ground in the game world.
 */
const char sqlWorldItemTable[] =
  "create table tmw_world_items ("
    "id        int          not null,"
    "map       text,"
    "x         int          not null,"  // Location of item on map
    "y         int          not null,"
    "deathtime int          not null,"  // Time to die (UNIX time)
    "primary key(id, map),"
    "foreign key(id) references tmw_items(id)"
  ")";

/*
 * Character Inventory
 */
const char sqlInventoryTable[] =
  "create table tmw_inventory ("
    "id        int          primary key not null," // Item ID
    "owner_id  int          not null," // Owner character ID
    "foreign key(id) references tmw_items(id),"
    "foreign key(owner_id) references tmw_characters(id)"
  ")";


SQLiteStorage::SQLiteStorage()
{
    // Open database
    if (sqlite.Open("tmw.db")) {
        std::cout << "Database: tmw.db created or opened" << std::endl;
    }
    else {
        std::cout << "Database: couldn't open tmw.db" << std::endl;
    }

    // Create tables
    createTablesIfNecessary();
}

SQLiteStorage::~SQLiteStorage()
{
    // Make sure any changes have been written
    flush();

    // Close database
    if (sqlite.Close())
    {
        std::cout << "Database: tmw.db closed" << std::endl;
    }
    else
    {
        std::cout << "Database: couldn't close tmw.db" << std::endl;
    }

    // Clean up managed accounts & characters
    for (unsigned int i = 0; i < accounts.size(); i++)
    {
        delete accounts[i];
    }
    for (unsigned int i = 0; i < characters.size(); i++)
    {
        delete characters[i];
    }
}

void SQLiteStorage::createTablesIfNecessary()
{
    SQLiteWrapper::ResultTable r;

    if (!sqlite.SelectStmt("select count(*) from sqlite_master where tbl_name='topics' and type='table'", r))
    {
        std::cout << "Error with select count(*) [createTablesIfNecessary]" << sqlite.LastError().c_str() << std::endl;
    }

    if (r.records_[0].fields_[0] != "0") {
        return;
    }

    sqlite.Begin();

    // Accounts table
    if (sqlite.DirectStatement(sqlAccountTable)) {
        std::cout << "Database: table tmw_accounts created" << std::endl;
    }
    else {
        std::cout << "Database: table tmw_accounts exists" << std::endl;
    }

    // Characters table
    if (sqlite.DirectStatement(sqlCharacterTable)) {
        std::cout << "Database: table tmw_characters created" << std::endl;
    }
    else {
        std::cout << "Database: table tmw_characters exists" << std::endl;
    }

    // Items table
    if (sqlite.DirectStatement(sqlItemTable)) {
        std::cout << "Database: table tmw_items created" << std::endl;
    }
    else {
        std::cout << "Database: table tmw_items exists" << std::endl;
    }

    // World Items table
    if (sqlite.DirectStatement(sqlWorldItemTable)) {
        std::cout << "Database: table tmw_world_items created" << std::endl;
    }
    else {
        std::cout << "Database: table tmw_world_items exists" << std::endl;
    }

    // Character Inventory
    if (sqlite.DirectStatement(sqlInventoryTable)) {
        std::cout << "Database: table tmw_inventory created" << std::endl;
    }
    else {
        std::cout << "Database: table tmw_inventory exists" << std::endl;
    }

    // Populate table for the hell of it ;)
    sqlite.DirectStatement("insert into tmw_accounts values (0, 'nym', 'tHiSiSHaShEd', 'nym@test', 1, 0)");
    sqlite.DirectStatement("insert into tmw_accounts values (1, 'Bjorn', 'tHiSiSHaShEd', 'bjorn@test', 1, 0)");
    sqlite.DirectStatement("insert into tmw_accounts values (2, 'Usiu', 'tHiSiSHaShEd', 'usiu@test', 1, 0)");
    sqlite.DirectStatement("insert into tmw_accounts values (3, 'ElvenProgrammer', 'tHiSiSHaShEd', 'elven@test', 1, 0)");
    sqlite.DirectStatement("insert into tmw_characters values (0, 0, 'Nym the Great', 0, 99, 1000000, 0, 0, 'main.map', 1, 2, 3, 4, 5, 6)");

    flush();
}

void SQLiteStorage::flush()
{
    sqlite.Commit();
}

unsigned int SQLiteStorage::getAccountCount()
{
    SQLiteWrapper::ResultTable r;
    std::stringstream s;
    unsigned int v;
    sqlite.SelectStmt("select count(*) from tmw_accounts", r);
    s << r.records_[0].fields_[0];
    s >> v;
    return v;
}

Account* SQLiteStorage::getAccount(const std::string &username)
{
    // Make sure account isn't loaded already
    for (unsigned int i = 0; i < accounts.size(); i++)
    {
        if (accounts[i]->getName() == username)
        {
            return accounts[i];
        }
    }

    // Give caller a initialized AccountData structure
    SQLiteWrapper::ResultTable r;

    std::string selectStatement =
        "select * from tmw_accounts where username = \"" + username + "\"";

    if (!sqlite.SelectStmt(selectStatement, r))
    {
        // Throwing an exception here could be good
        return NULL;
    }

    Account *acc = new Account();
    acc->setName(r.records_[0].fields_[1]);
    acc->setPassword(r.records_[0].fields_[2]);
    acc->setEmail(r.records_[0].fields_[3]);
    accounts.push_back(acc);

    std::string user_id = r.records_[0].fields_[0];

    // Load the characters associated with the account
    selectStatement =
        "select * from tmw_characters where id = \"" + user_id + "\"";

    if (!sqlite.SelectStmt(selectStatement, r))
    {
        // Throwing exception here could be good
        return NULL;
    }

    std::vector<Being*> beings;

    for (unsigned int i = 0; i < r.records_.size(); i++)
    {
        Being *being = new Being(
                r.records_[i].fields_[2], 1, 1, 1, 1, 1, 1, 1, 1);
        characters.push_back(being);
        beings.push_back(being);
    }

    acc->setCharacters(beings);

    return acc;
}