326 lines
8.5 KiB
C
326 lines
8.5 KiB
C
/*
|
|
* $Id$
|
|
*
|
|
Copyright 2006-2014 Chung, Hyung-Hwan.
|
|
This file is part of QSE.
|
|
|
|
QSE is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Lesser General Public License as
|
|
published by the Free Software Foundation, either version 3 of
|
|
the License, or (at your option) any later version.
|
|
|
|
QSE 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 Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public
|
|
License along with QSE. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "mod-str.h"
|
|
#include <qse/cmn/str.h>
|
|
#include <qse/cmn/chr.h>
|
|
#include "../cmn/mem.h"
|
|
#include "fnc.h"
|
|
|
|
static int fnc_normspace (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
/* normalize spaces
|
|
* - trim leading and trailing spaces
|
|
* - replace a series of spaces to a single space
|
|
*/
|
|
qse_cstr_t path;
|
|
qse_awk_val_t* retv;
|
|
qse_awk_val_t* a0;
|
|
|
|
a0 = qse_awk_rtx_getarg(rtx, 0);
|
|
|
|
path.ptr = qse_awk_rtx_getvalstr (rtx, a0, &path.len);
|
|
if (path.ptr)
|
|
{
|
|
path.len = qse_strxpac (path.ptr, path.len);
|
|
retv = qse_awk_rtx_makestrval (rtx, path.ptr, path.len);
|
|
qse_awk_rtx_freevalstr (rtx, a0, path.ptr);
|
|
if (retv) qse_awk_rtx_setretval (rtx, retv);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int trim (qse_awk_rtx_t* rtx, int flags)
|
|
{
|
|
qse_cstr_t path;
|
|
qse_char_t* npath;
|
|
qse_awk_val_t* retv;
|
|
qse_awk_val_t* a0;
|
|
|
|
a0 = qse_awk_rtx_getarg(rtx, 0);
|
|
|
|
path.ptr = qse_awk_rtx_getvalstr (rtx, a0, &path.len);
|
|
if (path.ptr)
|
|
{
|
|
npath = qse_strxtrmx (path.ptr, &path.len, flags);
|
|
retv = qse_awk_rtx_makestrval (rtx, npath, path.len);
|
|
qse_awk_rtx_freevalstr (rtx, a0, path.ptr);
|
|
if (retv) qse_awk_rtx_setretval (rtx, retv);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int fnc_trim (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
return trim (rtx, QSE_STRTRMX_LEFT | QSE_STRTRMX_RIGHT);
|
|
}
|
|
static int fnc_ltrim (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
return trim (rtx, QSE_STRTRMX_LEFT);
|
|
}
|
|
static int fnc_rtrim (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
return trim (rtx, QSE_STRTRMX_RIGHT);
|
|
}
|
|
|
|
static int is_class (qse_awk_rtx_t* rtx, qse_ctype_t ctype)
|
|
{
|
|
qse_awk_val_t* a0;
|
|
qse_char_t* str0;
|
|
qse_size_t len0;
|
|
int tmp;
|
|
|
|
a0 = qse_awk_rtx_getarg (rtx, 0);
|
|
|
|
str0 = qse_awk_rtx_getvalstr (rtx, a0, &len0);
|
|
if (str0 == QSE_NULL) return -1;
|
|
|
|
if (len0 <= 0) tmp = 0;
|
|
else
|
|
{
|
|
tmp = 1;
|
|
do
|
|
{
|
|
len0--;
|
|
if (!qse_isctype(str0[len0], ctype))
|
|
{
|
|
tmp = 0;
|
|
break;
|
|
}
|
|
}
|
|
while (len0 > 0);
|
|
qse_awk_rtx_freevalstr (rtx, a0, str0);
|
|
}
|
|
|
|
a0 = qse_awk_rtx_makeintval (rtx, tmp);
|
|
if (a0 == QSE_NULL) return -1;
|
|
|
|
qse_awk_rtx_setretval (rtx, a0);
|
|
return 0;
|
|
}
|
|
|
|
static int fnc_isalnum (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
return is_class (rtx, QSE_CTYPE_ALNUM);
|
|
}
|
|
|
|
static int fnc_isalpha (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
return is_class (rtx, QSE_CTYPE_ALPHA);
|
|
}
|
|
|
|
static int fnc_isblank (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
return is_class (rtx, QSE_CTYPE_BLANK);
|
|
}
|
|
|
|
static int fnc_iscntrl (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
return is_class (rtx, QSE_CTYPE_CNTRL);
|
|
}
|
|
|
|
static int fnc_isdigit (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
return is_class (rtx, QSE_CTYPE_DIGIT);
|
|
}
|
|
|
|
static int fnc_isgraph (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
return is_class (rtx, QSE_CTYPE_GRAPH);
|
|
}
|
|
|
|
static int fnc_islower (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
return is_class (rtx, QSE_CTYPE_LOWER);
|
|
}
|
|
|
|
static int fnc_isprint (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
return is_class (rtx, QSE_CTYPE_PRINT);
|
|
}
|
|
|
|
static int fnc_ispunct (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
return is_class (rtx, QSE_CTYPE_PUNCT);
|
|
}
|
|
|
|
static int fnc_isspace (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
return is_class (rtx, QSE_CTYPE_SPACE);
|
|
}
|
|
|
|
static int fnc_isupper (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
return is_class (rtx, QSE_CTYPE_UPPER);
|
|
}
|
|
|
|
static int fnc_isxdigit (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
return is_class (rtx, QSE_CTYPE_XDIGIT);
|
|
}
|
|
|
|
static int fnc_value (qse_awk_rtx_t* rtx, const qse_awk_fnc_info_t* fi)
|
|
{
|
|
/* return the numeric value for the first character */
|
|
qse_cstr_t path;
|
|
qse_awk_val_t* retv;
|
|
qse_awk_val_t* a0;
|
|
|
|
a0 = qse_awk_rtx_getarg(rtx, 0);
|
|
|
|
path.ptr = qse_awk_rtx_getvalstr (rtx, a0, &path.len);
|
|
if (path.ptr && path.len >= 1)
|
|
{
|
|
#if defined(QSE_CHAR_IS_MCHAR)
|
|
/* typecasting in case qse_mchar_t is signed */
|
|
retv = qse_awk_rtx_makeintval (rtx, (unsigned char)path.ptr[0]);
|
|
#else
|
|
retv = qse_awk_rtx_makeintval (rtx, path.ptr[0]);
|
|
#endif
|
|
qse_awk_rtx_freevalstr (rtx, a0, path.ptr);
|
|
if (retv) qse_awk_rtx_setretval (rtx, retv);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
typedef struct fnctab_t fnctab_t;
|
|
struct fnctab_t
|
|
{
|
|
const qse_char_t* name;
|
|
qse_awk_mod_sym_fnc_t info;
|
|
};
|
|
|
|
|
|
#define A_MAX QSE_TYPE_MAX(int)
|
|
|
|
static fnctab_t fnctab[] =
|
|
{
|
|
/* keep this table sorted for binary search in query(). */
|
|
{ QSE_T("gsub"), { { 2, 3, QSE_T("xvr")}, qse_awk_fnc_gsub, 0 } },
|
|
{ QSE_T("index"), { { 2, 3, QSE_NULL }, qse_awk_fnc_index, 0 } },
|
|
{ QSE_T("isalnum"), { { 1, 1, QSE_NULL }, fnc_isalnum, 0 } },
|
|
{ QSE_T("isalpha"), { { 1, 1, QSE_NULL }, fnc_isalpha, 0 } },
|
|
{ QSE_T("isblank"), { { 1, 1, QSE_NULL }, fnc_isblank, 0 } },
|
|
{ QSE_T("iscntrl"), { { 1, 1, QSE_NULL }, fnc_iscntrl, 0 } },
|
|
{ QSE_T("isdigit"), { { 1, 1, QSE_NULL }, fnc_isdigit, 0 } },
|
|
{ QSE_T("isgraph"), { { 1, 1, QSE_NULL }, fnc_isgraph, 0 } },
|
|
{ QSE_T("islower"), { { 1, 1, QSE_NULL }, fnc_islower, 0 } },
|
|
{ QSE_T("isprint"), { { 1, 1, QSE_NULL }, fnc_isprint, 0 } },
|
|
{ QSE_T("ispunct"), { { 1, 1, QSE_NULL }, fnc_ispunct, 0 } },
|
|
{ QSE_T("isspace"), { { 1, 1, QSE_NULL }, fnc_isspace, 0 } },
|
|
{ QSE_T("isupper"), { { 1, 1, QSE_NULL }, fnc_isupper, 0 } },
|
|
{ QSE_T("isxdigit"), { { 1, 1, QSE_NULL }, fnc_isxdigit, 0 } },
|
|
{ QSE_T("length"), { { 1, 1, QSE_NULL }, qse_awk_fnc_length, 0 } },
|
|
{ QSE_T("ltrim"), { { 1, 1, QSE_NULL }, fnc_ltrim, 0 } },
|
|
{ QSE_T("match"), { { 2, 3, QSE_T("vxv") }, qse_awk_fnc_match, 0 } },
|
|
{ QSE_T("normspace"), { { 1, 1, QSE_NULL }, fnc_normspace, 0 } },
|
|
{ QSE_T("printf"), { { 1, A_MAX, QSE_NULL }, qse_awk_fnc_sprintf, 0 } },
|
|
{ QSE_T("rindex"), { { 2, 3, QSE_NULL }, qse_awk_fnc_rindex, 0 } },
|
|
{ QSE_T("rtrim"), { { 1, 1, QSE_NULL }, fnc_rtrim, 0 } },
|
|
{ QSE_T("split"), { { 2, 3, QSE_T("vrx") }, qse_awk_fnc_split, 0 } },
|
|
{ QSE_T("sub"), { { 2, 3, QSE_T("xvr") }, qse_awk_fnc_sub, 0 } },
|
|
{ QSE_T("substr"), { { 2, 3, QSE_NULL }, qse_awk_fnc_substr, 0 } },
|
|
{ QSE_T("tolower"), { { 1, 1, QSE_NULL }, qse_awk_fnc_tolower, 0 } },
|
|
{ QSE_T("toupper"), { { 1, 1, QSE_NULL }, qse_awk_fnc_toupper, 0 } },
|
|
{ QSE_T("trim"), { { 1, 1, QSE_NULL }, fnc_trim, 0 } },
|
|
{ QSE_T("value"), { { 1, 1, QSE_NULL }, fnc_value, 0 } }
|
|
};
|
|
|
|
static int query (qse_awk_mod_t* mod, qse_awk_t* awk, const qse_char_t* name, qse_awk_mod_sym_t* sym)
|
|
{
|
|
qse_cstr_t ea;
|
|
int left, right, mid, n;
|
|
|
|
left = 0; right = QSE_COUNTOF(fnctab) - 1;
|
|
|
|
while (left <= right)
|
|
{
|
|
mid = (left + right) / 2;
|
|
|
|
n = qse_strcmp (fnctab[mid].name, name);
|
|
if (n > 0) right = mid - 1;
|
|
else if (n < 0) left = mid + 1;
|
|
else
|
|
{
|
|
sym->type = QSE_AWK_MOD_FNC;
|
|
sym->u.fnc = fnctab[mid].info;
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
#if 0
|
|
left = 0; right = QSE_COUNTOF(inttab) - 1;
|
|
while (left <= right)
|
|
{
|
|
mid = (left + right) / 2;
|
|
|
|
n = qse_strcmp (inttab[mid].name, name);
|
|
if (n > 0) right = mid - 1;
|
|
else if (n < 0) left = mid + 1;
|
|
else
|
|
{
|
|
sym->type = QSE_AWK_MOD_INT;
|
|
sym->u.in = inttab[mid].info;
|
|
return 0;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
ea.ptr = (qse_char_t*)name;
|
|
ea.len = qse_strlen(name);
|
|
qse_awk_seterror (awk, QSE_AWK_ENOENT, &ea, QSE_NULL);
|
|
return -1;
|
|
}
|
|
|
|
/* TODO: proper resource management */
|
|
|
|
static int init (qse_awk_mod_t* mod, qse_awk_rtx_t* rtx)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
static void fini (qse_awk_mod_t* mod, qse_awk_rtx_t* rtx)
|
|
{
|
|
/* TODO: anything */
|
|
}
|
|
|
|
static void unload (qse_awk_mod_t* mod, qse_awk_t* awk)
|
|
{
|
|
/* TODO: anything */
|
|
}
|
|
|
|
int qse_awk_mod_str (qse_awk_mod_t* mod, qse_awk_t* awk)
|
|
{
|
|
mod->query = query;
|
|
mod->unload = unload;
|
|
|
|
mod->init = init;
|
|
mod->fini = fini;
|
|
/*
|
|
mod->ctx...
|
|
*/
|
|
|
|
return 0;
|
|
}
|
|
|