/*
 * UserCommands.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 "UserCommands.h"
#include "Application.h"
#include "CommandParser.h"

#include <Common/RequestManager.h>

#include <Common/Requests/GroupListRequest.h>
#include <Common/Requests/GroupUserListRequest.h>
#include <Common/Requests/UserInfoRequest.h>
#include <Common/Requests/UserListRequest.h>
#include <Common/Requests/UserGroupListRequest.h>

#include <iostream>


namespace Mad {
namespace Client {

void UserCommands::userInfoCommand(CommandParser *commandParser, const std::vector<std::string> &args) {
  if(args.size() == 1) {
    std::cerr << args[0] << ": No user id given." << std::endl;
    commandParser->printUsage("user_info");
    return;
  }
  if(args.size() > 2) {
    std::cerr << args[0] << ": Too many arguments." << std::endl;
    commandParser->printUsage("user_info");
    return;
  }

  char *endptr;
  unsigned long uid = std::strtoul(args[1].c_str(), &endptr, 10);
  if(args[1].empty() || *endptr != '\0') {
    std::cerr << args[0] << ": Unable to parse user id." << std::endl;
    commandParser->printUsage("user_info");
    return;
  }

  boost::shared_ptr<Common::Requests::UserInfoRequest> request(new Common::Requests::UserInfoRequest(commandParser->application, uid));
  commandParser->application->getRequestManager()->sendRequest(commandParser->connection, request);
  request->wait();

  std::pair<boost::shared_ptr<const Common::XmlPacket>, Core::Exception> result = request->getResult();

  if(!result.first || result.second) {
    std::cerr << "An error occurred during your request: " << result.second.strerror() << "." << std::endl;
  }
  else {
    const Common::XmlPacket &packet = *result.first;

    std::cout << "  " << packet.get<unsigned long>("uid") << ", " << packet.get<unsigned long>("gid") << ", "
    << packet.get<const std::string&>("username") << ", " << packet.get<const std::string&>("fullName") << std::endl << std::endl;
  }
}

void UserCommands::listUsersCommand(CommandParser *commandParser, const std::vector<std::string> &args _UNUSED_PARAMETER_) {
  boost::shared_ptr<Common::Requests::UserListRequest> request(new Common::Requests::UserListRequest(commandParser->application));

  commandParser->application->getRequestManager()->sendRequest(commandParser->connection, request);
  request->wait();

  std::pair<boost::shared_ptr<const Common::XmlPacket>, Core::Exception> result = request->getResult();

  if(!result.first || result.second) {
    std::cerr << "An error occurred during your request: " << result.second.strerror() << "." << std::endl;
  }
  else {
    const Common::XmlPacket::List *users = result.first->getList("users");

    if(!users || users->isEmpty()) {
      std::cout << "User list is empty." << std::endl;
    }
    else {
      std::cout << "Found " << users->getSize() << " user" << (users->getSize()==1 ? "" : "s") << ":" << std::endl;


      for(Common::XmlPacket::List::const_iterator user = users->begin(); user != users->end(); ++user)
        std::cout << "  " << user->get<unsigned long>("uid") << ", " << user->get<unsigned long>("gid")
        << ", " << user->get<const std::string&>("username") << ", " << user->get<const std::string&>("fullName") << std::endl;
    }

    std::cout << std::endl;
  }
}

void UserCommands::listUserGroupsCommand(CommandParser *commandParser, const std::vector<std::string> &args) {
  if(args.size() == 1) {
    std::cerr << args[0] << ": No user id given." << std::endl;
    commandParser->printUsage("list_user_groups");
    return;
  }
  if(args.size() > 2) {
    std::cerr << args[0] << ": Too many arguments." << std::endl;
    commandParser->printUsage("list_user_groups");
    return;
  }

  char *endptr;
  unsigned long uid = std::strtoul(args[1].c_str(), &endptr, 10);
  if(args[1].empty() || *endptr != '\0') {
    std::cerr << args[0] << ": Unable to parse user id." << std::endl;
    commandParser->printUsage("list_user_groups");
    return;
  }

  boost::shared_ptr<Common::Requests::UserGroupListRequest> request(new Common::Requests::UserGroupListRequest(commandParser->application, uid));
  commandParser->application->getRequestManager()->sendRequest(commandParser->connection, request);
  request->wait();

  std::pair<boost::shared_ptr<const Common::XmlPacket>, Core::Exception> result = request->getResult();

  if(!result.first || result.second) {
    std::cerr << "An error occurred during your request: " << result.second.strerror() << "." << std::endl;
  }
  else {
    const Common::XmlPacket::List *groups = result.first->getList("groups");

    if(!groups || groups->isEmpty()) {
      std::cout << "The user isn't member of any group." << std::endl;
    }
    else {
      std::cout << "User is member of " << groups->getSize() << " group" << (groups->getSize()==1 ? "" : "s") << ":" << std::endl;


      for(Common::XmlPacket::List::const_iterator group = groups->begin(); group != groups->end(); ++group)
        std::cout << "  " << group->get<unsigned long>("gid") << std::endl;
    }

    std::cout << std::endl;
  }
}

void UserCommands::listGroupsCommand(CommandParser *commandParser, const std::vector<std::string> &args _UNUSED_PARAMETER_) {
  boost::shared_ptr<Common::Requests::GroupListRequest> request(new Common::Requests::GroupListRequest(commandParser->application));

  commandParser->application->getRequestManager()->sendRequest(commandParser->connection, request);
  request->wait();

  std::pair<boost::shared_ptr<const Common::XmlPacket>, Core::Exception> result = request->getResult();

  if(!result.first || result.second) {
    std::cerr << "An error occurred during your request: " << result.second.strerror() << "." << std::endl;
  }
  else {
    const Common::XmlPacket::List *groups = result.first->getList("groups");

    if(!groups || groups->isEmpty()) {
      std::cout << "Group list is empty." << std::endl;
    }
    else {
      std::cout << "Found " << groups->getSize() << " group" << (groups->getSize()==1 ? "" : "s") << ":" << std::endl;


      for(Common::XmlPacket::List::const_iterator group = groups->begin(); group != groups->end(); ++group)
        std::cout << "  " << group->get<unsigned long>("gid") << ", " << group->get<const std::string&>("name") << std::endl;
    }

    std::cout << std::endl;
  }
}

void UserCommands::listGroupUsersCommand(CommandParser *commandParser, const std::vector<std::string> &args) {
  if(args.size() == 1) {
    std::cerr << args[0] << ": No group id given." << std::endl;
    commandParser->printUsage("list_group_users");
    return;
  }
  if(args.size() > 2) {
    std::cerr << args[0] << ": Too many arguments." << std::endl;
    commandParser->printUsage("list_group_users");
    return;
  }

  char *endptr;
  unsigned long gid = std::strtoul(args[1].c_str(), &endptr, 10);
  if(args[1].empty() || *endptr != '\0') {
    std::cerr << args[0] << ": Unable to parse group id." << std::endl;
    commandParser->printUsage("list_group_users");
    return;
  }

  boost::shared_ptr<Common::Requests::GroupUserListRequest> request(new Common::Requests::GroupUserListRequest(commandParser->application, gid));
  commandParser->application->getRequestManager()->sendRequest(commandParser->connection, request);
  request->wait();

  std::pair<boost::shared_ptr<const Common::XmlPacket>, Core::Exception> result = request->getResult();

  if(!result.first || result.second) {
    std::cerr << "An error occurred during your request: " << result.second.strerror() << "." << std::endl;
  }
  else {
    const Common::XmlPacket::List *users = result.first->getList("users");

    if(!users || users->isEmpty()) {
      std::cout << "The group doesn't have any members." << std::endl;
    }
    else {
      std::cout << "The group has " << users->getSize() << " member" << (users->getSize()==1 ? "" : "s") << ":" << std::endl;


      for(Common::XmlPacket::List::const_iterator user = users->begin(); user != users->end(); ++user)
        std::cout << "  " << user->get<unsigned long>("uid") << std::endl;
    }

    std::cout << std::endl;
  }
}

}
}