/*-------------------------------------------------------------------------*/

/**

   @file    iniparser.c

   @author  N. Devillard

   @brief   Parser for ini files.

*/

/*--------------------------------------------------------------------------*/

/*---------------------------- Includes ------------------------------------*/

#include <ctype.h>

#include "iniparser.h"

/*---------------------------- Defines -------------------------------------*/

#define ASCIILINESZ         (1024)

#define INI_INVALID_KEY     ((char*)-1)

/*---------------------------------------------------------------------------

                        Private to this module

 ---------------------------------------------------------------------------*/

/**

 * This enum stores the status for each parsed line (internal use only).

 */

typedef enum _line_status_ {

    LINE_UNPROCESSED,

    LINE_ERROR,

    LINE_EMPTY,

    LINE_COMMENT,

    LINE_SECTION,

    LINE_VALUE

} line_status ;

/*-------------------------------------------------------------------------*/

/**

  @brief    Convert a string to lowercase.

  @param    s   String to convert.

  @return   ptr to statically allocated string.

  This function returns a pointer to a statically allocated string

  containing a lowercased version of the input string. Do not free

  or modify the returned string! Since the returned string is statically

  allocated, it will be modified at each function call (not re-entrant).

 */

/*--------------------------------------------------------------------------*/

static char * strlwc(const char * s)

{

    static char l[ASCIILINESZ+1];

    int i ;

    if (s==NULL) return NULL ;

    memset(l, 0, ASCIILINESZ+1);

    i=0 ;

    while (s[i] && i<ASCIILINESZ) {

        l[i] = (char)tolower((int)s[i]);

        i++ ;

    }

    l[ASCIILINESZ]=(char)0;

    return l ;

}

/*-------------------------------------------------------------------------*/

/**

  @brief    Remove blanks at the beginning and the end of a string.

  @param    s   String to parse.

  @return   ptr to statically allocated string.

  This function returns a pointer to a statically allocated string,

  which is identical to the input string, except that all blank

  characters at the end and the beg. of the string have been removed.

  Do not free or modify the returned string! Since the returned string

  is statically allocated, it will be modified at each function call

  (not re-entrant).

 */

/*--------------------------------------------------------------------------*/

static char * strstrip(const char * s)

{

    static char l[ASCIILINESZ+1];

    char * last ;

    if (s==NULL) return NULL ;

    while (isspace((int)*s) && *s) s++;

    memset(l, 0, ASCIILINESZ+1);

    strcpy(l, s);

    last = l + strlen(l);

    while (last > l) {

        if (!isspace((int)*(last-1)))

            break ;

        last -- ;

    }

    *last = (char)0;

    return (char*)l ;

}

/*-------------------------------------------------------------------------*/

/**

  @brief    Get number of sections in a dictionary

  @param    d   Dictionary to examine

  @return   int Number of sections found in dictionary

  This function returns the number of sections found in a dictionary.

  The test to recognize sections is done on the string stored in the

  dictionary: a section name is given as "section" whereas a key is

  stored as "section:key", thus the test looks for entries that do not

  contain a colon.

  This clearly fails in the case a section name contains a colon, but

  this should simply be avoided.

  This function returns -1 in case of error.

 */

/*--------------------------------------------------------------------------*/

int iniparser_getnsec(dictionary * d)

{

    int i ;

    int nsec ;

    if (d==NULL) return -1 ;

    nsec=0 ;

    for (i=0 ; i<d->size ; i++) {

        if (d->key[i]==NULL)

            continue ;

        if (strchr(d->key[i], ':')==NULL) {

            nsec ++ ;

        }

    }

    return nsec ;

}

/*-------------------------------------------------------------------------*/

/**

  @brief    Get name for section n in a dictionary.

  @param    d   Dictionary to examine

  @param    n   Section number (from 0 to nsec-1).

  @return   Pointer to char string

  This function locates the n-th section in a dictionary and returns

  its name as a pointer to a string statically allocated inside the

  dictionary. Do not free or modify the returned string!

  This function returns NULL in case of error.

 */

