/simpletypesystem/trunk

/*

Copyright (c) 2013-2014 Gustav Hartvigsson

Permission is hereby granted, free of charge, to any person obtaining a copy

of this software and associated documentation files (the "Software"), to deal

in the Software without restriction, including without limitation the rights

to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

copies of the Software, and to permit persons to whom the Software is

furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in

all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN

THE SOFTWARE.

*/

#include "utils.h"

#include <string.h>

#include <stdlib.h>

#include <stdio.h>

#include <stdarg.h>

#include <time.h>

#include <wchar.h>

#include <locale.h>

schar *

s_string_new (const schar * s) {

  if (s == NULL) {

    return NULL;

  }

  size_t s_len = strlen (s);

  assert (s_len > 0);

  schar * ret_val = s_malloc (s_len + 1);

  strcpy (ret_val, s);

  ret_val[s_len] = '\0';

  return ret_val;

}

schar *

s_string_new_fmt (const schar * format, ...) {

  schar * buffer = s_malloc (strlen (format) + 512);

  schar * ret_val = NULL;

  va_list args;

  va_start(args, format);

  vsprintf (buffer, format, args);

  ret_val = s_string_new (buffer);

  va_end(args);

  s_free (buffer);

  return ret_val;

}

schar *

s_string_new_with_len (const schar * s, size_t len) {

  schar * ret_val = s_malloc (len + 1);

  ret_val[len + 1] = 0x0;

  strncpy (ret_val, s, len);

  return ret_val;

}

schar *

s_current_time (void) {

  schar * ret_val = s_malloc (21);

  time_t t = time (NULL);

  strftime (ret_val, 21, "%F %T", localtime (&t));

  return ret_val;

}

sboolean

s_string_is_equal (const schar * a, const schar * b) {

  if (strcmp (a, b)){

    return FALSE;

  } else {

    return TRUE;

  }

}

size_t

s_string_len (const schar * str) {

  return strlen (str);

}

size_t

s_ustring_len (const suchar * us) {

  if (!us) {

    return 0;

  }

  size_t ret_val = 0;

  suchar * tmp = (suchar *)us;

  while (*tmp) {

    ret_val++;

    us++;

  }

  return ret_val;

}

/******************************************************************************/

schar *

s_ustring_to_string (const suchar * us) {

  schar * buffer;

  schar * resized;

  size_t bufpos;

  mbstate_t mbstate;

  memset (&mbstate, 0, sizeof (mbstate));

  /* Thanks to Florian Philipp.

   * Thread: https://plus.google.com/u/0/+GustavHartvigsson/posts/4Wk7La1kWPP

   *

   * This is an adaptation of the original code to work with our suchar stuffs.

   */

  /* Save locale */

  schar * saved_locale;

  {

    schar * old_locale;

    size_t old_locale_len;

    old_locale = setlocale (LC_ALL, NULL);

    old_locale_len = strlen (old_locale) + 1;

    saved_locale = s_malloc (sizeof (char *) * old_locale_len);

    memcpy (saved_locale, old_locale, old_locale_len);

    /* set locale */

    setlocale (LC_ALL, "C.utf8");

  }

  buffer = malloc (sizeof (char32_t) * 4);

  schar out[MB_CUR_MAX];

  for(size_t n = 0, buflen = 4, bufpos = 0;

        n < s_ustring_len (us);

        n++, buflen += MB_CUR_MAX) { /* could be: buflen *= 2 ?*/

    sint rc = c32rtomb(out, us[n], &mbstate);

    if(! (resized = s_realloc (buffer, buflen))) {

      goto err;

    }

    buffer = resized;

    for (sint i = 0; i < rc; ++i) {

      buffer[bufpos] = out[i];

      bufpos++;

    }

  }

    /* reset locale */

  setlocale (LC_ALL, saved_locale);

  /* shrink buffer to actually required size */

  if (! (resized = s_realloc (buffer, bufpos + 1)))

    goto err;

  return resized;

err:

  s_free(buffer);

  return NULL;

}

/******************************************************************************/

suchar *

