path: root/src/Client/UserListCommands.cpp

/*
 * UserListCommands.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 "UserListCommands.h"
#include "Application.h"
#include "XLSReader.h"
#include "XLSSheet.h"
#include "Requests/UserLists/UserListListRequest.h"
#include "Requests/UserLists/UserListDownloadRequest.h"
#include "Requests/UserLists/UserListDiffListRequest.h"

#include <Common/RequestManager.h>
#include <Common/UserLists/UserList.h>
#include <Common/UserLists/Util.h>

#include <iostream>

namespace Mad {
namespace Client {

const UserListCommands::Command UserListCommands::commands[] = {
    {{"help", "?", 0}, "help [command]", "Display usage information about user list commands", "Display usage information about a user list command. If no command is given, display a list of all available user list commands.", &UserListCommands::helpCommand},
    {{"list", "ls", 0}, "list", "List the stored user lists", "List the stored user lists.", &UserListCommands::listCommand},
    {{"show_list", "sl", 0}, "show_list list", "Displays a user list", "Displays a user list.", &UserListCommands::showListCommand},
    {{"import", "im", 0}, "import filename", "Imports a user list file", "Imports a user list file. At the moment, this supports only XLS files.", &UserListCommands::importCommand},
    {{0}, 0, 0, 0, 0}
};

const UserListCommands::Command* UserListCommands::findCommand(const Core::String& command) {
  for(int i = 0; commands[i].commands[0] != 0; ++i) {
    for(int j = 0; commands[i].commands[j] != 0; ++j) {
      if(command == commands[i].commands[j]) {
        return &commands[i];
      }
    }
  }

  return 0;
}

void UserListCommands::printUsage(const Core::String& command) {
  const UserListCommands::Command *cmd = findCommand(command);

  if(cmd)
    std::cerr << "Usage: " << "user_lists " << cmd->cmdline << std::endl;
}


void UserListCommands::helpCommand(CommandParser* /*commandParser*/, const std::vector<Core::String> &args) {
  if(args.size() <= 2) {
    std::cout << "Available commands:" << std::endl << std::endl;

    for(int i = 0; commands[i].commands[0] != 0; ++i) {
      std::cout << commands[i].commands[0];

      for(int j = 1; commands[i].commands[j] != 0; ++j)
        std::cout << ", " << commands[i].commands[j];

      std::cout << std::endl << "\t" << commands[i].desc << std::endl;
    }
  }
  else if(args.size() == 3) {
    const Command* command = findCommand(args[2]);

    if(command) {
      std::cout << "Usage: " << "user_lists " << command->cmdline << std::endl << std::endl;
      std::cout << command->longdesc << std::endl << std::endl;
    }
    else
      std::cerr << args[0] << " " << args[1] << ": Command '" << args[2] << "' doesn't exist." << std::endl;
  }
  else {
    std::cerr << args[0] << " " << args[1] << ": Too many arguments." << std::endl;
    printUsage("help");
  }
}