/*--------------------------------------------------------------------------*/

char * iniparser_getsecname(dictionary * d, int n)

{

    int i ;

    int foundsec ;

    if (d==NULL || n<0) return NULL ;

    foundsec=0 ;

    for (i=0 ; i<d->size ; i++) {

        if (d->key[i]==NULL)

            continue ;

        if (strchr(d->key[i], ':')==NULL) {

            foundsec++ ;

            if (foundsec>n)

                break ;

        }

    }

    if (foundsec<=n) {

        return NULL ;

    }

    return d->key[i] ;

}

/*-------------------------------------------------------------------------*/

/**

  @brief    Dump a dictionary to an opened file pointer.

  @param    d   Dictionary to dump.

  @param    f   Opened file pointer to dump to.

  @return   void

  This function prints out the contents of a dictionary, one element by

  line, onto the provided file pointer. It is OK to specify @c stderr

  or @c stdout as output files. This function is meant for debugging

  purposes mostly.

 */

/*--------------------------------------------------------------------------*/

void iniparser_dump(dictionary * d, FILE * f)

{

    int     i ;

    if (d==NULL || f==NULL) return ;

    for (i=0 ; i<d->size ; i++) {

        if (d->key[i]==NULL)

            continue ;

        if (d->val[i]!=NULL) {

            fprintf(f, "[%s]=[%s]\n", d->key[i], d->val[i]);

        } else {

            fprintf(f, "[%s]=UNDEF\n", d->key[i]);

        }

    }

    return ;

}

/*-------------------------------------------------------------------------*/

/**

  @brief    Save a dictionary to a loadable ini file

  @param    d   Dictionary to dump

  @param    f   Opened file pointer to dump to

  @return   void

  This function dumps a given dictionary into a loadable ini file.

  It is Ok to specify @c stderr or @c stdout as output files.

 */

/*--------------------------------------------------------------------------*/

void iniparser_dump_ini(dictionary * d, FILE * f)

{

    int     i ;

    int     nsec ;

    char *  secname ;

    if (d==NULL || f==NULL) return ;

    nsec = iniparser_getnsec(d);

    if (nsec<1) {

        /* No section in file: dump all keys as they are */

        for (i=0 ; i<d->size ; i++) {

            if (d->key[i]==NULL)

                continue ;

            fprintf(f, "%s = %s\n", d->key[i], d->val[i]);

        }

        return ;

    }

    for (i=0 ; i<nsec ; i++) {

        secname = iniparser_getsecname(d, i) ;

        iniparser_dumpsection_ini(d, secname, f) ;

    }

    fprintf(f, "\n");

    return ;

}

/*-------------------------------------------------------------------------*/

/**

  @brief    Save a dictionary section to a loadable ini file

  @param    d   Dictionary to dump

  @param    s   Section name of dictionary to dump

  @param    f   Opened file pointer to dump to

  @return   void

  This function dumps a given section of a given dictionary into a loadable ini

  file.  It is Ok to specify @c stderr or @c stdout as output files.

 */

/*--------------------------------------------------------------------------*/

void iniparser_dumpsection_ini(dictionary * d, char * s, FILE * f)

{

    int     j ;

    char    keym[ASCIILINESZ+1];

    int     seclen ;

    if (d==NULL || f==NULL) return ;

    if (! iniparser_find_entry(d, s)) return ;

    seclen  = (int)strlen(s);

    fprintf(f, "\n[%s]\n", s);

    sprintf(keym, "%s:", s);

    for (j=0 ; j<d->size ; j++) {

        if (d->key[j]==NULL)

            continue ;

        if (!strncmp(d->key[j], keym, seclen+1)) {

            fprintf(f,

                    "%-30s = %s\n",

                    d->key[j]+seclen+1,

                    d->val[j] ? d->val[j] : "");

        }

    }

    fprintf(f, "\n");

    return ;

}

