/*
 * CommandParser.cpp
 *
 * Copyright (C) 2008 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 "CommandParser.h"
#include "InformationManager.h"
#include "Requests/DaemonFSInfoRequest.h"
#include "Requests/DaemonCommandRequest.h"
#include "Requests/DaemonStatusRequest.h"
#include <Common/Logger.h>
#include <Common/RequestManager.h>
#include <Common/Requests/FSInfoRequest.h>
#include <Common/Requests/DisconnectRequest.h>
#include <Common/Requests/GroupListRequest.h>
#include <Common/Requests/GroupUserListRequest.h>
#include <Common/Requests/StatusRequest.h>
#include <Common/Requests/UserInfoRequest.h>
#include <Common/Requests/UserListRequest.h>
#include <Common/Requests/UserGroupListRequest.h>
#include <Common/Tokenizer.h>

#include <iostream>
#include <cstdio>

#include <boost/bind.hpp>

namespace Mad {
namespace Client {

const CommandParser::Command CommandParser::commands[] = {
    {{"fsinfo", 0}, "fsinfo [host]", "Display file system usage information", "Display file system usage information of a host or the server (if no host is given).", &CommandParser::fsinfoCommand},
    {{"help", "?", 0}, "help [command]", "Display usage information about commands", "Display usage information about a command. If no command is given, display a list of all available commands.", &CommandParser::helpCommand},
    {{"list_hosts", "hosts", 0}, "list_hosts [-a]", "List the currently active hosts", "List the currently active hosts.\n\n -a\tAlso list inactive hosts", &CommandParser::listHostsCommand},
    {{"reboot", 0}, "reboot *|host...", "Reboot host", "Reboot hosts. * will reboot all hosts.", &CommandParser::rebootCommand},
    {{"shutdown", "halt", 0}, "shutdown *|host...", "Shut hosts down", "Shut hosts down. * will shut down all hosts.", &CommandParser::shutdownCommand},
    {{"status", "st", 0}, "status [host]", "Display status information", "Display host status information. If no host is given, display server status information.", &CommandParser::statusCommand},
    {{"user_info", "user", 0}, "user_info uid", "Search for a user id", "Search for a user id.", &CommandParser::userInfoCommand},
    {{"list_users", "users", 0}, "list_users", "Show the user account database", "Show the user account database.", &CommandParser::listUsersCommand},
    {{"list_user_groups", "user_groups", 0}, "list_user_groups uid", "List user's groups", "List the groups that the user is member of.", &CommandParser::listUserGroupsCommand},
    {{"list_groups", "groups", 0}, "list_groups", "Show the user group database", "Show the user group database.", &CommandParser::listGroupsCommand},
    {{"list_group_users", "group_users", 0}, "list_group_users gid", "List group's users", "List the users that are members of the group.", &CommandParser::listGroupUsersCommand},
    {{"exit", "quit", 0}, "exit", "Close the connection and quit the client", "Close the connection and quit the client.", &CommandParser::exitCommand},
    {{0}, 0, 0, 0, 0}
};

CommandParser CommandParser::commandParser;


const CommandParser::Command* CommandParser::findCommand(const std::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 CommandParser::printUsage(const std::string& command) {
  const CommandParser::Command *cmd = findCommand(command);

  if(cmd)
    Common::Logger::logf("Usage: %s\n", cmd->cmdline);
}


std::map<std::string, Common::HostInfo> CommandParser::parseHostList(const std::vector<std::string> &args, bool mustBeActive) {
  const std::map<std::string, Common::HostInfo>& hosts = InformationManager::get()->getDaemons();
  std::map<std::string, Common::HostInfo> ret;

  for(std::vector<std::string>::const_iterator arg = args.begin(); arg != args.end(); ++arg) {
    if(*arg == "*") {
      if(mustBeActive) {
        for(std::map<std::string, Common::HostInfo>::const_iterator host = hosts.begin(); host != hosts.end(); ++host) {
          if(host->second.getState() != Common::HostInfo::INACTIVE)
            ret.insert(*host);
        }

        if(ret.empty())
          Common::Logger::log("No hosts active.");
      }
      else {
        ret = hosts;
      }
    }
    else {
      std::map<std::string, Common::HostInfo>::const_iterator host = hosts.find(*arg);

      if(host == hosts.end())
        Common::Logger::logf(Common::Logger::ERROR, "Host '%s' doesn't exist.", arg->c_str());
      else if(mustBeActive && host->second.getState() == Common::HostInfo::INACTIVE)
        Common::Logger::logf(Common::Logger::WARNING, "Host '%s' is inactive.", arg->c_str());
      else
        ret.insert(*host);
    }
  }

  return ret;
}


void CommandParser::fsinfoCommand(const std::vector<std::string> &args) {
  if(args.size() == 1)
    Common::RequestManager::get()->sendRequest0<Common::Requests::FSInfoRequest>(connection, boost::bind(&CommandManager::fsInfoRequestFinished, CommandManager::get(), _1));
  else if(args.size() == 2)
    Common::RequestManager::get()->sendRequest1<Requests::DaemonFSInfoRequest>(connection, args[1], boost::bind(&CommandManager::daemonFSInfoRequestFinished, CommandManager::get(), _1));
  else {
    Common::Logger::logf(Common::Logger::ERROR, "%s: Too many arguments.", args[0].c_str());
    printUsage("fsinfo");
    return;
  }

  ++CommandManager::get()->activeRequests;
}

void CommandParser::helpCommand(const std::vector<std::string> &args) {
  if(args.size() == 1) {
    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() == 2) {
    const Command* command = findCommand(args[1]);

    if(command) {
      std::cout << "Usage: " << command->cmdline << std::endl << std::endl;
      std::cout << command->longdesc << std::endl << std::endl;
    }
    else
      Common::Logger::logf(Common::Logger::WARNING, "%s: Command '%s' doesn't exist.", args[0].c_str(), args[1].c_str());
  }
  else {
    Common::Logger::logf(Common::Logger::ERROR, "%s: Too many arguments.", args[0].c_str());
    printUsage("help");
  }
}

void CommandParser::listHostsCommand(const std::vector<std::string> &args) {
  const std::map<std::string, Common::HostInfo>& hosts = InformationManager::get()->getDaemons();

  if(args.size() == 1) {
    if(hosts.empty()) {
      std::cout << "The host list is empty." << std::endl << std::endl;
      return;
    }

    bool output = false;

    for(std::map<std::string, Common::HostInfo>::const_iterator host = hosts.begin(); host != hosts.end(); ++host) {
      if(host->second.getState() == Common::HostInfo::INACTIVE)
        continue;

      if(!output) {
        std::cout << "Active hosts:" << std::endl;
        output = true;
      }

      std::cout << "  " << host->first << std::endl;
    }

    if(!output)
      std::cout << "No active hosts." << std::endl;

    std::cout << std::endl;
  }
  else if(args.size() > 2) {
    Common::Logger::logf(Common::Logger::ERROR, "%s: Too many arguments.", args[0].c_str());
    printUsage("list_hosts");
  }
  else if(args[1] == "-a") {
    if(hosts.empty()) {
      std::cout << "The host list is empty." << std::endl << std::endl;
      return;
    }

    std::cout << "Host list:" << std::endl;

    for(std::map<std::string, Common::HostInfo>::const_iterator host = hosts.begin(); host != hosts.end(); ++host) {
      std::cout << "  " << host->first << " (" << (host->second.getState() == Common::HostInfo::RUNNING ? "running" : "inactive") << ")" << std::endl;
    }

    std::cout << std::endl;
  }
  else {
    Common::Logger::logf(Common::Logger::ERROR, "%s: Don't understand argument '%s'.", args[0].c_str(), args[1].c_str());
    printUsage("list_hosts");
  }
}

void CommandParser::rebootCommand(const std::vector<std::string> &args) {
  if(args.size() < 2) {
    Common::Logger::logf(Common::Logger::ERROR, "%s: No host given.", args[0].c_str());
    printUsage("reboot");
    return;
  }

  std::map<std::string, Common::HostInfo> hosts = parseHostList(std::vector<std::string>(args.begin()+1, args.end()), true);

  for(std::map<std::string, Common::HostInfo>::iterator host = hosts.begin(); host != hosts.end(); ++host) {
    Common::RequestManager::get()->sendRequest2<Requests::DaemonCommandRequest>(connection,
        host->first, true, boost::bind(&CommandManager::daemonCommandRequestFinished, CommandManager::get(), _1)
    );

    ++CommandManager::get()->activeRequests;
  }
}

void CommandParser::shutdownCommand(const std::vector<std::string> &args) {
  if(args.size() < 2) {
    Common::Logger::logf(Common::Logger::ERROR, "%s: No host given.", args[0].c_str());
    printUsage("shutdown");
    return;
  }

  std::map<std::string, Common::HostInfo> hosts = parseHostList(std::vector<std::string>(args.begin()+1, args.end()), true);

  for(std::map<std::string, Common::HostInfo>::iterator host = hosts.begin(); host != hosts.end(); ++host) {
    Common::RequestManager::get()->sendRequest2<Requests::DaemonCommandRequest>(connection,
        host->first, false, boost::bind(&CommandManager::daemonCommandRequestFinished, CommandManager::get(), _1)
    );

    ++CommandManager::get()->activeRequests;
  }
}

void CommandParser::statusCommand(const std::vector<std::string> &args) {
  if(args.size() == 1)
    Common::RequestManager::get()->sendRequest0<Common::Requests::StatusRequest>(connection,
        boost::bind(&CommandManager::statusRequestFinished, CommandManager::get(), _1));
  else if(args.size() == 2)
    Common::RequestManager::get()->sendRequest1<Requests::DaemonStatusRequest>(connection,
        args[1], boost::bind(&CommandManager::daemonStatusRequestFinished, CommandManager::get(), _1));
  else {
    Common::Logger::logf(Common::Logger::ERROR, "%s: Too many arguments.", args[0].c_str());
    printUsage("status");
    return;
  }

  ++CommandManager::get()->activeRequests;
}

void CommandParser::userInfoCommand(const std::vector<std::string> &args) {
  if(args.size() == 1) {
    Common::Logger::logf(Common::Logger::ERROR, "%s: No user id given.", args[0].c_str());
    printUsage("user_info");
    return;
  }
  if(args.size() > 2) {
    Common::Logger::logf(Common::Logger::ERROR, "%s: Too many arguments.", args[0].c_str());
    printUsage("user_info");
    return;
  }

  char *endptr;
  unsigned long uid = std::strtoul(args[1].c_str(), &endptr, 10);
  if(args[1].empty() || *endptr != '\0') {
    Common::Logger::logf(Common::Logger::ERROR, "%s: Unable to parse user id.", args[0].c_str());
    printUsage("user_info");
    return;
  }

  ++CommandManager::get()->activeRequests;

  Common::RequestManager::get()->sendRequest1<Common::Requests::UserInfoRequest>(connection,
      uid, boost::bind(&CommandManager::userInfoRequestFinished, CommandManager::get(), _1));
}

void CommandParser::listUsersCommand(const std::vector<std::string>&) {
  ++CommandManager::get()->activeRequests;

  Common::RequestManager::get()->sendRequest0<Common::Requests::UserListRequest>(connection,
      boost::bind(&CommandManager::userListRequestFinished, CommandManager::get(), _1));
}

void CommandParser::listUserGroupsCommand(const std::vector<std::string> &args) {
  if(args.size() == 1) {
    Common::Logger::logf(Common::Logger::ERROR, "%s: No user id given.", args[0].c_str());
    printUsage("list_user_groups");
    return;
  }
  if(args.size() > 2) {
    Common::Logger::logf(Common::Logger::ERROR, "%s: Too many arguments.", args[0].c_str());
    printUsage("list_user_groups");
    return;
  }

  char *endptr;
  unsigned long uid = std::strtoul(args[1].c_str(), &endptr, 10);
  if(args[1].empty() || *endptr != '\0') {
    Common::Logger::logf(Common::Logger::ERROR, "%s: Unable to parse user id.", args[0].c_str());
    printUsage("list_user_groups");
    return;
  }

  ++CommandManager::get()->activeRequests;

  Common::RequestManager::get()->sendRequest1<Common::Requests::UserGroupListRequest>(connection,
      uid, boost::bind(&CommandManager::userGroupListRequestFinished, CommandManager::get(), _1));
}

void CommandParser::listGroupsCommand(const std::vector<std::string>&) {
  ++CommandManager::get()->activeRequests;

  Common::RequestManager::get()->sendRequest0<Common::Requests::GroupListRequest>(connection,
      boost::bind(&CommandManager::groupListRequestFinished, CommandManager::get(), _1));
}

void CommandParser::listGroupUsersCommand(const std::vector<std::string> &args) {
  if(args.size() == 1) {
    Common::Logger::logf(Common::Logger::ERROR, "%s: No group id given.", args[0].c_str());
    printUsage("list_group_users");
    return;
  }
  if(args.size() > 2) {
    Common::Logger::logf(Common::Logger::ERROR, "%s: Too many arguments.", args[0].c_str());
    printUsage("list_group_users");
    return;
  }

  char *endptr;
  unsigned long gid = std::strtoul(args[1].c_str(), &endptr, 10);
  if(args[1].empty() || *endptr != '\0') {
    Common::Logger::logf(Common::Logger::ERROR, "%s: Unable to parse group id.", args[0].c_str());
    printUsage("list_group_users");
    return;
  }

  ++CommandManager::get()->activeRequests;

  Common::RequestManager::get()->sendRequest1<Common::Requests::GroupUserListRequest>(connection,
      gid, boost::bind(&CommandManager::groupUserListRequestFinished, CommandManager::get(), _1));
}

void CommandParser::exitCommand(const std::vector<std::string>&) {
  ++CommandManager::get()->activeRequests;

  Common::RequestManager::get()->sendRequest0<Common::Requests::DisconnectRequest>(connection,
      boost::bind(&CommandManager::disconnectRequestFinished, CommandManager::get(), _1));
}

bool CommandParser::parse(const std::string &cmd) {
  std::vector<std::string> splitCmd;

  Common::Tokenizer::tokenize(cmd, splitCmd);

  if(splitCmd.empty())
    return true;

  const Command* command = findCommand(splitCmd[0]);

  if(command)
    (this->*command->funcPtr)(splitCmd);
  else
    Common::Logger::logf(Common::Logger::ERROR, "Unknown command '%s'.", splitCmd[0].c_str());

  return true;
}

}
}