#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/file.h>
#include <sys/stat.h>
#include <sys/errno.h>
#include <limits.h>
#include <unistd.h>
#include <time.h>

#include <readline/readline.h>
#include <readline/history.h>

#define _GNU_SOURCE
#include <getopt.h>

#include <main.h>
#include <commands.h>
#include <completion.h>
#include <utils.h>

#include <shell.h>
#include <users.h>


char user_name[USER_NAME_SIZE];
char user_role[USER_ROLE_SIZE];

/* **************************************************************** */
/*                                                                  */
/*                       SILA Users Commands                        */
/*                                                                  */
/* **************************************************************** */

/* String to pass to system ().  This is for the LIST, VIEW and RENAME
   commands. */
static char syscom[PATH_MAX];

static int argc;
static char args[PATH_MAX];
static const char *argv[PATH_MAX];

/* Options: */
static int  opt_help = 0;
static char opt_name[USER_NAME_SIZE];
static char opt_role[USER_ROLE_SIZE];

void split_args( const char *name, char *arg )
{
  char *p = NULL;
  bzero( &args[0], PATH_MAX );
  bzero( (void *)argv, sizeof( argv ) );
  argv[0] = name;
  argc = 1;

  if( !arg )
  {
    return;
  }

  strncpy( args, arg, PATH_MAX - 1 );
  p = &args[0];
  while( *p )
  {
    while( *p && (*p == '\t' || *p == ' ') )
      ++p;

    if( *p )
    {
      argv[argc] = p;
      while( *p && *p != '\t' && *p != ' ' )
        ++p;
      *p = '\0';
      ++p;

      ++argc;
    }
  }
}

void get_args( int argc, char * const *argv )
{
  const char* short_options = "hn:r:";

  const struct option long_options[] =
  {
    { "help",        no_argument,       NULL, 'h' },
    { "name",        required_argument, NULL, 'n' },
    { "role",        required_argument, NULL, 'r' },
    { NULL,          0,                 NULL,  0  }
  };

  int ret;
  int option_index = 0;

  /* Reset options: */
  opt_help = 0;
  bzero( &opt_name[0], USER_NAME_SIZE );
  bzero( &opt_role[0], USER_ROLE_SIZE );

  while( (ret = getopt_long( argc, argv, short_options, long_options, &option_index )) != -1 )
  {
    switch( ret )
    {
      case 'h':
      {
        opt_help = 1;
        break;
      }
      case 'n':
      {
        if( optarg != NULL )
          strncpy( opt_name, (const char *)optarg, sizeof( opt_name ) - 1 );
        else
          opt_help = 1;
        break;
      }
      case 'r':
      {
        if( optarg != NULL )
          strncpy( opt_role, (const char *)optarg, sizeof( opt_role ) - 1 );
        else
          opt_help = 1;
        break;
      }
      case '?': default:
      {
        opt_help = 1;
        break;
      }
    }
  }

  /*
    Assume that the last argument is a user name:
   */
  if( optind < argc )
  {
    strncpy( opt_name, (const char *)argv[optind++], sizeof( opt_name ) - 1 );
  }

  /* This is important for call getopt_long() several times! */
  optind = 0;
}


/***************************************************************
  Users management commands:
 ***************************************************************/

static void passwd_usage( char *fname )
{
  fprintf( stdout, "\n" );
  fprintf( stdout, "Usage: %s [username]\n", fname );
  fprintf( stdout, "\n" );
  fprintf( stdout, "Set or change user password.\n" );
  fprintf( stdout, "\n" );
  fprintf( stdout, "Options:\n" );
  fprintf( stdout, "  -h,--help    Display this information.\n" );
  fprintf( stdout, "\n" );
  fprintf( stdout, "Parameter:\n" );
  fprintf( stdout, "  [username]   User name (default: %s).\n", user_name );
  fprintf( stdout, "\n" );
}

int com_passwd( char *arg )
{
  split_args( "passwd", arg );

  get_args( argc, (char * const *)&argv[0] );

  if( opt_help )
  {
    passwd_usage( "passwd" );
  }
  else
  {
    /* check args */
    if( !opt_name[0] )
    {
      sprintf( syscom, "passwd %s", user_name );
    }
    else
    {
      if( !strncmp( user_role, ROLE_ADMIN, sizeof( ROLE_ADMIN ) ) )
        sprintf( syscom, "passwd %s", opt_name );
      else
        sprintf( syscom, "echo \"%s: Only users with role='%s' can change other user's password.\"",
                                "passwd",                  ROLE_ADMIN );
    }

    return( system( syscom ) );
  }

  return( 0 );
}


