/*
 * UserRequestHandlerGroup.cpp
 *
 * Copyright (C) 2009 Matthias Schiffer <matthias@gamezock.de>
 *
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by the
 * Free Software Foundation, either version 3 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "UserRequestHandlerGroup.h"
#include "../Application.h"

#include <Common/UserManager.h>

#include <boost/date_time/posix_time/posix_time.hpp>

namespace Mad {
namespace Server {
namespace RequestHandlers {

void UserRequestHandlerGroup::handleUserListRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  boost::posix_time::ptime timestamp(boost::posix_time::not_a_date_time);

  const std::string &timestr = packet->get<const std::string&>("timestamp");
  if(!timestr.empty()) {
    try {
      timestamp = boost::posix_time::from_iso_string(timestr);
    }
    catch(...) {}
  }

  boost::shared_ptr<const std::map<unsigned long, Common::UserInfo> > users = application->getUserManager()->getUserList(&timestamp);

  ret->setType("OK");

  if(!timestamp.is_not_a_date_time())
    ret->set("timestamp", boost::posix_time::to_iso_string(timestamp));

  if(users) {
    Common::XmlPacket::List *list = ret->createList("users");

    for(std::map<unsigned long, Common::UserInfo>::const_iterator user = users->begin(); user != users->end(); ++user) {
      Common::XmlPacket::List::iterator entry = list->addEntry();

      entry->set("uid", user->second.getUid());
      entry->set("gid", user->second.getGid());
      entry->set("username", user->second.getUsername());
      entry->set("fullName", user->second.getFullName());
    }
  }
}

void UserRequestHandlerGroup::handleUserInfoRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  boost::posix_time::ptime timestamp(boost::posix_time::not_a_date_time);

  const std::string &timestr = packet->get<const std::string&>("timestamp");
  if(!timestr.empty()) {
    try {
      timestamp = boost::posix_time::from_iso_string(timestr);
    }
    catch(...) {}
  }

  boost::shared_ptr<const Common::UserInfo> info;

  unsigned long uid = packet->get<unsigned long>("uid");
  if(uid)
    info = application->getUserManager()->getUserInfo(uid, &timestamp);
  else
    info = application->getUserManager()->getUserInfoByName(packet->get<const std::string&>("name"), &timestamp);

  ret->setType("OK");

  if(!timestamp.is_not_a_date_time())
    ret->set("timestamp", boost::posix_time::to_iso_string(timestamp));

  if(info) {
    ret->set("uid", info->getUid());
    ret->set("gid", info->getGid());
    ret->set("username", info->getUsername());
    ret->set("fullName", info->getFullName());
  }
}

void UserRequestHandlerGroup::handleUserGroupListRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  boost::posix_time::ptime timestamp(boost::posix_time::not_a_date_time);

  const std::string &timestr = packet->get<const std::string&>("timestamp");
  if(!timestr.empty()) {
    try {
      timestamp = boost::posix_time::from_iso_string(timestr);
    }
    catch(...) {}
  }

  boost::shared_ptr<const std::set<unsigned long> > groups = application->getUserManager()->getUserGroupList(packet->get<unsigned long>("uid"), &timestamp);

  ret->setType("OK");

  if(!timestamp.is_not_a_date_time())
    ret->set("timestamp", boost::posix_time::to_iso_string(timestamp));

  if(groups) {
    Common::XmlPacket::List *list = ret->createList("groups");

    for(std::set<unsigned long>::const_iterator group = groups->begin(); group != groups->end(); ++group) {
      Common::XmlPacket::List::iterator entry = list->addEntry();

      entry->set("gid", *group);
    }
  }
}

void UserRequestHandlerGroup::handleGroupListRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  boost::posix_time::ptime timestamp(boost::posix_time::not_a_date_time);

  const std::string &timestr = packet->get<const std::string&>("timestamp");
  if(!timestr.empty()) {
    try {
      timestamp = boost::posix_time::from_iso_string(timestr);
    }
    catch(...) {}
  }

  boost::shared_ptr<const std::map<unsigned long, Common::GroupInfo> > groups = application->getUserManager()->getGroupList(&timestamp);

  ret->setType("OK");

  if(!timestamp.is_not_a_date_time())
    ret->set("timestamp", boost::posix_time::to_iso_string(timestamp));

  if(groups) {
    Common::XmlPacket::List *list = ret->createList("groups");

    for(std::map<unsigned long, Common::GroupInfo>::const_iterator group = groups->begin(); group != groups->end(); ++group) {
      Common::XmlPacket::List::iterator entry = list->addEntry();

      entry->set("gid", group->second.getGid());
      entry->set("name", group->second.getName());
    }
  }
}

void UserRequestHandlerGroup::handleGroupInfoRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  boost::posix_time::ptime timestamp(boost::posix_time::not_a_date_time);

  const std::string &timestr = packet->get<const std::string&>("timestamp");
  if(!timestr.empty()) {
    try {
      timestamp = boost::posix_time::from_iso_string(timestr);
    }
    catch(...) {}
  }

  boost::shared_ptr<const Common::GroupInfo> info;

  unsigned long gid = packet->get<unsigned long>("gid");
  if(gid)
    info = application->getUserManager()->getGroupInfo(gid, &timestamp);
  else
    info = application->getUserManager()->getGroupInfoByName(packet->get<const std::string&>("name"), &timestamp);

  ret->setType("OK");

  if(!timestamp.is_not_a_date_time())
    ret->set("timestamp", boost::posix_time::to_iso_string(timestamp));

  if(info) {
    ret->set("gid", info->getGid());
    ret->set("name", info->getName());
  }
}

void UserRequestHandlerGroup::handleGroupUserListRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  boost::posix_time::ptime timestamp(boost::posix_time::not_a_date_time);

  const std::string &timestr = packet->get<const std::string&>("timestamp");
  if(!timestr.empty()) {
    try {
      timestamp = boost::posix_time::from_iso_string(timestr);
    }
    catch(...) {}
  }

  boost::shared_ptr<const std::set<unsigned long> > users = application->getUserManager()->getGroupUserList(packet->get<unsigned long>("gid"), &timestamp);

  ret->setType("OK");

  if(!timestamp.is_not_a_date_time())
    ret->set("timestamp", boost::posix_time::to_iso_string(timestamp));

  Common::XmlPacket::List *list = ret->createList("users");

  for(std::set<unsigned long>::const_iterator user = users->begin(); user != users->end(); ++user) {
    Common::XmlPacket::List::iterator entry = list->addEntry();

    entry->set("uid", *user);
  }
}

void UserRequestHandlerGroup::handleFullUserGroupListRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  boost::posix_time::ptime timestamp(boost::posix_time::not_a_date_time);

  const std::string &timestr = packet->get<const std::string&>("timestamp");
  if(!timestr.empty()) {
    try {
      timestamp = boost::posix_time::from_iso_string(timestr);
    }
    catch(...) {}
  }

  boost::shared_ptr<const std::multimap<unsigned long, unsigned long> > userGroups = application->getUserManager()->getFullUserGroupList(&timestamp);

  ret->setType("OK");

  if(!timestamp.is_not_a_date_time())
    ret->set("timestamp", boost::posix_time::to_iso_string(timestamp));

  if(userGroups) {
    Common::XmlPacket::List *list = ret->createList("userGroupList");

    for(std::multimap<unsigned long, unsigned long>::const_iterator userGroup = userGroups->begin(); userGroup != userGroups->end(); ++userGroup) {
      Common::XmlPacket::List::iterator entry = list->addEntry();

      entry->set("uid", userGroup->first);
      entry->set("gid", userGroup->second);
    }
  }
}


void UserRequestHandlerGroup::handleUserInfoCheckRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  Common::UserInfo userInfo(packet->get<unsigned long>("uid"), packet->get<const std::string&>("username"));
  userInfo.setGid(packet->get<unsigned long>("gid"));
  userInfo.setFullName(packet->get<const std::string&>("fullName"));

  application->getUserManager()->checkUserInfo(userInfo);

  ret->setType("OK");
}

void UserRequestHandlerGroup::handleUserAddRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  Common::UserInfo userInfo(packet->get<unsigned long>("uid"), packet->get<const std::string&>("username"));
  userInfo.setGid(packet->get<unsigned long>("gid"));
  userInfo.setFullName(packet->get<const std::string&>("fullName"));

  application->getUserManager()->addUser(userInfo);

  ret->setType("OK");
}

void UserRequestHandlerGroup::handleUserUpdateRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  Common::UserInfo userInfo(packet->get<unsigned long>("uid"), packet->get<const std::string&>("username"));
  userInfo.setGid(packet->get<unsigned long>("gid"));
  userInfo.setFullName(packet->get<const std::string&>("fullName"));

  application->getUserManager()->updateUser(packet->get<unsigned long>("origUid"), userInfo);

  ret->setType("OK");
}

void UserRequestHandlerGroup::handleUserDeleteRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  application->getUserManager()->deleteUser(packet->get<unsigned long>("uid"));

  ret->setType("OK");
}


void UserRequestHandlerGroup::handleGroupInfoCheckRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  application->getUserManager()->checkGroupInfo(Common::GroupInfo(packet->get<unsigned long>("gid"), packet->get<const std::string&>("name")));

  ret->setType("OK");
}

void UserRequestHandlerGroup::handleGroupAddRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  application->getUserManager()->addGroup(Common::GroupInfo(packet->get<unsigned long>("gid"), packet->get<const std::string&>("name")));

  ret->setType("OK");
}

void UserRequestHandlerGroup::handleGroupUpdateRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  application->getUserManager()->updateGroup(packet->get<unsigned long>("origGid"),
      Common::GroupInfo(packet->get<unsigned long>("gid"), packet->get<const std::string&>("name")));

  ret->setType("OK");
}

void UserRequestHandlerGroup::handleGroupDeleteRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  application->getUserManager()->deleteGroup(packet->get<unsigned long>("gid"));

  ret->setType("OK");
}


void UserRequestHandlerGroup::handleAddUserToGroupRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  application->getUserManager()->addUserToGroup(packet->get<unsigned long>("uid"), packet->get<unsigned long>("gid"));

  ret->setType("OK");
}

void UserRequestHandlerGroup::handleDeleteUserFromGroupRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  application->getUserManager()->deleteUserFromGroup(packet->get<unsigned long>("uid"), packet->get<unsigned long>("gid"));

  ret->setType("OK");
}

void UserRequestHandlerGroup::handlePasswordSetRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret,
    Common::Connection* /*connection*/) {
  application->getUserManager()->setPassword(packet->get<unsigned long>("uid"), packet->get<const std::string&>("password"));

  ret->setType("OK");
}