void UserListCommands::listCommand(CommandParser *commandParser, const std::vector<Core::String>& /*args*/) {
  boost::shared_ptr<Requests::UserLists::UserListListRequest> userListRequest(new Requests::UserLists::UserListListRequest(commandParser->application));
  boost::shared_ptr<Requests::UserLists::UserListDiffListRequest> userListDiffRequest(new Requests::UserLists::UserListDiffListRequest(commandParser->application));

  commandParser->application->getRequestManager()->sendRequest(commandParser->connection, userListRequest);
  commandParser->application->getRequestManager()->sendRequest(commandParser->connection, userListDiffRequest);

  userListRequest->wait();
  userListDiffRequest->wait();

  std::pair<boost::shared_ptr<const Common::XmlData>, Core::Exception> userListResult = userListRequest->getResult();
  std::pair<boost::shared_ptr<const Common::XmlData>, Core::Exception> userListDiffResult = userListDiffRequest->getResult();

  if(userListResult.second) {
    std::cerr << "An error occurred during your request: " << userListResult.second.what() << "." << std::endl;
    return;
  }
  else if(!userListResult.first || !userListDiffResult.first || userListDiffResult.second) {
    std::cerr << "An error occurred during your request: " << userListDiffResult.second.what() << "." << std::endl;
    return;
  }

  const Common::XmlData::List *userLists = userListResult.first->getList("userLists");
  const Common::XmlData::List *userListDiffs = userListDiffResult.first->getList("userListDiffs");

  if(userLists->isEmpty()) {
    std::cout << "There aren't any user lists stored on the server." << std::endl;
  }
  else {
    if(userLists->getSize() == 1)
      std::cout << "There is 1 user list stored on the server:" << std::endl;
    else
      std::cout << "There are " << userLists->getSize() << " user lists stored on the server:" << std::endl;

    for(Common::XmlData::List::const_iterator userList = userLists->begin(); userList != userLists->end(); ++userList) {
      std::cout << "  " << userList->get<const Core::String&>("name") << std::endl;
    }
  }

  std::cout << std::endl;

  if(!userListDiffs->isEmpty()) {
    if(userListDiffs->getSize() == 1)
      std::cout << "There is 1 user list difference record stored on the server:" << std::endl;
    else
      std::cout << "There are " << userListDiffs->getSize() << " user list difference records stored on the server:" << std::endl;

    for(Common::XmlData::List::const_iterator diff = userListDiffs->begin(); diff != userListDiffs->end(); ++diff) {
      std::cout << "  " << diff->get<const Core::String&>("name") << std::endl;
    }
  }
}

void UserListCommands::showListCommand(CommandParser *commandParser, const std::vector<Core::String> &args) {
  if(args.size() < 3) {
    std::cerr << args[0] << " " << args[1] << ": No list name given." << std::endl;
    printUsage("show_list");
    return;
  }
  else if(args.size() == 3) {
    boost::shared_ptr<Requests::UserLists::UserListDownloadRequest> request(
        new Requests::UserLists::UserListDownloadRequest(commandParser->application, args[2]));
    commandParser->application->getRequestManager()->sendRequest(commandParser->connection, request);
    request->wait();

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

    if(!result.first || result.second) {
      std::cerr << "An error occurred during your request: " << result.second.what() << "." << std::endl;
      return;
    }

    boost::shared_ptr<Common::UserLists::UserList> userList = Common::UserLists::Util::deserializeUserList(result.first.get());

    std::cout << "User list '" << args[2];

    if(userList->isEmpty()) {
      std::cout << " is empty." << std::endl;
    }
    else {
      std::cout << "' contains " << userList->getLength() << " user";
      if(userList->getLength() > 1)
        std::cout << "s";
      std::cout << "." << std::endl << std::endl;

      printUserList(*userList);
    }

    std::cout << std::endl;
  }
  else {
    std::cerr << args[0] << " " << args[1] << ": Too many arguments." << std::endl;
    printUsage("show_list");
    return;
  }
}

