/* Copyright (C) 1997, 1998 Free Software Foundation, Inc.
   This file is part of the GNU C Library.

   The GNU C Library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Library General Public License as
   published by the Free Software Foundation; either version 2 of the
   License, or (at your option) any later version.

   The GNU C Library 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
   Library General Public License for more details.

   You should have received a copy of the GNU Library General Public
   License along with the GNU C Library; see the file COPYING.LIB.  If not,
   write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
   Boston, MA 02111-1307, USA.  */

#include <sys/stat.h>
#include <sys/types.h>
#include <fcntl.h>
#include <unistd.h>
#include <pwd.h>
#include <stdio.h>
#include <stdlib.h>
#include <wordexp.h>

#define IFS ", \n\t"

struct test_case_struct
{
  int retval;
  const char *env;
  const char *words;
  int flags;
  int wordc;
  const char *wordv[10];
} test_case[] =
  {
    /* Simple field-splitting */
    { 0, NULL, "one", 0, 1, { "one", } },
    { 0, NULL, "one two", 0, 2, { "one", "two", } },
    { 0, NULL, "one two three", 0, 3, { "one", "two", "three", } },
    { 0, NULL, " \tfoo\t\tbar ", 0, 2, { "foo", "bar", } },
    { 0, NULL, "  red  , white blue", 0, 3, { "red", "white", "blue", } },

    /* Simple parameter expansion */
    { 0, "foo", "${var}", 0, 1, { "foo", } },
    { 0, "foo", "$var", 0, 1, { "foo", } },

    /* Simple quote removal */
    { 0, NULL, "\"quoted\"", 0, 1, { "quoted", } },
    { 0, "foo", "\"$var\"\"$var\"", 0, 1, { "foofoo", } },
    { 0, NULL, "'singly-quoted'", 0, 1, { "singly-quoted", } },

    /* Simple command substitution */
    { 0, NULL, "$(echo hello)", 0, 1, { "hello", } },
    { 0, NULL, "$( (echo hello) )", 0, 1, { "hello", } },
    { 0, NULL, "$((echo hello);(echo there))", 0, 2, { "hello", "there", } },
    { 0, NULL, "`echo one two`", 0, 2, { "one", "two", } },

    /* Simple arithmetic expansion */
    { 0, NULL, "$((1 + 1))", 0, 1, { "2", } },
    { 0, NULL, "$((2-3))", 0, 1, { "-1", } },
    { 0, NULL, "$((-1))", 0, 1, { "-1", } },
    { 0, NULL, "$[50+20]", 0, 1, { "70", } },

    /* Advanced parameter expansion */
    { 0, NULL, "${var:-bar}", 0, 1, { "bar", } },
    { 0, NULL, "${var-bar}", 0, 1, { "bar", } },
    { 0, "", "${var:-bar}", 0, 1, { "bar", } },
    { 0, "foo", "${var:-bar}", 0, 1, { "foo", } },
    { 0, "", "${var-bar}", 0, 0, { NULL, } },
    { 0, NULL, "${var:=bar}", 0, 1, { "bar", } },
    { 0, NULL, "${var=bar}", 0, 1, { "bar", } },
    { 0, "", "${var:=bar}", 0, 1, { "bar", } },
    { 0, "foo", "${var:=bar}", 0, 1, { "foo", } },
    { 0, "", "${var=bar}", 0, 0, { NULL, } },
    { 0, "foo", "${var:?bar}", 0, 1, { "foo", } },
    { 0, NULL, "${var:+bar}", 0, 0, { NULL, } },
    { 0, NULL, "${var+bar}", 0, 0, { NULL, } },
    { 0, "", "${var:+bar}", 0, 0, { NULL, } },
    { 0, "foo", "${var:+bar}", 0, 1, { "bar", } },
    { 0, "", "${var+bar}", 0, 1, { "bar", } },
    { 0, "12345", "${#var}", 0, 1, { "5", } },

    { 0, "banana", "${var%na*}", 0, 1, { "bana", } },
    { 0, "banana", "${var%%na*}", 0, 1, { "ba", } },
    { 0, "borabora-island", "${var#*bora}", 0, 1, { "bora-island", } },
    { 0, "borabora-island", "${var##*bora}", 0, 1, {"-island", } },

    /* Pathname expansion */
    { 0, NULL, "???", 0, 2, { "one", "two", } },
    { 0, NULL, "[ot]??", 0, 2, { "one", "two", } },
    { 0, NULL, "t*", 0, 2, { "three", "two", } },
    { 0, NULL, "\"t\"*", 0, 2, { "three", "two", } },

    /* Nested constructs */
    { 0, "one two", "$var", 0, 2, { "one", "two", } },
    { 0, "one two three", "$var", 0, 3, { "one", "two", "three", } },
    { 0, " \tfoo\t\tbar ", "$var", 0, 2, { "foo", "bar", } },
    { 0, "  red  , white blue", "$var", 0, 3, { "red", "white", "blue", } },
    { 0, "  red  , white blue", "\"$var\"", 0, 1, { "  red  , white blue", } },
    { 0, NULL, "\"$(echo hello there)\"", 0, 1, { "hello there", } },
    { 0, NULL, "\"$(echo \"hello there\")\"", 0, 1, { "hello there", } },
    { 0, NULL, "${var=one two} \"$var\"", 0, 3, { "one", "two", "one two", } },
    { 0, "1", "$(( $(echo 3)+$var ))", 0, 1, { "4", } },
    { 0, NULL, "\"$(echo \"*\")\"", 0, 1, { "*", } },

    /* Other things that should succeed */
    { 0, NULL, "\\*\"|&;<>\"\\(\\)\\{\\}", 0, 1, { "*|&;<>(){}", } },
    { 0, "???", "$var", 0, 1, { "???", } },
    { 0, NULL, "$var", 0, 0, { NULL, } },
    { 0, NULL, "\"\\n\"", 0, 1, { "\\n", } },
    { 0, NULL, "", 0, 0, { NULL, } },

    /* Things that should fail */
    { WRDE_BADCHAR, NULL, "new\nline", 0, 0, { NULL, } },
    { WRDE_BADCHAR, NULL, "pipe|symbol", 0, 0, { NULL, } },
    { WRDE_BADCHAR, NULL, "&ampersand", 0, 0, { NULL, } },
    { WRDE_BADCHAR, NULL, "semi;colon", 0, 0, { NULL, } },
    { WRDE_BADCHAR, NULL, "<greater", 0, 0, { NULL, } },
    { WRDE_BADCHAR, NULL, "less>", 0, 0, { NULL, } },
    { WRDE_BADCHAR, NULL, "(open-paren", 0, 0, { NULL, } },
    { WRDE_BADCHAR, NULL, "close-paren)", 0, 0, { NULL, } },
    { WRDE_BADCHAR, NULL, "{open-brace", 0, 0, { NULL, } },
    { WRDE_BADCHAR, NULL, "close-brace}", 0, 0, { NULL, } },
    { WRDE_CMDSUB, NULL, "$(ls)", WRDE_NOCMD, 0, { NULL, } },
    { WRDE_BADVAL, NULL, "$var", WRDE_UNDEF, 0, { NULL, } },
    { WRDE_SYNTAX, NULL, "$[50+20))", 0, 0, { NULL, } },
    { WRDE_SYNTAX, NULL, "${%%noparam}", 0, 0, { NULL, } },
    { WRDE_SYNTAX, NULL, "${missing-brace", 0, 0, { NULL, } },
    { WRDE_SYNTAX, NULL, "$((2+))", 0, 0, { NULL, } },
    { WRDE_SYNTAX, NULL, "`", 0, 0, { NULL, } },

    { -1, NULL, NULL, 0, 0, { NULL, } },
  };

