//------------------------------------------------------------------------------
// @file: com_proto_group.cc
// @author: Fabio Luchetti - CERN
//------------------------------------------------------------------------------

/************************************************************************
 * EOS - the CERN Disk Storage System                                   *
 * Copyright (C) 2018 CERN/Switzerland                                  *
 *                                                                      *
 * 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 token) 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 "common/StringTokenizer.hh"
#include "common/Path.hh"
#include "console/ConsoleMain.hh"
#include "console/commands/ICmdHelper.hh"

extern int com_group(char*);
void com_group_help();

//------------------------------------------------------------------------------
//! Class GroupHelper
//------------------------------------------------------------------------------
class GroupHelper : public ICmdHelper
{
public:
  //----------------------------------------------------------------------------
  //! Constructor
  //!
  //! @param opts global options
  //----------------------------------------------------------------------------
  GroupHelper(const GlobalOptions& opts):
    ICmdHelper(opts)
  {}

  //----------------------------------------------------------------------------
  //! Destructor
  //----------------------------------------------------------------------------
  ~GroupHelper() override = default;

  //----------------------------------------------------------------------------
  //! Parse command line input
  //!
  //! @param arg input
  //!
  //! @return true if successful, otherwise false
  //----------------------------------------------------------------------------
  bool ParseCommand(const char* arg) override;
};

//------------------------------------------------------------------------------
// Parse command line input
//------------------------------------------------------------------------------
bool GroupHelper::ParseCommand(const char* arg)
{
  eos::console::GroupProto* group = mReq.mutable_group();
  eos::common::StringTokenizer tokenizer(arg);
  tokenizer.GetLine();
  std::string token;

  if (!tokenizer.NextToken(token)) {
    return false;
  }

  // one of { ls, rm, set }
  if (token == "ls") {
    eos::console::GroupProto_LsProto* ls = group->mutable_ls();

    while (tokenizer.NextToken(token)) {
      if (token == "-s") {
        mIsSilent = true;
      } else if (token == "-g") {
        if (!tokenizer.NextToken(token) ||
            !eos::common::StringTokenizer::IsUnsignedNumber(token)) {
          std::cerr << "error: geodepth was not provided or it does not have "
                    << "the correct value: geodepth should be a positive "
                    << "integer\n";
          return false;
        }

        try {
          ls->set_outdepth(std::stoi(token));
        } catch (const std::exception& e) {
          std::cerr <<
                    "error: argument to geodepth (-g flag) needs to be a positive integer\n";
          return false;
        }
      } else if (token == "-b" || token == "--brief") {
        ls->set_outhost(true);
      } else if (token == "-m") {
        ls->set_outformat(eos::console::GroupProto_LsProto::MONITORING);
      } else if (token == "-l") {
        ls->set_outformat(eos::console::GroupProto_LsProto::LISTING);
      } else if (token == "--io") {
        ls->set_outformat(eos::console::GroupProto_LsProto::IOGROUP);
      } else if (token == "--IO") {
        ls->set_outformat(eos::console::GroupProto_LsProto::IOFS);
      } else if (token.find('-') != 0) { // does not begin with "-"
        ls->set_selection(token);
      } else {
        return false;
      }
    }
  } else if (token == "rm") {
    if (!tokenizer.NextToken(token)) {
      return false;
    }

    eos::console::GroupProto_RmProto* rm = group->mutable_rm();
    rm->set_group(token);
  } else if (token == "set") {
    if (!tokenizer.NextToken(token)) {
      return false;
    }

    eos::console::GroupProto_SetProto* set = group->mutable_set();
    set->set_group(token);

    if (!tokenizer.NextToken(token)) {
      return false;
    }

    if (token == "on" || token == "off" || token == "drain") {
      set->set_group_state(token);
    } else {
      return false;
    }
  } else { // no proper subcommand
    return false;
  }

  return true;
}

//------------------------------------------------------------------------------
// Group command entry point
//------------------------------------------------------------------------------
int com_protogroup(char* arg)
{
  if (wants_help(arg)) {
    com_group_help();
    global_retc = EINVAL;
    return EINVAL;
  }

  GroupHelper group(gGlobalOpts);

  if (!group.ParseCommand(arg)) {
    com_group_help();
    global_retc = EINVAL;
    return EINVAL;
  }

  global_retc = group.Execute();
  return global_retc;
}

//------------------------------------------------------------------------------
// Print help message
//------------------------------------------------------------------------------
void com_group_help()
{
  std::ostringstream oss;
  oss
      << " usage:\n"
      << std::endl
      << "group ls [-s] [-g <depth>] [-b|--brief] [-m|-l|--io] [<groups>] : list groups\n"
      << "\t <groups> : list <groups> only, where <groups> is a substring match and can be a comma seperated list\n"
      << "\t       -s : silent mode\n"
      << "\t       -g : geo output - aggregate group information along the instance geotree down to <depth>\n"
      << "\t       -b : brief output\n"
      << "\t       -m : monitoring key=value output format\n"
      << "\t       -l : long output - list also file systems after each group\n"
      << "\t     --io : print IO statistics for the group\n"
      << "\t     --IO : print IO statistics for each filesystem\n"
      << std::endl
      << "group rm <group-name> : remove group\n"
      << std::endl
      << "group set <group-name> on|drain|off : activate/drain/deactivate group\n"
      << "\t  => when a group is (re-)enabled, the drain pull flag is recomputed for all filesystems within a group\n"
      << "\t  => when a group is (re-)disabled, the drain pull flag is removed from all members in the group\n"
      << "\t  => when a group is in drain, all the filesystems in the group will be drained to other groups\n"
      << std::endl;
  std::cerr << oss.str() << std::endl;
}