void UserListCommands::importCommand(CommandParser* /*commandParser*/, const std::vector<Core::String> &args) {
  if(args.size() < 3) {
    std::cerr << args[0] << " " << args[1] << ": No filename given." << std::endl;
    printUsage("import");
    return;
  }
  else if(args.size() == 3) {
    try {
      XLSReader reader(args[2].toLocale());
      const std::list<boost::shared_ptr<XLSSheet> > &sheets = reader.getSheets();

      if(sheets.empty()) {
        std::cerr << "There aren't any worksheets in the file." << std::endl;
        return;
      }

      unsigned long sheetNum = 1;

      if(sheets.size() > 1) {
        std::cout << "There is more than one worksheet in the file. Which one do you want to import?" << std::endl << std::endl;

        while(true) {
          int i = 1;
          for(std::list<boost::shared_ptr<XLSSheet> >::const_iterator sheet = sheets.begin(); sheet != sheets.end(); ++sheet, ++i) {
            std::cout << "  " << i << ") " << (*sheet)->getTitle() << std::endl;
          }

          std::cout << std::endl;
          std::cout << "Worksheet number: " << std::flush;

          std::string input;
          std::getline(std::cin, input);
          sheetNum = std::strtoul(input.c_str(), 0, 10);

          if(sheetNum > 0 && sheetNum <= sheets.size())
            break;

          std::cout << "Invalid selection. Please choose one of:" << std::endl << std::endl;
        }
      }

      std::list<boost::shared_ptr<XLSSheet> >::const_iterator sheet = sheets.begin();
      std::advance(sheet, sheetNum-1);

      std::cout << "Imported worksheet:" << std::endl;
      printSheet(**sheet, 10);

      std::cout << std::endl << std::endl;
      std::cout << "Does the first line of the worksheet contain the column names? (Y/n) " << std::flush;

      std::string input;
      std::getline(std::cin, input);
      if(input.empty() || (input[0] != 'n' && input[0] != 'N')) {
        (*sheet)->useFirstRowAsColumnNames(true);
      }

      std::cout << std::endl << std ::endl;
      printSheet(**sheet, 10);
      std::cout << std::endl << std ::endl;

      Common::UserLists::UserList userList;
      const std::vector<XLSSheet::RowType> &rows = (*sheet)->getRows();
      unsigned id = 1;

      for(std::vector<XLSSheet::RowType>::const_iterator rowIt = rows.begin(); rowIt != rows.end(); ++rowIt, ++id) {
        const std::vector<Core::String> &row = **rowIt;
        Common::UserLists::UserListEntry entry;

        std::ostringstream stream;
        stream << id;
        entry.setName(stream.str().c_str());

        for(unsigned col = 0; col < (*sheet)->getColumnCount(); ++col) {
          entry.setDetail((*sheet)->getColumnName(col), row[col]);
        }

        userList.addUser(entry);
      }

      printUserList(userList);
      std::cout << std::endl << std ::endl;
    }
    catch(Core::Exception e) {
      std::cerr << "The file can't be read: " << e.what() << "." << std::endl;
    }
  }
  else {
    std::cerr << args[0] << " " << args[1] << ": Too many arguments." << std::endl;
    printUsage("import");
    return;
  }
}


