// Cyphesis Online RPG Server and AI Engine
// Copyright (C) 2000 Alistair Riddoch
//
// This program 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
// (at your option) any later version.
// 
// This program 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 this program; if not, write to the Free Software Foundation,
// Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA


#include "Py_CreatorClient.h"

#include "CreatorClient.h"
#include "ClientConnection.h"

#include "rulesets/Py_Operation.h"
#include "rulesets/Py_RootEntity.h"
#include "rulesets/Py_WorldTime.h"
#include "rulesets/Py_Location.h"
#include "rulesets/Py_Message.h"
#include "rulesets/Py_Thing.h"
#include "rulesets/Py_MemMap.h"

#include "common/debug.h"
#include "common/id.h"
#include "common/TypeNode.h"

using Atlas::Message::Element;
using Atlas::Message::MapType;

static const bool debug_flag = false;

static PyObject * CreatorClient_as_entity(PyCreatorClient * self, PyObject *)
{
#ifndef NDEBUG
    if (self->m_mind.c == nullptr) {
        PyErr_SetString(PyExc_AssertionError, "nullptr CreatorClient in CreatorClient.as_entity");
        return nullptr;
    }
#endif // NDEBUG
    PyMessage * ret = newPyMessage();
    if (ret != nullptr) {
        ret->m_obj = new Element(MapType());
        self->m_mind.c->addToMessage(ret->m_obj->asMap());
    }
    return (PyObject *)ret;
}

static PyObject * CreatorClient_make(PyCreatorClient * self,
                                     PyRootEntity * entity)
{
#ifndef NDEBUG
    if (self->m_mind.c == nullptr) {
        PyErr_SetString(PyExc_AssertionError, "nullptr CreatorClient in CreatorClient.make");
        return nullptr;
    }
#endif // NDEBUG
    if (!PyRootEntity_Check(entity)) {
        PyErr_SetString(PyExc_TypeError, "Can only make Atlas entity");
        return nullptr;
    }
    LocatedEntity * retval = self->m_mind.a->make(entity->entity);
    if (retval == nullptr) {
        PyErr_SetString(PyExc_RuntimeError, "Entity creation failed");
        return nullptr;
    }
    PyEntity * ret = newPyLocatedEntity();
    if (ret != nullptr) {
        ret->m_entity.l = retval;
    }
    return (PyObject *)ret;
}

static PyObject * CreatorClient_set(PyCreatorClient * self, PyObject * args)
{
#ifndef NDEBUG
    if (self->m_mind.c == nullptr) {
        PyErr_SetString(PyExc_AssertionError, "nullptr CreatorClient in CreatorClient.set");
        return nullptr;
    }
#endif // NDEBUG
    PyRootEntity * entity = nullptr;
    char * id = nullptr;
    if (!PyArg_ParseTuple(args, "sO", &id, &entity)) {
        return nullptr;
    }
    if (!PyRootEntity_Check(entity)) {
        PyErr_SetString(PyExc_TypeError, "Can only set Atlas entity");
        return nullptr;
    }
    self->m_mind.a->sendSet(id, entity->entity);
    Py_INCREF(Py_None);
    return Py_None;
}

static PyObject * CreatorClient_look(PyCreatorClient * self, PyObject * py_id)
{
#ifndef NDEBUG
    if (self->m_mind.c == nullptr) {
        PyErr_SetString(PyExc_AssertionError, "nullptr CreatorClient in CreatorClient.look");
        return nullptr;
    }
#endif // NDEBUG
    if (!PyUnicode_CheckExact(py_id)) {
        PyErr_SetString(PyExc_TypeError, "CreatorClient.look must be a string");
        return nullptr;
    }
    char * id = PyUnicode_AsUTF8(py_id);
    LocatedEntity * retval = self->m_mind.c->look(id);
    if (retval == nullptr) {
        PyErr_SetString(PyExc_RuntimeError, "Entity look failed");
        return nullptr;
    }
    PyEntity * ret = newPyLocatedEntity();
    if (ret != nullptr) {
        ret->m_entity.l = retval;
    }
    return (PyObject *)ret;
}
static PyObject * CreatorClient_look_for(PyCreatorClient * self,
                                         PyRootEntity * ent)
{
#ifndef NDEBUG
    if (self->m_mind.c == nullptr) {
        PyErr_SetString(PyExc_AssertionError, "nullptr CreatorClient in CreatorClient.look_for");
        return nullptr;
    }
#endif // NDEBUG
    if (!PyRootEntity_Check(ent)) {
        PyErr_SetString(PyExc_TypeError, "Can only look for Atlas description");
        return nullptr;
    }
    LocatedEntity * retval = self->m_mind.c->lookFor(ent->entity);
    if (retval == nullptr) {
        Py_INCREF(Py_None);
        return Py_None;
    }
    PyEntity * ret = newPyLocatedEntity();
    if (ret != nullptr) {
        ret->m_entity.l = retval;
    }
    return (PyObject *)ret;
}

