/*
 Copyright (C) 2013 Erik Ogenvik

 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., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

#include "ProxyMind.h"
#include "rules/Script.h"

#include "common/custom.h"
#include "common/operations/Think.h"
#include "common/debug.h"

#include <Atlas/Objects/Operation.h>

using Atlas::Message::Element;
using Atlas::Objects::Root;
using Atlas::Objects::Operation::Look;
using Atlas::Objects::Entity::RootEntity;
using Atlas::Objects::Entity::Anonymous;

using Atlas::Objects::smart_dynamic_cast;

static const bool debug_flag = false;

ProxyMind::ProxyMind(const std::string& id, long intId, LocatedEntity& ownerEntity) :
    m_ownerEntity(ownerEntity)
{

}

void ProxyMind::thinkSetOperation(const Operation& op, OpVector& res)
{
    //If it's a Set op named "persistthoughts" it's coming from an AI client, and is meant
    //for persisting thoughts. We should remove all existing thoughts first.
    if (!op->isDefaultName() && op->getName() == "persistthoughts") {
        m_randomThoughts.clear();
        m_thoughtsWithId.clear();
    }
    const std::vector<Root>& args = op->getArgs();
    for (const Root& arg : args) {
        if (arg->isDefaultId()) {
            m_randomThoughts.push_back(arg);
        } else {
            m_thoughtsWithId[arg->getId()] = arg;
        }
    }
    m_ownerEntity.addFlags(entity_dirty_thoughts);
    m_ownerEntity.onUpdated();
}

void ProxyMind::thinkDeleteOperation(const Operation& op, OpVector& res)
{
    const std::vector<Root>& args = op->getArgs();
    if (args.empty()) {
        //No args means "delete all"
        m_thoughtsWithId.clear();
        m_randomThoughts.clear();
    } else {
        for (const Root& arg : args) {
            if (arg->isDefaultId()) {
                log(WARNING, "Thought in Delete operation had no id set, ignoring.");
            } else {
                m_thoughtsWithId.erase(arg->getId());
            }
        }
    }
    m_ownerEntity.addFlags(entity_dirty_thoughts);
    m_ownerEntity.onUpdated();
}

void ProxyMind::thinkGetOperation(const Operation& op, OpVector& res)
{
    Atlas::Objects::Operation::Think think;
    Atlas::Objects::Operation::Set set;
    if (!op->isDefaultSerialno()) {
        think->setRefno(op->getSerialno());
    }
    std::vector<Root> thoughts;
    if (op->getArgs().empty()) {
        thoughts = getThoughts();
    } else {
        //If there's any arguments, we'll use the key of the first entry to filter on predicates
        auto frontArg = op->getArgs().front();
        Atlas::Message::MapType frontArgMap;
        frontArg->addToMessage(frontArgMap);
        if (!frontArgMap.empty()) {
            auto searchTerm = frontArgMap.begin()->first;
            for (auto& thought : m_thoughtsWithId) {
                if (thought.second->hasAttr(searchTerm)) {
                    thoughts.push_back(thought.second);
                }
            }
            for (auto& thought : m_randomThoughts) {
                if (thought->hasAttr(searchTerm)) {
                    thoughts.push_back(thought);
                }
            }
        }
    }
    set->setArgs(thoughts);
    think->setArgs1(set);
    res.push_back(think);
}

void ProxyMind::thinkLookOperation(const Operation& op, OpVector& res)
{
    Atlas::Objects::Operation::Think think;
    Atlas::Objects::Operation::Info info;
    if (!op->isDefaultSerialno()) {
        think->setRefno(op->getSerialno());
    }
    std::vector<Root> thoughts;
    if (op->getArgs().empty()) {
        thoughts = getThoughts();
    } else {
        //If there's any arguments, we'll use the key of the first entry to filter on predicates
        auto frontArg = op->getArgs().front();
        Atlas::Message::MapType frontArgMap;
        frontArg->addToMessage(frontArgMap);
        if (!frontArgMap.empty()) {
            auto searchTerm = frontArgMap.begin()->first;
            for (auto& thought : m_thoughtsWithId) {
                if (thought.second->hasAttr(searchTerm)) {
                    thoughts.push_back(thought.second);
                }
            }
            for (auto& thought : m_randomThoughts) {
                if (thought->hasAttr(searchTerm)) {
                    thoughts.push_back(thought);
                }
            }
        }
    }
    info->setArgs(thoughts);
    think->setArgs1(info);
    res.push_back(think);
}

std::vector<Atlas::Objects::Root> ProxyMind::getThoughts() const
{
    std::vector<Atlas::Objects::Root> thoughts = m_randomThoughts;
    for (auto& thought : m_thoughtsWithId) {
        thoughts.push_back(thought.second);
    }
    return thoughts;
}

void ProxyMind::clearThoughts()
{
    m_randomThoughts.clear();
    m_thoughtsWithId.clear();
}

void ProxyMind::operation(const Operation& op, OpVector& res)
{
    //For proxy minds we're only interested in Think ops; all others are ignored.
    auto op_no = op->getClassNo();
    if (op_no == Atlas::Objects::Operation::THINK_NO) {
        ThinkOperation(op, res);
    }

}

void ProxyMind::ThinkOperation(const Operation& op, OpVector& res)
{
    //Get the contained op
    debug_print("ProxyMind::ThinkOperation(Think)");
    const std::vector<Root>& args = op->getArgs();
    if (args.empty()) {
        debug_print(" no args!");
        return;
    }
    const Root& arg = args.front();
    Operation op2(Atlas::Objects::smart_dynamic_cast<Operation>(arg));
    if (op2.isValid()) {
        debug_print(" args is an op!");
        std::string event_name("think_");
        event_name += op2->getParent();

        OpVector mres;

        int op2ClassNo = op2->getClassNo();
        switch (op2ClassNo) {
            case Atlas::Objects::Operation::SET_NO:
                thinkSetOperation(op2, mres);
                break;
            case Atlas::Objects::Operation::DELETE_NO:
                thinkDeleteOperation(op2, mres);
                break;
            case Atlas::Objects::Operation::GET_NO:
                thinkGetOperation(op2, mres);
                break;
            case Atlas::Objects::Operation::LOOK_NO:
                thinkLookOperation(op2, mres);
                break;
            default:
                log(WARNING, "Got invalid Think operation. We only support 'Set', 'Delete', 'Get' and 'Look'.");
                break;
        }

        if (!mres.empty() && !op->isDefaultSerialno()) {
            mres.front()->setRefno(op->getSerialno());
        }
        for (auto& resOp : mres) {
            res.push_back(resOp);
        }

    }
}