static void useradd_usage( char *fname )
{
  fprintf( stdout, "\n" );
  fprintf( stdout, "Usage: %s [options] [username]\n", fname );
  fprintf( stdout, "\n" );
  fprintf( stdout, "Delete user.\n" );
  fprintf( stdout, "\n" );
  fprintf( stdout, "Options:\n" );
  fprintf( stdout, "  -h,--help    Display this information.\n" );
  fprintf( stdout, "  -n,--name    User Name.\n" );
  fprintf( stdout, "  -r,--role    User Role (default: %s).\n", ROLE_USER );
  fprintf( stdout, "\n" );
  fprintf( stdout, "Parameter:\n" );
  fprintf( stdout, "  [username]   User name (if not set by -n option).\n" );
  fprintf( stdout, "\n" );
}

int com_useradd( char *arg )
{
  split_args( "useradd", arg );

  get_args( argc, (char * const *)&argv[0] );

  if( opt_help )
  {
    useradd_usage( "useradd" );
  }
  else
  {
    /* check args */
    if( !opt_name[0] || !opt_role[0] )
    {
      useradd_usage( "useradd" );
    }
    else
    {
      /* temporary STUB. Should use busctl or REST */
      sprintf( syscom, "echo 'useradd --name %s --role %s (command is not implemented yet).'", opt_name, opt_role );
      return( system( syscom ) );
    }
  }

  return( 0 );

}


static void userdel_usage( char *fname )
{
  fprintf( stdout, "\n" );
  fprintf( stdout, "Usage: %s [options] [username]\n", fname );
  fprintf( stdout, "\n" );
  fprintf( stdout, "Delete user.\n" );
  fprintf( stdout, "\n" );
  fprintf( stdout, "Options:\n" );
  fprintf( stdout, "  -h,--help    Display this information.\n" );
  fprintf( stdout, "  -n,--name    User Name.\n" );
  fprintf( stdout, "\n" );
  fprintf( stdout, "Parameter:\n" );
  fprintf( stdout, "  [username]   User name (if not set by -n option).\n" );
  fprintf( stdout, "\n" );
}

int com_userdel( char *arg )
{
  split_args( "userdel", arg );

  get_args( argc, (char * const *)&argv[0] );

  if( opt_help )
  {
    userdel_usage( "userdel" );
  }
  else
  {
    /* check args */
    if( !opt_name[0] )
    {
      useradd_usage( "userdel" );
    }
    else
    {
      /* temporary STUB. Should use busctl or REST */
      sprintf( syscom, "echo 'userdel --name %s (command is not implemented yet).'", opt_name );
      return( system( syscom ) );
    }
  }

  return( 0 );
}


static void userlist_usage( char *fname )
{
  fprintf( stdout, "\n" );
  fprintf( stdout, "Usage: %s [options]\n", fname );
  fprintf( stdout, "\n" );
  fprintf( stdout, "List users.\n" );
  fprintf( stdout, "\n" );
  fprintf( stdout, "Options:\n" );
  fprintf( stdout, "  -h,--help    Display this information.\n" );
  fprintf( stdout, "\n" );
}

int com_userlist( char *arg )
{
  split_args( "list", arg );

/* test:
  printf( "argc: %d\n", argc );
  if( argc > 0 )
  {
    int i = 0;
    while( i < argc )
    {
      printf( "arg[%d] = '%s';\n", i, argv[i] );
      ++i;
    }
  }
 */

  get_args( argc, (char * const *)&argv[0] );

  if( opt_help )
  {
    userlist_usage( "list" );
  }
  else
  {
    /*
      list command:
     */
    sprintf( syscom, "busctl --list tree %s | grep %s/ | sed 's,%s/,  ,'",
                                         USERS_DBUS_SERVICE,
                                                   USERS_DBUS_ROOT_PATH,
                                                                USERS_DBUS_ROOT_PATH );
    return( system( syscom ) );
  }

  return( 0 );
}