/* Removes leading and/or trailing whitespaces
   Copyright (C) 2006-2008 Free Software Foundation, Inc.

   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/>.  */

/* Written by Davide Angelocola <davide.angelocola@gmail.com> */

#include <config.h>

/* Specification.  */
#include "trim.h"

#include <ctype.h>
#include <string.h>

#if HAVE_MBRTOWC 
# include <stddef.h>
# include <stdlib.h>
# include "mbchar.h"
# include "mbiter.h"
#endif

#include "xalloc.h"

char *
trim2(const char *s, int how)
{
  char *d;
    
  d = strdup(s);

  if (!d)
    xalloc_die();
  
#if HAVE_MBRTOWC
  if (MB_CUR_MAX > 1)
    {
      mbi_iterator_t i;
      
      /* Trim leading whitespaces. */
      if (how != TRIM_TRAILING) 
	{
	  mbi_init (i, d, strlen (d));
      
	  for (; mbi_avail (i) && mb_isspace (mbi_cur (i)); mbi_advance (i))
	    ;
	  
	  memmove (d, mbi_cur_ptr (i), strlen (mbi_cur_ptr (i)) + 1);
	}
  
      /* Trim trailing whitespaces. */
      if (how != TRIM_LEADING) 
	{
	  int state = 0;
	  char *r; /* used only while state = 2 */
	  
	  mbi_init (i, d, strlen (d));

	  for (; mbi_avail (i); mbi_advance (i)) 
	    {
	      if (state == 0 && mb_isspace (mbi_cur (i))) 
		{
		  state = 0;
		  continue;
		}
	      
	      if (state == 0 && !mb_isspace (mbi_cur (i)))
		{
		  state = 1;
		  continue;
		}
	      
	      if (state == 1 && !mb_isspace (mbi_cur (i)))
		{
		  state = 1;
		  continue;
		}
	      
	      if (state == 1 && mb_isspace (mbi_cur (i))) 
		{
		  state = 2;
		  r = (char *) mbi_cur_ptr (i);
		}
	      else if (state == 2 && mb_isspace (mbi_cur (i))) 
		{
		  state = 2;
		} 
	      else 
		{
		  state = 1;
		}
	    }
	  
	  if (state == 2) 
	    *r = '\0';
	}
    }
  else
#endif /* HAVE_MBRTOWC */
    {
      char *p;
      
      /* Trim leading whitespaces. */
      if (how != TRIM_TRAILING) {
	for (p = d; *p && isspace ((unsigned char) *p); p++)
	  ;			

	memmove (d, p, strlen (p) + 1);
      }

      /* Trim trailing whitespaces. */
      if (how != TRIM_LEADING) {
	for (p = d + strlen (d) - 1; p >= d && isspace ((unsigned char) *p); p--)
	  *p = '\0';
      }
    }
  
  return d;
}