s_string_to_ustring (const schar * str) {

  suchar * buffer;

  suchar * resized;

  size_t bufpos;

  mbstate_t mbstate;

  memset (&mbstate, 0, sizeof (mbstate));

  suint slen = s_string_len (str);

  /* Addaptation of the code in s_wstring_to_string to make it convert

   * mb strings to char32_t strings.

   */

  /* Save locale */

  schar * saved_locale;

  {

    schar * old_locale;

    size_t old_locale_len;

    old_locale = setlocale (LC_ALL, NULL);

    old_locale_len = strlen (old_locale) + 1;

    saved_locale = s_malloc (sizeof (char *) * old_locale_len);

    memcpy (saved_locale, old_locale, old_locale_len);

    /* set locale */

    setlocale (LC_ALL, "C.utf8");

  }

  buffer = malloc (sizeof (char32_t) * 4);

  suchar * out = s_malloc (sizeof (suchar));

  for(size_t n = 0, buflen = 4, bufpos = 0;

        n < slen;

        buflen += 2) {

    sint rc = mbrtoc32(out, &str[n], slen ,&mbstate);

    n += rc; // Get next mb in string...?

    if(! (resized = s_realloc (buffer, buflen * sizeof (suchar)))){

      goto err;

    }

    buffer = resized;

    buffer[bufpos] = *out;

    bufpos++;

  }

  s_free (out);

  /* reset locale */

  setlocale (LC_ALL, saved_locale);

  /* shrink buffer to actually required size */

  if (! (resized = s_realloc (buffer, (bufpos + 1) * sizeof (suchar)))) {

    goto err;

  }

  return resized;

err:

  s_free(buffer);

  return NULL;

}

/******************************************************************************/

/*

 * This should not be used. If you need to use this, uncomment it.

 * I am conveting to something more sane than wchar_t.

 */

schar *

s_wstring_to_string (const wchar_t * ws) {

  s_err_print ("Using depricated function: s_wstring_to_string ().\n"

               "Do not use s_wstring_to_string, as it is platform "

               "specific and may, or may not couse troubble.\n"

               "Use uchar strings instead of wchar_t strings, and "

               "use s_ustring_to_string instead of this.\n");

  size_t buflen;

  schar * buffer;

  schar * resized;

  size_t bufpos;

  mbstate_t mbstate;

  memset (&mbstate, 0, sizeof (mbstate));

  /* Save locale */

  schar * saved_locale;

  {

    schar * old_locale;

    size_t old_locale_len;

    old_locale = setlocale (LC_ALL, NULL);

    old_locale_len = strlen (old_locale) + 1;

    saved_locale = s_malloc (sizeof (char *) * old_locale_len);

    memcpy (saved_locale, old_locale, old_locale_len);

    /* set locale */

    setlocale (LC_ALL, "C.utf8");

  }

  /* Thanks to Florian Philipp.

   * Thread: https://plus.google.com/u/0/+GustavHartvigsson/posts/4Wk7La1kWPP

   *

   * to avoid parsing the string twice, we allocate memory speculatively

   * with exponential growth. Then we shrink it at the end.

   *

   * Alternative:  use

   * const wchar_t* tmp = ws;

   * size_t buflen = wcsrtombs(NULL, &tmp, 0, mbstate);

   * buffer = malloc(buflen);

   * wcsrtombs(buffer, &ws, 0, mbstate);

   * return buffer;

   */

  for (buflen = 4, buffer = NULL, bufpos = 0; ws; buflen *= 2) {

    size_t converted;

    if(! (resized = s_realloc (buffer, buflen)))

      goto err;

    buffer = resized;

    if((converted = wcsrtombs (buffer + bufpos, &ws, buflen - bufpos, &mbstate))

       == (size_t) -1)

      goto err;

    bufpos += converted;

  }

  /* reset locale */

  setlocale (LC_ALL, saved_locale);

  /* shrink buffer to actually required size */

  if (! (resized = s_realloc (buffer, bufpos + 1))) {

    goto err;

  }

  return resized;

err:

  s_free(buffer);

  return NULL;

}