static int testit (struct test_case_struct *tc);

void
command_line_test (const char *words)
{
  wordexp_t we;
  int i;
  int retval = wordexp (words, &we, 0);
  printf ("wordexp returned %d\n", retval);
  for (i = 0; i < we.we_wordc; i++)
    printf ("we_wordv[%d] = \"%s\"\n", i, we.we_wordv[i]);
}

int
main (int argc, char *argv[])
{
  char tmpdir[32];
  struct passwd *pw;
  int test;
  int fail = 0;
  int fd;

  if (argc > 1)
    {
      command_line_test (argv[1]);
      return 0;
    }

  setenv ("IFS", IFS, 1);

  /* Set up arena for pathname expansion */
  tmpnam (tmpdir);
  if (mkdir (tmpdir, S_IRWXU) ||
      chdir (tmpdir) ||
      (fd = creat ("one", S_IRWXU)) == -1 || close (fd) ||
      (fd = creat ("two", S_IRWXU)) == -1 || close (fd) ||
      (fd = creat ("three", S_IRWXU)) == -1 || close (fd))
    return -1;

  for (test = 0; test_case[test].retval != -1; test++)
    if (testit (&test_case[test]))
      ++fail;

  pw = getpwnam ("root");
  if (pw != NULL)
    {
      struct test_case_struct ts;

      ts.retval = 0;
      ts.env = NULL;
      ts.words = "~root";
      ts.flags = 0;
      ts.wordc = 1;
      ts.wordv[0] = pw->pw_dir;

      if (testit (&ts))
	++fail;
    }

  return fail != 0;
}


static int
testit (struct test_case_struct *tc)
{
  static int test;
  int retval;
  wordexp_t we;
  int bzzzt = 0;
  int i;

  if (tc->env)
    setenv ("var", tc->env, 1);
  else
    unsetenv ("var");

  printf ("Test %d: ", ++test);
  retval = wordexp (tc->words, &we, tc->flags);

  if (retval != tc->retval || (retval != 0 && we.we_wordc != tc->wordc))
    bzzzt = 1;
  else
    for (i = 0; i < we.we_wordc; ++i)
      if (strcmp (tc->wordv[i], we.we_wordv[i]) != 0)
	{
	  bzzzt = 1;
	  break;
	}

  if (bzzzt)
    {
      printf ("FAILED\n");
      printf ("Test words: <%s>, need retval %d, wordc %d\n",
	      tc->words, tc->retval, tc->wordc);
      printf ("Got retval %d, wordc %d: ", retval, we.we_wordc);
      for (i = 0; i < we.we_wordc; ++i)
	printf ("<%s> ", we.we_wordv[i]);
      printf ("\n");
    }
  else
    printf ("OK\n");

  wordfree (&we);

  return bzzzt;
}