diff options
| -rw-r--r-- | source4/scripting/ejs/smbcalls_string.c | 103 |
1 files changed, 103 insertions, 0 deletions
diff --git a/source4/scripting/ejs/smbcalls_string.c b/source4/scripting/ejs/smbcalls_string.c index 7817de1b0d..b52a6d2182 100644 --- a/source4/scripting/ejs/smbcalls_string.c +++ b/source4/scripting/ejs/smbcalls_string.c @@ -59,6 +59,106 @@ static int ejs_strupper(MprVarHandle eid, int argc, char **argv) return 0; } +/* + usage: + list = split(".", "a.foo.bar"); + + NOTE: does not take a regular expression, unlink perl split() +*/ +static int ejs_split(MprVarHandle eid, int argc, char **argv) +{ + const char *separator; + char *s, *p; + struct MprVar ret; + int count = 0; + TALLOC_CTX *tmp_ctx = talloc_new(mprMemCtx()); + if (argc != 2) { + ejsSetErrorMsg(eid, "split invalid arguments"); + return -1; + } + separator = argv[0]; + s = argv[1]; + + ret = mprObject("list"); + + while ((p = strstr(s, separator))) { + char *s2 = talloc_strndup(tmp_ctx, s, (int)(p-s)); + mprAddArray(&ret, count++, mprString(s2)); + talloc_free(s2); + s = p + strlen(separator); + } + if (*s) { + mprAddArray(&ret, count++, mprString(s)); + } + talloc_free(tmp_ctx); + mpr_Return(eid, ret); + return 0; +} + + +/* + usage: + str = join("DC=", list); +*/ +static int ejs_join(MprVarHandle eid, int argc, struct MprVar **argv) +{ + int i; + const char *separator; + char *ret = NULL; + const char **list; + TALLOC_CTX *tmp_ctx = talloc_new(mprMemCtx()); + if (argc != 2 || + argv[0]->type != MPR_TYPE_STRING || + argv[1]->type != MPR_TYPE_OBJECT) { + ejsSetErrorMsg(eid, "join invalid arguments"); + return -1; + } + + separator = mprToString(argv[0]); + list = mprToList(tmp_ctx, argv[1]); + + if (list == NULL || list[0] == NULL) { + talloc_free(tmp_ctx); + mpr_Return(eid, mprString(NULL)); + return 0; + } + + ret = talloc_strdup(tmp_ctx, list[0]); + if (ret == NULL) { + goto failed; + } + for (i=1;list[i];i++) { + ret = talloc_asprintf_append(ret, "%s%s", separator, list[i]); + if (ret == NULL) { + goto failed; + } + } + mpr_Return(eid, mprString(ret)); + talloc_free(tmp_ctx); + return 0; +failed: + ejsSetErrorMsg(eid, "out of memory"); + return -1; +} + +/* + load a file as a string + usage: + string = FileLoad(filename); +*/ +static int ejs_FileLoad(MprVarHandle eid, int argc, char **argv) +{ + char *s; + if (argc != 1) { + ejsSetErrorMsg(eid, "FileLoad invalid arguments"); + return -1; + } + + s = file_load(argv[0], NULL, mprMemCtx()); + mpr_Return(eid, mprString(s)); + talloc_free(s); + return 0; +} /* setup C functions that be called from ejs @@ -67,4 +167,7 @@ void smb_setup_ejs_string(void) { ejsDefineStringCFunction(-1, "strlower", ejs_strlower, NULL, MPR_VAR_SCRIPT_HANDLE); ejsDefineStringCFunction(-1, "strupper", ejs_strupper, NULL, MPR_VAR_SCRIPT_HANDLE); + ejsDefineStringCFunction(-1, "split", ejs_split, NULL, MPR_VAR_SCRIPT_HANDLE); + ejsDefineStringCFunction(-1, "FileLoad", ejs_FileLoad, NULL, MPR_VAR_SCRIPT_HANDLE); + ejsDefineCFunction(-1, "join", ejs_join, NULL, MPR_VAR_SCRIPT_HANDLE); } |