static PyObject * CreatorClient_send(PyCreatorClient * self, PyOperation * op)
{
#ifndef NDEBUG
    if (self->m_mind.c == nullptr) {
        PyErr_SetString(PyExc_AssertionError, "nullptr CreatorClient in CreatorClient.send");
        return nullptr;
    }
#endif // NDEBUG
    if (!PyOperation_Check(op)) {
        PyErr_SetString(PyExc_TypeError, "Can only send Atlas operation");
        return nullptr;
    }
    self->m_mind.c->send(op->operation);
    Py_INCREF(Py_None);
    return Py_None;
}

static PyObject * CreatorClient_delete(PyCreatorClient * self, PyObject * py_id)
{
#ifndef NDEBUG
    if (self->m_mind.c == nullptr) {
        PyErr_SetString(PyExc_AssertionError, "nullptr CreatorClient in CreatorClient.send");
        return nullptr;
    }
#endif // NDEBUG
    if (!PyUnicode_CheckExact(py_id)) {
        PyErr_SetString(PyExc_TypeError, "CreatorClient.delete must be a string");
        return nullptr;
    }
    char * id = PyUnicode_AsUTF8(py_id);
    self->m_mind.a->del(id);
    Py_INCREF(Py_None);
    return Py_None;
}

#if 0
static PyMethodDef CharacterClient_methods[] = {
        {"as_entity",      (PyCFunction)CreatorClient_as_entity, METH_NOARGS},
        {"make",           (PyCFunction)CreatorClient_make,      METH_O},
        {"look",           (PyCFunction)CreatorClient_look,      METH_O},
        {"look_for",       (PyCFunction)CreatorClient_look_for,  METH_O},
        {"send",           (PyCFunction)CreatorClient_send,      METH_O},
        {nullptr,          nullptr}           /* sentinel */
};
#endif

static PyMethodDef CreatorClient_methods[] = {
        {"as_entity",      (PyCFunction)CreatorClient_as_entity, METH_NOARGS},
        {"make",           (PyCFunction)CreatorClient_make,      METH_O},
        {"set",            (PyCFunction)CreatorClient_set,       METH_VARARGS},
        {"look",           (PyCFunction)CreatorClient_look,      METH_O},
        {"look_for",       (PyCFunction)CreatorClient_look_for,  METH_O},
        {"send",           (PyCFunction)CreatorClient_send,      METH_O},
        {"delete",         (PyCFunction)CreatorClient_delete,    METH_O},
        {nullptr,          nullptr}           /* sentinel */
};

static PyObject * CreatorClient_getattro(PyCreatorClient *self,
                                         PyObject *oname)
{
    // Fairly major re-write of this to use operator[] of CreatorClient base class
#ifndef NDEBUG
    if (self->m_mind.c == nullptr) {
        PyErr_SetString(PyExc_AssertionError, "nullptr CreatorClient in CreatorClient.getattr");
        return nullptr;
    }
#endif // NDEBUG
    char * name = PyUnicode_AsUTF8(oname);
    // If operation search gets to here, it goes no further
    if (strstr(name, "_operation") != nullptr) {
        PyErr_SetString(PyExc_AttributeError, name);
        return nullptr;
    }
    if (strcmp(name, "type") == 0) {
        PyObject * list = PyList_New(0);
        if (list == nullptr) {
            return nullptr;
        }
        PyList_Append(list, PyUnicode_FromString(self->m_mind.c->getType()->name().c_str()));
        return list;
    }
    if (strcmp(name, "map") == 0) {
        PyMemMap * map = newPyMemMap();
        if (map != nullptr) {
            map->m_map = self->m_mind.c->getMap();
        }
        return (PyObject *)map;
    }
    if (strcmp(name, "location") == 0) {
        PyLocation * loc = newPyLocation();
        if (loc != nullptr) {
            loc->location = &self->m_mind.c->m_location;
            loc->owner = self->m_mind.c;
        }
        return (PyObject *)loc;
    }
    if (strcmp(name, "time") == 0) {
        PyWorldTime * worldtime = newPyWorldTime();
        if (worldtime != nullptr) {
            worldtime->time = self->m_mind.c->getTime();
        }
        return (PyObject *)worldtime;
    }
    LocatedEntity * thing = self->m_mind.c;
    Element attr;
    if (thing->getAttr(name, attr) == 0) {
        PyObject * ret = MessageElement_asPyObject(attr);
        // FIXME Set an error
        return ret;
    }
    return PyObject_GenericGetAttr((PyObject *)self, oname);
}

static int CreatorClient_setattro(PyCreatorClient *self,
                                  PyObject * oname,
                                  PyObject *v)
{
    char * name = PyUnicode_AsUTF8(oname);
    if (self->m_mind.c == nullptr) {
        // FIXME Set an error
        return -1;
    }
    if (strcmp(name, "map") == 0) {
        PyErr_SetString(PyExc_AttributeError, "map attribute forbidden");
        return -1;
    }
    LocatedEntity * thing = self->m_mind.c;
    //std::string attr(name);
    //if (v == nullptr) {
        //thing->attributes.erase(attr);
        //return 0;
    //}
    Element obj;
    if (PyObject_asMessageElement(v, obj) == 0 && !obj.isMap() && !obj.isList()) {
        thing->setAttr(name, obj);
        return 0;
    }
    PyErr_SetString(PyExc_ValueError, "attribute must be a simple type");
    return -1;
}