void UserListCommands::printUserList(const Common::UserLists::UserList &list) {
  std::map<Core::String, boost::int32_t> lengths;

  static const Core::String USER_NAME = "User name";
  static const Core::String GROUP_NAME = "Group name";

  lengths.insert(std::make_pair(USER_NAME, USER_NAME.length()));
  lengths.insert(std::make_pair(GROUP_NAME, GROUP_NAME.length()));

  std::set<Core::String> details = list.getDetails();
  for(std::set<Core::String>::iterator detail = details.begin(); detail != details.end(); ++detail) {
    lengths.insert(std::make_pair(*detail, detail->length()));
  }

  for(Common::UserLists::UserList::const_iterator user = list.begin(); user != list.end(); ++user) {
    std::map<Core::String, boost::int32_t>::iterator it = lengths.find(USER_NAME);
    if(user->getName().length() > it->second)
      it->second = user->getName().length();

    it = lengths.find(GROUP_NAME);
    if(user->getGroup().length() > it->second)
      it->second = user->getGroup().length();

    for(std::set<Core::String>::iterator detail = details.begin(); detail != details.end(); ++detail) {
      it = lengths.find(*detail);
      Core::String detailStr = user->getDetail(*detail);
      if(detailStr.length() > it->second)
        it->second = detailStr.length();
    }
  }

  std::cout << " ";
  printPadded(USER_NAME, lengths.find(USER_NAME)->second);
  std::cout << "   ";
  printPadded(GROUP_NAME, lengths.find(GROUP_NAME)->second);
  for(std::set<Core::String>::iterator detail = details.begin(); detail != details.end(); ++detail) {
    std::cout << "   ";
    printPadded(*detail, lengths.find(*detail)->second);
  }
  std::cout << std::endl;

  std::cout << "-";
  printDelimiter(lengths.find(USER_NAME)->second);
  std::cout << "- -";
  printDelimiter(lengths.find(GROUP_NAME)->second);
  for(std::set<Core::String>::iterator detail = details.begin(); detail != details.end(); ++detail) {
    std::cout << "- -";
    printDelimiter(lengths.find(*detail)->second);
  }
  std::cout << "-" << std::endl;

  for(Common::UserLists::UserList::const_iterator user = list.begin(); user != list.end(); ++user) {
    static const Core::String UNSET("<unset>");

    std::cout << " ";

    if(!user->getName().isEmpty())
      printPadded(user->getName(), lengths.find(USER_NAME)->second);
    else
      printPadded(UNSET, lengths.find(USER_NAME)->second);

    std::cout << "   ";

    if(!user->getGroup().isEmpty())
      printPadded(user->getGroup(), lengths.find(GROUP_NAME)->second);
    else
      printPadded(UNSET, lengths.find(GROUP_NAME)->second);

    for(std::set<Core::String>::iterator detail = details.begin(); detail != details.end(); ++detail) {
      std::cout << "   ";
      printPadded(user->getDetail(*detail), lengths.find(*detail)->second);
    }

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

void UserListCommands::printSheet(const XLSSheet &sheet, unsigned rowCount) {
  const std::vector<XLSSheet::RowType> &rows = sheet.getRows();
  std::vector<boost::int32_t> lengths(sheet.getColumnCount());

  for(unsigned col = 0; col < sheet.getColumnCount(); ++col) {
    lengths[col] = sheet.getColumnName(col).length();
  }

  for(std::vector<XLSSheet::RowType>::const_iterator rowIt = rows.begin(); rowIt != rows.end() && (rowCount == 0 || rowIt != rows.begin()+rowCount); ++rowIt) {
    const std::vector<Core::String> &row = **rowIt;

    for(unsigned col = 0; col < sheet.getColumnCount(); ++col) {
      if(row[col].length() > lengths[col])
        lengths[col] = row[col].length();
    }
  }

  for(unsigned col = 0; col < sheet.getColumnCount(); ++col) {
    std::cout << "   ";
    printPadded(sheet.getColumnName(col), lengths[col]);
  }
  std::cout << std::endl;

  std::cout << " ";
  for(unsigned col = 0; col < sheet.getColumnCount(); ++col) {
    std::cout << " -";
    printDelimiter(lengths[col]);
    std::cout << "-";
  }
  std::cout << std::endl;

  for(std::vector<XLSSheet::RowType>::const_iterator rowIt = rows.begin(); rowIt != rows.end() && (rowCount == 0 || rowIt != rows.begin()+rowCount); ++rowIt) {
    const std::vector<Core::String> &row = **rowIt;

    for(unsigned col = 0; col < sheet.getColumnCount(); ++col) {
      std::cout << "   ";
      printPadded(row[col], lengths[col]);
    }

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

  if(rowCount > 0 && rows.size() > rowCount) {
    for(unsigned col = 0; col < sheet.getColumnCount(); ++col) {
      std::cout << "   ";
      printPadded("...", lengths[col]);
    }
  }
}

void UserListCommands::printPadded(const Core::String &str, boost::int32_t length) {
  std::cout << str;
  if(str.length() < length)
    std::cout << std::string(length-str.length(), ' ');
}

void UserListCommands::printDelimiter(boost::int32_t length) {
  printPadded(Core::String(std::string(length, '-').c_str()), length);
}


void UserListCommands::userListCommand(CommandParser *commandParser, const std::vector<Core::String> &args) {
  Core::String cmd;

  if(args.size() < 2)
    cmd = "help";
  else
    cmd = args[1];

  const Command* command = findCommand(cmd);

  if(command)
    command->function(commandParser, args);
  else
    std::cerr << args[0] << ": Unknown command '" << cmd << "'." << std::endl;
}

}
}