// Copyright (C) 1999,2000 Bruce Guenter <bruceg@em.ca>
//
// 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 2 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, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

#include <config.h>
#include "daemon.h"
#include "misc/lookup.h"

// declare the commands
extern CMD(adduser2);
extern CMD(adduser3);
extern CMD(autoresponse);
extern CMD(chattr);
extern CMD(check);
extern CMD(deluser);
extern CMD(listdomain);
extern CMD(lookup);
extern CMD(stat);

#ifdef TEST_DAEMON
CMD(echo)
{
  logcommand(args);
  mystring msg;
  for(unsigned i = 0; i < args.count(); i++) {
    if(i > 0) msg += " ";
    msg += args[i];
  }
  RETURN(ok, msg);
}
CMD(fecho)
{
  logcommand(args);
  mystring msg;
  for(unsigned i = 0; i < args.count(); i++) {
    if(i > 0) msg += " ";
    msg += args[i];
  }
  RETURN(ok, msg);
}
#endif // TEST_DAEMON

//CMD(stat);
//CMD(statall);

struct dispatch 
{
  const char* name;
  response (*function)(command&, int);
  unsigned arg_min;
  unsigned arg_max;
  bool decode_virtual;
  //unsigned count;
};

#define ENTRY(NAME,MIN,MAX,DV) { #NAME , NAME##_cmd , unsigned(MIN), unsigned(MAX), DV }
dispatch dispatch_table[] = {
  ENTRY(lookup,       3,  3, true),
  ENTRY(check,        3,  3, true),
  ENTRY(chattr,       5, -1, true),
  ENTRY(adduser2,     4, -1, true),
  ENTRY(adduser3,     5, -1, true),
  ENTRY(autoresponse, 4,  5, true),
  ENTRY(deluser,      3,  3, true),
  ENTRY(stat,         3,  3, true),
  ENTRY(listdomain,   2,  2, false),
#ifdef TEST_DAEMON
  ENTRY(echo,         0, -1, false), // For testing purposes only
  ENTRY(fecho,        0, -1, false), // For testing purposes only
#endif // TEST_DAEMON
  { "", 0, 0, false, 0 }
};
//ENTRY(stat, 1, , false),
//ENTRY(statall, 0, 0, false),

static dispatch* find_dispatch(mystring name)
{
  for(dispatch* ptr = dispatch_table; ptr->function != 0; ptr++) {
    if(name == ptr->name)
      return ptr;
  }
  return 0;
}

#if 0
CMD(stat)
  // Usage: stat function
  // Returns: count
{
  logcommand(args);
  const dispatch* ptr = find_dispatch(args[0]);
  if(!ptr)
    RETURN(err, "Unknown operation to stat");
  RETURN(ok, itoa(ptr->count));
}

CMD(statall)
  // Usage: statall
  // Returns: name:count;name:count...
{
  logcommand(args);
  mystring r;
  for(const dispatch* ptr = dispatch_table; ptr->function != 0; ptr++) {
    if(!r.empty())
      r += ";";
    r += ptr->name;
    r += ":";
    r += itoa(ptr->count);
  }
  RETURN(ok, r);
}
#endif

response dispatch_cmd(command& args, int fd)
{
  dispatch* ptr = find_dispatch(args.name());
  if(ptr) {
    if(args.count() < ptr->arg_min ||
       (ptr->arg_max != unsigned(-1) && args.count() > ptr->arg_max))
      RETURN(bad, "Incorrect number of parameters to command " + args.name());
    //++ptr->count;
    if(ptr->decode_virtual) {
      if(is_local(args[0]))
	args.replace_first_two(args[1]);
      else {
	mystring baseuser(find_virtual(args[0]));
	if(baseuser.empty())
	  RETURN(err, "Invalid or unknown domain name: " + args[0]);
	else if(!args[1])
	  args.replace_first_two(baseuser);
	else
	  args.replace_first_two(baseuser + "-" + args[1]);
      }
    }
    return ptr->function(args, fd);
  }
  else
    RETURN(bad, "Invalid operation");
}