/*-------------------------------------------------------------------------*/

/**

  @brief    Get the number of keys in a section of a dictionary.

  @param    d   Dictionary to examine

  @param    s   Section name of dictionary to examine

  @return   Number of keys in section

 */

/*--------------------------------------------------------------------------*/

int iniparser_getsecnkeys(dictionary * d, char * s)

{

    int     seclen, nkeys ;

    char    keym[ASCIILINESZ+1];

    int j ;

    nkeys = 0;

    if (d==NULL) return nkeys;

    if (! iniparser_find_entry(d, s)) return nkeys;

    seclen  = (int)strlen(s);

    sprintf(keym, "%s:", s);

    for (j=0 ; j<d->size ; j++) {

        if (d->key[j]==NULL)

            continue ;

        if (!strncmp(d->key[j], keym, seclen+1)) 

            nkeys++;

    }

    return nkeys;

}

/*-------------------------------------------------------------------------*/

/**

  @brief    Get the number of keys in a section of a dictionary.

  @param    d   Dictionary to examine

  @param    s   Section name of dictionary to examine

  @return   pointer to statically allocated character strings

  This function queries a dictionary and finds all keys in a given section.

  Each pointer in the returned char pointer-to-pointer is pointing to

  a string allocated in the dictionary; do not free or modify them.

  This function returns NULL in case of error.

 */

/*--------------------------------------------------------------------------*/

char ** iniparser_getseckeys(dictionary * d, char * s)

{

    char **keys;

    int i, j ;

    char    keym[ASCIILINESZ+1];

    int     seclen, nkeys ;

    keys = NULL;

    if (d==NULL) return keys;

    if (! iniparser_find_entry(d, s)) return keys;

    nkeys = iniparser_getsecnkeys(d, s);

    keys = (char**) malloc(nkeys*sizeof(char*));

    seclen  = (int)strlen(s);

    sprintf(keym, "%s:", s);

    i = 0;

    for (j=0 ; j<d->size ; j++) {

        if (d->key[j]==NULL)

            continue ;

        if (!strncmp(d->key[j], keym, seclen+1)) {

            keys[i] = d->key[j];

            i++;

        }

    }

    return keys;

}

/*-------------------------------------------------------------------------*/

/**

  @brief    Get the string associated to a key

  @param    d       Dictionary to search

  @param    key     Key string to look for

  @param    def     Default value to return if key not found.

  @return   pointer to statically allocated character string

  This function queries a dictionary for a key. A key as read from an

  ini file is given as "section:key". If the key cannot be found,

  the pointer passed as 'def' is returned.

  The returned char pointer is pointing to a string allocated in

  the dictionary, do not free or modify it.

 */

/*--------------------------------------------------------------------------*/

char * iniparser_getstring(dictionary * d, const char * key, char * def)

{

    char * lc_key ;

    char * sval ;

    if (d==NULL || key==NULL)

        return def ;

    lc_key = strlwc(key);

    sval = dictionary_get(d, lc_key, def);

    return sval ;

}

/*-------------------------------------------------------------------------*/

/**

  @brief    Get the string associated to a key, convert to an int

  @param    d Dictionary to search

  @param    key Key string to look for

  @param    notfound Value to return in case of error

  @return   integer

  This function queries a dictionary for a key. A key as read from an

  ini file is given as "section:key". If the key cannot be found,

  the notfound value is returned.

  Supported values for integers include the usual C notation

  so decimal, octal (starting with 0) and hexadecimal (starting with 0x)

  are supported. Examples:

  "42"      ->  42

  "042"     ->  34 (octal -> decimal)

  "0x42"    ->  66 (hexa  -> decimal)

  Warning: the conversion may overflow in various ways. Conversion is

  totally outsourced to strtol(), see the associated man page for overflow

  handling.

  Credits: Thanks to A. Becker for suggesting strtol()

 */