static PyObject* CreatorClient_compare(PyObject *a, PyObject *b, int op)
{
    auto self = (PyCreatorClient*)a;
    if (PyCreatorClient_Check(b)) {
        auto other = (PyCreatorClient*)b;
        if (op == Py_EQ) {
            return self->m_mind.c == other->m_mind.c ? Py_True : Py_False;
        }
    }

    return Py_NotImplemented;
}

static int CreatorClient_init(PyCreatorClient * self,
                              PyObject * args, PyObject * kwds)
{
    char * id = nullptr;

    if (!PyArg_ParseTuple(args, "s", &id)) {
        return -1;
    }
    long intId = integerId(id);
    if (intId == -1L) {
        PyErr_SetString(PyExc_ValueError, "CreatorClient() requires string/int ID");
        return -1;
    }
    // This is just here for testing, as there is currently no suppor in the
    // API for creating this in a meaningful way. A CreatorClient must have
    // a connection, so we create one here even though it is not in a connected
    // state. The connection is stubbed out in the tests, so this doesn't
    // matter.
    self->m_mind.c = new CreatorClient(id, intId, *new ClientConnection);
    return 0;
}

PyTypeObject PyCharacterClient_Type = {
        PyVarObject_HEAD_INIT(&PyType_Type, 0)
        "CreatorClient",                        /*tp_name*/
        sizeof(PyCreatorClient),                /*tp_basicsize*/
        0,                                      /*tp_itemsize*/
        /* methods */
        0,                                      /*tp_dealloc*/
        0,                                      /*tp_print*/
        0,                                      /*tp_getattr*/
        0,                                      /*tp_setattr*/
        0,                                      /*tp_compare*/
        0,                                      /*tp_repr*/
        0,                                      /*tp_as_number*/
        0,                                      /*tp_as_sequence*/
        0,                                      /*tp_as_mapping*/
        0,                                      /*tp_hash*/
        0,                                      // tp_call
        0,                                      // tp_str
        (getattrofunc)CreatorClient_getattro,   // tp_getattro
        (setattrofunc)CreatorClient_setattro,   // tp_setattro
        0,                                      // tp_as_buffer
        Py_TPFLAGS_DEFAULT,                     // tp_flags
        "CreatorClient objects",                // tp_doc
        0,                                      // tp_travers
        0,                                      // tp_clear
        (richcmpfunc)CreatorClient_compare,     // tp_richcompare
        0,                                      // tp_weaklistoffset
        0,                                      // tp_iter
        0,                                      // tp_iternext
        0,                                      // tp_methods
        0,                                      // tp_members
        0,                                      // tp_getset
        0,                                      // tp_base
        0,                                      // tp_dict
        0,                                      // tp_descr_get
        0,                                      // tp_descr_set
        0,                                      // tp_dictoffset
        (initproc)CreatorClient_init,           // tp_init
        0,                                      // tp_alloc
        0,                                      // tp_new
};

PyTypeObject PyCreatorClient_Type = {
        PyVarObject_HEAD_INIT(&PyType_Type, 0)
        "CreatorClient",                     /*tp_name*/
        sizeof(PyCreatorClient),                /*tp_basicsize*/
        0,                                      /*tp_itemsize*/
        /* methods */
        0,                                      /*tp_dealloc*/
        0,                                      /*tp_print*/
        0,                                      /*tp_getattr*/
        0,                                      /*tp_setattr*/
        0,         /*tp_compare*/
        0,                                      /*tp_repr*/
        0,                                      /*tp_as_number*/
        0,                                      /*tp_as_sequence*/
        0,                                      /*tp_as_mapping*/
        0,                                      /*tp_hash*/
        0,                                      // tp_call
        0,                                      // tp_str
        (getattrofunc)CreatorClient_getattro,   // tp_getattro
        (setattrofunc)CreatorClient_setattro,   // tp_setattro
        0,                                      // tp_as_buffer
        Py_TPFLAGS_DEFAULT,                     // tp_flags
        "CreatorClient objects",                // tp_doc
        0,                                      // tp_travers
        0,                                      // tp_clear
        (richcmpfunc)CreatorClient_compare,     // tp_richcompare
        0,                                      // tp_weaklistoffset
        0,                                      // tp_iter
        0,                                      // tp_iternext
        CreatorClient_methods,                  // tp_methods
        0,                                      // tp_members
        0,                                      // tp_getset
        0,                                      // tp_base
        0,                                      // tp_dict
        0,                                      // tp_descr_get
        0,                                      // tp_descr_set
        0,                                      // tp_dictoffset
        (initproc)CreatorClient_init,           // tp_init
        0,                                      // tp_alloc
        0,                                      // tp_new
};

PyCreatorClient * newPyCreatorClient()
{
    return (PyCreatorClient *)PyCreatorClient_Type.tp_new(&PyCreatorClient_Type, 0, 0);
}