UserRequestHandlerGroup::UserRequestHandlerGroup(Application *application0) : application(application0) {
  registerHandler("ListUsers", boost::bind(&UserRequestHandlerGroup::handleUserListRequest, this, _1, _2, _3));
  registerHandler("GetUserInfo", boost::bind(&UserRequestHandlerGroup::handleUserInfoRequest, this, _1, _2, _3));
  registerHandler("ListUserGroups", boost::bind(&UserRequestHandlerGroup::handleUserGroupListRequest, this, _1, _2, _3));

  registerHandler("ListGroups", boost::bind(&UserRequestHandlerGroup::handleGroupListRequest, this, _1, _2, _3));
  registerHandler("GetGroupInfo", boost::bind(&UserRequestHandlerGroup::handleGroupInfoRequest, this, _1, _2, _3));
  registerHandler("ListGroupUsers", boost::bind(&UserRequestHandlerGroup::handleGroupUserListRequest, this, _1, _2, _3));

  registerHandler("GetFullUserGroupList", boost::bind(&UserRequestHandlerGroup::handleFullUserGroupListRequest, this, _1, _2, _3));

  registerHandler("CheckUserInfo", boost::bind(&UserRequestHandlerGroup::handleUserInfoCheckRequest, this, _1, _2, _3));
  registerHandler("AddUser", boost::bind(&UserRequestHandlerGroup::handleUserAddRequest, this, _1, _2, _3));
  registerHandler("UpdateUser", boost::bind(&UserRequestHandlerGroup::handleUserUpdateRequest, this, _1, _2, _3));
  registerHandler("DeleteUser", boost::bind(&UserRequestHandlerGroup::handleUserDeleteRequest, this, _1, _2, _3));

  registerHandler("CheckGroupInfo", boost::bind(&UserRequestHandlerGroup::handleGroupInfoCheckRequest, this, _1, _2, _3));
  registerHandler("AddGroup", boost::bind(&UserRequestHandlerGroup::handleGroupAddRequest, this, _1, _2, _3));
  registerHandler("UpdateGroup", boost::bind(&UserRequestHandlerGroup::handleGroupUpdateRequest, this, _1, _2, _3));
  registerHandler("DeleteGroup", boost::bind(&UserRequestHandlerGroup::handleGroupDeleteRequest, this, _1, _2, _3));

  registerHandler("AddUserToGroup", boost::bind(&UserRequestHandlerGroup::handleAddUserToGroupRequest, this, _1, _2, _3));
  registerHandler("DeleteUserFromGroup", boost::bind(&UserRequestHandlerGroup::handleDeleteUserFromGroupRequest, this, _1, _2, _3));

  registerHandler("SetPassword", boost::bind(&UserRequestHandlerGroup::handlePasswordSetRequest, this, _1, _2, _3));
}

}
}
}