/*
 * 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 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "UserRequestHandlerGroup.h"

#include <Common/Logger.h>
#include "../UserManager.h"

namespace Mad {
namespace Server {
namespace RequestHandlers {

void UserRequestHandlerGroup::handleUserListRequest(boost::shared_ptr<const Common::XmlPacket> packet _UNUSED_PARAMETER_, Common::XmlPacket *ret) {
  boost::shared_ptr<std::map<unsigned long, Common::UserInfo> > info = UserManager::get()->getUserList();

  ret->setType("OK");
  ret->addList("users");

  for(std::map<unsigned long, Common::UserInfo>::iterator user = info->begin(); user != info->end(); ++user) {
    (*ret)["users"].addEntry();
    Common::XmlPacket::Entry &entry = (*ret)["users"].back();

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

void UserRequestHandlerGroup::handleUserInfoRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret) {
  boost::shared_ptr<Common::UserInfo> info = UserManager::get()->getUserInfo((*packet)["uid"]);

  ret->setType("OK");

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

void UserRequestHandlerGroup::handleUserGroupListRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret) {
  boost::shared_ptr<std::set<unsigned long> > groups = UserManager::get()->getUserGroupList((unsigned long)(*packet)["uid"]);

  ret->setType("OK");
  ret->addList("groups");

  for(std::set<unsigned long>::iterator group = groups->begin(); group != groups->end(); ++group) {
    (*ret)["groups"].addEntry();
    Common::XmlPacket::Entry &entry = (*ret)["groups"].back();

    entry.add("gid", *group);
  }
}

void UserRequestHandlerGroup::handleGroupListRequest(boost::shared_ptr<const Common::XmlPacket> packet _UNUSED_PARAMETER_, Common::XmlPacket *ret) {
  boost::shared_ptr<std::map<unsigned long, Common::GroupInfo> > info = UserManager::get()->getGroupList();

  ret->setType("OK");
  ret->addList("groups");

  for(std::map<unsigned long, Common::GroupInfo>::iterator group = info->begin(); group != info->end(); ++group) {
    (*ret)["groups"].addEntry();
    Common::XmlPacket::Entry &entry = (*ret)["groups"].back();

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

void UserRequestHandlerGroup::handleGroupUserListRequest(boost::shared_ptr<const Common::XmlPacket> packet, Common::XmlPacket *ret) {
  boost::shared_ptr<std::set<unsigned long> > users = UserManager::get()->getGroupUserList((unsigned long)(*packet)["gid"]);

  ret->setType("OK");
  ret->addList("users");

  for(std::set<unsigned long>::iterator user = users->begin(); user != users->end(); ++user) {
    (*ret)["users"].addEntry();
    Common::XmlPacket::Entry &entry = (*ret)["users"].back();

    entry.add("uid", *user);
  }
}

UserRequestHandlerGroup::UserRequestHandlerGroup() {
  registerHandler("ListUsers", &UserRequestHandlerGroup::handleUserListRequest);
  registerHandler("GetUserInfo", &UserRequestHandlerGroup::handleUserInfoRequest);
  registerHandler("ListUserGroups", &UserRequestHandlerGroup::handleUserGroupListRequest);
  registerHandler("ListGroups", &UserRequestHandlerGroup::handleGroupListRequest);
  registerHandler("ListGroupUsers", &UserRequestHandlerGroup::handleGroupUserListRequest);
}

}
}
}