/*--------------------------------------------------------------------------*/

int iniparser_getint(dictionary * d, const char * key, int notfound)

{

    char    *   str ;

    str = iniparser_getstring(d, key, INI_INVALID_KEY);

    if (str==INI_INVALID_KEY) return notfound ;

    return (int)strtol(str, NULL, 0);

}

/*-------------------------------------------------------------------------*/

/**

  @brief    Get the string associated to a key, convert to a double

  @param    d Dictionary to search

  @param    key Key string to look for

  @param    notfound Value to return in case of error

  @return   double

  This function queries a dictionary for a key. A key as read from an

  ini file is given as "section:key". If the key cannot be found,

  the notfound value is returned.

 */

/*--------------------------------------------------------------------------*/

double iniparser_getdouble(dictionary * d, const char * key, double notfound)

{

    char    *   str ;

    str = iniparser_getstring(d, key, INI_INVALID_KEY);

    if (str==INI_INVALID_KEY) return notfound ;

    return atof(str);

}

/*-------------------------------------------------------------------------*/

/**

  @brief    Get the string associated to a key, convert to a boolean

  @param    d Dictionary to search

  @param    key Key string to look for

  @param    notfound Value to return in case of error

  @return   integer

  This function queries a dictionary for a key. A key as read from an

  ini file is given as "section:key". If the key cannot be found,

  the notfound value is returned.

  A true boolean is found if one of the following is matched:

  - A string starting with 'y'

  - A string starting with 'Y'

  - A string starting with 't'

  - A string starting with 'T'

  - A string starting with '1'

  A false boolean is found if one of the following is matched:

  - A string starting with 'n'

  - A string starting with 'N'

  - A string starting with 'f'

  - A string starting with 'F'

  - A string starting with '0'

  The notfound value returned if no boolean is identified, does not

  necessarily have to be 0 or 1.

 */

/*--------------------------------------------------------------------------*/

int iniparser_getboolean(dictionary * d, const char * key, int notfound)

{

    char    *   c ;

    int         ret ;

    c = iniparser_getstring(d, key, INI_INVALID_KEY);

    if (c==INI_INVALID_KEY) return notfound ;

    if (c[0]=='y' || c[0]=='Y' || c[0]=='1' || c[0]=='t' || c[0]=='T') {

        ret = 1 ;

    } else if (c[0]=='n' || c[0]=='N' || c[0]=='0' || c[0]=='f' || c[0]=='F') {

        ret = 0 ;

    } else {

        ret = notfound ;

    }

    return ret;

}

/*-------------------------------------------------------------------------*/

/**

  @brief    Finds out if a given entry exists in a dictionary

  @param    ini     Dictionary to search

  @param    entry   Name of the entry to look for

  @return   integer 1 if entry exists, 0 otherwise

  Finds out if a given entry exists in the dictionary. Since sections

  are stored as keys with NULL associated values, this is the only way

  of querying for the presence of sections in a dictionary.

 */

/*--------------------------------------------------------------------------*/

int iniparser_find_entry(

    dictionary  *   ini,

    const char  *   entry

)

{

    int found=0 ;

    if (iniparser_getstring(ini, entry, INI_INVALID_KEY)!=INI_INVALID_KEY) {

        found = 1 ;

    }

    return found ;

}

/*-------------------------------------------------------------------------*/

/**

  @brief    Set an entry in a dictionary.

  @param    ini     Dictionary to modify.

  @param    entry   Entry to modify (entry name)

  @param    val     New value to associate to the entry.

  @return   int 0 if Ok, -1 otherwise.

  If the given entry can be found in the dictionary, it is modified to

  contain the provided value. If it cannot be found, -1 is returned.

  It is Ok to set val to NULL.

 */

/*--------------------------------------------------------------------------*/

int iniparser_set(dictionary * ini, const char * entry, const char * val)

{

    return dictionary_set(ini, strlwc(entry), val) ;

}